golang处理系统信号

最新推荐文章于 2024-04-26 14:57:40 发布

原创最新推荐文章于 2024-04-26 14:57:40 发布 · 668 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#go

go 同时被 2 个专栏收录

29 篇文章

订阅专栏

golang

15 篇文章

订阅专栏

今天学了一下golang处理系统信号。

和在后台开发来说，应该是个很常规的操作。

先定义一个通道 sigs ，用来和系统交互，系统把信号存放在这个通道里面，怎么让系统知道要存放到这个通道里呢？调用系统函数：

signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)

后面两个是我们定义的要捕获的信号。也就是说系统会将这两种信号捕获给我们的程序，其他的信号不会捕获给我们，除非我们继续注册其他的信号。

接下来，我们要开一个goroutine专门用来等待系统给我们捕获的信号

go handleSigs()

在handleSigs()里，如果系统没有捕获我们想要的信号，这个goroutine会一直阻塞在读通道sigs操作上。一旦捕获到了，那么handleSigs()会根据信号类型分别进行相应的处理。这里可以直接用switch结构，也可以用map[os.Signal]func()把信号处理函数注册到这个map里来，然后根据信号类型调用注册的处理函数。其实还可以抽象出一个interface,map[os.Signal]SigHandler。

详细代码如下：

var (
	sigs chan os.Signal
	done chan bool
)

func init() {
	sigs = make(chan os.Signal, 1)
	done = make(chan bool, 1)
}

func main() {
	signalTest()
}

func signalTest() {
	signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)

	go handleSigs()

	fmt.Println("awaiting signal")
	<-done
	fmt.Println("exiting")
}

func handleSigs() {
	sig := <-sigs
	fmt.Println("acquire signal:",sig)
	switch sig {
	case syscall.SIGINT: handleSigint()
	case syscall.SIGTERM: handleTerm()
	default:
		panic(errors.New("unsupport signal handle"))
	}
}

func handleSigint() {
	done <- true
}

func handleTerm() {
	done <- true
}

（全文完）