使用 Golang 的 exec.CommandContext 执行带有超时的 "su -c" 命令

Golang小白一枚，正在不断学习积累知识，现将学习到的知识记录一下，也是将我的所得分享给大家！而今天这篇文章《使用 Golang 的 exec.CommandContext 执行带有超时的 "su -c" 命令》带大家来了解一下##content_title##，希望对大家的知识积累有所帮助，从而弥补自己的不足，助力实战开发！

此代码工作正常，它打印 timeout：

ctx, cancel := context.withtimeout(context.background(), 1000 * time.millisecond)
process := exec.commandcontext(ctx, "sleep", "5")

processoutbytes, err := process.output()

if ctx.err() == context.deadlineexceeded {
    fmt.println("timeout")
}

但是，只是另一个命令，则不然。进程可以无限工作！

ctx, cancel := context.WithTimeout(context.Background(), 1000 * time.Millisecond)
process := exec.CommandContext(ctx, "su", "-", "myuser", "-c", "python3 main.py")

processOutBytes, err := process.Output()

if ctx.Err() == context.DeadlineExceeded {
    fmt.Println("Timeout")
}

在基于 ubuntu:20.04 的 docker 容器内的 ubuntu 20.04 上运行。为什么会这样以及如何使第二个代码工作？

正确答案

超时仅适用于exec启动的进程，它不会杀死任何子进程。在您的情况下，它将杀死 su 但不会杀死下一个 python3 进程。

要杀死给定进程启动的所有子进程，您可以在新进程组中启动它，然后通过将 sigkill 发送到 -pid （负 pid）来杀死整个组，如下所示：

ctx, cancel := context.WithTimeout(context.Background(), 1000 * time.Millisecond)
process := exec.CommandContext(ctx, "su", "-", "myuser", "-c", "python3 main.py")

process.SysProcAttr = &syscall.SysProcAttr{Setpgid: true}
go func() {
    <-ctx.Done()
    if ctx.Err() == context.DeadlineExceeded {
        syscall.Kill(-process.Process.Pid, syscall.SIGKILL)
    }
}()

processOutBytes, err := process.Output()
cancel()

if ctx.Err() == context.DeadlineExceeded {
    fmt.Println("Timeout")
}

另请注意，依赖 syscall 的代码不可移植；例如，它甚至无法在 windows 上编译。

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。