network_impl.hpp

Go to the documentation of this file.

#ifndef IROHA_NETWORK_IMPL_HPP
#define IROHA_NETWORK_IMPL_HPP

#include "consensus/yac/transport/yac_network_interface.hpp"  // for YacNetwork
#include "yac.grpc.pb.h"

#include <memory>
#include <mutex>
#include <unordered_map>

#include "consensus/yac/outcome_messages.hpp"
#include "consensus/yac/vote_message.hpp"
#include "interfaces/common_objects/peer.hpp"
#include "interfaces/common_objects/types.hpp"
#include "logger/logger_fwd.hpp"
#include "network/impl/async_grpc_client.hpp"

namespace iroha {
  namespace consensus {
    namespace yac {

      class NetworkImpl : public YacNetwork, public proto::Yac::Service {
       public:
        explicit NetworkImpl(
            std::shared_ptr<network::AsyncGrpcClient<google::protobuf::Empty>>
                async_call,
            std::function<std::unique_ptr<proto::Yac::StubInterface>(
                const shared_model::interface::Peer &)> client_creator,
            logger::LoggerPtr log);

        void subscribe(
            std::shared_ptr<YacNetworkNotifications> handler) override;

        void sendState(const shared_model::interface::Peer &to,
                       const std::vector<VoteMessage> &state) override;

        grpc::Status SendState(
            ::grpc::ServerContext *context,
            const ::iroha::consensus::yac::proto::State *request,
            ::google::protobuf::Empty *response) override;

        void stop() override;

       private:
        void createPeerConnection(const shared_model::interface::Peer &peer);

        std::unordered_map<shared_model::interface::types::AddressType,
                           std::unique_ptr<proto::Yac::StubInterface>>
            peers_;

        std::weak_ptr<YacNetworkNotifications> handler_;

        std::shared_ptr<network::AsyncGrpcClient<google::protobuf::Empty>>
            async_call_;

        std::function<std::unique_ptr<proto::Yac::StubInterface>(
            const shared_model::interface::Peer &)>
            client_creator_;

        std::mutex stop_mutex_;
        bool stop_requested_{false};

        logger::LoggerPtr log_;
      };

    }  // namespace yac
  }    // namespace consensus
}  // namespace iroha

#endif  // IROHA_NETWORK_IMPL_HPP