Line data    Source code

       1             : /**
       2             :  * Copyright Soramitsu Co., Ltd. 2018 All Rights Reserved.
       3             :  * http://soramitsu.co.jp
       4             :  *
       5             :  * Licensed under the Apache License, Version 2.0 (the "License");
       6             :  * you may not use this file except in compliance with the License.
       7             :  * You may obtain a copy of the License at
       8             :  *
       9             :  *        http://www.apache.org/licenses/LICENSE-2.0
      10             :  *
      11             :  * Unless required by applicable law or agreed to in writing, software
      12             :  * distributed under the License is distributed on an "AS IS" BASIS,
      13             :  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
      14             :  * See the License for the specific language governing permissions and
      15             :  * limitations under the License.
      16             :  */
      17             : 
      18             : #ifndef IROHA_GRPC_CHANNEL_BUILDER_HPP
      19             : #define IROHA_GRPC_CHANNEL_BUILDER_HPP
      20             : 
      21             : #include <grpc++/grpc++.h>
      22             : 
      23             : namespace iroha {
      24             :   namespace network {
      25             : 
      26             :     /**
      27             :      * Creates client which is capable of sending and receiving
      28             :      * messages of INT_MAX bytes size
      29             :      * @tparam T type for gRPC stub, e.g. proto::Yac
      30             :      * @param address ip address for connection, ipv4:port
      31             :      * @return gRPC stub of parametrized type
      32             :      */
      33             :     template <typename T>
      34             :     auto createClient(const grpc::string& address) {
      35             :       // in order to bypass built-in limitation of gRPC message size
      36        1737 :       grpc::ChannelArguments args;
      37        1737 :       args.SetMaxSendMessageSize(INT_MAX);
      38        1737 :       args.SetMaxReceiveMessageSize(INT_MAX);
      39             : 
      40        1737 :       return
      41        1737 :           T::NewStub(grpc::CreateCustomChannel(
      42        1737 :               address, grpc::InsecureChannelCredentials(), args));
      43        1737 :     }
      44             :   } // namespace network
      45             : } // namespace iroha
      46             : 
      47             : #endif  // IROHA_GRPC_CHANNEL_BUILDER_HPP