写在最前面

select为golang提供了多路IO复用机制，和其他IO复用一样，用于检测是否有读写事件是否ready。

本文将介绍一下golang的select的用法和实现原理。

实现原理

golang实现select的时候，实际上为每一个case语句定义了一个数据结构，select语句块执行的时候，实际上可以类比成对一个case数组处理的代码块（或者函数），然后程序流程转到选中的case块。

case数据结构

src/runtime/select.go:scase

　　scase.c表示当前case语句操作的chan指针，这也表明一个case只能监听一个chan。

　　scase.kind表示当前的chan是可读还是可写channel或者是default。三种类型分别由常量定义：

  • caseRecv：case语句中尝试读取scase.c中的数据；
  • caseSend：case语句中尝试向scase.c中写入数据；
  • caseDefault： default语句

　　scase.elem表示缓冲区地址，跟据scase.kind不同，有不同的用途：

  • scase.kind == caseRecv ： scase.elem表示读出channel的数据存放地址；
  • scase.kind == caseSend ： scase.elem表示将要写入channel的数据存放地址；

select实现逻辑

src/runtime/select.go:selectgo()

　　其中数据结构hselect如下：

hselect.tcase存的是scase总数。

hselect.pollorder是保存scase的随机后的序列。以达到随机检测case的目的。

hselect.lockorder是保存的channel地址。所有case语句中channel序列，以达到去重防止对channel加锁时重复加锁的目的。

selectgo返回int，表示选中的scase，也就是ready的channel index。

该函数执行逻辑大致如下：

1. 锁定scase语句中所有的channel

2. 按照随机顺序检测scase中的channel是否ready

　　2.1 如果case可读，则读取channel中数据，解锁所有的channel，然后返回(case index)

　　2.2 如果case可写，则将数据写入channel，解锁所有的channel，然后返回(case index)

　　2.3 所有case都未ready，则解锁所有的channel，然后返回（default index）

3. 所有case都未ready，且没有default语句

　　 3.1 将当前协程加入到所有channel的等待队列

 　　3.2 当将协程转入阻塞，等待被唤醒

4. 唤醒后返回channel对应的case index

　　4.1 如果是读操作，解锁所有的channel，然后返回(case index)

　　4.2 如果是写操作，解锁所有的channel，然后返回(case index)