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 什么时候发生阻塞？

1. 向一个值为nil的管道写或读数据
2. 无缓冲区时单独的写或读数据
3. 缓冲区为空时进行读数据
4. 缓冲区满时进行写数据

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函数阻塞时如果没有其他协程还在运作，就会发生死锁

三、常见错误

1. 关闭值为nil的管道
2. 关闭已经关闭的管道
3. 往已经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函数内，就会有死锁的危险。