intermediate changes

ref:cde9a383711a11544ce7e107a78147fb96cc4029

1 files changed, 84 insertions, 0 deletions

diff --git a/library/cpp/grpc/client/grpc_common.h b/library/cpp/grpc/client/grpc_common.h
new file mode 100644
index 0000000000..ffcdafe045
--- /dev/null
+++ b/library/cpp/grpc/client/grpc_common.h
@@ -0,0 +1,84 @@
+#pragma once
+
+#include <grpc++/grpc++.h>
+#include <grpc++/resource_quota.h>
+
+#include <util/datetime/base.h>
+#include <unordered_map>
+#include <util/generic/string.h>
+
+constexpr ui64 DEFAULT_GRPC_MESSAGE_SIZE_LIMIT = 64000000;
+
+namespace NGrpc {
+
+struct TGRpcClientConfig {
+    TString Locator; // format host:port
+    TDuration Timeout = TDuration::Max(); // request timeout
+    ui64 MaxMessageSize = DEFAULT_GRPC_MESSAGE_SIZE_LIMIT; // Max request and response size
+    ui64 MaxInboundMessageSize = 0; // overrides MaxMessageSize for incoming requests
+    ui64 MaxOutboundMessageSize = 0; // overrides MaxMessageSize for outgoing requests
+    ui32 MaxInFlight = 0;
+    bool EnableSsl = false;
+    TString SslCaCert;  //Implicitly enables Ssl if not empty
+    grpc_compression_algorithm CompressionAlgoritm = GRPC_COMPRESS_NONE;
+    ui64 MemQuota = 0;
+    std::unordered_map<TString, TString> StringChannelParams;
+    std::unordered_map<TString, int> IntChannelParams;
+    TString LoadBalancingPolicy = { };
+    TString SslTargetNameOverride = { };
+
+    TGRpcClientConfig() = default;
+    TGRpcClientConfig(const TGRpcClientConfig&) = default;
+    TGRpcClientConfig(TGRpcClientConfig&&) = default;
+    TGRpcClientConfig& operator=(const TGRpcClientConfig&) = default;
+    TGRpcClientConfig& operator=(TGRpcClientConfig&&) = default;
+
+    TGRpcClientConfig(const TString& locator, TDuration timeout = TDuration::Max(),
+            ui64 maxMessageSize = DEFAULT_GRPC_MESSAGE_SIZE_LIMIT, ui32 maxInFlight = 0, TString caCert = "",
+            grpc_compression_algorithm compressionAlgorithm = GRPC_COMPRESS_NONE, bool enableSsl = false)
+        : Locator(locator)
+        , Timeout(timeout)
+        , MaxMessageSize(maxMessageSize)
+        , MaxInFlight(maxInFlight)
+        , EnableSsl(enableSsl)
+        , SslCaCert(caCert)
+        , CompressionAlgoritm(compressionAlgorithm)
+    {}
+};
+
+inline std::shared_ptr<grpc::ChannelInterface> CreateChannelInterface(const TGRpcClientConfig& config, grpc_socket_mutator* mutator = nullptr){
+    grpc::ChannelArguments args;
+    args.SetMaxReceiveMessageSize(config.MaxInboundMessageSize ? config.MaxInboundMessageSize : config.MaxMessageSize);
+    args.SetMaxSendMessageSize(config.MaxOutboundMessageSize ? config.MaxOutboundMessageSize : config.MaxMessageSize);
+    args.SetCompressionAlgorithm(config.CompressionAlgoritm);
+
+    for (const auto& kvp: config.StringChannelParams) {
+        args.SetString(kvp.first, kvp.second);
+    }
+
+    for (const auto& kvp: config.IntChannelParams) {
+        args.SetInt(kvp.first, kvp.second);
+    }
+
+    if (config.MemQuota) {
+        grpc::ResourceQuota quota;
+        quota.Resize(config.MemQuota);
+        args.SetResourceQuota(quota);
+    }
+    if (mutator) {
+        args.SetSocketMutator(mutator);
+    }
+    if (!config.LoadBalancingPolicy.empty()) {
+        args.SetLoadBalancingPolicyName(config.LoadBalancingPolicy);
+    }
+    if (!config.SslTargetNameOverride.empty()) {
+        args.SetSslTargetNameOverride(config.SslTargetNameOverride);
+    }
+    if (config.EnableSsl || config.SslCaCert) {
+        return grpc::CreateCustomChannel(config.Locator, grpc::SslCredentials(grpc::SslCredentialsOptions{config.SslCaCert, "", ""}), args);
+    } else {
+        return grpc::CreateCustomChannel(config.Locator, grpc::InsecureChannelCredentials(), args);
+    }
+}
+
+} // namespace NGRpc