文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

一文说通异步 LINQ

2024-12-02 21:17

关注

LINQ 这个东西,出来很早了,写过几年代码的兄弟们,或多或少都用过一些。

早期的 LINQ,主要是同步的,直到 C# 8.0 加入 IAsyncEnumerable,LINQ 才真正转向异步。这本来是个非常好的改变,配合 System.Linq.Async 库提供的扩展,可以在诸如 Where、Select、GroupBy 等各种地方用到异步。

但事实上,在我 Review 代码时,见了很多人的代码,并没有按异步的规则去使用,出现了很多的坑。

举个简单的例子:

  1. static async TaskWhere(this IAsyncEnumerable source, Func predicate) 
  2.     var filteredItems = new List(); 
  3.     await foreach (var item in source) 
  4.     { 
  5.         if (predicate(item)) 
  6.         { 
  7.             filteredItems.Add(item); 
  8.         } 
  9.     } 
  10.  
  11.     return filteredItems; 

这样的写法,看着是用到了 async / await 对,但实际上并没有实现异步,程序依然是按照同步在运行。换句话说,这只是一个样子上的异步,实际没有任何延迟执行的效果。

1. 延迟执行

其实,这儿正确的写法也挺简单,用到的就是个异步的迭代器(关于异步迭代器,如果需要了解,可以看我的另一篇推文):

  1. static async IAsyncEnumerable Where(this IAsyncEnumerable source, Func predicate) 
  2.     await foreach (var item in source) 
  3.     { 
  4.         if (predicate(item)) 
  5.         { 
  6.             yield return item; 
  7.         } 
  8.     } 

这种写法下,编译器会将方法转了状态机,并在实际调用时,才通过枚举器返回异步枚举项。

看看调用过程:

  1. IAsyncEnumerable<User> users = ... 
  2. IAsyncEnumerable<User> filteredUsers = users.Where(User => User.Name == "WangPlus"); 
  3.  
  4. await foreach (User user in filteredUsers) 
  5.     Console.WriteLine(user.Age); 

在这个调用的例子中,在 Where 时,实际方法并不会马上开始。只有在下面 foreach 时,才真正开始执行 Where 方法。

延迟执行,这是异步 LINQ 的第一个优势。

2. 流执行

流执行,依托的也是异步迭代器。

所谓流执行,其实就是根据调用的要求,一次返回一个对象。通过使用异步迭代器,可以不用一次返回所有的对象,而是一个一个地返回单个的对象,直到枚举完所有的对象。

流执行需要做个技巧性的代码,需要用到一个 C# 8.0 的新特性:局部方法。

看代码:

  1. static IAsyncEnumerable Where(this IAsyncEnumerable source, Func predicate) 
  2.     return Core(); 
  3.  
  4.     async IAsyncEnumerable Core() 
  5.     { 
  6.         await foreach (var item in source) 
  7.         { 
  8.             if (predicate(item)) 
  9.             { 
  10.                 yield return item; 
  11.             } 
  12.         } 
  13.     } 

3. 取消异步 LINQ

前面两个小节,写的是异步 LINQ 的执行。

通常使用异步 LINQ 的原因,就是因为执行时间长,一般需要一段时间来完成。因此,取消异步 LINQ 就很重要。想象一下,一个长的 DB 查询已经超时了的情况,该怎么处理?

为了支持取消,IAsyncEnumerable.GetEnumerator 本身接受一个 CancellationToken 参数来中止任务,并用一个扩展方法挂接到 foreach 调用:

  1. CancellationToken cancellationToken = ... 
  2. IAsyncEnumerable<User> users = ... 
  3. IAsyncEnumerable<User> filteredUsers = users.Where(User => User.Name == "WangPlus"); 
  4.  
  5. await foreach (var User in filteredUsers.WithCancellation(cancellationToken)) 
  6.     Console.WriteLine(User.Age); 

