浅谈Golang中select的用法

2024年3月29日 138次阅读 来源: 达闻西

在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。

    原文作者：达闻西
    原文地址: https://segmentfault.com/a/1190000015036739
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。
点赞