【Golang】图解channel之读、写、关闭

幼麟实验室 2020-11-30

1189

select {
case ch <- 12:
    ... foo
default:
    ... bar
}

为什么这样写就不会阻塞了呢？因为上面的代码会被编译器转换为对runtime.selectnbsend的调用。

//编译后
if selectnbsend(ch, 12) {
    ... foo
} else {
    ... bar
}

runtime.selectnbsend也仅仅是调用了runtime.chansend，但是会标记上非阻塞。

func selectnbsend(c *hchan, elem unsafe.Pointer) (selected bool) {
   return chansend(c, elem, false, getcallerpc())
}


func chansend(c *hchan, ep unsafe.Pointer, block bool, callerpc uintptr) bool

所以，如果缓冲区指望不上，又没有谁在等着recv，又或者channel为nil......

总而言之，send任务不能马上执行，需要等待。那么selectnbsend函数就会返回false，对应的协程也就不会进到sendq里等待了。

非阻塞式的读操作，会被编译器转换为对runtime.selectnbrecv或selectnbrecv2的调用，而它们也仅仅是调用了runtime.chanrecv，只不过在需要等待时会返回false。

select {
case v = <-c:
    ... foo
default:
    ... bar
}


// 被编译器转化为
if selectnbrecv(&v, c) {
    ... foo
} else {
    ... bar
}

//comma ok写法
select {
case v, ok = <-c:
    ... foo
default:
    ... bar
}


// 被编译器转化为
if c != nil && selectnbrecv2(&v, &ok, c) {
    ... foo
} else {
    ... bar
}