TransactionChangeTracker.h

Go to the documentation of this file.

 
#pragma once
#include "ContainerTypes.h"
#include "EntityInfo.h"
 
namespace catapult { namespace model {
 
    class TransactionChangeTracker {
    public:
        const TransactionInfosSet& addedTransactionInfos() const {
            return m_addedTransactionInfos;
        }
 
        const TransactionInfosSet& removedTransactionInfos() const {
            return m_removedTransactionInfos;
        }
 
    public:
        void add(const TransactionInfo& transactionInfo) {
            // if the info can be found in the set of deleted transactions, it means that
            // it is currently in the collection so removing the info from the set means (re)adding it
            auto iter = m_removedTransactionInfos.find(transactionInfo);
            if (m_removedTransactionInfos.cend() != iter)
                m_removedTransactionInfos.erase(iter);
            else
                m_addedTransactionInfos.emplace(transactionInfo.copy());
        }
 
        void remove(const TransactionInfo& transactionInfo) {
            // if the info can be found in the set of added transactions, simply remove it
            // this only works because transactions are immutable (there are no upserts)
            // so a newly saved transaction cannot already be in the database
            auto iter = m_addedTransactionInfos.find(transactionInfo);
            if (m_addedTransactionInfos.cend() != iter)
                m_addedTransactionInfos.erase(iter);
            else
                m_removedTransactionInfos.emplace(transactionInfo.copy());
        }
 
        void reset() {
            m_addedTransactionInfos.clear();
            m_removedTransactionInfos.clear();
        }
 
    private:
        TransactionInfosSet m_addedTransactionInfos;
        TransactionInfosSet m_removedTransactionInfos;
    };
}}