protocol_req_collation.hpp

Go to the documentation of this file.

#ifndef KAGOME_NETWORK_REQCOLLATIONPROTOCOL
#define KAGOME_NETWORK_REQCOLLATIONPROTOCOL

#include "network/protocol_base.hpp"

#include <memory>

#include <libp2p/connection/stream.hpp>
#include <libp2p/host/host.hpp>

#include "application/app_configuration.hpp"
#include "application/chain_spec.hpp"
#include "log/logger.hpp"
#include "network/impl/stream_engine.hpp"
#include "network/peer_manager.hpp"
#include "network/protocols/req_collation_protocol.hpp"
#include "network/types/collator_messages.hpp"
#include "network/types/roles.hpp"
#include "network/types/status.hpp"
#include "utils/non_copyable.hpp"

namespace kagome::network {

  struct ReqCollationProtocolImpl;

  class ReqCollationProtocol final : public IReqCollationProtocol,
                                     NonCopyable,
                                     NonMovable {
   public:
    ReqCollationProtocol() = delete;
    ~ReqCollationProtocol() override = default;

    ReqCollationProtocol(libp2p::Host &host,
                         application::AppConfiguration const &app_config,
                         application::ChainSpec const &chain_spec,
                         std::shared_ptr<ReqCollationObserver> observer);

    const Protocol &protocolName() const override;

    bool start() override;
    bool stop() override;

    void onIncomingStream(std::shared_ptr<Stream> stream) override;
    void newOutgoingStream(
        const PeerInfo &peer_info,
        std::function<void(outcome::result<std::shared_ptr<Stream>>)> &&cb)
        override;

    void request(const PeerId &peer_id,
                 CollationFetchingRequest request,
                 std::function<void(outcome::result<CollationFetchingResponse>)>
                     &&response_handler) override;

   private:
    std::shared_ptr<ReqCollationProtocolImpl> impl_;
  };

}  // namespace kagome::network

#endif  // KAGOME_NETWORK_REQCOLLATIONPROTOCOL