Go 语言 select 的实现原理解析

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的数据存放地址；
}

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)
    }
    // 代码可能继续执行后续操作
}

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)
}

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

//改写
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
}