golang信号处理

来源:互联网 时间:1970-01-01

go语言中处理信号很简单,并且不会中断程序的正常运行逻辑。想监听一个信号只需要调用Notify函数 **func Notify(c chan<- os.Signal, sig ...os.Signal)**+ sig 表示你希望监听的信号,如果不设置表示监听所有的信号。+ c 表示当信号发生时,系统会往该channel中写入发生的信号。如果不想监听信号则可以调用stop**func Stop(c chan<- os.Signal)**##实例解析 ```exitChan := make(chan struct{})func main() { //开启一个goroutine专门处理信号 go HanleSignal() <-exitChan //释放程序占用资源,优雅退出}func HandleSignal() { ch := make(chan os.Signal)signal.Notify(ch,syscall.SIGINT,syscall.SIGTERM,syscall.SIGHUP)for {sig := <-chfmt.Println("reveive signal", sig.String())switch sig {case syscall.SIGHUP:// 接收到SIGHUP时可以重新加载配置case syscall.SIGINT:close(exitChan)case syscall.SIGQUIT: close(exitChan)} }} ```##golang实现源码``` var handlers struct { // key为信号发生时接收channel //value 为channel感兴趣的信号集合(当集合中的信号发生时需要往channel中写数据)m map[chan<- os.Signal]*handler}type handler struct {mask [(numSig + 31) / 32]uint32}func (h *handler) want(sig int) bool {return (h.mask[sig/32]>>uint(sig&31))&1 != 0}func (h *handler) set(sig int) {h.mask[sig/32] |= 1 << uint(sig&31)}func Notify(c chan<- os.Signal, sig ...os.Signal) {h := handlers.m[c]add := func(n int) {if !h.want(n) {h.set(n)}}//sig 为空时表示监听所有的信号if len(sig) == 0 {for n := 0; n < numSig; n++ {add(n)}} else {for _, s := range sig {add(signum(s))}}}func Stop(c chan<- os.Signal) { h := handlers.m[c] delete(handlers.m, c)}//信号发生时 遍历map往所有对该信号感兴趣的channel写入数据,往channel写入数据不会发生阻塞func process(sig os.Signal) {n := signum(sig)for c, h := range handlers.m {if h.want(n) {// send but do not block for itselect {case c <- sig:default:}}}}//在os/signal/signal_unix.go的init函数中开启一个goroutine监听用户感兴趣的所有信号func loop() {for {process(syscall.Signal(signal_recv()))}}func init() {go loop()}```

相关阅读:
Top