如何在Docker Compose启动Postgres容器后执行psql命令

Postgres官方Docker镜像会在首次初始化数据库时，自动执行/docker-entrypoint-initdb.d/目录下的.sql、.sh或.sql.gz脚本。你可以将需要执行的命令整理成脚本，通过Docker Compose挂载到该目录实现自动执行。

方法1：使用SQL脚本 #

创建init-db.sql文件，写入所有SQL操作：

-- 注：默认已存在postgres用户，若无需额外配置可省略CREATE USER语句
CREATE USER postgres;

CREATE DATABASE DB;
CREATE DATABASE TEST_DB;

GRANT CREATE ON DATABASE DB TO postgres;
GRANT CREATE ON DATABASE TEST_DB TO postgres;

-- 切换到DB数据库并添加扩展
\c DB
CREATE EXTENSION IF NOT EXISTS "uuid-ossp";
CREATE EXTENSION IF NOT EXISTS "pg_trgm";

-- 切换到TEST_DB数据库并添加扩展
\c TEST_DB
CREATE EXTENSION IF NOT EXISTS "uuid-ossp";
CREATE EXTENSION IF NOT EXISTS "pg_trgm";

修改你的docker-compose.yml，添加脚本挂载配置：

version: '3.8'
services:
  postgres:
    image: postgres:latest
    environment:
      POSTGRES_USER: postgres
      POSTGRES_PASSWORD: your_db_password  # 替换为你的数据库密码
    volumes:
      - ./init-db.sql:/docker-entrypoint-initdb.d/init-db.sql
      - postgres_data:/var/lib/postgresql/data  # 持久化数据库数据
    ports:
      - "5432:5432"

volumes:
  postgres_data:

方法2：使用Shell脚本 #

创建init-db.sh文件，写入shell命令：

#!/bin/bash
set -e

# 检查postgres用户是否存在，不存在则创建
psql -tAc "SELECT 1 FROM pg_roles WHERE rolname='postgres'" | grep -q 1 || createuser postgres

createdb DB
createdb TEST_DB

# 配置DB数据库权限与扩展
psql -d DB -c "GRANT CREATE ON DATABASE \"DB\" TO postgres"
psql -d DB -c "CREATE EXTENSION IF NOT EXISTS \"uuid-ossp\""
psql -d DB -c "CREATE EXTENSION IF NOT EXISTS \"pg_trgm\""

# 配置TEST_DB数据库权限与扩展
psql -d TEST_DB -c "GRANT CREATE ON DATABASE \"TEST_DB\" TO postgres"
psql -d TEST_DB -c "CREATE EXTENSION IF NOT EXISTS \"uuid-ossp\""
psql -d TEST_DB -c "CREATE EXTENSION IF NOT EXISTS \"pg_trgm\""

给脚本添加执行权限：

chmod +x init-db.sh

修改docker-compose.yml挂载该脚本：

version: '3.8'
services:
  postgres:
    image: postgres:latest
    environment:
      POSTGRES_USER: postgres
      POSTGRES_PASSWORD: your_db_password
    volumes:
      - ./init-db.sh:/docker-entrypoint-initdb.d/init-db.sh
      - postgres_data:/var/lib/postgresql/data
    ports:
      - "5432:5432"

volumes:
  postgres_data:

关键注意点 #

• 脚本仅在首次初始化数据库时运行（即数据卷postgres_data为空时）。若需重新执行，需删除数据卷：docker-compose down -v
• 确保POSTGRES_USER环境变量与脚本中的用户一致，避免权限错误
• 默认已存在postgres用户，直接执行createuser postgres会报错，建议添加存在性判断（如Shell脚本中的写法）

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