如何在单个 SSH 会话中连续执行多条命令-Golang-PHP中文网

如何在单个 SSH 会话中连续执行多条命令

本文详解如何在 Go 的 golang.org/x/crypto/ssh 包中，通过拼接 Shell 命令的方式，在同一个 SSH Session 中安全、高效地执行多个依赖顺序的命令（如 Docker 登录后立即查询容器状态），规避 session.Run() 仅允许调用一次的限制。

本文详解如何在 go 的 `golang.org/x/crypto/ssh` 包中，通过拼接 shell 命令的方式，在同一个 ssh session 中安全、高效地执行多个依赖顺序的命令（如 docker 登录后立即查询容器状态），规避 `session.run()` 仅允许调用一次的限制。

在使用 Go 的 SSH 客户端库（golang.org/x/crypto/ssh）时，一个常见误区是试图对同一 *ssh.Session 多次调用 session.Run()。但正如源码注释明确指出的：一个 Session 实例仅允许调用一次 Run、Start、Shell、Output 或 CombinedOutput。这是因为 Run 方法内部会复用并最终关闭底层的 channel 和 stdin/stdout 流，重复调用将导致 EOF、write on closed pipe 等错误——这正是你在执行第二条 docker ps 时遇到失败的根本原因。

✅ 正确解法是：将多个命令合并为一条 Shell 命令字符串，通过分号 ; 或逻辑操作符（如 &&）连接后，一次性传递给 session.Run()。这种方式复用了同一 SSH 会话的上下文（包括已加载的环境变量、登录态、Docker 凭据等），完全满足“先登录再操作”的业务逻辑。

以下为优化后的完整示例：

package main

import (
    "bytes"
    "fmt"
    "strings"
    "golang.org/x/crypto/ssh"
)

func main() {
    config := &ssh.ClientConfig{
        User: USERNAME,
        Auth: []ssh.AuthMethod{
            ssh.PublicKeys(pem),
        },
        HostKeyCallback: ssh.InsecureIgnoreHostKey(), // 生产环境请替换为安全校验
    }

    client, err := ssh.Dial("tcp", HOSTNAME+":"+PORT, config)
    if err != nil {
        panic("Failed to dial: " + err.Error())
    }
    defer client.Close()

    session, err := client.NewSession()
    if err != nil {
        panic("Failed to create session: " + err.Error())
    }
    defer session.Close()

    // ✅ 关键：将多条命令拼接为单个字符串
    commands := []string{
        "docker login -u $USER -p $PASS registry.example.com", // 替换为实际凭证
        "docker ps --format "table {{.ID}}\t{{.Image}}\t{{.Status}}"",
    }
    fullCommand := strings.Join(commands, " && ") // 使用 && 可确保前序命令成功才执行后续

    var output bytes.Buffer
    session.Stdout = &output
    session.Stderr = &output // 建议同时捕获 stderr 便于调试

    if err := session.Run(fullCommand); err != nil {
        panic("Command failed: " + fullCommand + "
Error: " + err.Error() + "
Output:
" + output.String())
    }

    fmt.Println("Execution succeeded. Output:")
    fmt.Println(output.String())
}

? 关键注意事项：

优先使用 && 而非 ;：&& 具有短路语义，当前命令失败时后续命令不会执行，避免误操作（例如未登录成功就拉取镜像）；而 ; 会无条件执行所有命令。
注意 Shell 解析上下文：session.Run() 默认调用的是远程主机的默认 Shell（通常是 /bin/sh），并非 Bash。若需 Bash 特性（如数组、扩展 glob），应显式调用：bash -c "cmd1 && cmd2"。
敏感信息处理：避免在命令字符串中硬编码密码。推荐使用 docker login 的 -p 参数配合 stdin 输入，或提前配置 ~/.docker/config.json。
错误诊断增强：务必同时设置 session.Stderr，否则命令报错时 output.String() 可能为空，导致难以定位问题。