Go 信号处理

有的时候我们希望Go能够智能地处理Unix信号。例如我们希望一个server接收到一个SIGTERM的信号时，能够自动地停止；或者一个命令行工具接收到一个SIGINT信号时，能够停止接收输入。现在我们来看下如何使用channel来处理信号。

  1. package main
  3. import "fmt"
  4. import "os"
  5. import "os/signal"
  6. import "syscall"
  8. func main() {
  10.     // Go信号通知通过向一个channel发送``os.Signal`来实现。
  11.     // 我们将创建一个channel来接受这些通知，同时我们还用
  12.     // 一个channel来在程序可以退出的时候通知我们
  13.     sigs := make(chan os.Signal, 1)
  14.     done := make(chan bool, 1)
  16.     // `signal.Notify`在给定的channel上面注册该channel
  17.     // 可以接受的信号
  18.     signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
  20.     // 这个goroutine阻塞等待信号的到来，当信号到来的时候，
  21.     // 输出该信号，然后通知程序可以结束了
  22.     go func() {
  23.         sig := <-sigs
  24.         fmt.Println()
  25.         fmt.Println(sig)
  26.         done <- true
  27.     }()
  29.     // 程序将等待接受信号，然后退出
  30.     fmt.Println("awaiting signal")
  31.     <-done
  32.     fmt.Println("exiting")
  33. }

当运行程序的时候，程序将阻塞等待信号的到来，我们可以使用CTRL+C来发送一个SIGINT信号，这样程序就会输出interrupt后退出。

  1. awaiting signal
  3. interrupt
  4. exiting