MongoStorageContext.h

Go to the documentation of this file.

 
#pragma once
#include "MongoDatabase.h"
#include "MongoErrorPolicy.h"
 
namespace catapult { namespace mongo { class MongoBulkWriter; } }
 
namespace catapult { namespace mongo {
 
    class MongoStorageContext {
    public:
        MongoStorageContext() = default;
 
        MongoStorageContext(
                const mongocxx::uri& uri,
                const std::string& databaseName,
                const std::shared_ptr<MongoBulkWriter>& pBulkWriter,
                MongoErrorPolicy::Mode errorPolicyMode)
                : m_connectionPool(uri)
                , m_databaseName(databaseName)
                , m_pBulkWriter(pBulkWriter)
                , m_errorPolicyMode(errorPolicyMode)
        {}
 
    public:
        MongoDatabase createDatabaseConnection() {
            return MongoDatabase(m_connectionPool, m_databaseName);
        }
 
        MongoErrorPolicy createCollectionErrorPolicy(const std::string& collectionName) {
            return MongoErrorPolicy(collectionName, m_errorPolicyMode);
        }
 
        MongoBulkWriter& bulkWriter() const {
            return *m_pBulkWriter;
        }
 
    private:
        mongocxx::pool m_connectionPool;
        std::string m_databaseName;
        std::shared_ptr<MongoBulkWriter> m_pBulkWriter;
        MongoErrorPolicy::Mode m_errorPolicyMode;
    };
}}