golang的Channel
Channel 是 golang 一个非常重要的概念,如果你是刚开始使用 golang 的开发者,你可能还没有真正接触这一概念,本篇我们将分析 golang 的Channel
1. 引入
要讲 Channel 就避不开 Goroutine -- 协程。闲话不说, 直接上个例子
Goroutine demo:
package main
import (
"fmt"
"time"
)
func main() {
origin := 1
go doSomeCompute(origin)
time.Sleep(5 * time.Second)
}
func doSomeCompute(num int) {
result := num * 2
fmt.Println(result)
return
}
简单来说,例子中有一个 main 的协程,一个 doSomeCompute 的协程。还有个延时退出的方法等待计算协程计算结果。我们尝试思考这个例子的一些问题:
a. 如何获取 doSomeCompute 的计算结果?
b. 如何获取 doSomeCompute 的执行状态,当前是执行完了还是执行中?
如何解决这种问题呢?
Channel!
2. Channel
Channel怎么处理上面的问题?我们直接上代码:
举例
package main
import (
"fmt"
)
func main() {
origin := 1
//一个无缓冲Channel
res := make(chan int)
go doSomeCompute(origin, res)
fmt.Println(<-res)
}
func doSomeCompute(num int, res chan int) {
result := num * 2
res <- result
}
例子中,Channel 充当起了协程间通信的桥梁。Channel 可以传递到协程说明它是线程安全的,事实也是如此。Channel 可以理解为管道,协程 doSomeCompute 向 Channel 写入结果, main 中读取结果。注意,例子中读取 Channel 的地方会阻塞直到拿到计算结果,这样就解决了问题 a 和 b。
2. Channel 的方向性
上面的例子中,计算协程是负责计算并将计算结果写入 Channel ,如果我们希望保证计算协程不会从 Channel 中读取数据该怎么处理?很简单,看例子:
func doSomeCompute(num int, res chan<- int) {
result := num * 2
res <- result
}
这个参数的声明 chan<- int
就表示该函数只能讲数据写入 Channel ,而不能从中读取数据。后面的 int
表示 Channel 中数据的格式。同样的, 只可以读取数据的 Channel 可以声明为 <-chan int
。而例子中不带有方向声明的 Channel 则既可以写入也可以读取。
3. 阻塞性质
Channel 的读取和写入操作在各自的协程内部都是阻塞的。比如例子中 fmt.Println(<-res)
, 这一语句会阻塞直至计算协程将计算结果放入,可以读出。也就是说,协程会阻塞直至从 res
中读出数据。
注意,无缓冲的 Channel 的读写都是阻塞的,有缓冲的 Channel 可以一直向里面写数据,直到缓存满才会阻塞。读取数据同理, 直至 Channel 为空才阻塞。
用一个典型的例子来说明缓冲和非缓冲 Channel 的区别:
package main
import "fmt"
func doSomeCompute(ch chan int) {
fmt.Println("deadlock test")
<- ch
}
func main() {
ch := make(chan int)
ch <- 1
go doSomeCompute(ch)
}
例子中,main 协程会向 ch
写入数据, 这一过程是阻塞的,也就是说,doSomeCompute
协程无法执行,程序死锁。输出如下:
fatal error: all goroutines are asleep - deadlock!
goroutine 1 [chan send]:
main.main()
C:/mygo/src/demo/blog.go:12 +0x73
exit status 2
如果改成有缓冲的 Channel :
package main
import (
"fmt"
"time"
)
func doSomeCompute(ch chan int) {
fmt.Println("deadlock test")
<- ch
}
func main() {
ch := make(chan int, 1)
ch <- 1
go doSomeCompute(ch)
time.Sleep(1 * time.Second)
}
有与有缓冲的 Channel 写入后不阻塞(下一次写入才会阻塞),程序会继续执行。
4. Channel 的数据结构
Channel 在 golang 中实际上就是个数据结构。在 Golang 源码中, Channel 的数据结构 Hchan 的定义如下:
struct Hchan
{
uint32 qcount; // total data in the q
uint32 dataqsiz; // size of the circular q
uint16 elemsize;
bool closed;
uint8 elemalign;
Alg* elemalg; // interface for element type
uint32 sendx; // send index
uint32 recvx; // receive index
WaitQ recvq; // list of recv waiters
WaitQ sendq; // list of send waiters
Lock;
};
时间仓促,这次对 Channel 介绍写的有点简单粗暴,下次再写。