PostgreSQL中如何仅在模式和表存在时添加外键约束？

PostgreSQL的CREATE TABLE语句本身不支持在约束定义中加入条件判断，要实现“仅当依赖表存在时才添加外键约束”的需求，需要借助PL/pgSQL的动态逻辑来完成。以下是具体实现方案：

方法一：分步骤创建表并条件添加约束 #

先创建基础表（不含外键约束），再通过条件判断添加外键：

-- 1. 创建second.second_table表（确保表存在，无外键约束）
CREATE TABLE IF NOT EXISTS second.second_table
(
    id serial NOT NULL,
    another_id integer NOT NULL,
    some_column boolean NOT NULL,
    CONSTRAINT other_pkey PRIMARY KEY (id)
);

-- 2. 条件添加外键约束
DO $$
BEGIN
    -- 检查first模式下的first_table是否存在
    IF EXISTS (
        SELECT 1 
        FROM information_schema.tables 
        WHERE table_schema = 'first' 
          AND table_name = 'first_table'
    ) THEN
        -- 先确认约束尚未存在，避免重复添加报错
        IF NOT EXISTS (
            SELECT 1 
            FROM information_schema.table_constraints 
            WHERE constraint_schema = 'second'
              AND table_name = 'second_table'
              AND constraint_name = 'fk_other_another_id'
        ) THEN
            ALTER TABLE second.second_table
            ADD CONSTRAINT fk_other_another_id 
            FOREIGN KEY (another_id) REFERENCES first.first_table (id)
            MATCH SIMPLE;
        END IF;
    END IF;
END $$;

方法二：用单个DO块封装完整逻辑 #

如果想把创建表和约束的逻辑放在一个脚本块中，可以用以下写法：

DO $$
BEGIN
    -- 创建表（如果不存在）
    CREATE TABLE IF NOT EXISTS second.second_table
    (
        id serial NOT NULL,
        another_id integer NOT NULL,
        some_column boolean NOT NULL,
        CONSTRAINT other_pkey PRIMARY KEY (id)
    );

    -- 检查依赖表是否存在，存在则添加外键
    IF EXISTS (
        SELECT 1 
        FROM information_schema.tables 
        WHERE table_schema = 'first' 
          AND table_name = 'first_table'
    ) THEN
        IF NOT EXISTS (
            SELECT 1 
            FROM information_schema.table_constraints 
            WHERE constraint_schema = 'second'
              AND table_name = 'second_table'
              AND constraint_name = 'fk_other_another_id'
        ) THEN
            ALTER TABLE second.second_table
            ADD CONSTRAINT fk_other_another_id 
            FOREIGN KEY (another_id) REFERENCES first.first_table (id)
            MATCH SIMPLE;
        END IF;
    END IF;
END $$;

关键说明 #

• 借助information_schema系统视图检查表和约束的存在性，这是PostgreSQL 11.5完全支持的标准方式，兼容性好。
• IF NOT EXISTS的使用确保脚本可以重复执行而不会抛出“表已存在”或“约束已存在”的错误，适合在测试、开发环境反复执行。
• 生产环境中依赖的first.first_table存在时，外键约束会自动添加；测试/本地环境若未创建该依赖表，约束创建逻辑会被跳过，避免执行报错。

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