Golang 捕捉信号

Question

我想在 Go 中实现一个“流程包装器”。基本上它会做的是启动一个进程（比如说一个节点服务器）并监视它（捕获信号，如 SIGKILL、SIGTERM ...）

我认为这样做的方法是使用 syscall.Exec 在 go 例程中启动节点服务器：

func launchCmd(path string, args []string) {
  err := syscall.Exec(path, args, os.Environ())
  if err != nil {
    panic(err)
  }
}

然后我想捕获由syscall 执行的命令生成的所有可能信号。我对 Go 很陌生，任何帮助将不胜感激。

Answer 1

Go 中有三种执行程序的方式：

1. syscall 与 syscall.Exec、syscall.ForkExec、syscall.StartProcess 的包
2. os 与os.StartProcess 一起打包
3. os/exec 包与exec.Command

syscall.StartProcess 为低电平。它返回一个uintptr 作为句柄。

os.StartProcess 给你一个很好的os.Process 结构，你可以调用Signal。 os/exec 给你io.ReaderWriter 在管道上使用。两者都在内部使用syscall。

读取从您自己的进程以外的进程发送的信号似乎有点棘手。如果可能的话，syscall 可以做到。我在更高级别的包中看不到任何明显的东西。

要接收信号，您可以像这样使用signal.Notify：

sigc := make(chan os.Signal, 1)
signal.Notify(sigc,
    syscall.SIGHUP,
    syscall.SIGINT,
    syscall.SIGTERM,
    syscall.SIGQUIT)
go func() {
    s := <-sigc
    // ... do something ...
}()

您只需要更改您有兴趣收听的信号即可。如果不指定信号，它将捕获所有可以捕获的信号。

您可以使用syscall.Kill 或Process.Signal 来映射信号。您可以从 Process.Pid 或 syscall.StartProcess 获取 pid。

Answer 2

你可以使用signal.Notify：

import (
"os"
"os/signal"
"syscall"
)

func main() {
    signalChannel := make(chan os.Signal, 2)
    signal.Notify(signalChannel, os.Interrupt, syscall.SIGTERM)
    go func() {
        sig := <-signalChannel
        switch sig {
        case os.Interrupt:
            //handle SIGINT
        case syscall.SIGTERM:
            //handle SIGTERM
        }
    }()
    // ...
}