use request timeout setting

ydb-platform · Oct 22, 2024 · 21ebd5a · 21ebd5a
1 parent 8a0e007
commit 21ebd5a
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/ydb/library/grpc/actor_client/grpc_service_client.h b/ydb/library/grpc/actor_client/grpc_service_client.h
@@ -71,10 +71,6 @@ class TGrpcServiceClient  {
         }
     };
 
-    void SetTimeout(TDuration timeout) {
-        Config.Timeout = timeout;
-    }
-
     template <typename TCallType>
     void MakeCall(typename TCallType::TRequestEventType::TPtr ev) {
         using TRequestType = decltype(typename TCallType::TRequestEventType().Request);
@@ -121,7 +117,8 @@ class TGrpcServiceClient  {
     }
 
     static NYdbGrpc::TGRpcClientConfig InitGrpcConfig(const NGrpcActorClient::TGrpcClientSettings& settings) {
-        NYdbGrpc::TGRpcClientConfig config(settings.Endpoint, DEFAULT_TIMEOUT, NYdbGrpc::DEFAULT_GRPC_MESSAGE_SIZE_LIMIT, 0, settings.CertificateRootCA);
+        const TDuration effectiveTimeout = settings.RequestTimeoutMs ? TDuration::Milliseconds(settings.RequestTimeoutMs) : DEFAULT_TIMEOUT;
+        NYdbGrpc::TGRpcClientConfig config(settings.Endpoint, effectiveTimeout, NYdbGrpc::DEFAULT_GRPC_MESSAGE_SIZE_LIMIT, 0, settings.CertificateRootCA);
         config.EnableSsl = settings.EnableSsl;
         config.IntChannelParams[GRPC_ARG_KEEPALIVE_TIME_MS] = settings.GrpcKeepAliveTimeMs;
         config.IntChannelParams[GRPC_ARG_KEEPALIVE_TIMEOUT_MS] = settings.GrpcKeepAliveTimeoutMs;