storage_impl.hpp

Go to the documentation of this file.

#ifndef IROHA_STORAGE_IMPL_HPP
#define IROHA_STORAGE_IMPL_HPP

#include "ametsuchi/storage.hpp"

#include <atomic>
#include <shared_mutex>

#include <soci/soci.h>
#include <boost/optional.hpp>
#include <rxcpp/rx-lite.hpp>
#include "ametsuchi/block_storage_factory.hpp"
#include "ametsuchi/impl/pool_wrapper.hpp"
#include "ametsuchi/key_value_storage.hpp"
#include "ametsuchi/ledger_state.hpp"
#include "ametsuchi/reconnection_strategy.hpp"
#include "interfaces/permission_to_string.hpp"
#include "logger/logger_fwd.hpp"
#include "logger/logger_manager_fwd.hpp"

namespace shared_model {
  namespace interface {
    class QueryResponseFactory;
  }  // namespace interface
}  // namespace shared_model
namespace iroha {

  class PendingTransactionStorage;

  namespace ametsuchi {

    class AmetsuchiTest;
    class PostgresOptions;
    class VmCaller;

    class StorageImpl : public Storage {
     public:
      static expected::Result<std::shared_ptr<StorageImpl>, std::string> create(
          const PostgresOptions &postgres_options,
          std::shared_ptr<PoolWrapper> pool_wrapper,
          std::shared_ptr<shared_model::interface::PermissionToString>
              perm_converter,
          std::shared_ptr<PendingTransactionStorage> pending_txs_storage,
          std::shared_ptr<shared_model::interface::QueryResponseFactory>
              query_response_factory,
          std::unique_ptr<BlockStorageFactory> temporary_block_storage_factory,
          std::shared_ptr<BlockStorage> persistent_block_storage,
          std::optional<std::reference_wrapper<const VmCaller>> vm_caller_ref,
          logger::LoggerManagerTreePtr log_manager,
          size_t pool_size = 10);

      expected::Result<std::unique_ptr<CommandExecutor>, std::string>
      createCommandExecutor() override;

      std::unique_ptr<TemporaryWsv> createTemporaryWsv(
          std::shared_ptr<CommandExecutor> command_executor) override;

      std::unique_ptr<MutableStorage> createMutableStorage(
          std::shared_ptr<CommandExecutor> command_executor) override;

      boost::optional<std::shared_ptr<PeerQuery>> createPeerQuery()
          const override;

      boost::optional<std::shared_ptr<BlockQuery>> createBlockQuery()
          const override;

      boost::optional<std::unique_ptr<SettingQuery>> createSettingQuery()
          const override;

      iroha::expected::Result<std::unique_ptr<QueryExecutor>, std::string>
      createQueryExecutor(
          std::shared_ptr<PendingTransactionStorage> pending_txs_storage,
          std::shared_ptr<shared_model::interface::QueryResponseFactory>
              response_factory) const override;

      bool insertBlock(
          std::shared_ptr<const shared_model::interface::Block> block) override;

      expected::Result<void, std::string> insertPeer(
          const shared_model::interface::Peer &peer) override;

      std::unique_ptr<MutableStorage> createMutableStorage(
          std::shared_ptr<CommandExecutor> command_executor,
          BlockStorageFactory &storage_factory) override;

      void resetPeers() override;

      expected::Result<void, std::string> dropBlockStorage() override;

      boost::optional<std::shared_ptr<const iroha::LedgerState>>
      getLedgerState() const override;

      void freeConnections() override;

      CommitResult commit(
          std::unique_ptr<MutableStorage> mutable_storage) override;

      bool preparedCommitEnabled() const override;

      CommitResult commitPrepared(
          std::shared_ptr<const shared_model::interface::Block> block) override;

      std::shared_ptr<WsvQuery> getWsvQuery() const override;

      std::shared_ptr<BlockQuery> getBlockQuery() const override;

      rxcpp::observable<std::shared_ptr<const shared_model::interface::Block>>
      on_commit() override;

      void prepareBlock(std::unique_ptr<TemporaryWsv> wsv) override;

      ~StorageImpl() override;

     protected:
      StorageImpl(
          boost::optional<std::shared_ptr<const iroha::LedgerState>>
              ledger_state,
          const PostgresOptions &postgres_options,
          std::shared_ptr<BlockStorage> block_store,
          std::shared_ptr<PoolWrapper> pool_wrapper,
          std::shared_ptr<shared_model::interface::PermissionToString>
              perm_converter,
          std::shared_ptr<PendingTransactionStorage> pending_txs_storage,
          std::shared_ptr<shared_model::interface::QueryResponseFactory>
              query_response_factory,
          std::unique_ptr<BlockStorageFactory> temporary_block_storage_factory,
          size_t pool_size,
          std::optional<std::reference_wrapper<const VmCaller>> vm_caller,
          logger::LoggerManagerTreePtr log_manager);

     private:
      using StoreBlockResult = iroha::expected::Result<void, std::string>;

      friend class ::iroha::ametsuchi::AmetsuchiTest;

      StoreBlockResult storeBlock(
          std::shared_ptr<const shared_model::interface::Block> block);

      void tryRollback(soci::session &session);

      std::shared_ptr<BlockStorage> block_store_;

      std::shared_ptr<PoolWrapper> pool_wrapper_;

      std::shared_ptr<soci::connection_pool> &connection_;

      rxcpp::composite_subscription notifier_lifetime_;
      rxcpp::subjects::subject<
          std::shared_ptr<const shared_model::interface::Block>>
          notifier_;

      std::shared_ptr<shared_model::interface::PermissionToString>
          perm_converter_;

      std::shared_ptr<PendingTransactionStorage> pending_txs_storage_;

      std::shared_ptr<shared_model::interface::QueryResponseFactory>
          query_response_factory_;

      std::unique_ptr<BlockStorageFactory> temporary_block_storage_factory_;

      std::optional<std::reference_wrapper<const VmCaller>> vm_caller_ref_;

      logger::LoggerManagerTreePtr log_manager_;
      logger::LoggerPtr log_;

      mutable std::shared_timed_mutex drop_mutex_;

      const size_t pool_size_;

      bool prepared_blocks_enabled_;

      std::atomic<bool> block_is_prepared_;

      std::string prepared_block_name_;

      boost::optional<std::shared_ptr<const iroha::LedgerState>> ledger_state_;
    };
  }  // namespace ametsuchi
}  // namespace iroha

#endif  // IROHA_STORAGE_IMPL_HPP