读取Go中StdoutPipe()的内容

最近发现不少小伙伴都对Golang很感兴趣，所以今天继续给大家介绍Golang相关的知识，本文《读取Go中StdoutPipe()的内容》主要内容涉及到等等知识点，希望能帮到你！当然如果阅读本文时存在不同想法，可以在评论中表达，但是请勿使用过激的措辞~

我试图从命令的标准输出中读取数据，但每（大约）50 次它就会冻结。

func runProcess(process *exec.Cmd) (string, string, error) {
    var stdout strings.Builder
    var stderr string
    process := exec.Command(programPath, params...)

    go func() {
        pipe, err := process.StderrPipe()
        if err != nil {
            return
        }

        buf, err := io.ReadAll(pipe)
        if err != nil {
            log.Warn("Error reading stderr: %v", err)
        }
        stderr = string(buf)
    }()

    pipe, err := process.StdoutPipe()
    if err = process.Start(); err != nil {
        return "", "", err
    }

    buf := make([]byte, 1024)
    read, err := pipe.Read(buf)            // it reads correctly from the pipe
    for err == nil && read > 0 {
        _, err = stdout.Write(buf[:read])
        read, err = pipe.Read(buf)         // this is where is stalls
    }
    if err = process.Wait(); err != nil {
        return stdout.String(), stderr, err
    }

    return stdout.String(), stderr, nil
}

我尝试使用 stdout, err := io.readall(pipe) 一次读取所有内容而不是读取块，但我得到了相同的行为。 调用的程序似乎执行成功。它的日志文件已创建并且完整。另外，当我第一次从管道读取时（在循环之前），所有输出都在那里。但在循环内部，当第二次调用 .read() 并且它应该返回 eof（输出小于 1024 字节）时，它会冻结。

正确答案

这段代码中有很多竞争条件。一般来说，如果你创建一个 goroutine，应该有某种同步——比如 chan、sync.mutex、sync.waitgroup 或原子。

修复竞争条件。

1. 在调用 start() 之前调用 stderrpipe()。该代码不会执行此操作。

2. 等待 goroutine 完成后再返回。

竞争条件可能会破坏 exec.cmd 结构...这可能意味着它泄漏管道，这可以解释为什么 read() 挂起（因为管道的写入端未关闭）。 p>

根据经验，始终修复竞争条件。将它们视为高优先级错误。

以下是如何在没有竞争条件的情况下编写它的草图：

func runprocess(process *exec.cmd) (stdout, stderr string, err error) {
    outpipe, err := process.stdoutpipe()
    if err != nil {
        return "", "", err
    }

    // call stderrpipe before start().
    // easy way to do it: outside the goroutine.
    errpipe, err := process.stderrpipe()
    if err != nil {
        return "", "", err
    }

    // start process.
    if err := process.start(); err != nil {
        return "", "", err
    }

    // read stderr in goroutine.
    var wg sync.waitgroup
    var stderrerr error
    wg.add(1)
    go func() {
        defer wg.done()
        data, err := ioutil.readall(errpipe)
        if err != nil {
            stderrerr = err
        } else {
            stderr = string(data)
        }
    }()

    // read stdout in main thread.
    data, stdouterr := ioutil.readall(outpipe)

    // wait until we are done reading stderr.
    wg.wait()

    // wait for process to finish.
    if err := process.wait(); err != nil {
        return "", "", err
    }

    // handle error from reading stdout.
    if stdouterr != nil {
        return "", "", stderrerr
    }
    // handle error from reading stderr.
    if stderrerr != nil {
        return "", "", stderrerr
    }

    stdout = string(data)
    return stdout, stderr, nil
}

代码更简单

所有这些都是由 os/exec 包自动完成的。您可以将任何 io.writer 用于 stdout 和 stderr，您不限于 *os.file。

func runProcess(process *exec.Cmd) (stdout, stderr string, err error) {
    var stdoutbuf, stderrbuf bytes.Buffer
    process.Stdout = &stdoutbuf
    process.Stderr = &stderrbuf
    if err := process.Run(); err != nil {
        return "", "", err
    }
    return stdoutbuf.String(), stderrbuf.String(), nil
}

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《读取Go中StdoutPipe()的内容》文章吧，也可关注golang学习网公众号了解相关技术文章。