moredoc/util/command/command_linux.go

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
	})
}