PL/pgSQL函数中如何通过参数指定表名实现查询？

你说得完全正确：PostgreSQL函数的返回类型必须在定义阶段明确指定，不能用函数参数来指代（毕竟参数值是运行时才传入的），所以RETURNS SETOF table_name会被数据库当成要查找名为table_name的自定义类型，这自然会报错“type does not exist”。

要实现传入表名动态查询的需求，有几种实用的方案，下面逐一说明：

方案1：返回SETOF record，调用时指定列结构 #

这种方法最直接，函数返回通用的record类型，调用时手动指定目标表的列结构即可：

CREATE OR REPLACE FUNCTION select_from_table(table_name varchar(63)) 
RETURNS SETOF record 
AS $$
DECLARE
    -- 用quote_ident转义表名，防止SQL注入和特殊字符/关键字问题
    query TEXT := 'SELECT * FROM ' || quote_ident(table_name);
BEGIN
    RETURN QUERY EXECUTE query;
END;
$$ LANGUAGE plpgsql;

调用示例 #

假设你要查询的表是users，结构为id int, name text, email varchar，调用时需要明确指定返回的列结构：

SELECT * FROM select_from_table('users') AS users(id int, name text, email varchar);

注意：必须用quote_ident（或者format('%I', table_name)）处理表名，否则如果表名包含空格、特殊字符或者是SQL关键字，会导致语法错误，更重要的是能防止SQL注入攻击。

方案2：使用多态函数（推荐） #

利用PostgreSQL的anyelement多态类型，可以让函数自动适配目标表的结构，不需要手动写列定义，体验更友好：

CREATE OR REPLACE FUNCTION select_from_table(p_table regclass, p_sample anyelement) 
RETURNS SETOF anyelement 
AS $$
BEGIN
    -- regclass类型会自动处理表名转义，比手动拼接更安全
    RETURN QUERY EXECUTE format('SELECT * FROM %s', p_table);
END;
$$ LANGUAGE plpgsql;

调用示例 #

调用时只需传入表名（转成regclass类型）和一个目标表类型的示例值（用NULL::表名即可）：

-- 查询users表
SELECT * FROM select_from_table('users'::regclass, NULL::users);

-- 查询products表
SELECT * FROM select_from_table('products'::regclass, NULL::products);

这种方法的优势是：

• 不需要手动维护列结构，数据库会自动通过p_sample推断返回类型
• regclass类型会自动处理表名的转义、schema前缀等问题，安全性更高

额外注意事项 #

• 权限控制：确保函数的调用者拥有目标表的SELECT权限，否则会出现权限错误
• SQL注入防护：永远不要直接拼接未处理的用户输入作为表名，一定要用quote_ident、format('%I')或者regclass类型来处理

内容的提问来源于stack exchange，提问作者Subtle Development Space