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!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值