db.hpp

Go to the documentation of this file.

 
#ifndef CCFCOMMON_DB_HPP_
#define CCFCOMMON_DB_HPP_
 
#include <string>
 
#include <mal/Cii.hpp>
#include <mal/utility/LoadMal.hpp>
 
#include <rad/topicPub.hpp>
#include <rad/mal/publisher.hpp>
#include <rad/dbAdapterRedis.hpp>
#include <rad/dbAdapter.hpp>
#include <rad/smAdapter.hpp>
 
#include <Stdif.hpp>
 
#include <ccf/common/base.hpp>
#include <ccf/common/config.hpp>
 
namespace ccf::common {
 
class InternalDb : public rad::DbAdapter {
   public:
 
    InternalDb() {m_connected = false;};
 
    ~InternalDb() {};
 
    void Config(const std::string& endpoint, const timeval& timeout);
 
    int Delete(const std::vector<std::string>& keys);
 
    bool Exists(const std::string& key);
 
    std::string Get(const std::string& key);
 
    std::vector<std::string> MultiGet(const std::vector<std::string>& keys);
 
    std::string Get(const std::string& key, const std::string& field);
 
    std::vector<std::string> MultiGet(const std::string& key,
                                      const std::vector<std::string>& fields);
 
    std::map<std::string, std::string> Scan(const std::string& key, const std::string& pattern);
 
    void Set(const std::string& key, const std::string& value);
 
    void MultiSet(const std::vector<std::string>& kvs);
 
    void Set(const std::string& key, const std::string& field, const std::string& value);
 
    void MultiSet(const std::string& key, const std::vector<std::string>& fvs);
 
    void Connect();
 
    void Disconnect();
 
    bool IsConnected();
 
    InternalDb(const InternalDb&) = delete;             
    InternalDb& operator=(const InternalDb&) = delete;  
 
   protected:
   private:
      std::map<std::string, std::string> m_db;
      bool m_connected;
};
 
 
class Db {
   public:
 
   static Db* s_instance;
 
   static Db& Instance();
 
   static void Simulate();
 
   Db(const std::string& prefix);
 
   virtual ~Db();
 
   void Disconnect();
 
   void Connect();
 
   std::string GenDbPath(const std::vector<std::string>& keys,
                         const std::string& base_path = "");
 
   std::map<std::string, std::string> Scan(const std::string& key, const std::string& pattern);
 
   bool IsConnected();
 
   std::string GetSmState();
 
   std::string GetSmStatusState();
 
   std::string GetSmStatusSubstate();
 
   void UpdateSmStatus(std::list<scxml4cpp::State*>& status);
 
   void UpdateSmStatus(const std::string& status);
 
   std::string Get(const std::string& key);
 
   std::string Get2(const std::vector<std::string>& keys,
                    const std::string& base_path = "");
 
   std::vector<std::string> MultiGet(const std::vector<std::string>& keys);
 
   bool Exists(const std::string& key);
 
   int Delete(const std::vector<std::string>& keys);
 
   void Set(const std::string& key, const std::string& value);
 
   template <class TYPE>
   void Set(const std::string& key, const TYPE value) {
      LOG4CPLUS_TRACE_METHOD(Logger(), __PRETTY_FUNCTION__);
      std::string str_value = std::to_string(value);
      Set(key, str_value);
   }
 
   void Set2(const std::vector<std::string>& keys,
             const std::string& value,
             const std::string& base_path = "");
 
   template <class TYPE>
   void Set2(const std::vector<std::string>& keys,
             const TYPE value,
             const std::string& base_path = "") {
      LOG4CPLUS_TRACE_METHOD(Logger(), __PRETTY_FUNCTION__);
      std::string str_value = std::to_string(value);
      Set2(keys, str_value, base_path);
   }
 
   const std::string& GetAddress(const std::string& key);
 
   Db(const Db&) = delete;             
   Db& operator=(const Db&) = delete;  
 
   rad::DbAdapter& GetDb();
 
 protected:
 
 private:
   static bool                        s_simulation;
 
   std::string                        m_prefix;
   rad::DbAdapterRedis                m_redis_db;
   InternalDb                         m_internal_db;
   std::map<std::string, std::string> m_key_address_map;
 
   bool m_with_pubsub;
   std::unique_ptr<rad::cii::Publisher<stdif::Status>> m_status_publisher;
 
   std::string  m_current_status; // Current state + substate.
   std::string  m_sm_state;
   std::string  m_sm_status_state;
   std::string  m_sm_status_substate;
 
   std::string  m_server_id;
};
 
} // namespace ccf::common
 
namespace ccf {
   inline ccf::common::Db& GetDb() {
      return ccf::common::Db::Instance();
   }
}
 
#endif // CCFCOMMON_DB_HPP_