如何将表名传递给 plpgsql 函数

Question

我正在尝试运行代码：

CREATE OR REPLACE FUNCTION anly_work_tbls.testfncjh (tablename text) returns int
AS $$
DECLARE 
        counter int;
        rec record;
        tname text;
BEGIN
counter = 0;
tname := tablename; 
FOR rec IN
        select *
        from tname
    loop
        counter = counter + 1;
    end loop;
RETURN counter;
END; 
$$ 
LANGUAGE 'plpgsql' IMMUTABLE
SECURITY DEFINER;

此代码的目标是返回您输入的表中的行数。我知道这可能不是完成这项任务的最佳方式，但这个函数的结构可以很好地扩展到我试图解决的另一个问题。每次我运行代码，我都会得到错误：

  ERROR: syntax error at or near "$1"

我找到的所有在线资源都告诉我如何在 EXECUTE 块中使用输入变量，但在上述情况下没有。

当前运行 PostgreSQL 8.2.15。

Answer 1

CREATE OR REPLACE FUNCTION anly_work_tbls.testfncjh (tbl regclass, OUT row_ct int) AS
$func$
BEGIN    
   EXECUTE 'SELECT count(*) FROM '|| tbl
   INTO row_ct;
END
$func$ LANGUAGE plpgsql IMMUTABLE SECURITY DEFINER;

呼叫：

SELECT anly_work_tbls.testfncjh('anly_work_tbls.text_tbl');

这应该适用于 Postgres 8.2，但无论如何你 consider upgrading to a current version。

• 我将表名作为object identifier type regclass 传递，它负责自动引用并使用模式限定名称。细节：
  • Table name as a PostgreSQL function parameter

• 使用OUT parameter 可以简化函数。

• 不要引用语言名称。这是一个标识符。

如果你确实需要遍历动态查询的结果：

CREATE OR REPLACE FUNCTION anly_work_tbls.testfncjh (tbl regclass)
  RETURNS int AS
$func$
DECLARE 
   counter int := 0;  -- init at declaration time
   rec record;
BEGIN
   FOR rec IN EXECUTE
      'SELECT * FROM ' || tbl
   LOOP
      counter := counter + 1;  -- placeholder for some serious action
   END LOOP;

   RETURN counter;
END
$func$  LANGUAGE plpgsql IMMUTABLE SECURITY DEFINER;

• 阅读手册中的这一章：Looping Through Query Results

• plpgsql 中记录的赋值运算符是:=:

  • The forgotten assignment operator "=" and the commonplace ":="

Answer 2

是的确实不是最好的方法，但这会起作用：

CREATE OR REPLACE FUNCTION testfncjh (tablename text) returns int
AS $$
DECLARE 
        counter int;
        rec record;        
BEGIN
counter = 0;
FOR rec IN
        EXECUTE 'select * from '||quote_ident(tablename) loop
        counter = counter + 1;
    end loop;
RETURN counter;
END; 
$$ 
LANGUAGE 'plpgsql' IMMUTABLE
SECURITY DEFINER;

这样会更好：

CREATE OR REPLACE FUNCTION testfncjh (tablename text) returns int
AS $$
DECLARE _count INT;
BEGIN    
 EXECUTE 'SELECT count(*) FROM '|| quote_ident(tablename) INTO _count;
 RETURN _count;
END; 
$$ 
LANGUAGE 'plpgsql' IMMUTABLE
SECURITY DEFINER;