func Notify(c chan- os.Signal, sig ...os.Signal)
Notify函數(shù)讓signal包將輸入信號(hào)轉(zhuǎn)發(fā)到c。如果沒有列出要傳遞的信號(hào),會(huì)將所有輸入信號(hào)傳遞到c;否則只傳遞列出的輸入信號(hào)。
signal包不會(huì)為了向c發(fā)送信息而阻塞(就是說如果發(fā)送時(shí)c阻塞了,signal包會(huì)直接放棄):調(diào)用者應(yīng)該保證c有足夠的緩存空間可以跟上期望的信號(hào)頻率。對(duì)使用單一信號(hào)用于通知的通道,緩存為1就足夠了。
ch := make(chan os.Signal, 1) signal.Notify(ch, syscall.SIGHUP, syscall.SIGQUIT, syscall.SIGTERM, syscall.SIGSTOP, syscall.SIGUSR1) for { s := -ch switch s { case syscall.SIGQUIT: log.Infof("SIGSTOP") return case syscall.SIGSTOP: log.Infof("SIGSTOP") return case syscall.SIGHUP: log.Infof("SIGHUP") return case syscall.SIGKILL: log.Infof("SIGKILL") return case syscall.SIGUSR1: log.Infof("SIGUSR1") return default: log.Infof("default") return } }
以上代碼告訴 signal ,將對(duì)應(yīng)的信號(hào)通知 ch,然后在 for 循環(huán)中針對(duì)不同信號(hào)做不同的處理, for 循環(huán)為死循環(huán)。
補(bǔ)充:關(guān)于 signal.Notify 使用帶緩存的 channel
package main import ( "fmt" "os" "os/signal" ) func main() { // Set up channel on which to send signal notifications. // We must use a buffered channel or risk missing the signal // if we're not ready to receive when the signal is sent. c := make(chan os.Signal, 1) signal.Notify(c, os.Interrupt) // Block until a signal is received. s := -c fmt.Println("Got signal:", s) }
上面一段代碼是 signal.Notify 的事例代碼,注釋說:
我們得使用帶緩沖 channel
否則,發(fā)送信號(hào)時(shí)我們還沒有準(zhǔn)備好接收,就有丟失信號(hào)的風(fēng)險(xiǎn)
我一直沒理解這段注釋,于是翻看源碼 $GOROOT/src/os/signal/signal.go,有這樣一段代碼,并注釋有“發(fā)送但不阻塞”。這里應(yīng)該就是“有可能丟失信號(hào)”的原因了吧。
... for c, h := range handlers.m { if h.want(n) { // send but do not block for it select { case c - sig: default: } } } ...
于是,我寫了一段代碼進(jìn)行測(cè)試:
package main import ( "log" "os" "os/signal" "time" ) func main() { c := make(chan os.Signal) signal.Notify(c, os.Interrupt) time.Sleep(time.Second * 5) // 假裝 5 秒沒準(zhǔn)備好接收 s := -c log.Println(s) }
在使用不帶緩存的 channel 時(shí),5 秒的 sleep 期間無論按多少個(gè) control + c,sleep 結(jié)束都不會(huì)打印,也不會(huì)退出程序;
在使用帶緩存的 channel 時(shí),只要接收到一個(gè) SIGINT ,在 sleep 結(jié)束后也就是準(zhǔn)備好接收,便會(huì)打印并退出程序。
這就是 signal.Notify 使用帶緩存 channel 的作用。
以上為個(gè)人經(jīng)驗(yàn),希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。如有錯(cuò)誤或未考慮完全的地方,望不吝賜教。
標(biāo)簽:青海 海南 電子產(chǎn)品 物業(yè)服務(wù) 安康 儋州 遼寧 西雙版納
巨人網(wǎng)絡(luò)通訊聲明:本文標(biāo)題《golang 中signal包的Notify用法說明》,本文關(guān)鍵詞 golang,中,signal,包的,Notify,;如發(fā)現(xiàn)本文內(nèi)容存在版權(quán)問題,煩請(qǐng)?zhí)峁┫嚓P(guān)信息告之我們,我們將及時(shí)溝通與處理。本站內(nèi)容系統(tǒng)采集于網(wǎng)絡(luò),涉及言論、版權(quán)與本站無關(guān)。