database.h

Go to the documentation of this file.

 
#ifndef DATABASE_H
#define DATABASE_H
 
#include <memory>
#include <soci/soci.h>
#include <nlohmann/json.hpp>
#include <mantisbase/core/private-impl/soci_custom_types.hpp>
 
#ifdef MANTIS_SCRIPTING_ENABLED
#include <dukglue/dukglue.h>
#endif
 
#include "../mantisbase.h"
#include "../utils/utils.h"
#include "logger/logger.h"
 
namespace mb {
    using json = nlohmann::json;
 
    class Database {
    public:
        Database();
 
        ~Database();
 
        bool connect(const std::string &conn_str);
        const std::string& connectionStr() const;
 
 
        void disconnect();
 
        bool createSysTables() const;
 
        [[nodiscard]] std::shared_ptr<soci::session> session() const;
 
        [[nodiscard]] soci::connection_pool &connectionPool() const;
 
        [[nodiscard]] bool isConnected() const;
 
#ifdef MANTIS_SCRIPTING_ENABLED
        static void registerDuktapeMethods();
#endif
 
    private:
#ifdef MANTIS_SCRIPTING_ENABLED
        duk_ret_t query(duk_context *ctx);
#endif
 
        void writeCheckpoint() const;
 
        std::string m_connStr;
        std::unique_ptr<soci::connection_pool> m_connPool;
        const MantisBase &mbApp;
    };
 
    class MantisLoggerImpl : public soci::logger_impl {
    public:
        void start_query(std::string const &query) override {
            logger_impl::start_query(query);
            LogOrigin::dbTrace(fmt::format("$ sql << {}", query));
        }
 
    private:
        logger_impl *do_clone() const override {
            return new MantisLoggerImpl();
        }
    };
} // mb
 
#endif //DATABASE_H