流式命令输出进度

怎么入门Golang编程？需要学习哪些知识点？这是新手们刚接触编程时常见的问题；下面golang学习网就来给大家整理分享一些知识点，希望能够给初学者一些帮助。本篇文章就来介绍《流式命令输出进度》，涉及到golang，有需要的可以收藏一下

问题内容

我正在编写一项服务，该服务必须将已执行命令的输出流式传输到父级和日志。当有一个漫长的过程时，问题是cmd.StdoutPipe给了我一个最终的（字符串）结果。

是否可以提供正在发生的事情的部分输出，例如在 shell 中

func main() {
    cmd := exec.Command("sh", "-c", "some long runnig task")

    stdout, _ := cmd.StdoutPipe()
    cmd.Start()

    scanner := bufio.NewScanner(stdout)
    for scanner.Scan() {
        m := scanner.Text()
        fmt.Println(m)
        log.Printf(m)
    }

    cmd.Wait()
}

PS只是输出将是：

cmd.Stdout = os.Stdout

但就我而言，这还不够。

正确答案

您发布的代码有效（执行了合理的命令）。

这是一个用 Go 编写的简单的 “一些长时间运行的任务” ，供您调用和测试您的代码：

func main() {
    fmt.Println("Child started.")
    time.Sleep(time.Second*2)
    fmt.Println("Tick...")
    time.Sleep(time.Second*2)
    fmt.Println("Child ended.")
}

编译它并调用它作为你的命令。您将看到不同的行立即显示为子进程所写的“流式传输”。

它可能不适合您的原因

Scanner返回的读取bufio.NewScanner()整行，并且仅在遇到换行符时才返回某些内容（由bufio.ScanLines()函数定义）。

如果您执行的命令不打印换行符，则不会立即返回其输出（仅在打印换行符时，内部缓冲区已填满或进程结束）。

可能的解决方法

如果您不能保证子进程打印换行符但您仍想流式输出，则无法读取整行。一种解决方案是按单词阅读，甚至按字符（符文）阅读。您可以通过使用以下方法设置不同的拆分函数Scanner.Split()来实现此目的：

scanner := bufio.NewScanner(stdout)
scanner.Split(bufio.ScanRunes)

该函数通过sbufio.ScanRunes读取输入，因此只要有新的可用就会返回。runeScanner.Scan()rune

或者在没有 a 的情况下手动读取Scanner（在本例中是逐字节）：

oneByte := make([]byte, 1)
for {
    _, err := stdout.Read(oneByte)
    if err != nil {
        break
    }
    fmt.Printf("%c", oneByte[0])
}

请注意，上面的代码会rune错误地读取 UTF-8 编码中的多个字节。要读取多个 UTF-8 字节符文，我们需要更大的缓冲区：

oneRune := make([]byte, utf8.UTFMax)
for {
    count, err := stdout.Read(oneRune)
    if err != nil {
        break
    }
    fmt.Printf("%s", oneRune[:count])
}

要记住的事情

进程有标准输出和标准错误的默认缓冲区（通常大小为几 KB）。如果一个进程写入标准输出或标准错误，它会进入相应的缓冲区。如果此缓冲区已满，则进一步的写入 将阻塞 （在子进程中）。如果您不读取子进程的标准输出和标准错误，则如果缓冲区已满，您的子进程可能会挂起。

因此建议始终读取子进程的标准输出和错误。即使您知道该命令通常不会 写入 其标准错误，如果发生某些错误，它也可能会开始将错误消息转储到其标准错误。

编辑： 正如 Dave C 提到的，默认情况下，子进程的标准输出和错误流被丢弃，如果不读取也不会导致阻塞/挂起。但是，如果不读取错误流，您可能会错过流程中的一两件事。

文中关于golang的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《流式命令输出进度》文章吧，也可关注golang学习网公众号了解相关技术文章。