Go os.Signal、os.Interrupt、os.Kill 信号量的使用

使用示例：

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)
}

Notify 的函数定义如下：

func Notify(c chan<- os.Signal, sig ...os.Signal) 

os.Signal 表示信号量，信号量有多种，例如 Interrupt、Kill，因此需要使用通道接收，这样可以可以避免信号丢失。

    c := make(chan os.Signal, 1)

然后监听种类型的信号量：

    signal.Notify(c, os.Interrupt, os.Kill)

os.Interrupt 表示中断；
os.Kill 表示杀死；

正常情况下，进程被按下 Ctrl+C，会被终止。但是如果自己订阅了信号量，那么可以一直不停止。

下面示例可以接收两个信号量，然后执行某些操作再退出进程：

package main

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

func main() {
    ci := make(chan os.Signal, 1)
    ck := make(chan os.Signal, 1)
    sleep := make(chan bool, 1)

    signal.Notify(ci, os.Interrupt)
    signal.Notify(ck, os.Kill)
    go interrupt(ci, sleep)
    go kill(ck, sleep)

    <-sleep
}
func interrupt(ci chan os.Signal, sleep chan bool) {
    s := <-ci
    fmt.Println("进程被中断:", s)
    sleep <- true
}

func kill(ck chan os.Signal, sleep chan bool) {
    s := <-ck
    fmt.Println("进程被杀死", s)
    sleep <- true
}

按下 Ctrl+C：

当然，如果是 Kill，那么进程可能会被直接杀死，而不会出现提示。

任务管理器杀死进程：

如果是 Linux 中，则还能有点效果。

停止监听信号量：

signal.Stop(ci)

Go 中支持的信号量：

其他示例：

package main

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

func main() {
    SigChan := make(chan os.Signal, 1)

    signal.Notify(SigChan, syscall.SIGHUP, syscall.SIGINT, syscall.SIGTERM, syscall.SIGQUIT)

    sig := <-SigChan

    switch sig {
    case syscall.SIGHUP:
        fmt.Println("\nSIGHUP signal generated")

    case syscall.SIGINT:
        fmt.Println("\nSIGINT signal generated")

    case syscall.SIGTERM:
        fmt.Println("\nSIGTERM signal generated")

    case syscall.SIGQUIT:
        fmt.Println("\nSIGQUIT signal generated")

    default:
        fmt.Println("\nUNKNOWN signal generated")
    }
}