|
|
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
|
|
|
}
|
|
|
|
|
|
pid := 0
|
|
|
if cmd.Process != nil && cmd.Process.Pid != 0 {
|
|
|
pid = cmd.Process.Pid
|
|
|
pidMap.Store(pid, pid)
|
|
|
}
|
|
|
defer func() {
|
|
|
if pid != 0 {
|
|
|
pidMap.Delete(pid)
|
|
|
}
|
|
|
}()
|
|
|
|
|
|
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
|
|
|
}
|
|
|
|
|
|
// 当主程序退出时，从pidMap中获取所有的pid，然后kill掉
|
|
|
func CloseChildProccess() {
|
|
|
pidMap.Range(func(key, value interface{}) bool {
|
|
|
if pid, ok := value.(int); ok {
|
|
|
fmt.Println("kill pid:", pid)
|
|
|
syscall.Kill(-pid, syscall.SIGKILL)
|
|
|
}
|
|
|
return true
|
|
|
})
|
|
|
}
|
