任何区分取消和超时的方法

Question

我有一些代码通过调用许多其他服务来验证某些数据。我并行启动所有调用，然后等待至少其中一个调用完成。如果任何请求失败，我不关心其他调用的结果。

我使用HttpClient 拨打电话，我通过了HttpMessageHandler 进行了大量的日志记录。本质上：

protected override async Task<HttpResponseMessage> SendAsync(HttpRequestMessage request, CancellationToken cancellationToken)
{
    HttpResponseMessage response = null;

    try
    {
        response = await base.SendAsync(request, cancellationToken);
    }
    catch (OperationCanceledException ex)
    {
        LogTimeout(...);
        throw;
    }
    catch (Exception ex)
    {
        LogFailure(...);
        throw;
    }
    finally
    {
        LogComplete(...);
    }

    return response;
}

不，我遇到问题的部分是我取消请求时。当我取消一个请求时，我是故意这样做的，所以我不希望它被记录为超时，但取消和真正的超时之间似乎没有任何区别。

有没有办法做到这一点？

编辑：我需要稍微澄清一下。并行调用的服务是在超时的情况下传入 CancellationTokens：

var ct = new CancellationTokenSource(TimeSpan.FromSeconds(2));

所以当服务器响应时间超过两秒时，我会得到一个OperationCanceledException，如果我手动取消令牌源（比如因为另一台服务器在 1 秒后返回错误），那么我仍然会得到一个@987654326 @。理想情况下，我可以查看CancellationToken.IsCancellationRequested 来确定它是否由于超时而被取消，而不是明确要求取消，但无论如何，您似乎都获得了相同的值> 它被取消了。

Answer 1

如果要区分这两种取消类型，则需要使用两种不同的取消令牌。没有别的办法。这并不难，因为他们can be linked - 有点尴尬。

编写此 IMO 最简洁的方法是将超时代码移动到 SendAsync 方法而不是调用方法中：

protected override async Task<HttpResponseMessage> SendAsync(HttpRequestMessage request, CancellationToken cancellationToken)
{
  using (var cts = CancellationTokenSource.CreateLinkedTokenSource(cancellationToken))
  {
    cts.CancelAfter(TimeSpan.FromSeconds(2));
    try
    {
      return await base.SendAsync(request, cts.Token);
    }
    catch (OperationCanceledException ex)
    {
      if (cancellationToken.IsCancellationRequested)
        return null;
      LogTimeout(...);
      throw;
    }
    catch (Exception ex)
    {
      LogFailure(...);
      throw;
    }
    finally
    {
      LogComplete(...);
    }
  }
}

如果您不想将超时代码移动到 SendAsync，那么您也需要在该方法之外进行日志记录。

Answer 2

如果异常没有告诉您这两种情况之间的区别，那么您需要检查Task 或CancellationToken 以查看是否真的取消了。

如果抛出未处理的OperationCanceledException（最有可能在base.SendAsync 内使用CancellationToken.ThrowIfCancellationRequested），我倾向于询问Task，它的IsCanceled 属性将返回true。像这样的...

HttpResponseMessage response = null;
Task sendTask = null;

try
{
  sendTask = base.SendAsync(request, cancellationToken);
  await sendTask;
}
catch (OperationCanceledException ex)
{
  if (!sendTask.IsCancelled)
  {
    LogTimeout(...);
    throw;
  }
}

编辑

针对问题的更新，我想更新我的答案。你是对的，无论是在CancellationTokenSource 上特别要求取消还是由超时引起，都会导致完全相同的结果。如果你反编译CancellationTokenSource，你会看到它只是设置了一个Timer回调，当达到超时时会显式调用CancellationTokenSource.Cancel，所以两种方式最终都会调用相同的Cancel方法。

我认为如果你想区分你需要从 CancellationTokenSource 派生（它不是 sealed 类），然后添加你自己的自定义取消方法，该方法将设置一个标志让你知道您明确取消了操作，而不是让它超时。

这很不幸，因为您将同时使用自定义取消方法和原始Cancel 方法，并且必须确保使用自定义取消方法。您可能能够摆脱您的自定义逻辑，只需隐藏现有的 Cancel 操作，如下所示：

class CustomCancellationTokenSource : CancellationTokenSource
{
  public bool WasManuallyCancelled {get; private set;}

  public new void Cancel()
  {
    WasManuallyCancelled = true;
    base.Cancel();
  }
}

我认为隐藏基本方法会起作用，您可以试一试并找出答案。