```golang```中会经常遇到要 fork 子进程的需求。go 标准库为我们封装了 ```os/exec```标准包,当我们要运行外部命令时应该优先使用这个库。这里我简单结合```context``` 和 ```Cmd``` 模块写一个通用的执行 command 方法。代码如下:
```go
package main
import (
"context"
"os/exec"
"syscall"
)
func RunCmd(ctx context.Context, cmd *exec.Cmd) error {
cmd.SysProcAttr = &syscall.SysProcAttr{
Setpgid: true,
}
if err := cmd.Start(); err != nil {
return err
}
errCh := make(chan error, 1)
go func() {
errCh <- cmd.Wait()
}()
done := ctx.Done()
for {
select {
case <-done:
done = nil
pid := cmd.Process.Pid
if err := syscall.Kill(-1*pid, syscall.SIGKILL); err != nil {
return err
}
case err := <-errCh:
if done == nil {
return ctx.Err()
} else {
return err
}
}
}
}
```
说明:
* 可以通过 context 控制命令执行, 调用方可以调用```cancel``` 或者设置超时控制命令执行生命周期
* 如果进程执行失败, 应当 kill 整个进程组, 防止该进程 fork 的子进程逃逸
有疑问加站长微信联系(非本文作者))