最近在读nsq的源码,看到一个比较好的安全退出程序的方式,利用golang的channel,以下是实例代码,分享给大家,不喜勿喷。
package main
import(
"fmt"
"os"
"os/signal"
"syscall"
)
func main(){
signalChan := make(chan os.Signal,1) //创建一个信号量的chan,缓存为1,(0,1)意义不大
signal.Notify(signalChan, syscall.SIGINT, syscall.SIGTERM)//让进城收集信号量。
fmt.Println("i am workding!")
<-signalChan
ExitFunc()
}
func ExitFunc(){
fmt.Println("i am exiting!")
}
以上是对让主函数总到exit func的方案,生产环境里,exit func可能会做很多的关闭操作,而关闭的对象很可能是很多内嵌无限循环的goroutine,如何把退出信号传递给每一个goroutine呢?,我通常的做法是定义一个chan struct{} ,当调用close()函数关闭channel时,所有的<-chan操作会同时进行,这样就实现了将退出信号传递给每一个goroutine,而各个goroutine都可以通过select(默认linux下select调用的是epoll进行轮询)进行轮询,当获取到<-chan的时候,break循环