```go
package main
import (
"fmt"
"os"
"os/signal"
)
func main() {
// Set up channel on which to send signal notifications.
// We must use a buffered channel or risk missing the signal
// if we're not ready to receive when the signal is sent.
c := make(chan os.Signal, 1)
signal.Notify(c, os.Interrupt)
// Block until a signal is received.
s := <-c
fmt.Println("Got signal:", s)
}
```
上面一段代码是 `signal.Notify` 的事例代码,注释说:
> 我们得使用带缓冲 channel
>
> 否则,发送信号时我们还没有准备好接收,就有丢失信号的风险
我一直没理解这段注释,于是翻看源码 `$GOROOT/src/os/signal/signal.go`,有这样一段代码,并注释有“发送但不阻塞”。这里应该就是“有可能丢失信号”的原因了吧。
```go
...
for c, h := range handlers.m {
if h.want(n) {
// send but do not block for it
select {
case c <- sig:
default:
}
}
}
...
```
于是,我写了一段代码进行测试:
```go
package main
import (
"log"
"os"
"os/signal"
"time"
)
func main() {
c := make(chan os.Signal)
signal.Notify(c, os.Interrupt)
time.Sleep(time.Second * 5) // 假装 5 秒没准备好接收
s := <-c
log.Println(s)
}
```
在使用不带缓存的 channel 时,5 秒的 sleep 期间无论按多少个 control + c,sleep 结束都不会打印,也不会退出程序;
在使用带缓存的 channel 时,只要接收到一个 `SIGINT` ,在 sleep 结束后也就是准备好接收,便会打印并退出程序。
这就是 `signal.Notify` 使用带缓存 channel 的作用。
有疑问加站长微信联系(非本文作者))