storage_changes_tracker_impl.cpp

Go to the documentation of this file.

#include "storage/changes_trie/impl/storage_changes_tracker_impl.hpp"

#include "storage/predefined_keys.hpp"

namespace kagome::storage::changes_trie {

  StorageChangesTrackerImpl::StorageChangesTrackerImpl(
      primitives::events::StorageSubscriptionEnginePtr
          storage_subscription_engine,
      primitives::events::ChainSubscriptionEnginePtr chain_subscription_engine)
      : storage_subscription_engine_(std::move(storage_subscription_engine)),
        chain_subscription_engine_(std::move(chain_subscription_engine)),
        logger_{log::createLogger("Storage Changes Tracker", "changes_trie")} {}

  void StorageChangesTrackerImpl::onBlockExecutionStart(
      primitives::BlockHash new_parent_hash) {
    parent_hash_ = new_parent_hash;
    // new block -- new extrinsics
    actual_val_.clear();
    new_entries_.clear();
  }

  void StorageChangesTrackerImpl::onBlockAdded(
      const primitives::BlockHash &hash) {
    if (actual_val_.find(storage::kRuntimeCodeKey) != actual_val_.cend()) {
      chain_subscription_engine_->notify(
          primitives::events::ChainEventType::kNewRuntime, hash);
    }
    for (auto &pair : actual_val_) {
      if (pair.second) {
        SL_TRACE(logger_, "Key: {:l}; Value {:l};", pair.first, *pair.second);
      } else {
        SL_TRACE(logger_, "Key: {:l}; Removed;", pair.first);
      }
      storage_subscription_engine_->notify(
          pair.first, pair.second, parent_hash_);
    }
  }

  void StorageChangesTrackerImpl::onPut(const common::BufferView &key,
                                        const common::BufferView &value,
                                        bool is_new_entry) {
    auto it = actual_val_.find(key);
    if (it != actual_val_.end()) {
      it->second.emplace(value);
    } else {
      actual_val_.emplace(key, value);
      if (is_new_entry) {
        new_entries_.insert(common::Buffer{key});
      }
    }
  }

  void StorageChangesTrackerImpl::onRemove(const common::BufferView &key) {
    if (auto it = actual_val_.find(key); it != actual_val_.end()) {
      if (new_entries_.erase(it->first) != 0) {
        actual_val_.erase(it);
      } else {
        it->second.reset();
      }
    }
  }
}  // namespace kagome::storage::changes_trie