首页 >后端开发 >Golang >如何从 Go 中执行的命令流式传输实时输出?

如何从 Go 中执行的命令流式传输实时输出?

Mary-Kate Olsen
Mary-Kate Olsen原创
2024-11-24 05:48:09752浏览

How to Stream Live Output from Executed Commands in Go?

执行命令的实时输出

此代码片段展示了在 Go 中运行命令并检索其输出时经常遇到的问题:输出变为仅在命令执行完成后才可用。为了解决这个问题,提供了一个修改后的解决方案,允许实时流式传输命令的输出。

cmdParams := [][]string{
    {filepath.Join(dir, path), "npm", "install"},
    {filepath.Join(pdir, n.path), "gulp"},
}

for _, cp := range cmdParams {
    log.Printf("Starting %s in folder %s...", cp[1:], cp[0])
    cmd := exec.Command(cp[1], cp[2:]...)
    cmd.Dir = cp[0]

    stdout, err := cmd.StdoutPipe()
    if err != nil {
        log.Printf("%s cmd.StdoutPipe() error: %v\n", cp[1:], err)
        return
    }
    // Start command:
    if err = cmd.Start(); err != nil {
        log.Printf("%s start error: %v\n", cp[1:], err)
        return
    }

    // Stream command output:
    scanner := bufio.NewScanner(stdout)
    scanner.Split(bufio.ScanRunes)
    for scanner.Scan() {
        fmt.Print(scanner.Text())
    }
    if scanner.Err() != nil {
        log.Printf("Reading %s stdout error: %v\n", cp[1:], err)
        return
    }

    // Get execution success or failure:
    if err = cmd.Wait(); err != nil {
        log.Printf("Error running %s: %v\n", cp[1:], err)
        return
    }
    log.Printf("Finished %s", cp[1:])
}

解释:

  • cmd.StdoutPipe()创建一个管道,允许我们读取命令的标准输出实时。
  • bufio.NewScanner(stdout) 创建一个扫描仪,通过符文(字符)从输出管道读取。
  • scanner.Scan() 读取并打印每个字符。可用。
  • 循环继续读取和打印字符,直到命令完成或发生错误。
  • 命令之后完成后,打印剩余的输出,并检索命令的退出状态。

附加说明:

  • 此解决方案仅流式传输标准输出。标准错误也可以使用 Command.StderrPipe() 进行流式传输。
  • 某些命令可能不会将所有内容写入其标准输出或错误。在这种情况下,输出可能无法正确流式传输。

以上是如何从 Go 中执行的命令流式传输实时输出?的详细内容。更多信息请关注PHP中文网其他相关文章!

声明:
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn