mst_processor_impl.hpp

Go to the documentation of this file.

#ifndef IROHA_MST_PROCESSOR_IMPL_HPP
#define IROHA_MST_PROCESSOR_IMPL_HPP

#include <memory>
#include "cryptography/public_key.hpp"
#include "logger/logger_fwd.hpp"
#include "multi_sig_transactions/mst_processor.hpp"
#include "multi_sig_transactions/mst_propagation_strategy.hpp"
#include "multi_sig_transactions/mst_time_provider.hpp"
#include "multi_sig_transactions/storage/mst_storage.hpp"
#include "network/mst_transport.hpp"

namespace iroha {

  class FairMstProcessor : public MstProcessor,
                           public iroha::network::MstTransportNotification {
   public:
    FairMstProcessor(std::shared_ptr<iroha::network::MstTransport> transport,
                     std::shared_ptr<MstStorage> storage,
                     std::shared_ptr<PropagationStrategy> strategy,
                     std::shared_ptr<MstTimeProvider> time_provider,
                     logger::LoggerPtr log);

    ~FairMstProcessor();

    // ------------------------| MstProcessor override |------------------------

    auto propagateBatchImpl(const DataType &batch)
        -> decltype(propagateBatch(batch)) override;

    auto onStateUpdateImpl() const -> decltype(onStateUpdate()) override;

    auto onPreparedBatchesImpl() const
        -> decltype(onPreparedBatches()) override;

    auto onExpiredBatchesImpl() const -> decltype(onExpiredBatches()) override;

    bool batchInStorageImpl(const DataType &batch) const override;

    // ------------------| MstTransportNotification override |------------------

    void onNewState(const shared_model::crypto::PublicKey &from,
                    ConstRefState new_state) override;

    // ----------------------------| end override |-----------------------------

   private:
    // -----------------------------| private api |-----------------------------

    void onPropagate(const PropagationStrategy::PropagationData &data);

    void completedBatchesNotify(ConstRefState state) const;

    void updatedBatchesNotify(ConstRefState state) const;

    void expiredBatchesNotify(ConstRefState state) const;

    // -------------------------------| fields |--------------------------------
    std::shared_ptr<iroha::network::MstTransport> transport_;
    std::shared_ptr<MstStorage> storage_;
    std::shared_ptr<PropagationStrategy> strategy_;
    std::shared_ptr<MstTimeProvider> time_provider_;

    // rx subjects

    rxcpp::subjects::subject<std::shared_ptr<MstState>> state_subject_;

    rxcpp::subjects::subject<DataType> batches_subject_;

    rxcpp::subjects::subject<DataType> expired_subject_;


    rxcpp::composite_subscription propagation_subscriber_;

    logger::LoggerPtr log_;
  };
}  // namespace iroha

#endif  // IROHA_MST_PROCESSOR_IMPL_HPP