ChannelReader.WaitToReadAsync时如何避免抛出InvalidOperationException？答案

【问题标题】：How to avoid InvalidOperationException was thrown when ChannelReader.WaitToReadAsync?ChannelReader.WaitToReadAsync时如何避免抛出InvalidOperationException？
【发布时间】：2019-04-25 02:03:47
【问题描述】：

我使用 System.Threading.Channels 编写了异步队列。但是当我运行程序进行测试时，随机抛出以下异常并且工作线程被停止。

System.InvalidOperationException: The asynchronous operation has not completed.
   at System.Threading.Channels.AsyncOperation.ThrowIncompleteOperationException()
   at System.Threading.Channels.AsyncOperation`1.GetResult(Int16 token)
   at AsyncChannels.Worker() in g:\src\gitrepos\dotnet-sandbox\channelstest\AsyncChannelsTest.cs:line 26

如果异常被捕获并忽略，则代码正在运行。但是我想摆脱原因不明的错误。

这是我的环境和最少的代码。

TargetFramework = netcoreapp2.1
System.Threading.Channels 版本 = 4.5.0

using System.Threading.Channels;
using System.Threading;
using System.Threading.Tasks;
using System;
using System.Linq;

class AsyncChannels : IDisposable
{
    Channel<TaskCompletionSource<bool>> _Channel;
    Thread _Thread;
    CancellationTokenSource _Cancellation;
    public AsyncChannels()
    {
        _Channel = Channel.CreateUnbounded<TaskCompletionSource<bool>>();
        _Thread = new Thread(Worker);
        _Thread.Start();
        _Cancellation = new CancellationTokenSource();
    }
    private void Worker()
    {
        while (!_Cancellation.IsCancellationRequested)
        {
            // System.InvalidOperationException is thrown
            if (!_Channel.Reader.WaitToReadAsync(_Cancellation.Token).Result)
            {
                break;
            }
            while (_Channel.Reader.TryRead(out var item))
            {
                item.TrySetResult(true);
            }
        }
    }
    public void Dispose()
    {
        _Cancellation.Cancel();
        _Channel.Writer.TryComplete();
        _Thread.Join();
    }
    public Task<bool> Enqueue()
    {
        var tcs = new TaskCompletionSource<bool>(TaskCreationOptions.RunContinuationsAsynchronously);
        _Channel.Writer.TryWrite(tcs);
        return tcs.Task;
    }
    public static async Task Test()
    {
        using (var queue = new AsyncChannels())
        {
            for (int i = 0; i < 100000; i++)
            {
                await queue.Enqueue().ConfigureAwait(false);
            }
        }
    }
}

【问题讨论】：

你调试了吗，你确定在Dispose期间取消任务时不会出现异常吗？

标签： c# multithreading async-await system.threading.channels

【解决方案1】：

你can't directly block on a ValueTask<T>这样：

_Channel.Reader.WaitToReadAsync(_Cancellation.Token).Result

您只能对 ValueTask<T> 做两件事：await（仅一次），或者通过调用 AsTask() 将其转换为 Task<T>。如果你需要做一些复杂的事情，比如await不止一次或阻止它，那么你需要使用AsTask()。

或者，在这种情况下，只需在标准 Channels 消费模式中使用 await：

class AsyncChannels : IDisposable
{
  Channel<TaskCompletionSource<bool>> _Channel;
  Task _Thread;
  CancellationTokenSource _Cancellation;
  public AsyncChannels()
  {
    _Channel = Channel.CreateUnbounded<TaskCompletionSource<bool>>();
    _Thread = Task.Run(() => WorkerAsync());
    _Cancellation = new CancellationTokenSource();
  }
  private async Task WorkerAsync()
  {
    try
    {
      while (await _Channel.Reader.WaitToReadAsync(_Cancellation.Token))
        while (_Channel.Reader.TryRead(out var item))
          item.TrySetResult(true);
    }
    catch (OperationCanceledException)
    {
    }
  }
  public void Dispose()
  {
    _Cancellation.Cancel();
    _Channel.Writer.TryComplete();
  }
  ...
}

【讨论】：