Go 信号处理

优质
小牛编辑
126浏览
2023-12-01

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

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

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

  1. awaiting signal
  2. interrupt
  3. exiting