Golang之select

"select for ?"

Posted by Keal on July 15, 2021

Select

select的使用方式和switch有点类似,作用和io多路复用里的select函数类似.

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
func fibonacci(c, quit chan int) {
	x, y := 0, 1
	for {
		if x > 100 {
			quit <- 1
		}
		select {
		case c <- x:
			println(x)
			x, y = y, x+y
		case <-quit:
			fmt.Println("quit")
			time.Sleep(time.Second)
			return
		}
	}
}

func main() {
	c := make(chan int, 100)
	quit := make(chan int, 10)
	fibonacci(c, quit)
}

// 结果
0
1
1
2
3
5
8
13
21
34
55
89
quit

select会随机顺序遍历(防止饥饿问题)case条件里channel是否有满足,然后执行.如果同时满足,会随机选择一个执行.

需要注意的是,如果go死锁检测发现进行select的goroutine永远不可能被唤醒,则会报错退出.

例如声明变量c的时候,如果不指明缓冲区大小又没有程序从c中读取,则永远不会触发case <- x这一段.而case <- quit如果没有往quit通道中写入的话,也不会被触发. 空的select也同理

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
func fibonacci(c, quit chan int) {
   x, y := 0, 1
   for {
      if x > 10000 {
         quit <- 1
      }
      select {
      case c <- x:
         println(x)
         x, y = y, x+y
      case <-quit:
         fmt.Println("quit")
         time.Sleep(time.Second)
         return
      }
   }
}

func main() {
   c := make(chan int)
   quit := make(chan int)
   fibonacci(c, quit)
}

# go run main.go  触发deadlock
fatal error: all goroutines are asleep - deadlock!

Reference

golang的系列在draveness的博客中有一些列非常详细的介绍.本篇都只是描述一些个人认为的重点内容.如果想知道相关golang设计过程,源码,机器执行指令等内容.推荐直接看draveness的内容或者其他更为详细的博文.

https://draveness.me/golang/