浅谈Golang中select的用法

在go中,select主要是和channel有关,大概的格式如下:

select{
case <- ch1:
// do something
case i := <- ch2
// use i do something
default:
// ...
}

这个语法看起来和switch语句很接近,但是具体select有什么用呢?我们来看一段代码:

• code-1

func main() {
 ch := make(chan int)
 for i := 0; i < 10; i++{
 select {
 case ch <- i:
 case x := <- ch:
 fmt.Println(x)
 }
 }
}

这段代码里我们先创建了一个channel(管道),这个管道是无缓冲的,那么这段代码是否能正常的运行呢?

答案是不能的,这段代码会产生一个deadlock。

那么我们稍微修改一下这个代码,

ch := make(chan int, 1)

在这里我们将刚才无缓冲的管道改变为有缓冲且缓冲大小为1的管道,再次运行之后,会显示正确的结果,但是这个结果比较微妙:

0
2
4
6
8

因为这个管道的缓冲值只有1,那么同一时间只会有一个case执行,这个channel不是空的就是满的。
所以这个结果是比较固定的,但是为什么只会输出这么几个数字呢?那我们来一起看看具体是怎么运行的。

在第一次进入循环的时候,i为0,进入到select中,开始由上向下来发现哪一个case可以执行,当计算表达式
ch <- i,也就是向管道写入数据的时候,因为这个管道现在有缓冲,那么在向管道写完数据之后,此时的case便执行完成,然后就跳出select,开始进行下一次的循环,当i=1的时候(现在这个管道里面的数据是0),再次进入select中,此时还是开始计算ch <- i 表达式,但是现在管道里面是有数据的,再次向管道中写入数据,那么会使该发送操作阻塞,此时该case便无法再执行,那么select将会继续向下执行下一个case,在下一个case中,有一个管道的接收操作x := <- ch,在这里管道里有之前第一次循环的时候放入的0这个数据,那么在这里就会将管道的数据赋值给x,从而打印出第一个数据0,那么后面的数据就和之前的过程是一样的了。

看到这里,大概就能明白select的作用了,顺便说一下,select的case语句中,都是对应一个I/O操作,准确的说是对应一个channel的I/O操作,那么到这里也应该可以理解为什么在code-1中,一个无缓冲的channel能在那段代码中产生一个deadlock。

这里还引出几个概念:channel, buffer channel, goroutine等。

OK!今天就先到这里,下次再继续分享Golang的特色之一,goroutine。

本文来自:Segmentfault

感谢作者:funnyx

查看原文:浅谈Golang中select的用法

• 编辑
• 预览