Go 信号处理

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

package main

import "fmt"
import "os"
import "os/signal"
import "syscall"

func main() {

	// Go信号通知通过向一个channel发送``os.Signal`来实现。
	// 我们将创建一个channel来接受这些通知,同时我们还用
	// 一个channel来在程序可以退出的时候通知我们
	sigs := make(chan os.Signal, 1)
	done := make(chan bool, 1)

	// `signal.Notify`在给定的channel上面注册该channel
	// 可以接受的信号
	signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)

	// 这个goroutine阻塞等待信号的到来,当信号到来的时候,
	// 输出该信号,然后通知程序可以结束了
	go func() {
		sig := <-sigs
		fmt.Println()
		fmt.Println(sig)
		done <- true
	}()

	// 程序将等待接受信号,然后退出
	fmt.Println("awaiting signal")
	<-done
	fmt.Println("exiting")
}

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

awaiting signal

interrupt
exiting
文章导航