これは、BlockingCollection
待機をサポートする非常に基本的な実装ですが、多くの機能が不足しています。これはAsyncEnumerable
ライブラリを使用します。これにより、8.0より前のC#バージョンで非同期列挙が可能になります。
public class AsyncBlockingCollection<T>
{
private Queue<T> _queue = new Queue<T>();
private SemaphoreSlim _semaphore = new SemaphoreSlim(0);
private int _consumersCount = 0;
private bool _isAddingCompleted;
public void Add(T item)
{
lock (_queue)
{
if (_isAddingCompleted) throw new InvalidOperationException();
_queue.Enqueue(item);
}
_semaphore.Release();
}
public void CompleteAdding()
{
lock (_queue)
{
if (_isAddingCompleted) return;
_isAddingCompleted = true;
if (_consumersCount > 0) _semaphore.Release(_consumersCount);
}
}
public IAsyncEnumerable<T> GetConsumingEnumerable()
{
lock (_queue) _consumersCount++;
return new AsyncEnumerable<T>(async yield =>
{
while (true)
{
lock (_queue)
{
if (_queue.Count == 0 && _isAddingCompleted) break;
}
await _semaphore.WaitAsync();
bool hasItem;
T item = default;
lock (_queue)
{
hasItem = _queue.Count > 0;
if (hasItem) item = _queue.Dequeue();
}
if (hasItem) await yield.ReturnAsync(item);
}
});
}
}
使用例:
var abc = new AsyncBlockingCollection<int>();
var producer = Task.Run(async () =>
{
for (int i = 1; i <= 10; i++)
{
await Task.Delay(100);
abc.Add(i);
}
abc.CompleteAdding();
});
var consumer = Task.Run(async () =>
{
await abc.GetConsumingEnumerable().ForEachAsync(async item =>
{
await Task.Delay(200);
await Console.Out.WriteAsync(item + " ");
});
});
await Task.WhenAll(producer, consumer);
出力:
1 2 3 4 5 6 7 8 9 10
更新: C#8のリリースにより、非同期列挙は組み込み言語機能になりました。必要なクラス(IAsyncEnumerable
、IAsyncEnumerator
)は.NET Core 3.0に組み込まれており、.NET Framework 4.6.1+(Microsoft.Bcl.AsyncInterfaces)のパッケージとして提供されます。
これは、GetConsumingEnumerable
新しいC#8構文を特徴とする代替実装です。
public async IAsyncEnumerable<T> GetConsumingEnumerable()
{
lock (_queue) _consumersCount++;
while (true)
{
lock (_queue)
{
if (_queue.Count == 0 && _isAddingCompleted) break;
}
await _semaphore.WaitAsync();
bool hasItem;
T item = default;
lock (_queue)
{
hasItem = _queue.Count > 0;
if (hasItem) item = _queue.Dequeue();
}
if (hasItem) yield return item;
}
}
同じ方法でawait
とyield
が共存していることに注意してください。
使用例(C#8):
var consumer = Task.Run(async () =>
{
await foreach (var item in abc.GetConsumingEnumerable())
{
await Task.Delay(200);
await Console.Out.WriteAsync(item + " ");
}
});
のawait
前に注意してくださいforeach
。
await Task.Run(() => blockingCollection.Take())
タスクを使用すると他のスレッドで実行され、UIスレッドがブロックされないため、これはわかりません。それがポイントではありませんか?