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,如果可能,你可以看看我的源码,或许更加方便定位问题吧。


重新翻了我的旧代码,然后再拿出来测试了可以!

package main

import (
    "fmt"
    "log"
    "os"
    "os/exec"
    "os/signal"
    "syscall"
    "time"
)

func docmd(Num int) {
    time.Sleep(time.Second * 3)
    
    cmd := exec.Command("hostname")
    buf, err := cmd.Output()
    if err != nil {
        fmt.Println(err.Error())
        return
    }

    fmt.Printf("%d Result: %s", Num, buf)
}

func main() {
    signalChan := make(chan os.Signal, 1)
    signal.Notify(signalChan, syscall.SIGINT, syscall.SIGTERM)
    go func() {
        for {
            select {
            case <-signalChan:
                log.Println("正在结束!")
                docmd(10)
                os.Exit(0)
            }
        }
    }()

    docmd(1)
    fmt.Println("How much can I fuck you?")
    docmd(2)

    time.Sleep(time.Second * 10)
}



相关阅读:
the value for the useBean class attribute xxx is invalid
android中webview无网络时怎么实现点击加载?
tsconfig如何配置多个输出目录?
Reactjs 0.13是否没有过滤XSS攻击 ?
如何删除或修改 Git Credential Manager (GCM)缓存的账号和密码
请问如何解决windows环境下phantomjs不支持flash的问题?
react单页面应用中,点击浏览器回退按钮,怎么能使回来的页面保持离开前的状态?
前端测试(UI Recorder)webdriver启动firefox报错
百度坐标批量转换后画直线的问题
关于vuex的疑惑
微信内置浏览器跳转session丢失问题
rails 用mongoid Model 自关联问题
jquery easyui中有组件可以让一个下拉框控制两个输入框吗?
为什么没有延迟执行呢
node-gyp的build过程出错
用php写sokket连接第三方服务器,要保持连接不断,每三分钟向对方发一次心跳,php是单线程的,这个怎么才能实现呢?
android studio 如何添加依赖
php-fpm占用内存和CPU过高!!!
用SSE推送消息,数据库部分需要怎么优化?
手机端,点击缩略图,图片放大可以轮播效果,比如天猫APP的晒单



快速导航

Copyright © 2016 phpStudy |