首页 > 文章列表 > 如何在 Go select 语句的默认分支中接收 os.Signal?

如何在 Go select 语句的默认分支中接收 os.Signal?

130 2025-02-20

如何在 Go select 语句的默认分支中接收 os.Signal?

如何在 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("收到停止信号: %vn", s)
                os.Exit(0)
            default:
                time.Sleep(time.Second)
            }
        }
    }()

    // 主循环
    for {
        fmt.Println("正在运行")
        time.Sleep(time.Second)
    }
}

解决方案:

有两种方法解决这个问题:

  1. 移除 default 分支:
    删除 default: time.sleep(time.second) 分支。这将使程序持续监听信号,而不会执行任何其他操作。
  2. 使用缓冲通道:
    使用带有缓冲区的通道:ch := make(chan os.signal, 1)。当通道中有缓冲区时,发送操作将不会阻塞,即使没有 goroutine 正在从通道中接收。因此,当收到信号时,它将被存储在缓冲区中,而 select 语句将可以接收它。
来源:1730109004