C#,BlockingCollection:如何等到集合少于N个项目
||
大家。
我在传统的生产者-消费者方案中使用BlockingCollection。要一个接一个地处理集合中的项目,我必须编写以下代码:
while (...)
{
var item = collection.Take(cancellationTokenSource.Token);
ProcessItem(item);
}
但是如何处理一批N项(等到收集少于N项)?
我的解决方案是使用一些临时缓冲区:
var buffer = new List<MyType>(N);
while (...)
{
var item = collection.Take(cancellationTokenSource.Token);
buffer.Add(item);
if (buffer.Count == N)
{
foreach (var item in items)
{
ProcessItem(item);
}
buffer.Clear();
}
}
但是在我看来,这非常丑陋……有更好的方法吗?
[更新]:
这是扩展方法的原型,它使解决方案更具可读性。也许有人会发现它有用:
public static class BlockingCollectionExtensions
{
public static IEnumerable<T> TakeBuffer<T>(this BlockingCollection<T> collection,
CancellationToken cancellationToken, Int32 bufferSize)
{
var buffer = new List<T>(bufferSize);
while (buffer.Count < bufferSize)
{
try
{
buffer.Add(collection.Take(cancellationToken));
}
catch (OperationCanceledException)
{
// we need to handle the rest of buffer,
// even if the task has been cancelled.
break;
}
}
return buffer;
}
}
和用法:
foreach (var item in collection.TakeBuffer(cancellationTokenSource.Token, 5))
{
// TODO: process items here...
}
当然,这不是一个完整的解决方案:例如,我将添加任何超时支持-如果没有足够的项目,但是经过了一段时间,我们需要停止等待并处理已经添加到缓冲区的项目。
没有找到相关结果
已邀请:
3 个回复
碧肮械淌设
类中,但除此之外,我并没有真正看到这种方法的问题。
吐兄
痰降锭骂奸