Merge pull request #1009 from sitkoru/grpc/client/deadlines

Grpc/client/deadlines
sitkoru · Aug 15, 2024 · 28ad3e3 · 28ad3e3
2 parents 8881bf4 + 821da4d
commit 28ad3e3
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/src/Sitko.Core.Grpc.Client/GrpcClientModule.cs b/src/Sitko.Core.Grpc.Client/GrpcClientModule.cs
@@ -26,10 +26,22 @@ public override void ConfigureServices(IApplicationContext applicationContext, I
                 "System.Net.Http.SocketsHttpHandler.Http2UnencryptedSupport", true);
         }
 
+        services.TryAddSingleton(TimeProvider.System);
         services.TryAddSingleton<GrpcClientActivator<TClient>>();
         services.TryAddSingleton<GrpcCallInvokerFactory>();
         services.TryAddTransient<global::Grpc.Net.ClientFactory.GrpcClientFactory, GrpcClientFactory>();
-        var builder = services.AddGrpcClient<TClient>();
+        var builder = services.AddGrpcClient<TClient>((provider, options) =>
+        {
+            options.CallOptionsActions.Add(context =>
+            {
+                if (startupOptions.DefaultDeadline is not null)
+                {
+                    context.CallOptions = context.CallOptions.WithDeadline(provider.GetRequiredService<TimeProvider>()
+                        .GetUtcNow()
+                        .Add(startupOptions.DefaultDeadline.Value).UtcDateTime);
+                }
+            });
+        });
         services.AddTransient<IGrpcClientProvider<TClient>, GrpcClientProvider<TClient>>();
         RegisterResolver(services, startupOptions);
         startupOptions.ConfigureClient(services, builder);

diff --git a/src/Sitko.Core.Grpc.Client/GrpcClientModuleOptions.cs b/src/Sitko.Core.Grpc.Client/GrpcClientModuleOptions.cs
@@ -17,6 +17,7 @@ public class GrpcClientModuleOptions<TClient> : BaseModuleOptions where TClient
     private readonly List<Action<IServiceCollection>> configureServicesActions = new();
     public bool EnableHttp2UnencryptedSupport { get; set; }
     public bool DisableCertificatesValidation { get; set; }
+    public TimeSpan? DefaultDeadline { get; set; } = TimeSpan.FromMinutes(30);
     [JsonIgnore] public Action<GrpcChannelOptions>? ConfigureChannelOptions { get; set; }
     [JsonIgnore] public Func<HttpClientHandler, HttpMessageHandler>? ConfigureHttpHandler { get; set; }