函數(shù)聲明為:
func Notify(c chan- os.Signal, sig ...os.Signal)
官方描述:
Notify函數(shù)讓signal包將輸入信號轉(zhuǎn)發(fā)到c。如果沒有列出要傳遞的信號,會將所有輸入信號傳遞到c;否則只傳遞列出的輸入信號。
signal包不會為了向c發(fā)送信息而阻塞(就是說如果發(fā)送時(shí)c阻塞了,signal包會直接放棄):調(diào)用者應(yīng)該保證c有足夠的緩存空間可以跟上期望的信號頻率。對使用單一信號用于通知的通道,緩存為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 ,將對應(yīng)的信號通知 ch,然后在 for 循環(huán)中針對不同信號做不同的處理, 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ā)送信號時(shí)我們還沒有準(zhǔn)備好接收,就有丟失信號的風(fēng)險(xiǎn)
我一直沒理解這段注釋,于是翻看源碼 $GOROOT/src/os/signal/signal.go,有這樣一段代碼,并注釋有“發(fā)送但不阻塞”。這里應(yīng)該就是“有可能丟失信號”的原因了吧。
...
for c, h := range handlers.m {
if h.want(n) {
// send but do not block for it
select {
case c - sig:
default:
}
}
}
...
于是,我寫了一段代碼進(jìn)行測試:
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é)束都不會打印,也不會退出程序;
在使用帶緩存的 channel 時(shí),只要接收到一個(gè) SIGINT ,在 sleep 結(jié)束后也就是準(zhǔn)備好接收,便會打印并退出程序。
這就是 signal.Notify 使用帶緩存 channel 的作用。
以上為個(gè)人經(jīng)驗(yàn),希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。如有錯(cuò)誤或未考慮完全的地方,望不吝賜教。
您可能感興趣的文章:- golang strings包的Replace的使用說明
- golang 中strings包的Replace的使用說明
- Golang Printf,Sprintf,Fprintf 格式化詳解
- 完美解決go Fscanf 在讀取文件時(shí)出現(xiàn)的問題
- Golang 并發(fā)以及通道的使用方式
- golang 如何替換掉字符串里面的換行符\n
- golang監(jiān)聽文件變化的實(shí)例