command_windows.go

package command

import (
	"bytes"
	"errors"
	"fmt"
	"os"
	"os/exec"
	"strings"
	"syscall"
	"time"
)

// ExecCommand 执行cmd命令操作
func execCommand(name string, args []string, opt ...Option) (out string, err error) {
	var (
		stderr, stdout bytes.Buffer
		expire         = 30 * time.Minute
		errs           []string
		cmd            = exec.Command(name, args...)
	)

	if len(opt) > 0 {
		if opt[0].Timeout != nil {
			expire = *opt[0].Timeout
		}
		if opt[0].Dir != "" {
			cmd.Dir = opt[0].Dir
		}
	}

	// 需要加上这一行，以便在windows下能够正常执行，而不会不停弹窗
	cmd.SysProcAttr = &syscall.SysProcAttr{HideWindow: true}
	if len(opt) > 0 && opt[0].Stdout != nil {
		cmd.Stdout = opt[0].Stdout
	} else {
		cmd.Stdout = &stdout
	}

	if len(opt) > 0 && opt[0].Stderr != nil {
		cmd.Stderr = opt[0].Stderr
	} else {
		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)
		if len(opt) > 0 && opt[0].Callback != nil {
			opt[0].Callback(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())))
			if proc, errProc := os.FindProcess(cmd.Process.Pid); errProc == nil {
				proc.Kill()
				proc.Release()
			}
			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
}

func closeProcess(pid int) {
	fmt.Println("kill pid:", pid)
	if proc, err := os.FindProcess(pid); err == nil {
		proc.Kill()
		proc.Release()
		pidMap.Delete(pid)
	}
}