如何解决当枚举数集合未转换为列表时,IAsyncEnumerator.Current 返回 null
第一个函数旨在使 linq 能够安全地并行执行 lambda 函数(即使是异步无效函数)。
所以你可以做 collection.AsParallel().ForAllASync(async x => await x.Action).
第二个函数旨在使您能够并行组合和执行多个 IAsyncEnumerables,并尽快返回它们的结果。
public static async Task ForAllAsync<TSource>(
this ParallelQuery<TSource> source,Func<TSource,Task> selector,int? maxDegreeOfParallelism = null)
{
int maxAsyncThreadCount = maxDegreeOfParallelism ?? Math.Min(System.Environment.ProcessorCount,128);
using SemaphoreSlim throttler = new SemaphoreSlim(maxAsyncThreadCount,maxAsyncThreadCount);
IEnumerable<Task> tasks = source.Select(async input =>
{
await throttler.WaitAsync().ConfigureAwait(false);
try
{
await selector(input).ConfigureAwait(false);
}
finally
{
throttler.Release();
}
});
await Task.WhenAll(tasks).ConfigureAwait(true);
}
public static async IAsyncEnumerable<T> ForAllAsync<TSource,T>(
this ParallelQuery<TSource> source,IAsyncEnumerable<T>> selector,int? maxDegreeOfParallelism = null,[EnumeratorCancellation]CancellationToken cancellationToken = default)
where T : new()
{
IEnumerable<(IAsyncEnumerator<T>,bool)> enumerators =
source.Select(x => (selector.Invoke(x).GetAsyncEnumerator(cancellationToken),true)).ToList();
while (enumerators.Any())
{
await enumerators.AsParallel()
.ForAllAsync(async e => e.Item2 = (await e.Item1.MoveNextAsync()),maxDegreeOfParallelism)
.ConfigureAwait(false);
foreach (var enumerator in enumerators)
{
yield return enumerator.Item1.Current;
}
enumerators = enumerators.Where(e => e.Item2);
}
}
如果我从第二个函数中删除“ToList()”,yield return 开始返回 null,因为 enumerator.Item1.Current 往往为 null,尽管 enumerator.Item2(MoveNextAsync() 的结果)为真。
为什么?
解决方法
这是一个延迟执行的经典案例。每次在非物化的 IEnumerable<>
上调用评估方法时,它都会执行物化 IEnumerable 的工作。在这种情况下,它会重新调用您的选择器并创建等待 GetAsyncEnumerator 调用的任务的新实例。
通过调用 .ToList()
,您实现了 IEnumerable。没有它,每次调用 .Any()
、调用 ForAllAsync()
以及在您的 foreach
循环中都会发生具体化。
可以像这样最低限度地重现相同的行为:
var enumerable = new[] { 1 }.Select(_ => Task.Delay(10));
await Task.WhenAll(enumerable);
Console.WriteLine(enumerable.First().IsCompleted); // False
enumerable = enumerable.ToList();
await Task.WhenAll(enumerable);
Console.WriteLine(enumerable.First().IsCompleted); // True
在第一次调用 enumerable.First()
时,我们最终得到的任务实例与我们在它之前的行中等待的任务实例不同。
在第二次调用中,我们使用了相同的实例,因为 Task 已经具体化到一个列表中。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。