golang使用stdio与子进程进行通信
在使用Cline调用本地MCP Server的时候,使用的是STDIO模式,也就是Cline启动一个子进程来运行MCP Server,然后通过STDIO来进行通信。这种方式即高效又安全。
import ("bufio""fmt""io""os/exec""time"
)func Run() {// 定义要执行的命令(例如:运行一个 Python 脚本)cmd := exec.Command("python", "-u", "-c", `user_input = input(); print("child get message: " + user_input)`)// 获取子进程的标准输入、输出和错误流stdin, err := cmd.StdinPipe()if err != nil {fmt.Println("Error creating stdin pipe:", err)return}defer stdin.Close()stdout, err := cmd.StdoutPipe()if err != nil {fmt.Println("Error creating stdout pipe:", err)return}// 启动子进程,异步的方式执行命令if err := cmd.Start(); err != nil {fmt.Println("Error starting command:", err)return}// 使用 bufio 读取子进程的输出go func() {reader := bufio.NewReader(stdout)for {line, err := reader.ReadString('\n')if err != nil && err != io.EOF {fmt.Println("Error reading from stdout:", err)break}if line != "" {fmt.Print(line)}if err == io.EOF {break}}}()// 向子进程的标准输入写入数据input := "Hello from Go!\n"fmt.Fprintf(stdin, input)// 等待子进程结束if err := cmd.Wait(); err != nil {fmt.Println("Error waiting for command:", err)}time.Sleep(3 * time.Second)
}
运行程序输出:child get message: Hello from Go!