主页 / coding / 问题 / 78940302
Accepted
dan-kli
Asked: 2024-09-02 19:47:59 +0800 CST

如何为 gRPC 客户端工厂创建的 gRPC 客户端设置通用截止时间？

  • 772

按照.NET 中的 gRPC 客户端工厂集成，我可以像这样注册一个 gRPC 客户端：

builder.Services.AddGrpcClient<Greeter.GreeterClient>(o =>
{
    o.Address = new Uri("https://localhost:5001");
});

现在我想为与该客户的所有通话设置一个通用截止时间。可能像这样：

int deadline = ReadFromConfig();

builder.Services.AddGrpcClient<Greeter.GreeterClient>(o =>
{
    o.Address = new Uri("https://localhost:5001");
    o.CallOptionsActions.Add(o => 
        o.CallOptions.Deadline = DateTime.UtcNow.AddSeconds(deadline))
});

但是，这里的调用选项中的截止日期是只读的。我似乎无法在这里设置它。我做错了吗，还是只能在发出请求时设置截止日期？

c#

1 个回答

  1. Best Answer

    完全未经测试，但是怎么样Interceptor？

    var channel = GrpcChannel.ForAddress("https://whatever");
var invoker = channel.Intercept(new DeadlineInterceptor(TimeSpan.FromSeconds(10)));
// TODO: create your client from "invoker", not "channel"


class DeadlineInterceptor(TimeSpan deadline) : Interceptor
{
    private void ApplyDeadline<TRequest, TResponse>(ref ClientInterceptorContext<TRequest, TResponse> context)
        where TRequest : class
        where TResponse : class
    {
        if (context.Options.Deadline is null)
        {
            context = new(context.Method, context.Host, context.Options.WithDeadline(DateTime.UtcNow.Add(deadline)));
        }
    }

    public override AsyncClientStreamingCall<TRequest, TResponse> AsyncClientStreamingCall<TRequest, TResponse>(ClientInterceptorContext<TRequest, TResponse> context, AsyncClientStreamingCallContinuation<TRequest, TResponse> continuation)
    {
        ApplyDeadline(ref context);
        return base.AsyncClientStreamingCall(context, continuation);
    }

    public override AsyncDuplexStreamingCall<TRequest, TResponse> AsyncDuplexStreamingCall<TRequest, TResponse>(ClientInterceptorContext<TRequest, TResponse> context, AsyncDuplexStreamingCallContinuation<TRequest, TResponse> continuation)
    {
        ApplyDeadline(ref context);
        return base.AsyncDuplexStreamingCall(context, continuation);
    }

    public override AsyncServerStreamingCall<TResponse> AsyncServerStreamingCall<TRequest, TResponse>(TRequest request, ClientInterceptorContext<TRequest, TResponse> context, AsyncServerStreamingCallContinuation<TRequest, TResponse> continuation)
    {
        ApplyDeadline(ref context);
        return base.AsyncServerStreamingCall(request, context, continuation);
    }

    public override AsyncUnaryCall<TResponse> AsyncUnaryCall<TRequest, TResponse>(TRequest request, ClientInterceptorContext<TRequest, TResponse> context, AsyncUnaryCallContinuation<TRequest, TResponse> continuation)
    {
        ApplyDeadline(ref context);
        return base.AsyncUnaryCall(request, context, continuation);
    }

    public override TResponse BlockingUnaryCall<TRequest, TResponse>(TRequest request, ClientInterceptorContext<TRequest, TResponse> context, BlockingUnaryCallContinuation<TRequest, TResponse> continuation)
    {
        ApplyDeadline(ref context);
        return base.BlockingUnaryCall(request, context, continuation);
    }

    // note no need to intercept server methods
}
    • 2

相关问题