gbase/cmd/command_line.go

package cmd

import (
	"context"
	"errors"
	"os/exec"
	"time"
)

//execute a command with timeout

func CommandWithTimeout(name string, arg ...string) (string, error) {
	ctxt, cancel := context.WithTimeout(context.Background(), 60*5*time.Second)
	defer cancel() //releases resources if slowOperation completes before timeout elapses

	cmd := exec.CommandContext(ctxt, name, arg...)
	//当经过Timeout时间后，程序依然没有运行完，则会杀掉进程，ctxt也会有err信息
	if out, err := cmd.Output(); err != nil {
		//检测报错是否是因为超时引起的
		if ctxt.Err() != nil && ctxt.Err() == context.DeadlineExceeded {
			return "", errors.New("command timeout")

		}
		return string(out), err
	} else {
		return string(out), nil
	}
}