MongoPluginManager.h

Go to the documentation of this file.

 
#pragma once
#include "ExternalCacheStorageBuilder.h"
#include "MongoReceiptPlugin.h"
#include "MongoStorageContext.h"
#include "MongoTransactionPlugin.h"
#include "catapult/model/NetworkInfo.h"
#include "catapult/plugins.h"
 
namespace catapult { namespace mongo {
 
    class MongoPluginManager {
    public:
        explicit MongoPluginManager(MongoStorageContext& mongoContext, model::NetworkIdentifier networkIdentifier)
                : m_mongoContext(mongoContext)
                , m_networkIdentifier(networkIdentifier)
        {}
 
    public:
        MongoStorageContext& mongoContext() const {
            return m_mongoContext;
        }
 
        model::NetworkIdentifier networkIdentifier() const {
            return m_networkIdentifier;
        }
 
    public:
        void addTransactionSupport(std::unique_ptr<MongoTransactionPlugin>&& pTransactionPlugin) {
            m_transactionRegistry.registerPlugin(std::move(pTransactionPlugin));
        }
 
        void addReceiptSupport(std::unique_ptr<MongoReceiptPlugin>&& pReceiptPlugin) {
            m_receiptRegistry.registerPlugin(std::move(pReceiptPlugin));
        }
 
        template<typename TStorage>
        void addStorageSupport(std::unique_ptr<TStorage>&& pStorage) {
            m_storageBuilder.add(std::move(pStorage));
        }
 
    public:
        const MongoTransactionRegistry& transactionRegistry() const {
            return m_transactionRegistry;
        }
 
        const MongoReceiptRegistry& receiptRegistry() const {
            return m_receiptRegistry;
        }
 
        std::unique_ptr<ExternalCacheStorage> createStorage() {
            return m_storageBuilder.build();
        }
 
    private:
        MongoStorageContext& m_mongoContext;
        model::NetworkIdentifier m_networkIdentifier;
        MongoTransactionRegistry m_transactionRegistry;
        MongoReceiptRegistry m_receiptRegistry;
        ExternalCacheStorageBuilder m_storageBuilder;
    };
}}
 
extern "C" PLUGIN_API
void RegisterMongoSubsystem(catapult::mongo::MongoPluginManager& manager);