Go管道阻塞、常见用法以及底层实现
channel数据结构
type hchan struct {
qcount uint //当前元素个数(len)
dataqsiz uint //队列长度(cap)
buf unsafe.Pointer //队列指针,指向队列内存位置
elemsize uint16 //单个元素大小
closed uint32 //标识通道状态
elemtype *_type //元素类型
sendx uint //下一个元素存放的位置
recvx uint //下一个元素读取的位置
recvq waitq //等待读消息的协程队列
sendq waitq //等待发消息的协程队列
lock mutex //互斥锁,保证不存在并发读写管道
//可以在GOROOT/src/runtime/chan.go里看到源码
}
一、管道状态
通过closed字段可知我们可以对管道是否关闭进行判断:
var ch=make(chan int)
go func() {
for i := 0; i < 5; i++ {
ch<-i
}
close(ch)
}()
for {
if v, ok := <-ch; ok { //v,ok:=<-ch,只有当管道关闭且数据全部读出时才返回false
fmt.Println(v)
}else {
break
}
}
二、阻塞
2.1 什么时候发生阻塞?
- 向一个值为nil的管道写或读数据
- 无缓冲区时单独的写或读数据
- 缓冲区为空时进行读数据
- 缓冲区满时进行写数据
2.2 阻塞的本质?
每次发生阻塞管道都会将发生阻塞所在的协程送入recvq或sendq队列
var ch=make(chan int,1)
for i:=0;i<3;i++ {
//每次循环开启协程Gi
go func(i int) {
ch<-i
}(i) //此时sendq=队列:[G1,G2]
}
<-ch //此时sendq队列:[G2]
<-ch //此时sendq队列:[]
recvq队列同理
另外main函数也可以视为一个协程,当main函数阻塞时如果没有其他协程还在运作,就会发生死锁
三、常见错误
- 关闭值为nil的管道
- 关闭已经关闭的管道
- 往已经close的管道发送数据
四、管道常见用法
4.1 单向管道
//只读管道
func readChan(ch chan int) <-chan int {
return ch
}
//只写管道
func writeChan(ch chan int) chan<- int {
return ch
}
可以增强数据安全性
4.2 select
var ch1 = make(chan int)
var ch2 = make(chan int)
select {
case <-ch1:
fmt.Println(1)
case <-ch2:
fmt.Println(2)
default:
fmt.Println(3)
}
select可以监控多个管道,当某个管道可操作时就会触发相应的case分支。
4.3 for-range
for v:=range ch{
Do(...)
}
完全等同于:
for {
if v,ok:=<-ch;!ok{ //当close(ch)且ch内数据全部读出时,ok==false
break
}else{
Do(...)
}
}
由此可见如果不主动close掉ch,读出ch全部数据后该协程就会阻塞(<-ch)。如果该for-range写在main函数内,就会有死锁的危险。