src/runtime/common/trie_storage_provider_impl.cpp

Namespaces

Name
sgns
sgns::runtime

Source code

#include "runtime/common/trie_storage_provider_impl.hpp"

namespace sgns::runtime {

  TrieStorageProviderImpl::TrieStorageProviderImpl(
      std::shared_ptr<storage::trie::TrieStorage> trie_storage)
      : trie_storage_(std::move(trie_storage)) {
    BOOST_ASSERT(trie_storage_ != nullptr);
  }

  outcome::result<void> TrieStorageProviderImpl::setToEphemeral() {
    OUTCOME_TRY((auto &&, batch), trie_storage_->getEphemeralBatch());
    current_batch_ = std::move(batch);
    return outcome::success();
  }

  outcome::result<void> TrieStorageProviderImpl::setToEphemeralAt(
      const base::Hash256 &state_root) {
    OUTCOME_TRY((auto &&, batch), trie_storage_->getEphemeralBatchAt(state_root));
    current_batch_ = std::move(batch);
    return outcome::success();
  }

  outcome::result<void> TrieStorageProviderImpl::setToPersistent() {
    if (persistent_batch_ == nullptr) {
      OUTCOME_TRY((auto &&, batch), trie_storage_->getPersistentBatch());
      persistent_batch_ = std::move(batch);
    }
    current_batch_ = persistent_batch_;
    return outcome::success();
  }

  outcome::result<void> TrieStorageProviderImpl::setToPersistentAt(
      const base::Hash256 &state_root) {
    OUTCOME_TRY((auto &&, batch), trie_storage_->getPersistentBatchAt(state_root));
    persistent_batch_ = std::move(batch);
    current_batch_ = persistent_batch_;
    return outcome::success();
  }

  std::shared_ptr<TrieStorageProviderImpl::Batch>
  TrieStorageProviderImpl::getCurrentBatch() const {
    return current_batch_;
  }

  boost::optional<std::shared_ptr<TrieStorageProviderImpl::PersistentBatch>>
  TrieStorageProviderImpl::tryGetPersistentBatch() const {
    return isCurrentlyPersistent() ? boost::make_optional(persistent_batch_)
                                   : boost::none;
  }

  bool TrieStorageProviderImpl::isCurrentlyPersistent() const {
    return std::dynamic_pointer_cast<PersistentBatch>(current_batch_)
           != nullptr;
  }

  outcome::result<base::Buffer> TrieStorageProviderImpl::forceCommit() {
    if(persistent_batch_ != nullptr) {
      return persistent_batch_->commit();
    }
    return base::Buffer{};
  }

}  // namespace sgns::runtime

---

Updated on 2026-03-04 at 13:10:45 -0800