linux - Golang如何阻止signal传递给os/exec执行的命令

问题描述

问 题

场景

项目中使用signal包捕获了关闭程序相关的信号(SIGHUP,SIGINT,SIGKILL)，在程序执行时，使用Ctrl+C结束进程，捕获这些信号后需要进行一些其它的处理，但是程序中使用os/exec包执行了额外的命令，本意是在控制台中执行Ctrl+C结束进程的时候能够将os/exec执行的命令执行完毕后进程再退出，实现平滑的退出。

但是问题是触发Ctrl+C的时候，虽然主程序捕获到了信号也能够对程序的行为进行控制，但是同时os/exec执行的命令也会受到这个信号退出，无法实现完全的平滑退出。

信号处理相关代码

signalChan := make(chan os.Signal, 1)
signal.Notify(
    signalChan,
    syscall.SIGHUP,
    syscall.SIGUSR2,
    syscall.SIGINT,
    syscall.SIGKILL,
)
go func() {

    for {
        sig := <-signalChan
        switch sig {
        case syscall.SIGUSR2, syscall.SIGHUP, syscall.SIGINT, syscall.SIGKILL:
            log.Debug("Received exit signal, waiting for exit...")

            for _, channel := range runtime.Channels {
                channel.StopChan <- struct{}{}
            }

            runtime.StopScheduler <- struct{}{}
            runtime.StopHTTPServer <- struct{}{}
        }
    }
}()

使用os/exec执行命令部分

cmd := exec.Command(params[0], params[1:]...)

问题

谷歌、百度许久没有找到相关的解决方案，查阅了官方文档关于os/signal的描述，其中有这么一段

Subprocesses started by os.Exec, or by the os/exec package, will inherit the modified signal mask.

有什么方法能够只让主进程收到信号，而阻止os/exec启动的子进程命令收到信号吗？或者是将收到的信号修改为其它信号，避免子进程退出？

项目地址为https://github.com/mylxsw/task-runner，如果可能，你可以看看我的源码，或许更加方便定位问题吧。

解决方案

半年前的提问，我现在回答会不会太晚。
刚好碰到这个问题, *nix环境中, 由exec.command开启的子进程的进程组ID默认为为父进程ID，需设置子进程的进程组ID为子进程自身ID

cmd.SysProcAttr = &syscall.SysProcAttr{
    Setpgid: true,
}

另外，SIGKILL相当于kill -9，不能被捕获

这篇关于linux - Golang如何阻止signal传递给os/exec执行的命令的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！