流命令输出 Goroutine 的进度

Question

流命令输出进度问题解决了长时间运行命令的打印进度问题。

我试图将打印代码放在一个 goroutine 中，但扫描器声称已经EOF立即命中并且永远不会执行 for 块。

第bufio.scan一次执行该Scan()方法时执行的代码是：

    // We cannot generate a token with what we are holding.
    // If we've already hit EOF or an I/O error, we are done.
    if s.err != nil {
        // Shut it down.
        s.start = 0
        s.end = 0
        return false
    }

如果我打印s.err输出是EOF.

我试图运行的代码是：

    // We cannot generate a token with what we are holding.
    // If we've already hit EOF or an I/O error, we are done.
    if s.err != nil {
        // Shut it down.
        s.start = 0
        s.end = 0
        return false
    }

这个想法是启动 Goroutine，获取cmd.stdout，等待cmd启动，然后开始处理它的输出。

结果是执行长命令并且程序等待其完成，但没有任何内容打印到终端。

知道为什么第一次scanner.Scan()调用stdout已经达到的时间EOF吗？

Answer 1

有一些问题：

• 在读取所有数据之前关闭管道。
• 始终检查错误
• cmd.Start()之后开始c <- struct{}{}并使用无缓冲通道c := make(chan struct{})

两个工作示例代码：

1：使用 channel 等待，然后在EOF使用后关闭管道defer func() { c <- struct{}{} }()，就像这个工作示例代码：

package main

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

func main() {
    cmd := exec.Command("Streamer")
    c := make(chan struct{})

    go run(cmd, c)

    c <- struct{}{}
    cmd.Start()

    <-c
    if err := cmd.Wait(); err != nil {
        fmt.Println(err)
    }
    fmt.Println("done.")
}

func run(cmd *exec.Cmd, c chan struct{}) {
    defer func() { c <- struct{}{} }()
    stdout, err := cmd.StdoutPipe()
    if err != nil {
        panic(err)
    }
    <-c
    scanner := bufio.NewScanner(stdout)
    for scanner.Scan() {
        m := scanner.Text()
        fmt.Println(m)
    }
    fmt.Println("EOF")
}

2：你也可以等待使用sync.WaitGroup，就像这个工作示例代码：

package main

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

var wg sync.WaitGroup

func main() {
    cmd := exec.Command("Streamer")
    c := make(chan struct{})
    wg.Add(1)
    go func(cmd *exec.Cmd, c chan struct{}) {
        defer wg.Done()
        stdout, err := cmd.StdoutPipe()
        if err != nil {
            panic(err)
        }
        <-c
        scanner := bufio.NewScanner(stdout)
        for scanner.Scan() {
            m := scanner.Text()
            fmt.Println(m)
        }
    }(cmd, c)

    c <- struct{}{}
    cmd.Start()

    wg.Wait()
    fmt.Println("done.")
}

和 Streamer 示例代码（仅用于测试）：

package main

import "fmt"
import "time"

func main() {
    for i := 0; i < 10; i++ {
        time.Sleep(1 * time.Second)
        fmt.Println(i, ":", time.Now().UTC())
    }
}

并查看func (c *Cmd) StdoutPipe() (io.ReadCloser, error) 文档：

StdoutPipe 返回一个管道，该管道将在命令启动时连接到命令的标准输出。

Wait 会在看到命令退出后关闭管道，所以大多数调用者不需要自己关闭管道；然而，这意味着在管道的所有读取完成之前调用 Wait 是不正确的。出于同样的原因，使用 StdoutPipe 时调用 Run 是不正确的。有关惯用用法，请参见示例。