如何在 go select 语句的默认分支中接收 os.signal？
在 go 中，通过使用 for 循环和 select 语句持续监听一个 os.signal 通道，可以捕获命令行的中断或终止信号。在默认分支（default）中可以执行代码块。然而，在终端命令行中终止程序时，无法接收相应信号，导致无法停止程序。
以下是代码示例：
package main
import (
"fmt"
"os"
"os/signal"
"time"
)
func main() {
// 创建一个 channel 接收 os.Signal
ch := make(chan os.Signal)
// 注册要关注的信号
signal.Notify(ch, os.Interrupt, os.Kill)
// 启动一个 goroutine 不断监听 channel
go func() {
for {
// 等待 channel 中的信号
select {
case s := <-ch:
fmt.Printf("收到停止信号: %v\n", s)
os.Exit(0)
default:
time.Sleep(time.Second)
}
}
}()
// 主循环
for {
fmt.Println("正在运行")
time.Sleep(time.Second)
}
}解决方案：
有两种方法解决这个问题：
-
移除 default 分支：
删除 default: time.sleep(time.second) 分支。这将使程序持续监听信号，而不会执行任何其他操作。 -
使用缓冲通道：
使用带有缓冲区的通道：ch := make(chan os.signal, 1)。当通道中有缓冲区时，发送操作将不会阻塞，即使没有 goroutine 正在从通道中接收。因此，当收到信号时，它将被存储在缓冲区中，而 select 语句将可以接收它。
