Go语言中与外部进程进行持久化交互：Stdin/Stdout管道的正确使用

// 错误示例：不适用于持续交互 cmd := exec.Command("your_program") // ... 设置StdoutPipe ... cmd.Start() for {     // ... 生成toProg ...     cmd.Stdin = strings.NewReader(toProg + "n") // 每次循环都重新赋值Stdin     // ... 读取输出 ... }

云雀语言模型

云雀是一款由字节跳动研发的语言模型，通过便捷的自然语言交互，能够高效的完成互动对话

54

查看详情

package main  import (     "bufio"     "fmt"     "os"     "strings" )  func main() {     reader := bufio.NewReader(os.Stdin)     for {         input, err := reader.ReadString('n') // 从标准输入读取一行         if err != nil {             // 通常在EOF或管道关闭时发生             fmt.Println("Echo程序读取失败:", err)             break // 退出循环         }          trimmedInput := strings.TrimSpace(input) // 移除输入中的空白字符          if trimmedInput == "exit" {             fmt.Println("Bye!") // 打印退出信息             return              // 退出程序         }          fmt.Print(input) // 将读取到的内容原样打印到标准输出     } }

package main  import (     "bufio"     "fmt"     "math/rand"     "os/exec"     "time" )  func main() {     // 确保 e 可执行文件在当前目录或 PATH 中     cmd := exec.Command("./e") // 启动我们编译的 echo 程序      // 获取外部程序的标准输出管道，用于读取其输出     progin, err := cmd.StdoutPipe()     if err != nil {         fmt.Println("获取 'e' 的标准输出管道失败:", err)         panic(err)     }      // 获取外部程序的标准输入管道，用于向其写入数据     progout, err := cmd.StdinPipe()     if err != nil {         fmt.Println("获取 'e' 的标准输入管道失败:", err)         panic(err)     }      // 启动外部程序     err = cmd.Start()     if err != nil {         fmt.Println("启动 'e' 程序失败:", err)         panic(err)     }      // 初始化随机数生成器，用于模拟输入     r := rand.New(rand.NewSource(time.Now().unixNano()))      // 使用 bufio.NewReader 包装 progin，以便按行读取     buf := bufio.NewReader(progin)      for i := 0; i < 10; i++ { // 循环10次进行交互         // 构造要发送给外部程序的数据         var toProg string         if r.Float64() < .2 { // 大约20%的概率发送 "exit"             toProg = "exit"         } else {             toProg = fmt.Sprintf("%d", r.Intn(100)) // 发送随机整数         }          fmt.Printf(">>> 发送: %sn", toProg)          // 向外部程序的标准输入写入数据，注意要加上换行符         _, err := progout.Write([]byte(toProg + "n"))         if err != nil {             fmt.Println("向 'e' 写入数据失败:", err)             break // 写入失败则退出循环         }          // 给予外部程序处理输入并生成输出的时间         // 在实际应用中，应避免使用硬编码的 Sleep，考虑更智能的同步机制         time.Sleep(100 * time.Millisecond)          // 从外部程序的标准输出读取一行数据         input, err := buf.ReadString('n')         if err != nil {             fmt.Println("<<< 从 'e' 读取数据失败:", err)             break // 读取失败则退出循环         }         fmt.Printf("<<< 接收: %s", input) // input 已包含换行符          if toProg == "exit" {             fmt.Println("检测到 'exit' 命令，退出交互循环。")             break         }     }      // 关闭输入管道，通知外部程序不再有更多输入     // 这有助于外部程序检测EOF并优雅退出     progout.Close()      // 等待外部程序完全退出，并获取其退出状态     err = cmd.Wait()     if err != nil {         fmt.Printf("外部程序 'e' 退出时发生错误: %vn", err)     } else {         fmt.Println("外部程序 'e' 已成功退出。")     } }