如何调用外部程序并处理其输出？

Question

我正在尝试调用外部命令（例如，seq 10）并获取其输出，处理输出，然后打印出处理后的结果。但是下面的代码不起作用。你能告诉我如何让它工作吗？

// vim: set noexpandtab tabstop=2:

package main

import (
    "bufio"
    "io"
    "os"
    "log"
    "os/exec"
)

func main() {
    cmd := exec.Command("seq", "10")
    stdin := bufio.NewReader(cmd.Stdout)
    err := cmd.Run()
    if err != nil {
        log.Fatalf("cmd.Run() failed with %s\n", err)
    }

    for {
        line, err := stdin.ReadBytes('\n')
        if err == io.EOF {
            if len(line) == 0 { break }
        } else {
            if err != nil { log.Fatal(err) }
            line = line[:(len(line)-1)]
        }
        os.Stdout.Write(line)
        os.Stdout.Write([]byte{'\n'})
    }
}

$ $ go run main.go 
# command-line-arguments
./main.go:15:30: cannot use cmd.Stdout (type io.Writer) as type io.Reader in argument to bufio.NewReader:
    io.Writer does not implement io.Reader (missing Read method)

编辑：我也试过这个。但它也有错误。谁能给我一个可行的例子。

// vim: set noexpandtab tabstop=2:

package main

import (
    "bufio"
    "io"
    "os"
    "log"
    "os/exec"
)

func main() {
    cmd := exec.Command("seq", "10")
    stdout, err := cmd.StdoutPipe()
    if err != nil { log.Fatal(err) }

    stdin := bufio.NewReader(stdout)
    err = cmd.Run()
    if err != nil {
        log.Fatalf("cmd.Run() failed with %s\n", err)
    }

    for {
        line, err := stdin.ReadBytes('\n')
        if err == io.EOF {
            if len(line) == 0 { break }
        } else {
            if err != nil { log.Fatal(err) }
            line = line[:(len(line)-1)]
        }
        os.Stdout.Write(line)
        os.Stdout.Write([]byte{'\n'})
    }
}

Answer 1

另一种方法（更简洁的方法）是使用bufio.Scanner 自动处理\n（或任何其他分隔符）。另一个优点是这种方法没有种族问题（去过那里，做过）：

package main

import (
    "bufio"
    "fmt"
    "log"
    "os/exec"
)

func main() {
    cmd := exec.Command("seq", "10")
    stdout, err := cmd.StdoutPipe()
    if err != nil {
        log.Fatal(err)
    }
    err = cmd.Start()
    if err != nil {
        log.Fatalf("cmd.Start() failed with %s\n", err)
    }

    stdin := bufio.NewScanner(stdout)
    for stdin.Scan() {
        fmt.Println(stdin.Text())
    }
    cmd.Wait()
}

stdin.Scan() 在EOF 上返回 false，这是在进程退出时给出的。 cmd.Wait() 将关闭 StdoutPipe，您可以阅读 err.(exec.ExitError).ExitCode() 以获取退出代码（如果退出非零）。

Answer 2

您需要使用.StdoutPipe() 将标准输出到阅读器，还需要使用exec.Command(..).Start() 进行增量阅读（.Run() 等待进程退出）。 这是工作代码：

// vim: set noexpandtab tabstop=2:

package main

import (
    "bufio"
    "fmt"
    "io"
    "log"
    "os"
    "os/exec"
)

func main() {
    cmd := exec.Command("seq", "10")
    cmdStdOut, err := cmd.StdoutPipe()
    cmdStdErr, err := cmd.StderrPipe()
    defer cmdStdOut.Close()
    if err != nil {
        log.Fatalf("command failed with %s\n", err)
    }
    stdoutReader := bufio.NewReader(cmdStdOut)
    stderrReader := bufio.NewReader(cmdStdErr)
    err = cmd.Start()
    if err != nil {
        log.Fatalf("cmd.Run() failed with %s\n", err)
    }
    // Read stdout
    for {
        line, err := stdoutReader.ReadBytes('\n')
        if err == io.EOF {
            if len(line) == 0 {
                break
            }
        } else {
            if err != nil {
                log.Fatal(err)
            }
            line = line[:(len(line) - 1)]
        }
        os.Stdout.Write(line)
        os.Stdout.Write([]byte{'\n'})
    }
    // Read stderr
    for {
        line, err := stderrReader.ReadBytes('\n')
        if err == io.EOF {
            if len(line) == 0 {
                break
            }
        } else {
            if err != nil {
                log.Fatal(err)
            }
            line = line[:(len(line) - 1)]
        }
        os.Stderr.Write(line)
        os.Stderr.Write([]byte{'\n'})
    }
    cmd.Wait()
    fmt.Println(cmd.ProcessState.ExitCode())
}