我有两个异步序列,我想成对地“压缩”,为此,我使用了System.Linq.Async包中的System.Linq.Async运算符。但是,这个操作符的行为方式并不理想,至少在我的情况下是这样的。它不是同时枚举这两个序列,而是按顺序枚举它们,结果是将延迟加起来。我的每个序列平均每一秒发出一个元素,我预计组合的序列也会每一秒发出一次压缩对,但实际上我每2秒就有一对。下面是一个演示此行为的最小示例:
static async IAsyncEnumerable<int> First()
{
for (int i = 1; i <= 5; i++) { await Task.Delay(1000); yield return i; }
}
static async IAsyncEnumerable<int> Second()
{
for (int i = 1; i <= 5; i++) { await Task.Delay(1000); yield return i; }
}
var stopwatch = Stopwatch.StartNew();
await foreach (var pair in First().Zip(Second()))
Console.WriteLine(pair);
Console.WriteLine($"Duration: {stopwatch.ElapsedMilliseconds:#,0} msec");输出:
(1, 1)
(2, 2)
(3, 3)
(4, 4)
(5, 5)
Duration: 10,155 msec我是否可以用程序在5秒而不是10秒内完成的方式来Zip这两个序列?我对自定义操作符感兴趣,或者对官方包中的操作符组合感兴趣,这些操作符具有理想的行为。
发布于 2022-01-05 17:09:29
像这样的事情似乎奏效了:
public static async IAsyncEnumerable<(TFirst, TSecond)> Zip<TFirst, TSecond>(this IAsyncEnumerable<TFirst> first, IAsyncEnumerable<TSecond> second)
{
await using var e1 = first.GetAsyncEnumerator();
await using var e2 = second.GetAsyncEnumerator();
while (true)
{
var t1 = e1.MoveNextAsync().AsTask();
var t2 = e2.MoveNextAsync().AsTask();
await Task.WhenAll(t1, t2);
if (!t1.Result || !t2.Result)
yield break;
yield return (e1.Current, e2.Current);
}
}请在dotnetfiddle.net上看到。
当然,这忽略了空检查之类的内容,因此可以通过一些改进来完成:这是留给读者的摘录。
我也不认为Task.WhenAll比这里的bool r1 = await t1; bool r2 = await t2; if (!r1 || !r2) yield break;更好。
https://stackoverflow.com/questions/70596735
复制相似问题