快捷導(dǎo)航

golang 中signal包的Notify用法說(shuō)明

更新時(shí)間：2021年03月22日 11:37:36 作者：kph_Hajash

這篇文章主要介紹了golang 中signal包的Notify用法說(shuō)明，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧

函數(shù)聲明為：

func Notify(c chan<- os.Signal, sig ...os.Signal)

官方描述：

Notify函數(shù)讓signal包將輸入信號(hào)轉(zhuǎn)發(fā)到c。如果沒(méi)有列出要傳遞的信號(hào)，會(huì)將所有輸入信號(hào)傳遞到c；否則只傳遞列出的輸入信號(hào)。

signal包不會(huì)為了向c發(fā)送信息而阻塞（就是說(shuō)如果發(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 的事例代碼，注釋說(shuō)：

我們得使用帶緩沖 channel

否則，發(fā)送信號(hào)時(shí)我們還沒(méi)有準(zhǔn)備好接收，就有丟失信號(hào)的風(fēng)險(xiǎn)

我一直沒(méi)理解這段注釋，于是翻看源碼 $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 秒沒(méi)準(zhǔn)備好接收
  s := <-c
  log.Println(s)
}

在使用不帶緩存的 channel 時(shí)，5 秒的 sleep 期間無(wú)論按多少個(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ò)誤或未考慮完全的地方，望不吝賜教。

您可能感興趣的文章: