By

文章目录

这几天在用golang写一个工具，要执行外部命令工具，而且还要将外部命令工具输出的日志也要输出出来。网上找了一下，资料很多，关键是执行的结果成功或失败状态没找到好的方法获取到。刚开始想的是看错误日志，如果有错误日志，那么就是执行失败。测试的时候发现这样不行，发现有些时候会用error输出日志，但不一定就是执行失败。后来想用日志中的关键字匹配，因为有些命令执行成功或失败都是有关键字输出的，测试发现也不太好。最后没办法，看了一下Cmd.Wait()方法的实现，突然眼前一亮，找到方法了，有一个Cmd.ProcessState结构体可以使用。于是整理了一下，贴一下代码实现：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39

func Exec(name string, args ...string) error {
	cmd := exec.Command(name, args...)
	stderr, _ := cmd.StderrPipe()
	stdout, _ := cmd.StdoutPipe()
	if err := cmd.Start(); err != nil {
		log.Println("exec the cmd ", name, " failed")
		return err
	}

	// 正常日志
	logScan := bufio.NewScanner(stdout)
	go func() {
		for logScan.Scan() {
			log.Println(logScan.Text())
		}
	}()

	// 错误日志
	errBuf := bytes.NewBufferString("")
	scan := bufio.NewScanner(stderr)

	for scan.Scan() {
		s := scan.Text()
		log.Println("build error: ", s)
		errBuf.WriteString(s)
		errBuf.WriteString("\n")
	}

	// 等待命令执行完
	cmd.Wait()

	if !cmd.ProcessState.Success() {
		// 执行失败，返回错误信息
		return errors.New(errBuf.String())
	}

	return nil
}

注意：上面代码中没有处理输入的情况，如果待执行的命令需要输入数据，则不适用。

文章目录