BaseTable.hpp

Go to the documentation of this file.

#pragma once
#ifndef _MDBX_CONTAINERS_BASE_DB_HPP_INCLUDED
#define _MDBX_CONTAINERS_BASE_DB_HPP_INCLUDED

 
namespace mdbxc {
    
    class BaseTable {
    public:
        explicit BaseTable(std::shared_ptr<Connection> connection,
                        std::string name,
                        MDBX_db_flags_t flags)
            : m_connection(std::move(connection)) {
            auto txn = m_connection->transaction();
            check_mdbx(
                mdbx_dbi_open(txn.handle(), name.c_str(), flags, &m_dbi),
                "Failed to open table"
            );
            txn.commit();
        }
        
        virtual ~BaseTable() = default;
        
        bool is_connected() const {
            return m_connection->is_connected();
        }

        void connect() {
            m_connection->connect();
        }

        void disconnect() {
            m_connection->disconnect();
        }
        
        void begin(TransactionMode mode = TransactionMode::WRITABLE) {
            m_connection->begin(mode);
        }

        void commit() {
            m_connection->commit();
        }

        void rollback() {
            m_connection->rollback();
        }
        
        template<typename Func>
        void execute_in_transaction(Func operation, TransactionMode mode = TransactionMode::WRITABLE) {
            auto txn = m_connection->transaction(mode);
            try {
                operation();
                txn.commit();
            } catch(...) {
                try {
                    txn.rollback();
                } catch(...) {}
                throw;
            }
        }
 
    protected:
        std::shared_ptr<Connection>  m_connection;   
        MDBX_dbi                     m_dbi{};         

        MDBX_txn* thread_txn() const {
                return m_connection->thread_txn();
        }
        
        MDBX_dbi handle() const { return m_dbi; }
    };
    
}; // namespace mdbxc
 
#endif // _MDBX_CONTAINERS_BASE_DB_HPP_INCLUDED