2017-05-05 05:44:44 +00:00
|
|
|
|
// +build windows
|
|
|
|
|
|
|
|
|
|
package utils
|
|
|
|
|
|
|
|
|
|
import (
|
2017-09-16 09:58:33 +00:00
|
|
|
|
"errors"
|
|
|
|
|
"golang.org/x/net/context"
|
|
|
|
|
"os/exec"
|
|
|
|
|
"strconv"
|
|
|
|
|
"syscall"
|
2017-05-05 05:44:44 +00:00
|
|
|
|
)
|
|
|
|
|
|
2017-05-26 13:59:01 +00:00
|
|
|
|
type Result struct {
|
2017-09-16 09:58:33 +00:00
|
|
|
|
output string
|
|
|
|
|
err error
|
2017-05-26 13:59:01 +00:00
|
|
|
|
}
|
|
|
|
|
|
2017-05-05 05:44:44 +00:00
|
|
|
|
// 执行shell命令,可设置执行超时时间
|
2017-09-16 09:58:33 +00:00
|
|
|
|
func ExecShell(ctx context.Context, command string) (string, error) {
|
|
|
|
|
cmd := exec.Command("cmd", "/C", command)
|
|
|
|
|
// 隐藏cmd窗口
|
|
|
|
|
cmd.SysProcAttr = &syscall.SysProcAttr{
|
|
|
|
|
HideWindow: true,
|
|
|
|
|
}
|
|
|
|
|
var resultChan chan Result = make(chan Result)
|
|
|
|
|
go func() {
|
|
|
|
|
output, err := cmd.CombinedOutput()
|
|
|
|
|
resultChan <- Result{string(output), err}
|
|
|
|
|
}()
|
|
|
|
|
select {
|
|
|
|
|
case <-ctx.Done():
|
|
|
|
|
if cmd.Process.Pid > 0 {
|
|
|
|
|
exec.Command("taskkill", "/F", "/T", "/PID", strconv.Itoa(cmd.Process.Pid)).Run()
|
|
|
|
|
cmd.Process.Kill()
|
|
|
|
|
}
|
|
|
|
|
return "", errors.New("timeout killed")
|
|
|
|
|
case result := <-resultChan:
|
|
|
|
|
return ConvertEncoding(result.output), result.err
|
|
|
|
|
}
|
2017-05-05 05:44:44 +00:00
|
|
|
|
}
|
|
|
|
|
|
2017-09-16 09:58:33 +00:00
|
|
|
|
func ConvertEncoding(outputGBK string) string {
|
|
|
|
|
// windows平台编码为gbk,需转换为utf8才能入库
|
|
|
|
|
outputUTF8, ok := GBK2UTF8(outputGBK)
|
|
|
|
|
if ok {
|
|
|
|
|
return outputUTF8
|
|
|
|
|
}
|
2017-05-26 10:09:07 +00:00
|
|
|
|
|
2017-09-16 09:58:33 +00:00
|
|
|
|
return "命令输出转换编码失败(gbk to utf8)"
|
|
|
|
|
}
|