Go 实用技巧:执行外部程序,像 tail -f 一样获取流式输出
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
}
}