golang訊號處理
來源:互聯網
上載者:User
這是一個建立於 的文章,其中的資訊可能已經有所發展或是發生改變。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()}```