Go 语言 select 的实现原理解析_Golang

介绍

select是go在语言层面提供的i/o多路复用的机制，其专门用来让goroutine同时等待多个channel是否准备完毕:可读或可写。在channel状态改变之前，select会一直阻塞当前线程或者goroutine。

特性：

case 必须是一个通信操作，主要是指对通道（channel）进行发送或者接收数据的操作。

select 语句中除 default 外，各 case 执行顺序是随机的。

select 语句中如果没有 default 语句，则会阻塞等待任意一个 case满足执行条件。

select 语句中除 default 外，每个 case 只能操作一个 channel，要么读要么写。

当 select 中的多个 case 同时被触发时，会随机执行其中的一个。

普通多线程

多路复用

数据结构

select在go语言的源代码中不存在对应的结构体，使用runtime.scase 结构体表示select控制结构里的case。

type scase struct {
	c    *hchan                    //case操作的通道     
    kind  uint16
    //表示该case的类型，分为读channel、写channel和default。
    //读channel、写channel和default三种类型分别由常量定义
    //caserecv：case语句中尝试读取scase.c中的数据。
    //casesend：case语句中尝试向scase.c中写入数据。
    //casedefault：default语句。
	elem unsafe.pointer 
    //scase.kind == caserecv ： scase.elem表示读出channel的数据存放地址；
    //scase.kind == casesend ： scase.elem表示将要写入channel的数据存放地址；
}

在select语句运行时，scase结构体的实例会被用来表示每个case。运行时根据c 字段找到对应的通道，根据elem字段来处理数据的发送或接收操作。

执行流程

实现过程和结果（穿插编译器的重写和优化）

单分支的select

只有一个 case 且不是 default，这种情况编译器会直接将其翻译成对管道的收发操作，并且还是阻塞式的，一直阻塞到操作可以完成。

对于接收操作（如 case val := <-ch），它会被转换为 val := <-ch，直接尝试从通道 ch 接收数据。

对于发送操作（如 case ch <- value）,它会被转换为 ch <- value，直接尝试向通道 ch 发送数据。

只包含default分支会直接执行default操作。

多路select

在编译器中会被转换为runtime.selectgo函数调用。

func selectgo(cas0 *scase, order0 *uint16, , ncases int) (int, bool) {
    pollorder := order1[:ncases:ncases]
    lockorder := order1[ncases:][:ncases:ncases]
    for i := 1; i < ncases; i++ {
        j := fastrandn(uint32(i + 1))
        pollorder[i] = pollorder[j]
        pollorder[j] = uint16(i)
    }
    // 代码可能继续执行后续操作
}

cas0，scase数组的头部指针，前半部分存放的是写管道 case，后半部分存放的读管道 case，以nsends来区分
order0，它的长度是scase数组的两倍，前半部分分配给pollorder数组（决定管道执行顺序），后半部分分配给lockorder数组（决定管道锁定顺序）
pollorder：每次selectgo执行都会把scase序列打乱，以达到随机检测case的目的。
lockorder：所有case语句中channel序列，以达到去重防止对channel加锁时重复加锁的目的。
ncases表示scase数组的长度

直接阻塞

1. select结构不包含任何case

在go编译器内部的代码如下

func walkselectcases(cases *nodes) []*node {
	n := cases.len()
	if n == 0 {
		return []*node{mkcall("block", nil, nil)}
	}
	...
}
func block() {
	gopark(nil, nil, waitreasonselectnocases, traceevgostop, 1)
}

walkselectcases的参数是一个select语句中的case元素的集合。当集合的长度为0时，表示当前select中无case会调用block函数，block函数会调用gopark让出goroutine对处理器的使用权并传入等待原因，暂停goroutine避免cpu空转。

2.当case中的channel是空指针

例：包含一个case且case中的channel是空指针，编译器会将select改写为if条件语句

//部分代码
if ch == nil {
    block()调用block，将goroutine陷入永久休眠
}

非阻塞操作

当select中包含default分支时，就会被编译器认为是一次非阻塞的收发操作。

示例：一个case分支一个default分支，对通道的读写操作

示例：一个case分支一个default分支，对通道的读写操作

写操作：编译器会使用条件语句和 runtime.selectnbsend 函数改写代码

//改写
if selectnbsend(ch, i) {
    ...
} else {
    ...
}
//false参数决定了这一次的发送是非阻塞的，所以如果存在缓冲区空间不足时，当前 goroutine 都不会阻塞而是会直接返回。
func selectnbsend(c *hchan, elem unsafe.pointer) (selected bool) {
	return chansend(c, elem, false, getcallerpc())
}

读操作：

// 改写前
select {
case v <- ch: // case v, ok <- ch:
    ......
default:
    ......
}
// 改写后
if selectnbrecv(&v, ch) { // if selectnbrecv2(&v, &ok, ch) {
    ...
} else {
    ...
}
//看读操作是否需要，第一个会忽略返回的布尔值，第二个会将布尔值传给调用方，block参数决定本次操作不阻塞
func selectnbrecv(elem unsafe.pointer, c *hchan) (selected bool) {
	selected, _ = chanrecv(c, elem, false)
	return
}
func selectnbrecv2(elem unsafe.pointer, received *bool, c *hchan) (selected bool) {
	selected, *received = chanrecv(c, elem, false)
	return
}

性能优化建议