如何在PostgreSQL触发器函数中检查指定列是否存在？

嘿，这个需求我之前帮人解决过，在PostgreSQL的PL/pgSQL触发器函数里检查NEW对象对应的列是否存在，其实可以借助PostgreSQL的系统函数和触发器内置的变量来轻松实现，我给你两种实用的方案：

方案1：使用pg_has_column系统函数（推荐） #

这个方案最简洁高效，PostgreSQL提供的pg_has_column函数可以直接判断指定模式、表中是否存在某列，结合触发器内置的TG_TABLE_SCHEMA（当前表的模式名）和TG_TABLE_NAME（当前表名）变量，无需硬编码表信息：

CREATE FUNCTION MyFunction() RETURNS trigger AS $$
BEGIN
    -- 检查当前触发的表中是否存在A列
    IF pg_has_column(TG_TABLE_SCHEMA, TG_TABLE_NAME, 'A') THEN
        -- 列存在时执行你的原有逻辑
        IF NEW.A >= 5 AND NEW.B <= 5 THEN
            -- 这里替换成你需要执行的操作
            RAISE NOTICE 'Column A exists and condition is satisfied!';
        END IF;
    ELSE
        -- 列不存在时的处理逻辑，可根据需求调整（比如抛出警告、直接返回等）
        RAISE NOTICE 'Column A does not exist in table %.%', TG_TABLE_SCHEMA, TG_TABLE_NAME;
    END IF;
    RETURN NEW;
END;
$$ LANGUAGE plpgsql;

代码解释： #

• pg_has_column(schema_name text, table_name text, column_name text)：返回布尔值，true表示列存在，false表示不存在。
• TG_TABLE_SCHEMA和TG_TABLE_NAME是触发器函数的内置变量，自动获取当前触发触发器的表所属模式和表名，适配多表绑定的场景。

方案2：查询information_schema.columns系统视图 #

如果你更习惯用标准SQL的方式查询系统信息，可以通过information_schema.columns视图来判断列是否存在，逻辑更直观：

CREATE FUNCTION MyFunction() RETURNS trigger AS $$
DECLARE
    column_exists BOOLEAN;
BEGIN
    -- 查询列是否存在
    SELECT EXISTS(
        SELECT 1 FROM information_schema.columns
        WHERE table_schema = TG_TABLE_SCHEMA
          AND table_name = TG_TABLE_NAME
          AND column_name = 'A'
    ) INTO column_exists;

    IF column_exists THEN
        -- 列存在时执行原有逻辑
        IF NEW.A >= 5 AND NEW.B <= 5 THEN
            -- 这里替换成你需要执行的操作
            RAISE NOTICE 'Column A exists and condition is satisfied!';
        END IF;
    ELSE
        -- 列不存在时的处理逻辑
        RAISE NOTICE 'Column A does not exist in table %.%', TG_TABLE_SCHEMA, TG_TABLE_NAME;
    END IF;
    RETURN NEW;
END;
$$ LANGUAGE plpgsql;

注意事项： #

• 两种方案都支持触发器绑定到多个表的场景，因为依赖的是触发器内置变量，而非硬编码表名。
• 确保执行该触发器函数的数据库用户拥有读取系统目录的权限，通常默认权限即可满足，若遇到权限问题，可联系数据库管理员调整权限。

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