KvStore.hpp

Go to the documentation of this file.

 
#ifndef FOCS_KVSTORE
#define FOCS_KVSTORE
 
#include "focs/FilterIterator.hpp"
 
#include <functional>
#include <iterator>
#include <string>
#include <unordered_map>
#include <vector>
 
namespace focs {
    class KvStore {
        using read_callback = std::function<bool(const std::pair<std::string, std::string>&)>;
        using group_filtered_iterator = focs::FilterIterator<std::unordered_map<std::string, std::string>::const_iterator, std::function<bool(const std::pair<std::string, std::string>&)>>;
        using group_iterator = std::vector<std::string>::const_iterator;
        using iterator = std::unordered_map<std::string, std::string>::iterator;
        using const_iterator = std::unordered_map<std::string, std::string>::const_iterator;
 
        public:
            KvStore() = default;
            explicit KvStore(const std::string& file);
 
            KvStore(const std::string& file, read_callback callback);
 
            bool load(const std::string& file);
 
            bool load(std::istream& in);
 
            bool load_line(const std::string& line);
 
            group_iterator group_begin() const;
            group_iterator group_end() const;
 
            iterator begin();
            iterator end();
 
            const_iterator cbegin() const;
            const_iterator cend() const;
 
            group_filtered_iterator begin(const std::string& group) const;
            group_filtered_iterator end(const std::string& group) const;
 
 
            std::string& operator[] (const std::string& k);
            std::string& operator[] (std::string&& k);
 
            std::string& at(const std::string& k);
            std::string& at(std::string&& k);
 
            KvStore::iterator find(const std::string& k);
            KvStore::iterator find(std::string&& k);
            KvStore::const_iterator find(std::string& k) const;
            KvStore::const_iterator find(std::string&& k) const;
 
            size_t count(const std::string& k) const;
            size_t count(std::string&& k) const;
 
 
            friend std::ostream& operator<<(std::ostream& os, const KvStore& kv);
 
            bool command(std::string&& cmd, std::string&& args);
            bool command_group(std::string&& group);
            bool command_include(std::string&& file);
            bool command_include_local(std::string&& file);
 
            void group_separator(const std::string& s){group_separator_ = s;}
            const std::string& group_separator(){return group_separator_;}
 
            void callback(read_callback callback){callback_ = callback;}
            read_callback callback(){return callback_;}
 
            void switch_group(const std::string& group);
 
        private:
            std::vector<std::string> groups_{};
            std::unordered_map<std::string, std::string> kv_{};
            std::string group_separator_{"."};
            std::string current_group_{""};
            std::string current_file_{""};
            std::string read_group_{""};
 
            read_callback callback_{};
 
            void add_group(const std::string& group);
 
    };
} // namespace focs
 
#endif // FOCS_KVSTORE