同时,在上面的 Where 定义中,也要响应 CancellationToken 参数:

  1. static IAsyncEnumerable Where(this IAsyncEnumerable source, Func predicate) 
  2.     return Core(); 
  3.  
  4.     async IAsyncEnumerable Core([EnumeratorCancellation] CancellationToken cancellationToken = default
  5.     { 
  6.         await foreach (var item in source.WithCancellation(cancellationToken)) 
  7.         { 
  8.             if (predicate(item)) 
  9.             { 
  10.                 yield return item; 
  11.             } 
  12.         } 
  13.     } 

多解释一下:在 Where 方法中,CancellationToken 只能加到局部函数 Core 中,一个简单的原因是 Where 本身并不是异步方法,而且,我们也不希望从 Where 往里传递。想象一下:

  1. Users.Where(xxx, cancellationToken).Select(xxx, cancellationToken).OrderBy(xxx, cancellationToken); 

这样的代码会让人晕死。

所以,我们会采用上面的方式,允许消费者在枚举数据时传递 CancellationToken 来达到取消异步操作的目的。

4. 处理ConfigureAwait(false)

这是另一个异步必须要注意的部分,其实就是上下文。

通常大多数的方法,我们不需要关注上下文,但总有一些需要,在等待的异步操作恢复后,需要返回到某个上下文的情况。这种情况在 UI 线程编码时通常都需要考虑。很多人提到的异步死锁,就是这个原因。

处理也很简单:

  1. static IAsyncEnumerable Where(this IAsyncEnumerable source, Func predicate) 
  2.     return Core(); 
  3.  
  4.     async IAsyncEnumerable Core([EnumeratorCancellation] CancellationToken cancellationToken = default
  5.     { 
  6.         await foreach (var item in source.WithCancellation(cancellationToken).ConfigureAwait(false)) 
  7.         { 
  8.             if (predicate(item)) 
  9.             { 
  10.                 yield return item; 
  11.             } 
  12.         } 
  13.     } 

这儿也多说两句:按微软的说法,await foreach 本身是基于模式的,WithCancellation 和 ConfigureAwait 返回同样的结构体 ConfiguredCancelableAsyncEnumerable。这个结构体没有实现 IAsyncEnumerable 接口,而是做了一个 GetAsyncEnumerator 方法,返回一个具有 MoveNextAsync、Current、DisposeAsync 的枚举器,因此可以 await foreach 。

5. 方法扩展

上面 4 个小节,我们完成了一个 Where 异步 LINQ 的全部内容。

不过,这个方法有一些限制和不足。熟悉异步的兄弟们应该已经看出来了,里面用了一个委托 predicate 来做数据过滤,而这个委托,是个同步的方法。

事实上,根据微软对异步 LINQ 的约定,每个操作符应该是三种重载:

参考微软的异步方法,基本上都是以这种结构来命名方法名称的。

下面,我们也按这个方式,来做一个 Where 方法的几个重载。

WhereAwait 方法

上面说了,这会是一个异步实现。所以,条件部分就不能用 Func

代码是这样:

  1. static IAsyncEnumerable WhereAwait(this IAsyncEnumerable source, Func> predicate) 
  2.     return Core(); 
  3.  
  4.     async IAsyncEnumerable Core([EnumeratorCancellation] CancellationToken cancellationToken = default
  5.     { 
  6.         await foreach (var item in source.WithCancellation(cancellationToken).ConfigureAwait(false)) 
  7.         { 
  8.             if (await predicate(item).ConfigureAwait(false)) 
  9.             { 
  10.                 yield return item; 
  11.             } 
  12.         } 
  13.     } 

调用时是这样:

  1. IAsyncEnumerable<User> filteredUsers = users.WhereAwait(async user => await someIfFunction()); 

在上面的基础上,又加了一个取消操作。

看代码:

  1. static IAsyncEnumerable WhereAwaitWithCancellation(this IAsyncEnumerable source, Func> predicate) 
  2.     return Core(); 
  3.  
  4.     async IAsyncEnumerable Core([EnumeratorCancellation] CancellationToken cancellationToken = default
  5.     { 
  6.         await foreach (var item in source.WithCancellation(cancellationToken).ConfigureAwait(false)) 
  7.         { 
  8.             if (await predicate(item, cancellationToken).ConfigureAwait(false)) 
  9.             { 
  10.                 yield return item; 
  11.             } 
  12.         } 
  13.     } 

调用时是这样:

IAsyncEnumerable filteredUsers = users.WhereAwaitWithCancellation(async (user, token) => await someIfFunction(user, token));

6. 总结

异步 LINQ,多数是在 LINQ 的扩展方法中使用,而不是我们通常习惯的 LINQ 直写。

 

事实上,异步 LINQ 的扩展,对 LINQ 本身是有比较大的强化作用的,不管从性能,还是可读性上,用多了,只会更爽。

 

来源:老王Plus内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