Golang Docker SDK 虚拟终端实现：使用 ContainerExecCreate、ContainerExecStart 和 ContainerExecAttach 方法

Golang Docker SDK 虚拟终端实现：使用 ContainerExecCreate、ContainerExecStart 和 ContainerExecAttach 方法

本文将介绍如何使用 Golang Docker SDK 中的 ContainerExecCreate、ContainerExecStart 和 ContainerExecAttach 方法创建一个虚拟终端，允许用户在容器内执行命令并获取输出结果。

示例代码

package main

import (
	"bufio"
	"context"
	"fmt"
	"io"
	"os"
	"strings"

	"github.com/docker/docker/api/types"
	"github.com/docker/docker/client"
)

func main() {
	// 创建docker客户端
	cli, err := client.NewEnvClient()
	if err != nil {
		panic(err)
	}

	// 获取容器ID
	containerID := "CONTAINER_ID"

	// 创建exec
	execConfig := types.ExecConfig{
		AttachStdout: true,
		AttachStderr: true,
		Tty:          true,
		Cmd:          []string{"bin/sh"},
	}
	respCreate, err := cli.ContainerExecCreate(context.Background(), containerID, execConfig)
	if err != nil {
		panic(err)
	}

	// 启动exec
	respStart, err := cli.ContainerExecStart(context.Background(), respCreate.ID, types.ExecStartCheck{})
	if err != nil {
		panic(err)
	}

	// 创建session
	in, out := io.Pipe()
	session := types.HijackedResponse{
		Conn: out,
	}

	// 将session与exec绑定
	errCh := make(chan error, 1)
	go func() {
		errCh <- cli.ContainerExecAttach(context.Background(), respCreate.ID, types.ExecConfig{
			Tty: true,
		}, session)
	}()

	// 读取输出
	go func() {
		scanner := bufio.NewScanner(in)
		for scanner.Scan() {
			fmt.Println(scanner.Text())
		}
	}()

	// 读取用户输入
	scanner := bufio.NewScanner(os.Stdin)
	for {
		fmt.Print("$ ")
		if !scanner.Scan() {
			break
		}
		cmd := scanner.Text()
		if strings.TrimSpace(cmd) == "" {
			continue
		}

		// 发送命令到容器
		_, err := fmt.Fprintln(session.Conn, cmd)
		if err != nil {
			panic(err)
		}
	}

	// 关闭session
	err = session.Close()
	if err != nil {
		panic(err)
	}

	// 等待exec结束
	err = <-errCh
	if err != nil {
		panic(err)
	}
}

代码解释

1. 创建 Docker 客户端：使用 client.NewEnvClient() 创建一个 Docker 客户端。
2. 获取容器 ID：将要进入的容器 ID 赋值给 containerID 变量。
3. 创建 Exec：使用 ContainerExecCreate 方法创建 Exec，并设置相关配置，例如：
   • AttachStdout 和 AttachStderr：表示将容器的标准输出和错误输出连接到虚拟终端。
   • Tty：设置为 true 表示创建交互式终端。
   • Cmd：要执行的命令，这里设置为 "bin/sh"，表示启动一个 shell。
4. 启动 Exec：使用 ContainerExecStart 方法启动 Exec。
5. 创建 Session：使用 io.Pipe() 创建一个管道，并将其写端作为 Session 的 Conn 字段，用于与 Exec 交互。
6. 将 Session 与 Exec 绑定：使用 ContainerExecAttach 方法将 Session 与 Exec 绑定，并在后台读取输出。
7. 读取输出：使用 bufio.NewScanner 从 Session 的 Conn 字段读取输出，并打印到控制台。
8. 读取用户输入：从标准输入读取用户输入的命令。
9. 发送命令到容器：将用户输入的命令写入 Session 的 Conn 字段。
10. 关闭 Session：在程序结束时关闭 Session。
11. 等待 Exec 结束：等待 Exec 结束，并处理可能的错误。

注意事项

• 由于我们在创建 Exec 时设置了 Tty 为 true，因此需要将 Session 与终端进行绑定，以便能够正确地读取输出和发送命令。
• 由于使用了 bufio.Scanner 方法读取输入和输出，因此需要在读取输出时使用 bufio.NewScanner 方法创建一个 Scanner 对象，以便能够逐行读取输出。

希望本文能够帮助您了解如何使用 Golang Docker SDK 创建虚拟终端，并实现容器内命令执行功能。如果您有任何问题，请随时留言。