You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
54 lines
1.1 KiB
54 lines
1.1 KiB
package command
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"fmt"
|
|
"os/exec"
|
|
"strings"
|
|
"syscall"
|
|
"time"
|
|
)
|
|
|
|
// ExecCommand 执行cmd命令操作
|
|
func ExecCommand(name string, args []string, timeout ...time.Duration) (out string, err error) {
|
|
var (
|
|
stderr, stdout bytes.Buffer
|
|
expire = 30 * time.Minute
|
|
errs []string
|
|
)
|
|
|
|
if len(timeout) > 0 {
|
|
expire = timeout[0]
|
|
}
|
|
|
|
cmd := exec.Command(name, args...)
|
|
cmd.SysProcAttr = &syscall.SysProcAttr{Setpgid: true}
|
|
cmd.Stdout = &stdout
|
|
cmd.Stderr = &stderr
|
|
err = cmd.Start()
|
|
if err != nil {
|
|
err = fmt.Errorf("%s\n%s", err.Error(), stderr.String())
|
|
return
|
|
}
|
|
|
|
time.AfterFunc(expire, func() {
|
|
if cmd.Process != nil && cmd.Process.Pid != 0 {
|
|
errs = append(errs, fmt.Sprintf("execute timeout: %d min.", int(expire.Minutes())))
|
|
syscall.Kill(-cmd.Process.Pid, syscall.SIGKILL)
|
|
cmd.Process.Kill()
|
|
}
|
|
})
|
|
|
|
err = cmd.Wait()
|
|
if err != nil {
|
|
errs = append(errs, err.Error(), stderr.String())
|
|
}
|
|
out = stdout.String()
|
|
if len(errs) > 0 {
|
|
errs = append(errs, out)
|
|
err = errors.New(strings.Join(errs, "\n\r"))
|
|
}
|
|
return
|
|
}
|