首页 > 编程笔记 > Go语言笔记

Go语言select case语句的用法

在Go语言中,除了 switch case 语句外,还有一种选择结构——select case。select 语句可以用于配合通道(channel)的读/写操作,用于多个 channel 的并发读/写操作。

select 语句类似于 switch 语句,switch 语句是按照顺序从上到下依次执行,而 select 是随机选择一个 case 执行。如果没有 case 可运行,它将阻塞,直到有 case 可运行。

Go语言 select case 语句的语法格式如下:
select {
    case:
        代码块1;
    case:
        代码块2;
    /*可以定义任意数量的 case*/
    default : /*可选*/
        代码块n;
}
在 select case 语句中:
请看下面的代码:
package main
import "fmt"
func main() {
    a := make(chan int, 1024)
    b := make(chan int, 1024)
    for i := 0; i < 10; i++ {
        fmt.Printf("第%d次,", i)
        a <- 1
        b <- 1
        select {
        case <-a:
            fmt.Println("from a")
        case <-b:
            fmt.Println("from b")
        }
    }
}
运行结果(每次返回的结果都是不同的):
第0次,from b
第1次,from a
第2次,from b
第3次,from b
第4次,from a
第5次,from b
第6次,from b
第7次,from b
第8次,from b
第9次,from b

在以上代码中,同时在 a 和 b 中进行选择,哪个有内容就从哪个读,由于 channel 的读/写操作是阻塞操作,使用 select 语句可以避免单个 channel 的阻塞。

此外,select 同样可以使用 default 代码块,避免所有 channel 同时阻塞。

相关文章