channel_factory.hpp

Go to the documentation of this file.

#ifndef IROHA_CHANNEL_FACTORY_HPP
#define IROHA_CHANNEL_FACTORY_HPP

#include "network/impl/channel_provider.hpp"

#include <memory>
#include <set>
#include <string>

#include <grpc++/grpc++.h>

#include "common/result.hpp"
#include "interfaces/common_objects/types.hpp"
#include "network/impl/grpc_channel_params.hpp"

namespace iroha {
  namespace network {

    std::unique_ptr<GrpcChannelParams> getDefaultChannelParams();

    grpc::ChannelArguments makeChannelArguments(
        const std::set<std::string> &services, const GrpcChannelParams &params);

    template <typename Service>
    grpc::ChannelArguments makeChannelArguments(
        const GrpcChannelParams &params) {
      return makeChannelArguments(Service::service_full_name(), params);
    }

    template <typename Service>
    std::unique_ptr<typename Service::StubInterface> createInsecureClient(
        const std::string &address, const GrpcChannelParams &params) {
      auto channel =
          createInsecureChannel(address, Service::service_full_name(), params);
      return Service::NewStub(std::move(channel));
    }

    template <typename Service>
    std::unique_ptr<typename Service::StubInterface> createInsecureClient(
        const std::string &ip, size_t port, const GrpcChannelParams &params) {
      return createInsecureClient<Service>(ip + ":" + std::to_string(port),
                                           params);
    }

    std::shared_ptr<grpc::Channel> createInsecureChannel(
        const shared_model::interface::types::AddressType &address,
        const std::string &service_full_name,
        const GrpcChannelParams &params);

    class ChannelFactory : public ChannelProvider {
     public:
      ChannelFactory(std::shared_ptr<const GrpcChannelParams> params);

      ~ChannelFactory() override;

      iroha::expected::Result<std::shared_ptr<grpc::Channel>, std::string>
      getChannel(const std::string &service_full_name,
                 const shared_model::interface::Peer &peer) override;

     protected:
      virtual iroha::expected::Result<std::shared_ptr<grpc::ChannelCredentials>,
                                      std::string>
      getChannelCredentials(const shared_model::interface::Peer &) const;

     private:
      class ChannelArgumentsProvider;
      std::unique_ptr<ChannelArgumentsProvider> args_;
    };

  }  // namespace network
}  // namespace iroha

#endif