处理前景中的信号

Question

我想在背景中听一个OS信号，但在前景中处理它。

这是示例代码。 Mainloop每一秒无限期地打印“ Boop”。 但是，当收到中断信号时，处理程序会打印“缓慢终止……”，然后等待五秒钟，然后终止程序。

package main

import (
    "fmt"
    "os"
    "os/signal"
    "time"
)

func main() {
    c := make(chan os.Signal)
    signal.Notify(c, os.Interrupt)
    go func() {
        <-c
        fmt.Println("Terminating slowly...")
        time.Sleep(time.Duration(5)*time.Second)
        os.Exit(0)
    }()

    for {
        fmt.Println("boop")
        time.Sleep(time.Duration(1)*time.Second)
    }
}

当信号处理时，我希望其他所有内容都可以阻止。 但是目前，在缓慢终止的五秒钟内，它一直在打印“ boop”。

我明白了：

boop
boop
^CTerminating slowly...
boop
boop
boop
boop
boop

我想要这样：

boop
boop
^CTerminating slowly...

真正的程序是一个基于堆栈的语言解释器，用户可以在终止时运行某些东西，但是当前Mainloop可以同时更改堆栈。

Answer 1

尽管选择答案绝对是最好的，但我只想补充说，如果您会翻转逻辑，则结果可能还可以：

package main

import (
    "fmt"
    "os"
    "os/signal"
    "time"
)

func main() {
    c := make(chan os.Signal)
    signal.Notify(c, os.Interrupt)
    go func() {
        for {
            fmt.Println("boop")
            time.Sleep(time.Duration(1) * time.Second)
        }
    }()

    <-c
    fmt.Println("Terminating slowly...")
    time.Sleep(time.Duration(5) * time.Second)
    os.Exit(0)
}

Answer 2

AS mh-cbon oper >选择语句是您需要的。一旦将循环的主体包裹在A 选择的默认情况下，就不再需要启动另一个goroutine;使用选择案例来完成此工作。

另外，您正在使用一个未封闭的通道，但是根据 notify notify ' s文档，该功能期望适当缓冲通道：

软件包信号不会阻止发送到C：呼叫者必须确保C具有足够的缓冲空间以跟上预期信号速率。对于仅通知一个信号值的通道，大小1的缓冲区就足够了。

package main

import (
    "fmt"
    "os"
    "os/signal"
    "time"
)

func main() {
    c := make(chan os.Signal, 1) // buffered
    signal.Notify(c, os.Interrupt)
    for {
        select {
        case <-c:
            fmt.Println("Terminating slowly...")
            time.Sleep(time.Duration(5) * time.Second)
            return
        default:
            fmt.Println("boop")
            time.Sleep(time.Duration(1) * time.Second)
        }
    }
}

一旦信号发送到频道c，非默认情况就会变为非阻滞，并获取选择 ed，函数结束。

这是程序的一个可能输出（在该特定执行期间，我在大约4秒后点击^c）：

boop
boop
boop
boop
^CTerminating slowly...