dbInterfaceRedis.hpp

Go to the documentation of this file.

 
#ifndef CORE_UTILS_BAT_DB_INTERFACE_REDIS_HPP_
#define CORE_UTILS_BAT_DB_INTERFACE_REDIS_HPP_
 
// System headers
#include <string>
#include <fmt/format.h>
 
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wpedantic"
 
#include <sw/redis++/redis++.h>
 
#pragma GCC diagnostic pop
 
// Local headers
#include <utils/bat/logger.hpp>
 
namespace utils::bat {
 
  // These parameters are required by the Redis++ library.
  const std::size_t REDIS_CONNECTION_POOL_SIZE = 15;
   const int64_t REDIS_CONNECTION_POOL_CONN_LIFETIME_MS = 30000;
   const int64_t REDIS_WAIT_TIMEOUT_MS = 10000;
   const int64_t REDIS_SOCKET_TIMEOUT_MS = 15000;
 
  class DbInterfaceRedis {
  public:
    DbInterfaceRedis(const std::string& prefix,
                const std::string& db_endpoint,
                const timeval& db_timeout);
 
    inline bool IsConnected() const {return m_connected;}
    void SetStates(const std::string& prefix,
                                 const std::string& state,
                                 const std::string& substate);
    void BatchSet(const utils::bat::DbVector& vec);
    template<typename T>
    void Set(const std::string& key,
               const T& value);
 
    DbInterfaceRedis(const DbInterfaceRedis&) = delete;       
    DbInterfaceRedis& operator=(const DbInterfaceRedis&) = delete;  
 
  private:
    std::string               m_prefix;     
    std::string               m_db_endpoint;
    timeval                   m_db_timeout;
    bool                      m_connected{false};
    std::unique_ptr<::sw::redis::Redis> m_runtime_db;
 
    log4cplus::Logger         m_logger;
  };
 
  template<typename T>
  void DbInterfaceRedis::Set(const std::string& key, const T& value) {
    std::ostringstream os;
    os << value;
 
    try {
    LOG4CPLUS_DEBUG(m_logger, fmt::format("Writing attribute <{}> ...", m_prefix+key));
    m_runtime_db->set(m_prefix+key, os.str());
    LOG4CPLUS_DEBUG(m_logger, fmt::format("Writing attribute <{}> finished !", key));
    } catch (const ::sw::redis::Error& err) {
      LOG4CPLUS_ERROR(m_logger, fmt::format("Problems writing attribute <{}>: <> !",
                                            key, err.what()));
      throw std::runtime_error (fmt::format("Problems writing attribute <{}>: <> !", key,
                                            err.what()));
    }
  }
 
}  // namespace
 
#endif  // CORE_UTILS_BAT_DB_INTERFACE_REDIS_HPP_