Go语言学习笔记--select简介

从不同的并发执行的协程中获取值可以通过关键字select来完成。select是Go中的一个控制结构，类似于switch语句，用于处理异步IO操作。select会监听case语句中channel的读写操作，当case中channel读写操作为非阻塞状态（即能读写）时，将会触发相应的动作。

基本语法：

select {
case u:= <- ch1:
...
case v:= <- ch2:
...
...
default: // no value ready to be received
...
}

其中default语句是可选的，不支持fallthrough行为。在任何case中执行break或者return，select就结束。

select的作用是：选择处理列出的多个通道中的一个。

• 如果有多个case都可以运行，select会随机公平地选出一个执行，其他不会执行。
• 如果没有可运行的case语句，且有default语句，那么就会执行default的动作。
• 如果没有可运行的case语句，且没有default语句，select将阻塞，直到某个case通信可以运行

select语句实现了一种监听模式，通常用于无限循环中，在某种情况下可以通过break语句来退出循环。

package main

import (
	"fmt"
	"time"
)

func main() {
	ch1 := make(chan int)
	ch2 := make(chan int)
	go pump1(ch1)
	go pump2(ch2)
	go suck(ch1, ch2)
	time.Sleep(1e9)
}
func pump1(ch chan int) {
	for i := 0; ; i++ {
		ch <- i + 1
	}
}
func pump2(ch chan int) {
	for i := 0; ; i++ {
		ch <- i + 5
	}
}
func suck(ch1, ch2 chan int) {
	for {
		select {
		case v := <-ch1:
			fmt.Printf("Received on channel 1: %d\n", v)
		case v := <-ch2:
			fmt.Printf("Received on channel 2: %d\n", v)
		}
	}
}

程序中有2个通道ch1和ch2，三个协程pump1,pump2和suck。这是一个典型的生产者消费者模式。在无限循环中，ch1和ch2通过pump1和pump2填充整数，suck也是在无限循环中轮询输入的，通过select语句获取ch1和ch2的整数并输出。选择哪个case，取决于马一个通道收到了信息。程序在main执行1s后结束。

有兴趣的同学可以自己执行下代码，看看执行结果。

复制代码在playground运行