C#如何接收已取消gRPC调用返回的部分处理结果？

核心结论 #

原生一元gRPC调用无法实现该需求。
原因：gRPC的超时/取消逻辑在客户端和服务端独立跟踪，当客户端设置的超时时间到达后，客户端gRPC库会直接终止当前调用，抛出状态码为DeadlineExceeded或Cancelled的RpcException，销毁对应调用上下文，不再接收服务端后续返回的任何报文。即便服务端捕获取消异常后生成了带部分结果的响应，该响应也无法被客户端正常接收处理。

可行替代方案 #

方案1：使用服务端流式gRPC调用 #

将原来的一元调用改为服务端流式调用，服务端可以在处理过程中任意时刻将结果写入流返回给客户端，即便是后续触发超时取消，客户端已经接收到的流数据会被保留，可以正常读取。

• 优点：保留gRPC原生超时能力，传输效率高，适合部分结果可以分段返回的场景
• 代码示例：
  proto定义：

service ServiceB {
  rpc GetNumber (GetNumberRequest) returns (stream GetNumberReply);
}

服务A调用代码：

public async Task<int> GetNumberFromB() 
{
    int result = 0;
    try 
    {
        var cts = new CancellationTokenSource(TimeSpan.FromSeconds(3));
        using var call = _clientB.GetNumber(new GetNumberRequest(), cancellationToken: cts.Token);
        // 读取服务端返回的所有流消息
        await foreach (var resp in call.ResponseStream.ReadAllAsync())
        {
            result = resp.Value;
        }
    }
    catch (RpcException e) when (e.StatusCode is StatusCode.DeadlineExceeded or StatusCode.Cancelled)
    {
        // 超时触发时result已经保存了服务端提前返回的部分结果
    }
    return result;
}

服务B实现代码：

public async Task GetNumber(GetNumberRequest request, IServerStreamWriter<GetNumberReply> responseStream, ServerCallContext context) 
{
    int value = 0;
    try 
    {
        // 模拟耗时10秒的处理逻辑
        await Task.Delay(TimeSpan.FromSeconds(10), context.CancellationToken);  
        value = 1;
    }
    catch (OperationCancelledException) 
    {
        value = 100;
    }
    // 将最终的结果写入流返回
    await responseStream.WriteAsync(new GetNumberReply { Value = value });
}

方案2：业务层自定义超时控制 #

将超时控制逻辑从gRPC框架层下沉到业务层实现，客户端不设置gRPC调用的CancellationToken超时，而是将超时时间作为请求参数传递给服务端，由服务端内部自行控制处理时长，超时后直接返回带部分结果的正常响应。

• 优点：改动最小，不需要修改gRPC方法类型，逻辑简单易维护
• 缺点：客户端不会主动终止调用，需要等待服务端返回结果，无法提前释放连接
• 代码示例：
  服务A调用代码：

public async Task<int> GetNumberFromB() 
{
    // 不在gRPC层设置超时，将超时时间放到业务请求参数中
    var result = await _clientB.GetNumber(new GetNumberRequest { TimeoutSeconds = 3 });
    return result.Value;
}

服务B实现代码：

public async Task<GetNumberReply> GetNumber(GetNumberRequest request, ServerCallContext context) 
{
    int value = 0;
    // 业务层自行创建超时令牌
    using var businessCts = new CancellationTokenSource(TimeSpan.FromSeconds(request.TimeoutSeconds));
    try 
    {
        await Task.Delay(TimeSpan.FromSeconds(10), businessCts.Token);  
        value = 1;
    }
    catch (OperationCancelledException) 
    {
        value = 100;
    }
    // 正常返回结果，客户端不会触发异常
    return new GetNumberReply { Value = value };
}

方案3：中间存储中转结果 #

对于处理周期很长的场景，可以为每个请求生成唯一ID，服务端处理过程中定时将部分结果写入Redis等共享中间存储，客户端超时后可以通过请求ID主动查询已生成的部分结果。

• 优点：适合异步长任务场景，客户端不需要保持连接等待
• 缺点：需要额外引入中间存储，增加架构复杂度

内容的提问来源于stack exchange，提问作者Jaka Konda