Go语言中与交互式命令行程序进行程序化交互的正确姿势

package main

import (
    "fmt"
    "log"
    "os"
    "time"

    "golang.org/x/crypto/ssh"
)

func main() {
    // 1. 配置SSH客户端
    // 这里使用密码认证，也可以使用ssh.PublicKeys(signer)进行密钥认证
    config := &ssh.ClientConfig{
        User: "your_username", // 替换为你的SSH用户名
        Auth: []ssh.AuthMethod{
            ssh.Password("your_password"), // 替换为你的SSH密码
        },
        // HostKeyCallback用于验证远程主机的密钥。
        // 在生产环境中，强烈建议不要使用ssh.InsecureIgnoreHostKey()，
        // 而是实现一个安全的HostKeyCallback，例如从known_hosts文件加载。
        HostKeyCallback: ssh.InsecureIgnoreHostKey(), 
        Timeout:         5 * time.Second,
    }

    // 2. 建立SSH连接
    client, err := ssh.Dial("tcp", "your_host:22", config) // 替换为你的SSH主机地址和端口
    if err != nil {
        log.Fatalf("无法连接SSH服务器: %v", err)
    }
    defer client.Close() // 确保连接在使用完毕后关闭

    // 3. 打开一个新的SSH会话
    session, err := client.NewSession()
    if err != nil {
        log.Fatalf("无法创建SSH会话: %v", err)
    }
    defer session.Close() // 确保会话在使用完毕后关闭

    // 4. 将远程命令的输出重定向到本地的标准输出和标准错误
    session.Stdout = os.Stdout
    session.Stderr = os.Stderr

    // 5. 执行远程命令
    remoteCommand := "ls -l /tmp" // 替换为你想执行的远程命令
    fmt.Printf("正在远程主机上执行命令: '%s'\n", remoteCommand)
    if err := session.Run(remoteCommand); err != nil {
        log.Fatalf("远程命令执行失败: %v", err)
    }
    fmt.Println("远程命令执行完成。")
}

package main

import (
    "bytes"
    "fmt"
    "io"
    "log"
    "os/exec"
)

func main() {
    // 1. 定义要执行的命令
    // 示例中使用"cat"，它会将其stdin的输入直接输出到stdout
    cmd := exec.Command("cat") 

    // 2. 获取命令的StdinPipe，用于向命令写入数据
    stdin, err := cmd.StdinPipe()
    if err != nil {
        log.Fatalf("获取StdinPipe失败: %v", err)
    }
    defer stdin.Close() // 确保管道在函数结束时关闭

    // 3. 捕获命令的输出
    var out bytes.Buffer
    cmd.Stdout = &out
    cmd.Stderr = &out // 也捕获标准错误

    // 4. 启动命令
    err = cmd.Start()
    if err != nil {
        log.Fatalf("命令启动失败: %v", err)
    }

    // 5. 向命令的stdin写入数据
    _, err = io.WriteString(stdin, "Hello from Go!\n")
    if err != nil {
        log.Fatalf("写入stdin失败: %v", err)
    }
    _, err = io.WriteString(stdin, "This is a test line.\n")
    if err != nil {
        log.Fatalf("写入stdin失败: %v", err)
    }

    // 6. 关闭stdin，表示所有输入已发送完毕。
    // 这对于某些命令（如cat）是必要的，它会等待EOF才结束。
    stdin.Close() 

    // 7. 等待命令执行完成
    err = cmd.Wait()
    if err != nil {
        log.Fatalf("命令执行失败: %v\n输出:\n%s", err, out.String())
    }

    // 8. 打印命令的输出
    fmt.Printf("命令输出:\n%s", out.String())
}