如何理解exec.cmd是否被取消

Question

当命令被上下文取消时，我试图返回特定的错误。经过调查，ProcessState了解到，如果exitCode中得到-1，则进程得到终止信号https://golang.org/pkg/os/#ProcessState.ExitCode但也许我们有更优雅的方式？也许我可以把这个错误放在取消函数中？也许exitCode还不足以理解命令是否被取消？

var (
CmdParamsErr = errors.New("failed to get params for execution command")
ExecutionCanceled = errors.New("command canceled")
)
func execute(m My) error {
filePath, args, err := cmdParams(m)
err = nil
if err != nil {
log.Infof("cmdParams: err: %vn, m: %vn", err, m)
return CmdParamsErr
}
var out bytes.Buffer
var errStd bytes.Buffer
ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
defer cancel()
cmd := exec.CommandContext(ctx, filePath, args...)
cmd.Stdout = &out
cmd.Stderr = &errStd
err = cmd.Run()
if err != nil {
if cmd.ProcessState.ExitCode() == -1 {
log.Warnf("execution was canceled by signal, err: %vn", err)
err = ExecutionCanceled
return err
} else {
log.Errorf("run failed, err: %v, filePath: %v, args: %vn", err, filePath, args)
return err
}
}
return err
}

Answer 1

exec.ExitError没有提供退出代码的任何原因(没有相关的结构字段，也没有Unwrap方法(，因此您必须直接检查上下文：

if ctx.Err() != nil {
log.Println("canceled")
}   

请注意，这是一个轻微的竞争，因为上下文可能在命令失败后由于其他原因而被取消，但对此您无能为力。

Answer 2

没有直接或优雅的方法来判断进程是否因上下文被取消而终止。最接近的是：

func run() error {
ctx, cancel := context.WithTimeout(context.Background(), 1*time.Second)
defer cancel()
cmd := exec.CommandContext(ctx, "bash", "-c", "exit 1")
// Start() returns an error if the process can't be started. It will return
// ctx.Err() if the context is expired before starting the process.
if err := cmd.Start(); err != nil {
return err
}
if err := cmd.Wait(); err != nil {
if e, ok := err.(*exec.ExitError); ok {
// If the process exited by itself, just return the error to the
// caller.
if e.Exited() {
return e
}
// We know now that the process could be started, but didn't exit
// by itself. Something must have killed it. If the context is done,
// we can *assume* that it has been killed by the exec.Command.
// Let's return ctx.Err() so our user knows that this *might* be
// the case.
select {
case <-ctx.Done():
return ctx.Err()
default:
return e
}
}
return err
}
return nil
}

这里的问题是可能存在竞争条件，因此返回ctx.Err()可能会产生误导。例如，想象以下场景：

1. 过程开始
2. 这个过程被一个外部参与者扼杀了
3. 上下文已取消
4. 您检查上下文

此时，上面的函数将返回ctx.Err()，但这可能会产生误导，因为进程被终止的原因并不是因为上下文被取消。如果您决定使用类似于上面函数的代码，请记住这个近似值。