Go Select使用
Go Select使用
Go中的select和channel配合使用,通过select可以监听多个channel的I/O读写事件,当 IO操作发生时,触发相应的动作。
基本用法
//select基本用法 select { case <- chan1: // 如果chan1成功读到数据,则进行该case处理语句 case chan2 <- 1: // 如果成功向chan2写入数据,则进行该case处理语句 default: // 如果上面都没有成功,则进入default处理流程
使用规则
1.如果没有default分支,select会阻塞在多个channel上,对多个channel的读/写事件进行监控。 2.如果有一个或多个IO操作可以完成,则Go运行时系统会随机的选择一个执行,否则的话,如果有default分支,则执行default分支语句,如果连default都没有,则select语句会一直阻塞,直到至少有一个IO操作可以进行。
快速返回
同时监听不同的channel,做同一件工作,可以最快的返回结果。
package main import ( "fmt" "github.com/kirinlabs/HttpRequest" ) func main() { ch1 := make(chan int) ch2 := make(chan int) ch3 := make(chan int) go Getdata("https://www.baidu.com",ch1) go Getdata("https://www.baidu.com",ch2) go Getdata("https://www.baidu.com",ch3) select{ case v:=<- ch1: fmt.Println(v) case v:=<- ch2: fmt.Println(v) case v:=<- ch3: fmt.Println(v) } } func Getdata(url string,ch chan int){ req,err := HttpRequest.Get(url) if err != nil{ }else{ ch <- req.StatusCode() } }
随机返回
同时监控不同的channel,配上default,select也不会阻塞。
package main import ( "fmt" "github.com/kirinlabs/HttpRequest" ) func main() { ch1 := make(chan int) ch2 := make(chan int) ch3 := make(chan int) go func(){ for { Getdata("https://www.baidu.com", ch1) Getdata("https://cn.bing.com", ch2) Getdata("https://cn.bing.com", ch3) } }() go func(){ for { select { case v := <-ch1: fmt.Println("信道1的结果:",v) case v := <-ch2: fmt.Println("信道2的结果:",v) case v := <-ch3: fmt.Println("信道3的结果:",v) default: continue } } }() select{} } func Getdata(url string,ch chan int){ req,err := HttpRequest.Get(url) if err != nil{ }else{ ch <- req.StatusCode() } }
通过select来检测channel的关闭事件
func TestSelect1() { start := time.Now() c := make(chan interface{}) go func() { time.Sleep(2*time.Second) close(c) }() fmt.Println("Blocking on read...") select { case <-c: fmt.Printf("Unblocked %v later.\n", time.Since(start)) } }
注意:当close channel时,读取channel的一方会从channel中读取到value,false,此时的value一般情况下为nil。
该例子也可以用来通知当不使用channel时,关闭channel的情况。
通过channel通知,从而退出死循环
func TestExitLoop() { done := make(chan interface{}) go func() { time.Sleep(2*time.Second) close(done) }() workCounter := 0 loop: for { select { case <-done: break loop default: } // Simulate work workCounter++ time.Sleep(1*time.Second) } fmt.Printf("在通知退出循环时,执行了%d次.\n", workCounter) }
启动一个goroutine,该goroutine在2s后,关闭channel。此时,主协程会在select中的case <-done分支中得到通知,跳出死循环。而在此之前,会执行default分支的代码,这里是什么都不做。
超时机制
package main import ( "fmt" "time" ) func main() { ch := make(chan int) quit := make(chan bool) //新开一个协程 go func() { for { select { case num := <-ch: fmt.Println("num = ", num) case <-time.After(3 * time.Second): fmt.Println("超时") quit <- true } } }() for i := 0; i < 5; i++ { ch <- i time.Sleep(time.Second) } <-quit fmt.Println("程序结束") }
死锁与默认情况
package main func main() { ch := make(chan string) select { case <-ch: } }
上面的程序中,我们在第 4 行创建了一个信道 ch
。我们在 select
内部(第 6 行),试图读取信道 ch。由于没有 Go 协程向该信道写入数据,因此 select
语句会一直阻塞,导致死锁。该程序会触发运行时 panic
,报错信息如下:
fatal error: all goroutines are asleep - deadlock!
goroutine 1 [chan receive]:
main.main()
/tmp/sandbox416567824/main.go:6 +0x80
如果存在默认情况,就不会发生死锁,因为在没有其他 case 准备就绪时,会执行默认情况。我们用默认情况重写后,程序如下:
package main import "fmt" func main() { ch := make(chan string) select { case <-ch: default: fmt.Println("default case executed") } }
以上程序会输出:
default case executed
如果 select
只含有值为 nil
的信道,也同样会执行默认情况。
package main import "fmt" func main() { var ch chan string select { case v := <-ch: fmt.Println("received value", v) default: fmt.Println("default case executed") } }
在线运行程序
在上面程序中,ch
等于 nil
,而我们试图在 select
中读取 ch
(第 8 行)。如果没有默认情况,select
会一直阻塞,导致死锁。由于我们在 select
内部加入了默认情况,程序会执行它,并输出:
default case executed
空 select
package main func main() { select {} }
我们已经知道,除非有 case
执行,select
语句就会一直阻塞着。在这里,select
语句没有任何 case
,因此它会一直阻塞,导致死锁。该程序会触发 panic
,输出如下:
fatal error: all goroutines are asleep - deadlock!
goroutine 1 [select (no cases)]:
main.main()
/tmp/sandbox299546399/main.go:4 +0x20