服务调用场景中如何区分CancellationToken取消与HttpClient超时？

要在服务B内部区分服务A传入的CancellationToken主动取消和HttpClient超时导致的取消，可以通过以下两种可行方案实现：

方案一：自定义请求头+中断时间判断 #

核心思路是让服务A传递超时信息给服务B，服务B通过请求中断的时间差来推断取消原因，这是无需依赖外部日志的本地判断方案。

步骤1：服务A端修改请求逻辑 #

在调用服务B时，添加自定义HTTP头传递超时时间（单位毫秒），同时建议用手动超时逻辑替代HttpClient的全局超时，便于后续区分取消源：

// 生成请求消息，添加自定义超时头
var requestMessage = new HttpRequestMessage(HttpMethod.Post, "api/serviceb/dostuff")
{
    Content = JsonContent.Create(requestParams)
};
// 传递服务A客户端的超时时间（这里是5秒=5000毫秒）
requestMessage.Headers.Add("X-Request-Timeout", "5000");

// 合并传入的CancellationToken和手动超时Token
using var cts = CancellationTokenSource.CreateLinkedTokenSource(cancellationToken);
cts.CancelAfter(TimeSpan.FromSeconds(5)); // 代替HttpClient的全局超时

// 发起请求
var response = await _httpClient.SendAsync(requestMessage, cts.Token)
    .ContinueWith(task =>
    {
        // 后续处理逻辑
    }, cancellationToken);

步骤2：服务B端识别取消原因 #

在服务B的控制器中，记录请求开始时间，捕获请求中断事件后，通过时间差和超时头信息判断取消类型：

[ApiController]
[Route("api/[controller]")]
public class ServiceBController : ControllerBase
{
    [HttpPost("DoStuff")]
    public async Task<IActionResult> DoStuff([FromBody] RequestParams request)
    {
        var startTime = DateTimeOffset.UtcNow;
        int? clientTimeoutMs = null;
        // 解析服务A传递的超时头
        if (Request.Headers.TryGetValue("X-Request-Timeout", out var timeoutHeader))
        {
            int.TryParse(timeoutHeader, out var timeout);
            clientTimeoutMs = timeout;
        }

        try
        {
            // 模拟业务处理逻辑
            await Task.Delay(TimeSpan.FromSeconds(10), HttpContext.RequestAborted);
            return Ok("处理完成");
        }
        catch (OperationCanceledException)
        {
            if (!HttpContext.RequestAborted.IsCancellationRequested)
            {
                return StatusCode(500, "服务内部取消");
            }

            // 计算请求从开始到中断的耗时
            var elapsedMs = (DateTimeOffset.UtcNow - startTime).TotalMilliseconds;
            string cancelReason;

            // 用误差范围（±500ms）判断是否为超时导致的取消
            if (clientTimeoutMs.HasValue && Math.Abs(elapsedMs - clientTimeoutMs.Value) < 500)
            {
                cancelReason = "服务A HttpClient超时导致取消";
            }
            else
            {
                cancelReason = "服务A传入的CancellationToken主动取消";
            }

            // 可将原因写入日志或返回给调用方
            Console.WriteLine($"请求取消：{cancelReason}，耗时：{elapsedMs:F0}ms");
            return StatusCode(499, cancelReason); // 用499表示客户端主动中断请求
        }
    }
}

方案二：请求ID+跨服务日志关联 #

如果对判断准确性要求更高，可以通过请求ID关联服务A和服务B的日志，直接在服务A端区分取消原因后记录，服务B只需记录中断事件即可。

步骤1：服务A端添加请求ID并记录取消原因 #

var requestId = Guid.NewGuid().ToString();
var requestMessage = new HttpRequestMessage(HttpMethod.Post, "api/serviceb/dostuff")
{
    Content = JsonContent.Create(requestParams)
};
requestMessage.Headers.Add("X-Request-Id", requestId);

using var cts = CancellationTokenSource.CreateLinkedTokenSource(cancellationToken);
cts.CancelAfter(TimeSpan.FromSeconds(5));

try
{
    var response = await _httpClient.SendAsync(requestMessage, cts.Token);
}
catch (TaskCanceledException)
{
    if (cancellationToken.IsCancellationRequested)
    {
        // 记录日志：Request {requestId} 被服务A的CancellationToken主动取消
        Console.WriteLine($"请求 {requestId}：被服务A主动取消");
    }
    else
    {
        // 记录日志：Request {requestId} 因HttpClient超时被取消
        Console.WriteLine($"请求 {requestId}：HttpClient超时取消");
    }
}

步骤2：服务B端记录中断事件 #

[HttpPost("DoStuff")]
public async Task<IActionResult> DoStuff([FromBody] RequestParams request)
{
    if (!Request.Headers.TryGetValue("X-Request-Id", out var requestId))
    {
        requestId = Guid.NewGuid().ToString();
    }

    try
    {
        await Task.Delay(TimeSpan.FromSeconds(10), HttpContext.RequestAborted);
        return Ok("处理完成");
    }
    catch (OperationCanceledException)
    {
        if (HttpContext.RequestAborted.IsCancellationRequested)
        {
            Console.WriteLine($"请求 {requestId}：被客户端中断");
        }
        return StatusCode(499, "请求被客户端中断");
    }
}

后续通过X-Request-Id关联服务A和服务B的日志，即可明确区分两种取消原因。

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