c# - 如何为不接受取消 token 的异步函数设置超时？

我的网络请求由这个代码处理；

Response = await Client.SendAsync(Message, HttpCompletionOption.ResponseHeadersRead, CToken);

在读取响应头之后和内容完成读取之前返回。当我打这个电话来获取内容时…

return await Response.Content.ReadAsStringAsync();

我想在X秒后停止它。但它不接受取消代币。

最佳答案

虽然您可以依赖WithCancellation进行重用，但对于超时（不抛出OperationCanceledException）的一个更简单的解决方案是使用Task.Delay创建一个超时任务，并使用Task.WhenAny等待第一个任务完成：

public static Task<TResult> WithTimeout<TResult>(this Task<TResult> task, TimeSpan timeout)
{
    var timeoutTask = Task.Delay(timeout).ContinueWith(_ => default(TResult), TaskContinuationOptions.ExecuteSynchronously);
    return Task.WhenAny(task, timeoutTask).Unwrap();
}

或者，如果希望在超时时抛出异常，而不是只返回默认值（即null）：

public static async Task<TResult> WithTimeout<TResult>(this Task<TResult> task, TimeSpan timeout)
{
    if (task == await Task.WhenAny(task, Task.Delay(timeout)))
    {
        return await task;
    }
    throw new TimeoutException();
}

用法是：

var content = await Response.Content.ReadAsStringAsync().WithTimeout(TimeSpan.FromSeconds(1));