Go语言中实现与外部进程的持久化交互教程

// e.go
package main

import (
    "bufio"
    "fmt"
    "os"
    "strings"
)

func main() {
    for {
        // 创建一个从标准输入读取的缓冲读取器
        buf := bufio.NewReader(os.Stdin)
        input, err := buf.ReadString('\n') // 读取一行直到换行符

        if err != nil {
            fmt.Println("Echo failed: ", input)
            panic(err) // 发生错误时退出
        }

        // 如果输入以"exit"开头，则退出程序
        if strings.HasPrefix(input, "exit") {
            fmt.Println("Bye!")
            return
        }

        // 将输入原样打印到标准输出
        fmt.Print(input)
    }
}

// main.go
package main

import (
    "bufio"
    "fmt"
    "math/rand"
    "os/exec"
    "time"
)

func main() {
    // 1. 定义要执行的外部命令
    // 注意：如果e不在当前目录，需要提供完整路径，例如 "./e" 或 "/path/to/e"
    cmd := exec.Command("./e") 

    // 2. 获取子进程的标准输出管道
    progin, err := cmd.StdoutPipe()
    if err != nil {
        fmt.Println("获取子进程Stdout管道失败:", err)
        panic(err)
    }

    // 3. 获取子进程的标准输入管道
    progout, err := cmd.StdinPipe()
    if err != nil {
        fmt.Println("获取子进程Stdin管道失败:", err)
        panic(err)
    }

    // 4. 启动子进程
    err = cmd.Start()
    if err != nil {
        fmt.Println("启动子进程失败:", err)
        panic(err)
    }

    // 初始化随机数生成器，用于模拟随机输入
    r := rand.New(rand.NewSource(time.Now().UnixNano()))

    // 创建一个缓冲读取器，用于从子进程的输出管道读取数据
    buf := bufio.NewReader(progin)

    // 5. 进入无限循环，进行读写交互
    for {
        // 写入数据到子进程
        var toProg string
        if r.Float64() < .1 { // 大约10%的概率发送"exit"命令
            toProg = "exit"
        } else {
            toProg = fmt.Sprintf("%d", r.Intn(100)) // 发送一个随机整数
        }
        fmt.Println("发送到子进程:", toProg)
        // 注意：必须加上换行符，因为子进程是按行读取的
        _, err = progout.Write([]byte(toProg + "\n"))
        if err != nil {
            fmt.Println("写入子进程Stdin管道失败:", err)
            panic(err)
        }

        // 如果发送了"exit"命令，则等待子进程退出并结束循环
        if toProg == "exit" {
            fmt.Println("发送'exit'，等待子进程退出...")
            _ = cmd.Wait() // 等待子进程完成
            break
        }

        // 给子进程一些时间来处理输入并生成输出
        time.Sleep(500 * time.Millisecond) 

        // 从子进程读取数据
        input, err := buf.ReadString('\n')
        if err != nil {
            // 注意：当子进程退出时，管道可能会关闭，导致io.EOF错误
            fmt.Println("从子进程Stdout管道读取失败:", err, "接收到:", input)
            // 如果是EOF错误，说明子进程可能已退出，可以安全地退出循环
            if err.Error() == "EOF" {
                break
            }
            panic(err)
        }
        fmt.Println("从子进程接收:", input)
    }

    fmt.Println("主程序退出。")
}