MongoDatabase.h

Go to the documentation of this file.

 
#pragma once
#include <bsoncxx/builder/stream/array.hpp>
#include <bsoncxx/builder/stream/document.hpp>
#include <mongocxx/client.hpp>
#include <mongocxx/pool.hpp>
#include <string>
 
namespace catapult { namespace mongo {
 
    class MongoDatabase {
    public:
        explicit MongoDatabase(mongocxx::pool& connectionPool, const std::string& databaseName)
                : m_pConnection(connectionPool.acquire())
                , m_database(m_pConnection->database(databaseName))
        {}
 
    public:
        operator const mongocxx::database&() const {
            return m_database;
        }
 
        operator mongocxx::database&() {
            return m_database;
        }
 
        const mongocxx::collection operator[](const std::string& collectionName) const {
            return m_database[collectionName];
        }
 
        mongocxx::collection operator[](const std::string& collectionName) {
            return m_database[collectionName];
        }
 
    private:
        mongocxx::pool::entry m_pConnection;
        mongocxx::database m_database;
    };
}}