当前位置:主页 > 脚本语言 > Golang >

golang执行命令获取执行结果状态(推荐)

时间: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
}

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

总结

您可能感兴趣的文章:

相关文章