golang 中signal包的Notify用法说明

 更新时间:2021年03月22日 11:37:36   作者:kph_Hajash  
这篇文章主要介绍了golang 中signal包的Notify用法说明,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧

函数声明为:

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

官方描述:

Notify函数让signal包将输入信号转发到c。如果没有列出要传递的信号,会将所有输入信号传递到c;否则只传递列出的输入信号。

signal包不会为了向c发送信息而阻塞(就是说如果发送时c阻塞了,signal包会直接放弃):调用者应该保证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 ,将对应的信号通知 ch,然后在 for 循环中针对不同信号做不同的处理, for 循环为死循环。

补充:关于 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

否则,发送信号时我们还没有准备好接收,就有丢失信号的风险

我一直没理解这段注释,于是翻看源码 $GOROOT/src/os/signal/signal.go,有这样一段代码,并注释有“发送但不阻塞”。这里应该就是“有可能丢失信号”的原因了吧。

  ...
  for c, h := range handlers.m {
    if h.want(n) {
      // send but do not block for it
      select {
      case c <- sig:
      default:
      }
    }
  }
  ...

于是,我写了一段代码进行测试:

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 秒没准备好接收
  s := <-c
  log.Println(s)
}

在使用不带缓存的 channel 时,5 秒的 sleep 期间无论按多少个 control + c,sleep 结束都不会打印,也不会退出程序;

在使用带缓存的 channel 时,只要接收到一个 SIGINT ,在 sleep 结束后也就是准备好接收,便会打印并退出程序。

这就是 signal.Notify 使用带缓存 channel 的作用。

以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。如有错误或未考虑完全的地方,望不吝赐教。

相关文章

  • 关于go语言编码需要放到src 文件夹下的问题

    关于go语言编码需要放到src 文件夹下的问题

    这篇文章主要介绍了go语言编码需要放到src 文件夹下的相关知识,本文给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
    2020-10-10
  • 简单了解Go语言中函数作为值以及函数闭包的使用

    简单了解Go语言中函数作为值以及函数闭包的使用

    这篇文章主要介绍了简单了解Go语言中函数作为值以及函数闭包的使用,是golang入门学习中的基础知识,需要的朋友可以参考下
    2015-10-10
  • Go实现map转json的示例详解

    Go实现map转json的示例详解

    这篇文章主要为大家详细介绍了如何利用Go语言实现map转json的功能,文中的示例代码讲解详细,感兴趣的小伙伴可以跟随小编一起学习一下
    2023-09-09
  • Go语言中使用urfave/cli命令行框架

    Go语言中使用urfave/cli命令行框架

    这篇文章介绍了Go语言中使用urfave/cli命令行框架的方法,文中通过示例代码介绍的非常详细。对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
    2022-07-07
  • Go语言模型:string的底层数据结构与高效操作详解

    Go语言模型:string的底层数据结构与高效操作详解

    这篇文章主要介绍了Go语言模型:string的底层数据结构与高效操作详解,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2020-12-12
  • 一文带你轻松学会Go语言动态调用函数

    一文带你轻松学会Go语言动态调用函数

    这篇文章主要是带大家学习一下Go语言是如何动态调用函数的,文中的示例代码讲解详细,对我们学习Go语言有一定的帮助,需要的可以参考下
    2022-11-11
  • 分析Go错误处理优化go recover机制缺陷

    分析Go错误处理优化go recover机制缺陷

    这篇文章主要为大家介绍了分析Go错误处理优化go recover机制缺陷示例详解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2022-07-07
  • go值赋值和引用赋值的使用

    go值赋值和引用赋值的使用

    本文将介绍Go语言中的值赋值和引用赋值,并比较它们之间的差异,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2023-10-10
  • Go开发go-optioner工具实现轻松生成函数选项模式代码

    Go开发go-optioner工具实现轻松生成函数选项模式代码

    go-optioner 是一个在 Go 代码中生成函数选项模式代码的工具,可以根据给定的结构定义自动生成相应的选项代码,下面就来聊聊go-optioner是如何使用的吧
    2023-07-07
  • golang中sync.Mutex的实现方法

    golang中sync.Mutex的实现方法

    本文主要介绍了golang中sync.Mutex的实现方法,mutex 主要有两个 method: Lock() 和 Unlock(),文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2022-04-04

最新评论