Go语言中实现与子进程的交互式双向通信

# add.py
import sys
while True:
    # 从标准输入读取一行
    line = sys.stdin.readline()
    if not line: # 如果读取到EOF，则退出循环
        break
    try:
        # 评估输入字符串，并将其结果写入标准输出
        sys.stdout.write('%s\n'%eval(line))
        # 立即刷新输出缓冲区，确保Go程序能及时收到数据
        sys.stdout.flush()
    except Exception as e:
        sys.stderr.write(f"Error processing input: {e}\n")
        sys.stderr.flush()
        break

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("Failed to get StdinPipe:", err)
    }

    // 获取子进程的标准输出管道
    so, err := c.StdoutPipe()
    if err != nil {
        log.Fatal("Failed to get StdoutPipe:", err)
    }

    // 使用bufio.NewReader包装StdoutPipe，以便按行读取
    reader := bufio.NewReader(so)

    // 启动子进程
    err = c.Start()
    if err != nil {
        log.Fatal("Failed to start command:", err)
    }

    // ... 后续交互代码 ...
}

    // 现在进行一些数学运算的交互
    for i := 0; i < 10; i++ {
        // 构造要发送给Python脚本的字符串，并加上换行符
        input := fmt.Sprintf("2+%d\n", i)

        // 将数据写入子进程的stdin
        _, err = si.Write([]byte(input))
        if err != nil {
            log.Fatal("Failed to write to stdin:", err)
        }

        // 从子进程的stdout读取一行数据，直到遇到换行符
        answer, err := reader.ReadString('\n')
        if err != nil {
            log.Fatal("Failed to read from stdout:", err)
        }

        // 打印交互结果
        fmt.Printf("Answer to %q is %q\n", input, answer)
    }

    // 关闭输入管道，通知子进程不再有更多输入
    si.Close()
    // 关闭输出管道
    so.Close()

    // 等待子进程退出，并处理其可能的错误码
    err = c.Wait()
    if err != nil {
        // 如果子进程非正常退出，这里会捕获错误
        log.Printf("Command finished with error: %v", err)
    } else {
        fmt.Println("Subprocess exited successfully.")
    }
}

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("Failed to get StdinPipe:", err)
    }

    // 获取子进程的标准输出管道
    so, err := c.StdoutPipe()
    if err != nil {
        log.Fatal("Failed to get StdoutPipe:", err)
    }

    // 使用bufio.NewReader包装StdoutPipe，以便按行读取
    reader := bufio.NewReader(so)

    // 启动子进程
    err = c.Start()
    if err != nil {
        log.Fatal("Failed to start command:", err)
    }

    // 现在进行一些数学运算的交互
    for i := 0; i < 10; i++ {
        // 构造要发送给Python脚本的字符串，并加上换行符
        input := fmt.Sprintf("2+%d\n", i)

        // 将数据写入子进程的stdin
        _, err = si.Write([]byte(input))
        if err != nil {
            log.Fatal("Failed to write to stdin:", err)
        }

        // 从子进程的stdout读取一行数据，直到遇到换行符
        answer, err := reader.ReadString('\n')
        if err != nil {
            log.Fatal("Failed to read from stdout:", err)
        }

        // 打印交互结果
        fmt.Printf("Answer to %q is %q\n", input, answer)
    }

    // 关闭输入管道，通知子进程不再有更多输入
    si.Close()
    // 关闭输出管道
    so.Close()

    // 等待子进程退出，并处理其可能的错误码
    err = c.Wait()
    if err != nil {
        log.Printf("Command finished with error: %v", err)
    } else {
        fmt.Println("Subprocess exited successfully.")
    }
}

Answer to "2+0\n" is "2\n"
Answer to "2+1\n" is "3\n"
Answer to "2+2\n" is "4\n"
Answer to "2+3\n" is "5\n"
Answer to "2+4\n" is "6\n"
Answer to "2+5\n" is "7\n"
Answer to "2+6\n" is "8\n"
Answer to "2+7\n" is "9\n"
Answer to "2+8\n" is "10\n"
Answer to "2+9\n" is "11\n"
Subprocess exited successfully.