使用Docker Go SDK无法在Postgres容器内执行SQL文件

核心问题排查与解决方向 #

1. Postgres服务未就绪就执行SQL #

Postgres容器启动后需要完成初始化流程（创建默认库、启动服务）才能接受连接，直接在容器启动后立刻执行SQL命令，会因服务未就绪导致命令静默失败，且SDK不会返回错误（容器内命令进程能启动，但SQL执行时连接失败）。

解决办法：添加就绪检查逻辑，反复验证Postgres可正常连接后再执行SQL。示例检查命令：psql -U postgres -c "SELECT 1;"。

2. 执行命令的上下文错误 #

• 必须指定正确的数据库和用户：默认用postgres用户连接postgres库，否则SQL会执行到错误的上下文。
• 检查容器内SQL文件权限：若文件属主为root，postgres用户可能无法读取，需修改权限为postgres:postgres。

3. 命令执行语法问题 #

执行SQL文件时，用sh -c包裹完整命令，避免参数拆分导致的解析错误，确保命令在容器shell环境中正确执行。

修改后的代码示例 #

package main

import (
	"bytes"
	"context"
	"fmt"
	"io"
	"io/ioutil"
	"time"

	"github.com/docker/docker/api/types"
	"github.com/docker/docker/api/types/container"
	"github.com/docker/docker/client"
	"github.com/docker/docker/pkg/stdcopy"
)

func main() {
	ctx := context.Background()
	cli, err := client.NewClientWithOpts(client.FromEnv, client.WithAPIVersionNegotiation())
	if err != nil {
		panic(err)
	}
	defer cli.Close()

	// 拉取postgres镜像
	_, err = cli.ImagePull(ctx, "postgres:14-alpine", types.ImagePullOptions{})
	if err != nil {
		panic(err)
	}

	// 创建并启动容器
	resp, err := cli.ContainerCreate(ctx, &container.Config{
		Image: "postgres:14-alpine",
		Env: []string{
			"POSTGRES_PASSWORD=testpass",
			"POSTGRES_USER=postgres",
			"POSTGRES_DB=postgres",
		},
		Cmd: []string{"postgres"},
	}, &container.HostConfig{
		AutoRemove: true,
	}, nil, nil, "test-postgres")
	if err != nil {
		panic(err)
	}

	if err := cli.ContainerStart(ctx, resp.ID, types.ContainerStartOptions{}); err != nil {
		panic(err)
	}
	defer cli.ContainerStop(ctx, resp.ID, container.StopOptions{})

	// 等待Postgres服务就绪
	ready := false
	for i := 0; i < 30; i++ {
		execResp, err := cli.ContainerExecCreate(ctx, resp.ID, types.ExecConfig{
			User: "postgres",
			Cmd:  []string{"psql", "-U", "postgres", "-c", "SELECT 1;"},
		})
		if err != nil {
			time.Sleep(1 * time.Second)
			continue
		}

		execStartResp, err := cli.ContainerExecStart(ctx, execResp.ID, types.ExecStartCheck{})
		if err != nil {
			time.Sleep(1 * time.Second)
			continue
		}

		// 读取执行输出，确认连接正常
		stdoutPipe, stderrPipe := io.Pipe()
		go func() {
			stdcopy.StdCopy(stdoutPipe, stderrPipe, execStartResp.Conn)
			stdoutPipe.Close()
			stderrPipe.Close()
		}()

		stderrBytes, _ := ioutil.ReadAll(stderrPipe)
		if len(stderrBytes) == 0 {
			ready = true
			break
		}
		time.Sleep(1 * time.Second)
	}

	if !ready {
		panic("Postgres容器未在指定时间内就绪")
	}

	// 复制SQL文件到容器内
	sqlContent, err := ioutil.ReadFile("./init.sql")
	if err != nil {
		panic(err)
	}

	err = cli.CopyToContainer(ctx, resp.ID, "/tmp/init.sql", bytes.NewReader(sqlContent), types.CopyToContainerOptions{
		AllowOverwriteDirWithFile: true,
	})
	if err != nil {
		panic(err)
	}

	// 修改SQL文件权限为postgres用户可读取
	execResp, err := cli.ContainerExecCreate(ctx, resp.ID, types.ExecConfig{
		User: "root",
		Cmd:  []string{"chown", "postgres:postgres", "/tmp/init.sql"},
	})
	if err != nil {
		panic(err)
	}
	if _, err := cli.ContainerExecStart(ctx, execResp.ID, types.ExecStartCheck{}); err != nil {
		panic(err)
	}

	// 执行SQL文件
	execResp, err = cli.ContainerExecCreate(ctx, resp.ID, types.ExecConfig{
		User: "postgres",
		Cmd: []string{"sh", "-c", "psql -U postgres -d postgres -f /tmp/init.sql"},
	})
	if err != nil {
		panic(err)
	}

	execStartResp, err := cli.ContainerExecStart(ctx, execResp.ID, types.ExecStartCheck{})
	if err != nil {
		panic(err)
	}

	// 读取执行输出，排查问题
	stdoutPipe, stderrPipe := io.Pipe()
	go func() {
		stdcopy.StdCopy(stdoutPipe, stderrPipe, execStartResp.Conn)
		stdoutPipe.Close()
		stderrPipe.Close()
	}()

	stdoutBytes, _ := ioutil.ReadAll(stdoutPipe)
	stderrBytes, _ := ioutil.ReadAll(stderrPipe)
	fmt.Println("SQL执行输出:", string(stdoutBytes))
	fmt.Println("SQL执行错误:", string(stderrBytes))

	// 验证表是否创建成功
	execResp, err = cli.ContainerExecCreate(ctx, resp.ID, types.ExecConfig{
		User: "postgres",
		Cmd:  []string{"psql", "-U", "postgres", "-c", "\\d tester"},
	})
	if err != nil {
		panic(err)
	}
	execStartResp, err = cli.ContainerExecStart(ctx, execResp.ID, types.ExecStartCheck{})
	if err != nil {
		panic(err)
	}

	stdoutPipe, stderrPipe = io.Pipe()
	go func() {
		stdcopy.StdCopy(stdoutPipe, stderrPipe, execStartResp.Conn)
		stdoutPipe.Close()
		stderrPipe.Close()
	}()
	stdoutBytes, _ = ioutil.ReadAll(stdoutPipe)
	stderrBytes, _ = ioutil.ReadAll(stderrPipe)
	fmt.Println("表验证输出:", string(stdoutBytes))
	fmt.Println("表验证错误:", string(stderrBytes))
}

额外说明 #

• 若将SQL文件放到/docker-entrypoint-initdb.d/目录，Postgres容器初始化阶段会自动执行脚本，但容器启动后再复制该文件，需手动执行（初始化流程已结束）。
• 执行命令必须指定User: "postgres"，避免root用户执行导致的权限或上下文偏差。
• 每次执行命令后读取stdout和stderr，可直接排查命令执行的实际问题，不要仅依赖SDK的错误返回。

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