标签 golang 下的文章

Go 在使用 os/exec 执行外部命令的时候，假如外部命令是持续输出的，该如何实时、持续获取外部命令的输出呢？

其实很简单，只需要通过 StdoutPipe() 创建一个管道接收外部命令的标准输出即可。

先来准备一个外部命令程序 ./hello/main.cc：

#include "iostream"
#include <thread>

int main() {
    while (true) {
        auto now = std::chrono::system_clock::now();
        std::time_t now_c = std::chrono::system_clock::to_time_t(now);
        std::cout << "Current time is: " << std::ctime(&now_c) << std::flush;
        // sleep for a second
        std::this_thread::sleep_for(std::chrono::seconds(1));
    }
}

然后用 g++ 编译：g++ -o hello main.cc 得到 ./hello/hello，这个程序会每隔一秒输出当前时间。

接下来开始写我们的 Go 程序：

package main

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

func main() {
    cmd := exec.Command("./hello/hello")

    cmdStdout, err := cmd.StdoutPipe() // 创建一个管道接收外部命令的标准输出
    if err != nil {
        fmt.Println("Error creating StdoutPipe:", err)
        return
    }

    if err := cmd.Start(); err != nil {
        fmt.Println("Error starting command:", err)
        return
    }

    // 用协程非阻塞接收外部命令的流式输出
    go func() {
        for {
            buf := make([]byte, 1024)
            n, err := cmdStdout.Read(buf)
            if err != nil {
                fmt.Println("Error reading from stdout:", err)
                break
            }

            if n == 0 {
                continue
            }

            // 按需把尾部的换行符、或者是 '\0' （在 Go 里是 "\x00"）去除掉
            output := strings.TrimSuffix(string(buf[:n]), "\n")
            fmt.Println("Output:", output)
        }
    }()

    if err := cmd.Wait(); err != nil {
        fmt.Println("Error waiting for command:", err)
        return
    }
}