如何在PostgreSQL的SQL查询中动态获取指定表主键列的所有值

核心逻辑分为两步：首先通过数据库系统表查询目标表的主键字段名，再动态拼接SQL语句执行，全程做标识符转义避免SQL注入风险。

PostgreSQL 实现示例 #

你提到用character varying类型存储SQL语句，以下是PL/pgSQL可直接使用的函数：

CREATE OR REPLACE FUNCTION get_primary_key_values(p_table_name text)
RETURNS SETOF text AS $$
DECLARE
    v_pk_column text;
    v_exec_sql character varying; -- 存储动态SQL的变量
BEGIN
    -- 查询目标表的主键列名
    SELECT a.attname
    INTO v_pk_column
    FROM pg_index i
    JOIN pg_attribute a ON a.attrelid = i.indrelid AND a.attnum = ANY(i.indkey)
    JOIN pg_class c ON c.oid = i.indrelid
    JOIN pg_namespace n ON n.oid = c.relnamespace
    WHERE c.relname = p_table_name
      AND i.indisprimary = true
      AND n.nspname = 'public' -- 可根据实际使用的schema调整
    LIMIT 1; -- 单字段主键场景，联合主键可自行扩展逻辑

    -- 异常校验
    IF v_pk_column IS NULL THEN
        RAISE EXCEPTION '表 % 不存在或未设置主键', p_table_name;
    END IF;

    -- 拼接动态SQL，%I会自动转义标识符避免注入
    v_exec_sql := format('SELECT %I FROM %I', v_pk_column, p_table_name);

    -- 执行SQL返回结果
    RETURN QUERY EXECUTE v_exec_sql;
END;
$$ LANGUAGE plpgsql;

调用方法 #

-- 查询car表所有主键值
SELECT * FROM get_primary_key_values('car');
-- 查询boat表所有主键值
SELECT * FROM get_primary_key_values('boat');

其他数据库适配说明 #

核心逻辑完全一致，仅需要调整查询主键列名的语句即可，以MySQL为例，查主键列的语句为：

SELECT COLUMN_NAME
FROM information_schema.KEY_COLUMN_USAGE
WHERE TABLE_NAME = '目标表名'
  AND CONSTRAINT_NAME = 'PRIMARY'
  AND TABLE_SCHEMA = '所属库名';

注意事项 #

• 禁止直接拼接用户输入的表名字符串，必须用数据库提供的标识符转义方法处理，避免SQL注入风险
• 若业务场景存在联合主键，调整查询主键列的逻辑，将所有主键字段查出后用逗号分隔拼接到SQL中即可

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