LockStatusAccountBalanceObserver.h

Go to the documentation of this file.

 
#pragma once
#include "catapult/cache_core/AccountStateCache.h"
 
namespace catapult { namespace observers {
 
    template<typename TTraits>
    void LockStatusAccountBalanceObserver(const typename TTraits::Notification& notification, ObserverContext& context) {
        auto& accountStateCache = context.Cache.sub<cache::AccountStateCache>();
        auto& cache = context.Cache.template sub<typename TTraits::CacheType>();
        const auto& key = TTraits::NotificationToKey(notification, context.Resolvers);
        auto lockInfoIter = cache.find(key);
        auto& lockInfo = lockInfoIter.get();
 
        auto accountStateIter = accountStateCache.find(TTraits::DestinationAccount(lockInfo));
        auto& accountState = accountStateIter.get();
 
        if (NotifyMode::Rollback == context.Mode) {
            lockInfo.Status = state::LockStatus::Unused;
            accountState.Balances.debit(lockInfo.MosaicId, lockInfo.Amount);
            return;
        }
 
        lockInfo.Status = state::LockStatus::Used;
        accountState.Balances.credit(lockInfo.MosaicId, lockInfo.Amount);
 
        model::BalanceChangeReceipt receipt(TTraits::Receipt_Type, accountState.PublicKey, lockInfo.MosaicId, lockInfo.Amount);
        context.StatementBuilder().addReceipt(receipt);
    }
}}