实时捕获命令执行stdout：聊天机器人输出功能技术求助

我明白你的问题所在——当前代码会等到脚本完全执行完才把所有输出一次性返回，这确实没法实现实时推送。问题根源在于你依赖return传递结果（但return只能返回一次），同时cmd.Run()会阻塞到命令结束，还把所有输出都攒在bytes.Buffer里了。

要实现实时输出，你需要调整代码逻辑：让程序在命令运行过程中逐行读取输出并立即发送到聊天频道，而不是等全部输出完成后再返回。下面是具体的修改方案：

核心思路 #

1. 放弃用return一次性返回结果的模式，改为在命令执行过程中直接调用Bot的消息发送方法（假设你的bot.Cmd对象有类似Reply的发送消息方法，这是大多数Bot框架的标准设计）。
2. 使用StdoutPipe()获取命令的实时输出流，配合bufio.Scanner逐行读取并发送。
3. 用cmd.Start()启动命令（非阻塞），再用cmd.Wait()等待命令执行完毕。

修改后的代码 #

import (
    "bufio"
    "fmt"
    "os/exec"
)

func reboot(command *bot.Cmd) error {
    // 先检查参数是否存在
    if len(command.Args) == 0 {
        command.Reply("请提供服务器参数")
        return fmt.Errorf("missing server argument")
    }
    server := command.Args[0]
    
    // 调用runcommand，传入实时发送消息的回调
    return runcommand(server, func(output string) {
        command.Reply(output)
    })
}

func runcommand(server string, sendOutput func(string)) error {
    // 正确拼接命令字符串（你原来的写法无法替换%s，这里用fmt.Sprintf处理）
    cmdStr := fmt.Sprintf("python test.py %s", server)
    cmd := exec.Command("/bin/bash", "-c", cmdStr)
    
    // 创建标准输出管道，用于实时读取输出
    stdout, err := cmd.StdoutPipe()
    if err != nil {
        sendOutput("创建输出管道失败: " + err.Error())
        return err
    }
    
    // 启动命令（非阻塞，不会等命令完成）
    if err := cmd.Start(); err != nil {
        sendOutput("启动命令失败: " + err.Error())
        return err
    }
    
    // 逐行读取输出并实时发送
    scanner := bufio.NewScanner(stdout)
    for scanner.Scan() {
        line := scanner.Text()
        sendOutput(line)
    }
    
    // 等待命令执行完成
    if err := cmd.Wait(); err != nil {
        sendOutput("命令执行出错: " + err.Error())
        return err
    }
    
    // 检查读取过程中是否有错误
    if err := scanner.Err(); err != nil {
        sendOutput("读取输出流出错: " + err.Error())
        return err
    }
    
    return nil
}

关键改动说明 #

• 回调函数sendOutput：把发送消息的逻辑封装成回调，让runcommand每读到一行输出就能立即调用它推送到聊天频道。
• StdoutPipe()替代bytes.Buffer：不再攒住所有输出，而是实时获取命令的输出流。
• cmd.Start() + cmd.Wait()：Start()启动命令后立即返回，不会阻塞；Wait()则会等待命令执行完毕，确保我们能读到所有输出。
• 修复命令拼接错误：你原来的exec.Command写法无法正确替换%s，现在用fmt.Sprintf先拼接好完整的命令字符串。

如果你的Bot框架对消息发送频率有限制，可能需要额外加个简单的限流逻辑（比如短时间内发送多条消息时做合并），但这属于优化范畴，核心的实时输出逻辑已经实现了。

内容的提问来源于stack exchange，提问作者Blooze