在go语言中实现捕获一个子进程的stdout并根据结果进行处理代码示例

Go  /  管理员 发布于 1年前   641

假设你想在主程序中执行一个子进程的shell命令，并想通过stdout管道捕获子进程的输出。

下面看看如何执行一个子进程，捕获输出，将输出转换成一个片断，然后根据子进程的输出进行处理。

示例代码：

 package main
 
 import (
         "bufio"
         "fmt"
         "io"
         "os"
         "os/exec"
         "time"
 )
 
 func handleError(err error) {
         if err != nil {
                 fmt.Println(err)
                 os.Exit(1)
         }
 }
 
 func main() {
         // ls -la命令可能不是最好的，因为它总是会返回一些东西
         // 并导致allText(下文)的长度超过1。
         cmd := exec.Command("ls", "-la")
         
         //捕获输出和错误管道
         stdout, err := cmd.StdoutPipe()
         handleError(err)
         stderr, err := cmd.StderrPipe()
         handleError(err)
         
         //开始
         err = cmd.Start()
         handleError(err)
         
         //在我们的命令运行完毕之前，不要让main()退出。
         defer cmd.Wait() // Doesn't block
         
         //非阻塞性地将命令输出回传到终端
         //go io.Copy(os.Stdout, stdout) // <----，因为我们将用buff.Scan()打印出来。
         go io.Copy(os.Stderr, stderr)
         
         // 这里是我们想知道孩子的输出的地方
         // 并根据输出的情况，显示不同的结果
         // 在主程序中
         buff := bufio.NewScanner(stdout)
         var allText []string
         for buff.Scan() {
                 allText = append(allText, buff.Text()+"\n")
         }
         
         if len(allText) > 1 {
         
                 // 如果allText有值，就进行倒计时
                 ticker := time.Tick(time.Second)
                 fmt.Println("扫描仪检测到了一些东西。寻找倒计时...\n\n")
                 for i := 9; i >= 0; i-- {
                         <-ticker
                         
                         //fmt.Printf("\x0cOn 10/%d", i) // 使用 \x0c 为 play.golang.org。
                         fmt.Printf("\r还在计算10/%d", i) //如果你在终端运行这个程序，请使用 \r
                 }
                 
                 fmt.Println("\n结束.\n")
         } else {
                 fmt.Println("返回零")
         }
 }