Go语言基础 — Channel
Golang Basic — Channel
基础
为什么需要Channel
在介绍goroutine的时候有提到过多个goroutine会产生竞争关系,如果使用不得当,轻则无法正确更新变量,重则挂掉程序。我们可以通过原子函数和互斥锁来解决竞争关系,但是还有一种更有趣的方法,那就是使用Channel。
当一个资源需要在goroutine之间共享时,Channel在goroutine之间架起了一个管道,并提供了 确保同步交换数据的机制。
定义Channel
从Channel里收发数据
无缓冲的Channel
无缓冲的Channel必须有接受者,如果没有接受者(worker)就会报错。
注意在把Channel当做函数参数使用的时候,可以定义Channel的数据走向chan<- int
,这样Channel就只能接受数据。
有缓冲的Channel
有缓冲的Channel就算没有接受者,只要没有超过预先设置的缓冲,就能保存传入的值。
如果不加worker,程序会报错,但是如果把c <- 'd'
也注释掉,程序时不会报错的。
关闭Channel
在传完4个值之后关闭了Channel,但是worker还在无限循环并且往外传Channel。但是由于已经没有值了,在1 毫秒的时间内,会生成下面的结果。
如何对应这个问题可以有2种写法。
写法1
写法2
如何等待Goroutine
在之前的代码里面都有使用time.Sleep(time.Millisecond)
来等待Channel的结束。但是这种方法的危险肉眼可见。所以我们需要修改一下。
修改点:
- 在执行打印的时候,不仅接受一个Channel的int,同时在打印完成后,对完传出一个done
- 由于有一对chan,将这一对chan提出来做一个struct
- 在Createworker的时候用worker构造体传值
- 同时在创建10个worker的时候也用构造体创建构造体的slice
- 在向内部传值后,同时加入接受done的处理
这个程序跑出来的结果如下:
可以看到程序并没有并发,而是传进去一条,然后等待done,然后再传下一条。如果并发效果,使用goroutine就没有了意义。所以这里需要继续改:
把<-worker.done也单独拿出来接收,但是这样其实是会报错的。程序只会打印出第一段小写的for,原因是所有done的接受都放在了最后,而Channel的是传值是阻塞式的,第一段for的Channel接受完值,但是没有收到done,而第二段done却已经开始执行要传值了,导致报错。这里有一种非常tricky的改法,就是把doworker函数里的done改为goroutine。
最后自己管理goroutine的结束实在太累了,所以go语言提供了一种叫做sync.WaitGroup
的语法糖。
WaitGroup的用法:
- 定义WaitGroup本身:
var wg sync.WaitGroup
- 定义需要等待的goroutine数量:
wg.Add(20)
- 等待:
wg.Wait()
- 在goroutine执行完成后done:
w.done()
在这个程序里还有一些需要说的点:
- WaitGroup作为参数的时候一定要用指针
- 这里利用函数式编程重构了一下worker构造,把done设置为函数,createworker的时候定义函数同时传入dowork里。
应用
Go语言有句名言:
Don’t communicate by sharing memory, share memory by communicating.
比如我们判断前后程序是否处理是否完成的时候经常设置一个flag变量,然后通过检测这个变量来进行判断,这就是所谓通过共享内存进行通信。但是go语言更提倡通过使用Channel,通过通信来共享内存。
其实说说直接一点就是作者是推荐使用Channel而不是Mutex的。但是还是要分场景,详细可以参照这篇文章。
利用Select
假如你需要从多个Channel接受值,但是你不知道哪一个Channel会先发数据给你。那你可以使用Select语句来完成此类需求。
执行结果:
假如我想把c1和c2的值传给之前写的worker。可以这样写。
但是这样写有个问题,由于generator需要等待,所以在一开始的时候,会走很多遍case w <- n
同时输出0。所以我们在传给worker之前,需要检查n是否为空。
这样就对了,这里有个知识点是nil chan永远不会被select到。用var activeworker chan<- int
定义的chan是nil,所以在最后的case activeworker <- n
的时候可以用来作为判断条件。
这里还有一个点需要考虑,现在我们的chan是从c1或者c2传进worker,但是中途经过int变量n,如果worker处理的速度过慢,而c1,c2传入过快,会导致n被覆盖从而无法传入全部的值到worker之中。
个人还想过为什么不直接把c1,c2传给worker,但是worker是个Channel类型,c1,c2也是Channel类型,所以不能直接
worker<-c1
这个时候我们需要把c1,c2传进来的数据储存起来。