时间:2021-09-23 08:46:05 | 栏目:Golang | 点击:次
这几天在用golang写一个工具,要执行外部命令工具,而且还要将外部命令工具输出的日志也要输出出来。网上找了一下,资料很多,关键是执行的结果成功或失败状态没找到好的方法获取到。
刚开始想的是看错误日志,如果有错误日志,那么就是执行失败。测试的时候发现这样不行,发现有些时候会用error输出日志,但不一定就是执行失败。后来想用日志中的关键字匹配,因为有些命令执行成功或失败都是有关键字输出的,测试发现也不太好。
最后没办法,看了一下Cmd.Wait()
方法的实现,突然眼前一亮,找到方法了,有一个Cmd.ProcessState
结构体可以使用。于是整理了一下,贴一下代码实现:
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 }
注意:上面代码中没有处理输入的情况,如果待执行的命令需要输入数据,则不适用。
总结