Question

我的接口INetwork有一个方法：

Task<bool> SendAsync(string messageToSend, CancellationToken ct)

接口的一个实现具有如下代码：

public async Task<bool> SendAsync(string messageToSend, CancellationToken ct)
{
  var udpClient = new UdpClient();
  var data = Encoding.UTF8.GetBytes (messageToSend);
  var sentBytes = await udpClient.SendAsync(data);
  return sentBytes == data.Length; 
}

很遗憾，SendAsync()课程的UdpClient不接受CancellationToken。

所以我开始将其改为：

public Task<bool> SendAsync(string messageToSend, CancellationToken ct)
{
  var udpClient = new UdpClient();
  var data = Encoding.UTF8.GetBytes (messageToSend);
  var sendTask = udpClient.SendAsync(data);
  sendTask.Wait(ct);

  if(sendTask.Status == RanToCompletion)
  {
    return sendTask.Result == data.Length;
  }
}

显然这不起作用，因为没有返回Task。但是，如果我返回任务，则签名不再匹配。 SendAsync()会返回Task<int>，但我需要Task<bool>。

现在我很困惑。 :-) 如何解决这个问题？

Answer 1

我知道这有点晚了，但我最近不得不取消UdpClient ReceiveAsync / SendAsync。

您的第一个代码块是在没有取消的情况下发送的（您的标题表示顺便收到...）。

你的第二个代码块绝对不是这样做的。您正在调用* Async，然后调用Task.Wait，它将一直阻塞，直到调用完成。这使得呼叫有效地同步，并且调用* Async版本没有意义。最好的解决方案是使用Async，如下所示：

...
var sendTask = udpClient.SendAsync(data);
var tcs = new TaskCompletionSource<bool>();
using( ct.Register( s => tcs.TrySetResult(true), null) )
{
    if( sendTask != await Task.WhenAny( task, tcs.Task) )
        // ct.Cancel() called
    else
        // sendTask completed first, so .Result will not block
}
...

在UdpClient上没有内置的取消方法（没有一个函数接受CancellationToken），但你可以利用Task.WhenAny等待多个任务的能力。这将返回第一个完成的任务（这也是使用Task.Delay()实现超时的简单方法）。然后我们只需要创建一个在取消CancellationToken时完成的任务，我们可以通过创建TaskCompletionSource并使用CancellationToken的回调设置它来完成。

一旦取消，我们可以关闭套接字以实际“取消”基础读/写。

最初的想法是来自处理文件句柄的另一个SO答案，但它也适用于套接字。我通常用这样的扩展方法将其包装起来：

public static class AsyncExtensions
{
    public static async Task<T> WithCancellation<T>( this Task<T> task, CancellationToken cancellationToken )
    {
        var tcs = new TaskCompletionSource<bool>();
        using( cancellationToken.Register( s => ( (TaskCompletionSource<bool>)s ).TrySetResult( true ), tcs ) )
        {
            if( task != await Task.WhenAny( task, tcs.Task ) )
            {
                throw new OperationCanceledException( cancellationToken );
            }
        }

        return task.Result;
    }
}

然后像这样使用它：

try
{
    var data = await client.ReceiveAsync().WithCancellation(cts.Token);
    await client.SendAsync(data.Buffer, data.Buffer.Length, toep).WithCancellation(cts.Token);
}
catch(OperationCanceledException)
{
    client.Close();
}

Answer 2

首先，如果您想要返回Task<bool>，您只需使用Task.FromResult()即可。但是你可能不应该这样做，拥有一个实际上是同步的异步方法没有多大意义。

除此之外，我还认为你不应该假装该方法被取消，即使它不是。您可以做的是在开始真实SendAsync()之前检查令牌，但就是这样。

如果你真的想假装该方法被尽快取消，你可以使用ContinueWith()取消：

var sentBytes = await sendTask.ContinueWith(t => t.Result, ct);

如何使UdpClient.ReceiveAsync（）可以取消？

2 个答案: