如何将游标的 table 名称作为函数参数传递？

Question

在 PostgreSQL 9.1 中，我需要将一个 table 名称作为参数传递给我的函数以在游标中使用 - 然后像这样调用函数：

select myfunction('tableA');

或：

select myfunction('tableB')

我试过这个：

 create or replace function myfunction(sid text) 
 returns integer as $BODY$
 declare
      bids cursor is select id from sid; --line 4
 begin
      for bid in bids loop
          --some logic
      end loop;
      return 1;
 end;
 $BODY$
 language plpgsql volatile;

我得到 error on line 4。我明白了，不允许将 table 名称作为函数参数传递给允许休息（选择、投影）的地方。

是否有其他方法可以实现此目的？

Answer 1

在SQL 命令中只能参数化值，不能参数化标识符或关键字。目前它仅适用于 SELECT、INSERT、UPDATE 和 DELETE 命令。详情：

您需要在 plpgsql 中使用 EXECUTE 的动态 SQL。
将其与 FOR 循环的隐式游标相结合，后者通常比显式游标更简单、更快。 Instructions in the manual.

我还建议使用 regclass 参数来安全地传递 有效 table 名称。

CREATE OR REPLACE FUNCTION myfunction(_tbl regclass) 
  RETURNS int  AS
$func$
DECLARE
   bid integer;  -- appropriate data type!
BEGIN
   FOR bid IN EXECUTE
      'SELECT id FROM ' || _tbl
   LOOP
      -- statements can use bid
      RAISE NOTICE '%', bid;
   END LOOP;

   RETURN 1;
END
$func$  LANGUAGE plpgsql;

通话：

SELECT myfunction('tableA');  -- careful: case sensitive!

更多解释：

Table name as a PostgreSQL function parameter

通常，有一个更快的基于集合的解决方案，根本不需要循环。

如果您确实需要一个游标，请声明一个未绑定游标并使用OPEN FOR EXECUTE:

CREATE OR REPLACE FUNCTION myfunction(_tbl text)  -- could be regclass again
  RETURNS int AS
$func$
DECLARE
   _bids refcursor;
   _rec  record;
BEGIN
   OPEN _bids FOR EXECUTE 'SELECT id FROM ' || quote_ident(_tbl);  -- text must be escaped

   LOOP
      FETCH NEXT FROM _bids INTO _rec;
      EXIT WHEN _rec IS NULL;

      -- some logic
      RAISE NOTICE '%', _rec.id;
   END LOOP;

   RETURN 1;
END
$func$  LANGUAGE plpgsql;

同样的电话。这个密切相关的答案中的更多详细信息：

Update record of a cursor where the table name is a parameter

如何将游标的 table 名称作为函数参数传递？

How to pass a table name for a cursor as function parameter?

postgresql

cursor

plpgsql

postgresql-9.1

dynamic-sql