在某些场景下我们需要同时从多个通道接收数据,通道在没有数据接收的时候就会发生阻塞,Go语言内置了select关键字,可以同时响应多个通道的操作,通过select可以监听多个通道上的数据流动,select与c/java中的switch分支语句的用法类似,由select开始一个新的选择块,每个选择条件由case语句来描述,所以包含若干个case分支和最后的default默认分支,每个case会对应一个通道的接收或发送的通信过程,如果其中的某些语句可以继续执行(即没有被阻塞),那么就从那些可以执行的语句中任意选择一条来执行,如果没有一条case语句可以执行判断是否有default语句,如果有default语句则执行default语句,否则执行,没有default语句那么select语句会被阻塞直到至少有一个通信可以执行下去:
package main
import (
"fmt"
"time"
)
func print1(c chan int, n int) {
c <- n
}
func print2(c chan int, n int) {
c <- n
}
func main() {
c1 := make(chan int)
c2 := make(chan int)
// 由于使用了go关键字所以会并发执行print1和print2函数
go print1(c1, 12)
go print2(c2, 23)
// 如果有多个case语句满足要求那么选择其中一条执行
select {
case t1 := <-c1:
fmt.Println(t1)
case t2 := <-c2:
fmt.Println(t2)
}
time.Sleep(time.Second * 5)
}
使用select语句可以判断通道是否为空,下面的程序需要向通道output写入和读取数据,由于需要并发执行向output写入和读取数据的操作,在main()函数中可以使用go关键字开启一个goroutine向通道output写入数据,可以使用一个死循环来执行select语句,这样可以多次执行select语句向通道output写入数据,在main()函数中则使用for-range循环从通道output中取出数据,虽然在write()函数中使用了死循环但是由于使用go关键字开启的另外一个goroutine,所以write()函数所在的协程与main()函数的主协程是并发执行的,由go语言的并发机制判断先执行哪个goroutine,也两个goroutine是并发执行的,两个goroutine会一直执行下去:
package main
import (
"fmt"
"time"
)
func main() {
// 创建通道
output := make(chan string, 10)
// 子协程写数据
go write(output)
// 取数据
for s := range output {
fmt.Println("res: ", s)
time.Sleep(time.Second)
}
}
func write(ch chan string) {
for {
select {
// 写数据
case ch <- "hello":
fmt.Println("write hello")
default:
fmt.Println("channel full")
}
time.Sleep(time.Second)
}
}