在Golang中终止以os / exec开头的进程


72

有没有办法终止Golang中以os.exec开始的进程?例如(来自http://golang.org/pkg/os/exec/#example_Cmd_Start),

cmd := exec.Command("sleep", "5")
err := cmd.Start()
if err != nil {
    log.Fatal(err)
}
log.Printf("Waiting for command to finish...")
err = cmd.Wait()
log.Printf("Command finished with error: %v", err)

是否可以提前3秒后终止该过程?

提前致谢

Answers:


134

终止运行exec.Process

// Start a process:
cmd := exec.Command("sleep", "5")
if err := cmd.Start(); err != nil {
    log.Fatal(err)
}

// Kill it:
if err := cmd.Process.Kill(); err != nil {
    log.Fatal("failed to kill process: ", err)
}

exec.Process超时后终止运行:

// Start a process:
cmd := exec.Command("sleep", "5")
if err := cmd.Start(); err != nil {
    log.Fatal(err)
}

// Wait for the process to finish or kill it after a timeout (whichever happens first):
done := make(chan error, 1)
go func() {
    done <- cmd.Wait()
}()
select {
case <-time.After(3 * time.Second):
    if err := cmd.Process.Kill(); err != nil {
        log.Fatal("failed to kill process: ", err)
    }
    log.Println("process killed as timeout reached")
case err := <-done:
    if err != nil {
        log.Fatalf("process finished with error = %v", err)
    }
    log.Print("process finished successfully")
}

该过程结束并且在done3秒钟内收到了错误(如果有的话),并且该程序在完成之前被终止了。


7
注意:杀死进程后,Wait()将返回。您应该从完成后撤出,err := cmd.Process.Kill()以防止内存泄漏。
史蒂芬·温伯格

3
@RhythmicFistman它将导致goroutine挂起,直到在没有人接收到它的情况下,试图发送的程序结束。更好的方法是将第一行更改为done := make(chan error, 1)。这将使发送立即成功,并且goroutine无需完成就退出。
Stephen Weinberg

1
@CharlieParker,可确保一个发送到通道的消息总是立即成功,并且发送者可以继续前进,或者在这种情况下退出。如果没有,它将等待某人接收。如果没有人这样做,它将永远等待并保留内存。
Stephen Weinberg 2014年

1
@KamilDziedzic,完成写入的目的是在继续执行程序之前确保程序已退出(等待返回)。在任何情况下,该选择语句之后程序都无法继续运行。如果您进行非阻塞写入,则将无法保证。
斯蒂芬·温伯格2014年

1
@StephenWeinberg我不明白。它等待的时间与原始代码完全相同cmd.Wait(),并且只有在以下情况下才会写入完成:尝试在主goroutine可以接收之前发送完成消息,因此,是的,需要缓冲chan。play.golang.org/p/Mx2gh9zMji但是我仍然不明白为什么你说它不会等待cmd完成
Kamil Dziedzic 2014年

13

关于调用的其他答案是正确的Kill(),但是有关超时后终止进程的部分现在已经过时了。

现在,可以使用context包和exec.CommandContext(从docs中的示例改编而成的示例)完成此操作:

func main() {
    ctx, cancel := context.WithTimeout(context.Background(), 100*time.Millisecond)
    defer cancel()

    if err := exec.CommandContext(ctx, "sleep", "5").Run(); err != nil {
        // This will fail after 100 milliseconds. The 5 second sleep
        // will be interrupted.
    }
}

从文档:

如果上下文在命令自行完成之前完成,则提供的上下文用于终止进程(通过调用os.Process.Kill)。

在之后Run()完成,你可以检查ctx.Err()。如果已达到超时,则返回的错误类型将为DeadLineExceeded。如果是nil,请检查errreturn byRun()以查看命令是否已完成而没有错误。


7

没有选择和渠道的简单版本。

func main() {
    cmd := exec.Command("cat", "/dev/urandom")
    cmd.Start()
    timer := time.AfterFunc(1*time.Second, func() {
        err := cmd.Process.Kill()
        if err != nil {
            panic(err) // panic as can't kill a process.
        }
    })
    err := cmd.Wait()
    timer.Stop()

    // read error from here, you will notice the kill from the 
    fmt.Println(err)
}

好吧,在咨询了一些有经验的go程序员之后,这显然不足以解决问题。因此,请参阅接受的答案。


这是一个更短的版本,非常简单。但是,如果超时时间较长,可能会有大量的goroutine挂起。

func main() {
    cmd := exec.Command("cat", "/dev/urandom")
    cmd.Start()
    go func(){
        time.Sleep(timeout)
        cmd.Process.Kill()
    }()
    return cmd.Wait()
}
By using our site, you acknowledge that you have read and understand our Cookie Policy and Privacy Policy.
Licensed under cc by-sa 3.0 with attribution required.