
本文详细介绍了如何在go语言中启动一个外部子进程(如python脚本),并通过标准输入(stdin)和标准输出(stdout)实现实时的双向通信。教程涵盖了使用`os/exec`包创建命令、建立输入输出管道、启动进程、进行循环读写操作以及正确处理错误和等待进程结束的关键步骤,旨在帮助开发者构建高效的进程间通信解决方案。
在现代软件开发中,Go程序经常需要与其他语言编写的工具或服务进行交互。其中一种常见的模式是通过启动子进程,并利用其标准输入(stdin)和标准输出(stdout)进行实时数据交换。本教程将以一个Go程序与Python脚本通信的例子,详细讲解如何实现这一机制。
首先,我们需要一个能够接收标准输入并返回结果的子进程。这里我们使用一个简单的Python脚本 add.py,它会持续读取一行输入,将其作为表达式求值,并将结果写入标准输出。
# add.py
import sys
while True:
try:
# 读取一行输入,移除末尾换行符,并进行求值
line = sys.stdin.readline().strip()
if not line: # 如果读取到空行,可能是输入结束,退出循环
break
result = eval(line)
sys.stdout.write('%s\n' % result)
sys.stdout.flush() # 立即刷新缓冲区,确保Go程序能及时读取
except Exception as e:
sys.stderr.write(f"Error: {e}\n")
sys.stderr.flush()
break # 发生错误时退出脚本说明:
接下来,我们将编写Go程序来启动 add.py 脚本,并与其进行交互。
立即学习“go语言免费学习笔记(深入)”;
使用os/exec包来创建命令对象,并建立与子进程标准输入输出对应的管道。
package main
import (
"bufio"
"fmt"
"log"
"os/exec"
)
func main() {
// 创建命令对象。-u 标志对于Python是重要的,它强制标准输出和标准错误不进行缓冲。
c := exec.Command("python", "-u", "add.py")
// 获取子进程的标准输入管道
si, err := c.StdinPipe()
if err != nil {
log.Fatal("获取StdinPipe失败:", err)
}
// 获取子进程的标准输出管道
so, err := c.StdoutPipe()
if err != nil {
log.Fatal("获取StdoutPipe失败:", err)
}
// 使用bufio.NewReader包装StdoutPipe,以便按行读取
reader := bufio.NewReader(so)
// 启动子进程
err = c.Start()
if err != nil {
log.Fatal("启动子进程失败:", err)
}
// ... 后续读写操作
}关键点:
一旦子进程启动,我们就可以在一个循环中向其写入数据,并读取其响应。
// ... (接上文代码)
// 现在进行一些数学运算,与子进程交互10次
for i := 0; i < 10; i++ {
// 构造要发送的表达式,并添加换行符
expr := fmt.Sprintf("2+%d\n", i)
// 写入数据到子进程的stdin
_, err = si.Write([]byte(expr))
if err != nil {
log.Fatal("写入stdin失败:", err)
}
// 从子进程的stdout读取一行响应
answer, err := reader.ReadString('\n')
if err != nil {
log.Fatal("读取stdout失败:", err)
}
// 打印结果
fmt.Printf("对表达式 %q 的答案是 %q\n", expr, answer)
}
// ... (后续关闭和等待)关键点:
完成所有交互后,必须关闭管道并等待子进程退出,以释放资源并避免僵尸进程。
// ... (接上文代码)
// 关闭输入管道,通知子进程不再有更多输入
// 这会导致Python脚本的sys.stdin.readline()返回空字符串,从而退出循环
err = si.Close()
if err != nil {
log.Println("关闭StdinPipe失败:", err) // 记录错误但继续
}
// 关闭输出管道(可选,但良好实践)
err = so.Close()
if err != nil {
log.Println("关闭StdoutPipe失败:", err) // 记录错误但继续
}
// 等待子进程退出并获取其退出状态
err = c.Wait()
if err != nil {
log.Fatal("等待子进程退出失败:", err)
}
fmt.Println("子进程已成功退出。")
}关键点:
将上述所有片段组合起来,就得到了一个完整的Go程序:
package main
import (
"bufio"
"fmt"
"log"
"os/exec"
)
func main() {
// 创建命令对象。-u 标志对于Python是重要的,它强制标准输出和标准错误不进行缓冲。
c := exec.Command("python", "-u", "add.py")
// 获取子进程的标准输入管道
si, err := c.StdinPipe()
if err != nil {
log.Fatal("获取StdinPipe失败:", err)
}
// 获取子进程的标准输出管道
so, err := c.StdoutPipe()
if err != nil {
log.Fatal("获取StdoutPipe失败:", err)
}
// 使用bufio.NewReader包装StdoutPipe,以便按行读取
reader := bufio.NewReader(so)
// 启动子进程
err = c.Start()
if err != nil {
log.Fatal("启动子进程失败:", err)
}
// 现在进行一些数学运算,与子进程交互10次
for i := 0; i < 10; i++ {
// 构造要发送的表达式,并添加换行符
expr := fmt.Sprintf("2+%d\n", i)
// 写入数据到子进程的stdin
_, err = si.Write([]byte(expr))
if err != nil {
log.Fatal("写入stdin失败:", err)
}
// 从子进程的stdout读取一行响应
answer, err := reader.ReadString('\n')
if err != nil {
log.Fatal("读取stdout失败:", err)
}
// 打印结果
fmt.Printf("对表达式 %q 的答案是 %q\n", expr, answer)
}
// 关闭输入管道,通知子进程不再有更多输入
err = si.Close()
if err != nil {
log.Println("关闭StdinPipe失败:", err)
}
// 关闭输出管道(可选,但良好实践)
err = so.Close()
if err != nil {
log.Println("关闭StdoutPipe失败:", err)
}
// 等待子进程退出并获取其退出状态
err = c.Wait()
if err != nil {
log.Fatal("等待子进程退出失败:", err)
}
fmt.Println("子进程已成功退出。")
}将上述Go代码保存为 main.go,Python代码保存为 add.py,并在同一个目录下运行 go run main.go,你将看到如下输出:
对表达式 "2+0\n" 的答案是 "2\n" 对表达式 "2+1\n" 的答案是 "3\n" 对表达式 "2+2\n" 的答案是 "4\n" 对表达式 "2+3\n" 的答案是 "5\n" 对表达式 "2+4\n" 的答案是 "6\n" 对表达式 "2+5\n" 的答案是 "7\n" 对表达式 "2+6\n" 的答案是 "8\n" 对表达式 "2+7\n" 的答案是 "9\n" 对表达式 "2+8\n" 的答案是 "10\n" 对表达式 "2+9\n" 的答案是 "11\n" 子进程已成功退出。
通过 os/exec 包,Go语言提供了强大且灵活的机制来与外部子进程进行交互。实现这种实时双向通信的关键在于:
掌握这些技术,可以让你在Go程序中无缝集成和控制各种外部工具和脚本,极大地扩展Go应用程序的功能边界。
以上就是Go语言实现与子进程的实时交互:通过标准输入输出进行通信的详细内容,更多请关注php中文网其它相关文章!
每个人都需要一台速度更快、更稳定的 PC。随着时间的推移,垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是,许多工具可以让 Windows 保持平稳运行。
Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号