建议先阅读 阻塞通道 和 非阻塞通道 小节。在前面的两个小节中, 为了最小化代码达到演示效果,省略了 关闭通道
的步骤, 正确的做法应该是在通道使用完成后关闭。
通过关键字 clsoe
关闭通道。
这里的规则不必死记硬背,笔者遇到的大多数情况属于第二种,也就是 重复关闭一个通道
, 读者做到实际开发中遇到 关闭通道
的场景时,联系上下文,确认通道不会出现重复关闭的情况即可。
package main func main() { var ch chan bool close(ch) } // $ go run main.go // 输出如下 /** panic: close of nil channel ... ... exit status 2 */
package main func main() { ch := make(chan bool) close(ch) close(ch) // 重复关闭 } // $ go run main.go // 输出如下 /** panic: close of nil channel ... ... exit status 2 */
package main func main() { ch := make(chan bool) close(ch) println("channel closed") } // $ go run main.go // 输出如下 /** channel closed */
通道的方向分为 发送
和 接收
。默认情况下,通道是双向的 (同时发送和接收),但是可以通过标识符指明通道为单向 (只读或只写)。
变量 := make(chan 数据类型) # 例子 ch := make(chan string)
变量 := make(<-chan 数据类型) # 例子 ch := make(<-chan string)
变量 := make(chan<- 数据类型) # 例子 ch := make(chan<- string)
双向通道可以转换为单向通道,但是单向通道无法转换为双向通道。
package main // 参数是一个写入通道 func ping(pings chan<- string) { //<-pings // 错误: pings 通道只能写入 pings <- "hello world" } func pong(pings <-chan string, pongs chan<- string) { //pings <- "hello world" // 错误: pings 通道只能读取 //<-pongs // 错误: pongs 通道只能写入 msg := <-pings pongs <- msg } func main() { pings := make(chan string) pongs := make(chan string) done := make(chan bool) go ping(pings) go pong(pings, pongs) go func() { msg := <-pongs println(msg) done <- true }() <-done close(pings) close(pongs) close(done) } // $ go run main.go // 输出如下 /** hello world */
Go 语言没有提供函数或方法判断一个通道是否关闭。因此只能使用一个变通的办法:接收通道元素,根据返回的布尔值确定通道是否关闭。
package main func main() { ch := make(chan string) close(ch) if _, open := <-ch; !open { println("channel closed") } } // $ go run main.go // 输出如下 /** channel closed */
package main import "time" func main() { ch := make(chan string) go func(c <-chan string) { if _, open := <-c; !open { println("channel closed") } }(ch) close(ch) time.Sleep(time.Second) } // $ go run main.go // 输出如下 /** channel closed */
对于只写通道,需要采用一个折中的办法:
• 尝试向通道写入数据
• 如果写入成功,说明通道还未关闭
• 写入失败,发生 panic
, 这时可以利用 defer
在 recover
中输出原因
package main import "time" func main() { ch := make(chan string) go func(c chan<- string) { defer func() { if err := recover(); err != nil { // 捕获到 panic println("channel closed") } }() c <- "hello world" }(ch) close(ch) time.Sleep(time.Second) } // $ go run main.go // 输出如下,你的输出可能和这里的不一样 /** channel closed */