c# - 异步网络操作永远不会完成

Question

我有几个异步网络操作返回一个可能永远不会完成的任务:

1. UdpClient.ReceiveAsync 不接受 CancellationToken
2. TcpClient.GetStream 返回一个 NetworkStream，它不遵守 Stream.ReadAsync 上的 CancellationToken(检查仅在操作开始时取消)

两者都在等待一条可能永远不会到来的消息(例如因为数据包丢失或没有响应)。这意味着我有永远不会完成的 幻影 任务，永远不会运行的延续和使用的套接字暂停。我知道我可以使用 TimeoutAfter ，但这只会解决延续问题。

那我该怎么办呢？

Answer 1

所以我在 IDisposable 上创建了一个扩展方法，它创建了一个 CancellationToken，它在超时时处理连接，因此任务完成，一切继续进行:

public static IDisposable CreateTimeoutScope(this IDisposable disposable, TimeSpan timeSpan)
{
    var cancellationTokenSource = new CancellationTokenSource(timeSpan);
    var cancellationTokenRegistration = cancellationTokenSource.Token.Register(disposable.Dispose);
    return new DisposableScope(
        () =>
        {
            cancellationTokenRegistration.Dispose();
            cancellationTokenSource.Dispose();
            disposable.Dispose();
        });
}

而且用法非常简单:

try
{
    var client = new UdpClient();
    using (client.CreateTimeoutScope(TimeSpan.FromSeconds(2)))
    {
        var result = await client.ReceiveAsync();
        // Handle result
    }
}
catch (ObjectDisposedException)
{
    return null;
}

---

额外信息:

public sealed class DisposableScope : IDisposable
{
    private readonly Action _closeScopeAction;
    public DisposableScope(Action closeScopeAction)
    {
        _closeScopeAction = closeScopeAction;
    }
    public void Dispose()
    {
        _closeScopeAction();
    }
}