脚本之家,脚本语言编程技术及教程分享平台!
分类导航

Python|VBS|Ruby|Lua|perl|VBA|Golang|PowerShell|Erlang|autoit|Dos|bat|

服务器之家 - 脚本之家 - Golang - 聊聊 Go 流水线编程模式

聊聊 Go 流水线编程模式

2021-11-08 23:06Golang技术分享机器铃砍菜刀 Golang

流水线工作模型在工业领域内十分常见,它将工作流程分为多个环节,每个环节根据工作强度安排合适的人员数量。良好的流水线设计尽量让各环节的流通率平衡,最大化提高产能效率。

聊聊 Go 流水线编程模式

流水线工作模型在工业领域内十分常见,它将工作流程分为多个环节,每个环节根据工作强度安排合适的人员数量。良好的流水线设计尽量让各环节的流通率平衡,最大化提高产能效率。

Go 是一门实用性语言,流水线工作模型与 Go 融合地非常融洽,只不过我们一般使用另一个名词来表示流水线:pipeline。

pipeline

pipeline 由多个环节组成,具体在 Go 中,环节之间通过 channel 通信,同一个环节任务可以由多个 goroutine 来同时处理。

聊聊 Go 流水线编程模式

pipeline

pipeline 的核心是数据,通过 channel 来保证数据流动,每个环节的数据处理由 goroutine 完成。

除了开始环节和结束环节,每个环节都有任意数量的输入 channel 和输出 channel。开始环节被称为发送者或生产者,结束环节被称为接收者或消费者。

下面我们来看一个简单的 pipeline 例子,分为三个环节。

第一个环节,generate 函数:它充当生产者角色,将数据写入 channel,并把该 channel 返回。当所有数据写入完毕,关闭 channel。

  1. funcgenerate(nums...int)<-chanint{
  2. out:=make(chanint)
  3. gofunc(){
  4. for_,n:=rangenums{
  5. out<-n
  6. }
  7. close(out)
  8. }()
  9. returnout
  10. }

第二个环节,square 函数:它是数据处理的角色,从开始环节中的 channel 取出数据,计算平方,将结果写入新的 channel ,并把该新的 channel 返回。当所有数据计算完毕,关闭该新 channel。

  1. funcsquare(in<-chanint)<-chanint{
  2. out:=make(chanint)
  3. gofunc(){
  4. forn:=rangein{
  5. out<-n*n
  6. }
  7. close(out)
  8. }()
  9. returnout
  10. }

main 函数负责编排整个 pipeline ,并充当消费者角色:读取第二个环节的 channel 数据,打印出来。

  1. funcmain(){
  2. //Setupthepipeline.
  3. c:=generate(2,3)
  4. out:=square(c)
  5. //Consumetheoutput.
  6. forn:=rangeout{
  7. fmt.Println(n)
  8. }
  9. }

Fan-out,fan-in

在上述例子中,环节之间通过非缓冲的 channel 传递数据,节点中的数据都是单个 goroutine 处理与消费。

这种工作模式并不高效,会让整个流水线的效率取决于最慢的环节。因为每个环节中的任务量是不同的,这意味着我们需要的机器资源是存在差异的。任务量小的环节,尽量占有少量的机器资源,任务量重的环节,需要更多线程并行处理。

以汽车组装为例,我们可以将组装轮胎的工作分发给 4 个人一起干,当轮胎组装完毕之后,再交由剩下的环节。

多个 goroutine 可以从同一个 channel 读取数据,直到该通道关闭,这称为 fan-out(扇出)。

这个称呼比较形象,它将数据进行分散,所以被称为扇出。扇出是一种分发任务的模式。

聊聊 Go 流水线编程模式

fan-out

单个 goroutine 可以从多个输入 channel 中读取数据,直到所有输入都关闭。具体做法是将输入 channel 多路复用到同一个 channel 上,当所有输入 channel 都关闭时,该 channel 也关闭,这称为 fan-in(扇入)。

它将数据进行聚合,所以被称为扇入。扇入是一种整合任务结果的模式。

聊聊 Go 流水线编程模式

fan-in

在汽车组装的例子中,分发轮胎任务给每个人是 Fan-out,合并轮胎组装结果就是 Fan-in。

channel 的多路复用

扇出的编码模型比较简单,本文不多研究,我们提供一个扇入编程示例。

创建一个生成器函数 generate,通过 interval 参数控制消息生成频率。生成器返回消息 channel mc与停止 channel sc,停止 channel 用于停止生成器任务。

  1. funcgenerate(messagestring,intervaltime.Duration)(chanstring,chanstruct{}){
  2. mc:=make(chanstring)
  3. sc:=make(chanstruct{})
  4. gofunc(){
  5. deferfunc(){
  6. close(sc)
  7. }()
  8. for{
  9. select{
  10. case<-sc:
  11. return
  12. default:
  13. time.Sleep(interval)
  14. mc<-message
  15. }
  16. }
  17. }()
  18. returnmc,sc
  19. }

stopGenerating 函数通过通过向 sc 中传入空结构体,通知 generate退出,调用 close(mc) 关闭消息 channel

  1. funcstopGenerating(mcchanstring,scchanstruct{}){
  2. sc<-struct{}{}
  3. close(mc)
  4. }

