PostgreSQL中如何动态访问存储在变量中的查询返回列

方法1：利用JSONB类型转换 #

将record对象转换为JSONB类型后，即可通过列名（键名）直接取值，无需额外扩展，实现直观：

CREATE FUNCTION my_func(col text) RETURNS integer LANGUAGE plpgsql
$$
DECLARE
val text;
rec record;
BEGIN
FOR rec IN
  -- 用format函数避免SQL注入，%I用于安全转义标识符
  EXECUTE format('SELECT %I FROM employee_tb WHERE sal > 10000', col)
LOOP
    -- 转换为jsonb后通过列名提取值
    val := (to_jsonb(rec) ->> col);
    -- 可添加对val的业务处理逻辑，比如打印日志
    RAISE NOTICE '当前值: %', val;
END LOOP;
RETURN 1;
$$;

-- 调用函数
select my_func('emp_code');

如果需要特定数据类型，可在取值后做类型转换，例如(to_jsonb(rec) ->> col)::integer。

方法2：使用hstore扩展（需先启用） #

若数据库已安装hstore扩展，可将record转为hstore类型后取值：

首先启用扩展：

CREATE EXTENSION IF NOT EXISTS hstore;

修改后的函数：

CREATE FUNCTION my_func(col text) RETURNS integer LANGUAGE plpgsql
$$
DECLARE
val text;
rec record;
BEGIN
FOR rec IN
  EXECUTE format('SELECT %I FROM employee_tb WHERE sal > 10000', col)
LOOP
    -- 转换为hstore后通过列名取值
    val := (hstore(rec) -> col);
    RAISE NOTICE '当前值: %', val;
END LOOP;
RETURN 1;
$$;

方法3：直接将动态列值存入变量（更高效） #

如果循环仅需该动态列的值，无需获取完整record，可直接在EXECUTE时将结果赋值给变量，代码更简洁且性能更优：

CREATE FUNCTION my_func(col text) RETURNS integer LANGUAGE plpgsql
$$
DECLARE
val text;
BEGIN
FOR val IN
  EXECUTE format('SELECT %I FROM employee_tb WHERE sal > 10000', col)
LOOP
    -- 直接使用val即可
    RAISE NOTICE '当前值: %', val;
END LOOP;
RETURN 1;
$$;

关键注意事项 #

• 避免SQL注入：永远不要直接拼接用户传入的列名到SQL语句中，使用format函数的%I占位符可自动转义标识符，杜绝注入风险。

内容的提问来源于stack exchange，提问作者simply-put