MongoReceiptPluginFactory.h

Go to the documentation of this file.

 
#pragma once
#include "MongoReceiptPlugin.h"
#include "catapult/model/Receipt.h"
#include "catapult/functions.h"
 
namespace catapult { namespace mongo {
 
    class MongoReceiptPluginFactory {
    private:
        using ReceiptPlugin = MongoReceiptPlugin;
 
    public:
        template<typename TReceipt, typename TStreamFunc>
        static std::unique_ptr<ReceiptPlugin> Create(model::ReceiptType type, TStreamFunc streamFunc) {
            return std::make_unique<ReceiptPluginT<TReceipt>>(type, streamFunc);
        }
 
    private:
        template<typename TReceipt>
        class ReceiptPluginT : public ReceiptPlugin {
        private:
            using StreamFunc = consumer<bsoncxx::builder::stream::document&, const TReceipt&>;
 
        public:
            explicit ReceiptPluginT(model::ReceiptType type, const StreamFunc& streamFunc)
                    : m_type(type)
                    , m_streamFunc(streamFunc)
            {}
 
        public:
            model::ReceiptType type() const override {
                return m_type;
            }
 
            void streamReceipt(bsoncxx::builder::stream::document& builder, const model::Receipt& receipt) const override {
                m_streamFunc(builder, static_cast<const TReceipt&>(receipt));
            }
 
        private:
            model::ReceiptType m_type;
            StreamFunc m_streamFunc;
        };
    };
 
#define DEFINE_MONGO_RECEIPT_PLUGIN_FACTORY(NAME, STREAM) \
    std::unique_ptr<MongoReceiptPlugin> Create##NAME##ReceiptMongoPlugin(model::ReceiptType type) { \
        return MongoReceiptPluginFactory::Create<model::NAME##Receipt>(type, STREAM); \
    }
 
    std::unique_ptr<MongoReceiptPlugin> CreateBalanceTransferReceiptMongoPlugin(model::ReceiptType type);
 
    std::unique_ptr<MongoReceiptPlugin> CreateBalanceChangeReceiptMongoPlugin(model::ReceiptType type);
 
    std::unique_ptr<MongoReceiptPlugin> CreateInflationReceiptMongoPlugin(model::ReceiptType type);
}}