多路复用函数 multiplex 创建并返回整合消息 channel 和控制并发的 wg。

  1. funcmultiplex(mcs...chanstring)(chanstring,*sync.WaitGroup){
  2. mmc:=make(chanstring)
  3. wg:=&sync.WaitGroup{}
  4. for_,mc:=rangemcs{
  5. wg.Add(1)
  6. gofunc(mcchanstring,wg*sync.WaitGroup){
  7. deferwg.Done()
  8. form:=rangemc{
  9. mmc<-m
  10. }
  11. }(mc,wg)
  12. }
  13. returnmmc,wg
  14. }

在 main 函数中,创建两个消息 channel 并复用它们生成 mmc ,打印来自 mmc 的每条消息。另外,我们还实现了接收系统断信号(终端上执行 CTRL+C 即可发送中断信号)的优雅的关闭机制。

  1. funcmain(){
  2. //createtwosamplemessageandstopchannels
  3. mc1,sc1:=generate("messagefromgenerator1",200*time.Millisecond)
  4. mc2,sc2:=generate("messagefromgenerator2",300*time.Millisecond)
  5. //multiplexmessagechannels
  6. mmc,wg1:=multiplex(mc1,mc2)
  7. //createerrschannelforgracefulshutdown
  8. errs:=make(chanerror)
  9. //waitforinterruptorterminatesignal
  10. gofunc(){
  11. sc:=make(chanos.Signal,1)
  12. signal.Notify(sc,syscall.SIGINT,syscall.SIGTERM)
  13. errs<-fmt.Errorf("%ssignalreceived",<-sc)
  14. }()
  15. //waitformultiplexedmessages
  16. wg2:=&sync.WaitGroup{}
  17. wg2.Add(1)
  18. gofunc(){
  19. deferwg2.Done()
  20. form:=rangemmc{
  21. fmt.Println(m)
  22. }
  23. }()
  24. //waitforerrors
  25. iferr:=<-errs;err!=nil{
  26. fmt.Println(err.Error())
  27. }
  28. //stopgenerators
  29. stopGenerating(mc1,sc1)
  30. stopGenerating(mc2,sc2)
  31. wg1.Wait()
  32. //closemultiplexedmessageschannel
  33. close(mmc)
  34. wg2.Wait()
  35. }

总结

本文简单介绍了流水线编程模式,它和我们熟悉的生产者-消费者模式非常相似。

具体到 Go 编程实践中,pipeline 将数据流分为多个环节,channel 用于数据流动,goroutine 用于处理数据。fan-out 用于分发任务,fan-in 用于数据整合,通过 FAN 模式可以让流水线更好地并发。

当然,还有些细节需要注意,例如停止通知机制,可参照本文 channel 的多路复用章节示例中的 stopGenerating 函数;如何通过 sync.WaitGroup 做好并发控制,这些都是需要读者在实际编码中去体会掌握的。

参考

Go Concurrency Patterns: Pipelines and cancellation:https://go.dev/blog/pipelines

Multiplexing Channels In Go:https://medium.com/@ermanimer/multiplexing-channels-in-go-a7dccdcc4134

原文链接:https://mp.weixin.qq.com/s/IuGaKbQvW7z1KsoMhB8mgg

延伸 · 阅读

精彩推荐
  • Golanggolang如何使用struct的tag属性的详细介绍

    golang如何使用struct的tag属性的详细介绍

    这篇文章主要介绍了golang如何使用struct的tag属性的详细介绍,从例子说起,小编觉得挺不错的,现在分享给大家,也给大家做个参考。一起跟随小编过来看...

    Go语言中文网11352020-05-21
  • Golanggo日志系统logrus显示文件和行号的操作

    go日志系统logrus显示文件和行号的操作

    这篇文章主要介绍了go日志系统logrus显示文件和行号的操作,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...

    SmallQinYan12302021-02-02
  • Golanggolang 通过ssh代理连接mysql的操作

    golang 通过ssh代理连接mysql的操作

    这篇文章主要介绍了golang 通过ssh代理连接mysql的操作,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...

    a165861639710342021-03-08
  • GolangGolang通脉之数据类型详情

    Golang通脉之数据类型详情

    这篇文章主要介绍了Golang通脉之数据类型,在编程语言中标识符就是定义的具有某种意义的词,比如变量名、常量名、函数名等等,Go语言中标识符允许由...

    4272021-11-24
  • Golanggolang json.Marshal 特殊html字符被转义的解决方法

    golang json.Marshal 特殊html字符被转义的解决方法

    今天小编就为大家分享一篇golang json.Marshal 特殊html字符被转义的解决方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧 ...

    李浩的life12792020-05-27
  • GolangGolang中Bit数组的实现方式

    Golang中Bit数组的实现方式

    这篇文章主要介绍了Golang中Bit数组的实现方式,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...

    天易独尊11682021-06-09
  • Golanggo语言制作端口扫描器

    go语言制作端口扫描器

    本文给大家分享的是使用go语言编写的TCP端口扫描器,可以选择IP范围,扫描的端口,以及多线程,有需要的小伙伴可以参考下。 ...

    脚本之家3642020-04-25
  • Golanggolang的httpserver优雅重启方法详解

    golang的httpserver优雅重启方法详解

    这篇文章主要给大家介绍了关于golang的httpserver优雅重启的相关资料,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,...

    helight2992020-05-14