处理信号量

信号是用户或操作系统杀死正在运行的应用程序的有效方式。有时，以更优雅的方式处理这些信号是有意义的。Go提供了一种捕获和处理信号的机制。在本节中，我们将通过使用Go例程的信号来讨论信号的处理。

实践

在通道的使用中，done的使用很常见，参见Concurrency in Go 中文笔记 or-done-channel

1.建立signals.go：

package main

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

// CatchSig 为SIGINT中断设置一个监听器
func CatchSig(ch chan os.Signal, done chan bool) {
    // 在等待信号时阻塞
    sig := <-ch
    // 当接收到信号时打印
    fmt.Println("\nsig received:", sig)

    // 对信号类型进行处理
    switch sig {
    case syscall.SIGINT:
        fmt.Println("handling a SIGINT now!")
    case syscall.SIGTERM:
        fmt.Println("handling a SIGTERM in an entirely different way!")
    default:
        fmt.Println("unexpected signal received")
    }

    // 终止
    done <- true
}

func main() {
    // 初始化通道
    signals := make(chan os.Signal)
    done := make(chan bool)

    // 将它们连接到信号lib
    signal.Notify(signals, syscall.SIGINT, syscall.SIGTERM)

    // 如果一个信号被这个go例程捕获，它将写入 done
    go CatchSig(signals, done)

    fmt.Println("Press ctrl-c to terminate...")
    // 程序会持续打印日志直到done通道被写入
    <-done
    fmt.Println("Done!")

}

2.运行并按下ctrl+c，会显示：

Press ctrl-c to terminate...
^C
sig received: interrupt
handling a SIGINT now!
Done!

3.尝试在另一个单独的命令行窗口运行，然后杀掉其PID，会显示：

$./signals
Press ctrl-c to terminate...
# in a separate terminal
$ ps -ef | grep signals
501 30777 26360 0 5:00PM ttys000 0:00.00 ./signals
$ kill -SIGTERM 30777
# in the original terminal
sig received: terminated
handling a SIGTERM in an entirely different way!
Done!

说明

本节使用了通道，第9章“并行和并发”会对此进行了更广泛的介绍。signal.Notify函数需要通道以发送信号通知。在命令行杀掉应用是测试向应用程序传递信号的好方法。我们用信号记录程序关心的信号类型。然后，在Go例程中处理传递给该函数的通道上的活动。一旦我们收到信号，就可以随心所欲地处理它。我们可以终止应用程序，使用消息进行响应，并针对不同的信号实施不同的行为。

我们还使用done通道阻止应用程序终止，直到收到信号。对于长期运行的应用程序（如Web应用程序），这是不必要的。创建适当的信号处理例程以进行清理非常有用，尤其是在持有大量状态的应用程序中。例如，在正常关闭时，允许当前处理程序完成其HTTP请求而非中途终止它们。