Docker部署PostgreSQL时，SQL脚本使用变量报错如何解决？

你的问题核心是混淆了不同SQL方言/执行上下文的变量用法：PostgreSQL的普通SQL脚本不支持MySQL风格的@变量，DECLARE关键字也只能在PL/pgSQL块（比如函数、DO块）内使用，直接在顶层SQL里用会报错。下面是几种可行的解决方式：

方法1：使用psql内置变量（最适合初始化脚本场景） #

Postgres镜像的初始化脚本是通过psql执行的，所以可以用psql的\set命令定义会话级变量，然后通过:变量名引用：

-- 定义字符串变量
\set SOME_VAR 'initial_value'
-- 定义数字变量
\set SOME_NUM 100

-- 创建表时使用变量
CREATE TABLE user_info (
    id SERIAL PRIMARY KEY,
    default_role VARCHAR(20) DEFAULT :'SOME_VAR',
    max_limit INT DEFAULT :SOME_NUM
);

-- 插入数据时复用变量
INSERT INTO user_info (default_role) VALUES (:'SOME_VAR');

注意：字符串变量引用时要加单引号包裹（:'SOME_VAR'），数字变量直接用:SOME_NUM即可。

方法2：用PL/pgSQL的DO块处理复杂逻辑 #

如果需要变量配合条件判断、循环等逻辑，用DO块包裹代码，在块内用DECLARE定义变量：

DO $$
DECLARE
    -- 定义变量并赋值
    DEFAULT_CATEGORY VARCHAR(30) := 'uncategorized';
    INITIAL_USER_COUNT INT := 5;
BEGIN
    -- 先创建表
    CREATE TABLE IF NOT EXISTS products (
        id SERIAL PRIMARY KEY,
        name VARCHAR(100),
        category VARCHAR(30)
    );

    -- 批量插入初始化数据
    FOR i IN 1..INITIAL_USER_COUNT LOOP
        INSERT INTO products (name, category) 
        VALUES ('product_' || i, DEFAULT_CATEGORY);
    END LOOP;
END $$;

如果需要动态拼接SQL（比如用变量作为表名），一定要用EXECUTE和quote_ident()/quote_literal()避免SQL注入：

DO $$
DECLARE
    TABLE_NAME VARCHAR(30) := 'products';
BEGIN
    EXECUTE 'CREATE TABLE IF NOT EXISTS ' || quote_ident(TABLE_NAME) || ' (id SERIAL PRIMARY KEY, name VARCHAR(100))';
END $$;

方法3：从Docker环境变量传递值 #

如果变量需要从外部配置（比如docker-compose.yml）传入，直接在脚本中引用环境变量：
在docker-compose.yml中定义环境变量：

services:
  postgres:
    image: postgres:15-alpine
    environment:
      POSTGRES_DB: mydb
      POSTGRES_USER: myuser
      POSTGRES_PASSWORD: mypass
      APP_DEFAULT_ROLE: 'admin'
    volumes:
      - ./init.sql:/docker-entrypoint-initdb.d/init.sql

然后在init.sql中通过psql的:envvar()语法引用：

\set DEFAULT_ROLE :envvar(APP_DEFAULT_ROLE)

CREATE TABLE users (
    id SERIAL PRIMARY KEY,
    username VARCHAR(50),
    role VARCHAR(20) DEFAULT :'DEFAULT_ROLE'
);

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