PL/pgSQL 函数中的动态 SELECT INTO

Question

如何在 Postgres 的 PL/pgSQL 函数中编写动态 SELECT INTO 查询？

假设我有一个名为tb_name 的变量，它填充在来自information_schema.tables 的FOR 循环中。现在我有一个名为tc 的变量，它将获取每个表的行数。我想要以下内容：

FOR tb_name in select table_name from information_schema.tables where table_schema='some_schema' and table_name like '%1%'
LOOP
EXECUTE FORMAT('select count(*) into' || tc 'from' || tb_name);
END LOOP

这种情况下tb_name和tc的数据类型应该是什么？

Answer 1

CREATE OR REPLACE FUNCTION myfunc(_tbl_pattern text, _schema text = 'public')
  RETURNS void AS  -- or whatever you want to return
$func$
DECLARE
   _tb_name information_schema.tables.table_name%TYPE;  -- currently varchar
   _tc      bigint;  -- count() returns bigint
BEGIN
   FOR _tb_name IN
      SELECT table_name
      FROM   information_schema.tables
      WHERE  table_schema = _schema
      AND    table_name   ~ _tbl_pattern  -- see below!
   LOOP
      EXECUTE format('SELECT count(*) FROM %I.%I', _schema, _tb_name)
      INTO _tc;      

      -- do something with _tc
   END LOOP;
END
$func$  LANGUAGE plpgsql;

注意事项

• 我在所有参数和变量前加上下划线 (_) 以避免与表列发生命名冲突。只是一个有用的约定。

• _tc 应该是 bigint，因为这是聚合函数 count() 返回的内容。

• _tb_name 的数据类型是从其父列动态派生的：information_schema.tables.table_name%TYPE。请参阅章节Copying Types in the manual。

• 您确定只想要information_schema.tables 中列出的表吗？有道理，但要注意含义。见：

  • How to check if a table exists in a given schema

• a_horse 已经是 pointed to the manual 和 Andy provided a code example。这就是您如何分配从带有EXECUTE 的动态查询返回的单个行或值到（行）变量。 single 列（如示例中的count）自动从行类型分解，因此我们可以直接分配给标量变量tc - 就像我们分配整行一样到记录或行变量。相关：

  • How to get the value of a dynamically generated field name in PL/pgSQL

• Schema-qualify动态查询中的表名。当前search_path 中可能还有其他同名的表，这将导致完全错误（并且非常混乱！）的结果没有模式限定。偷偷摸摸的虫子！或者 this 架构根本不在 search_path 中，这会使函数立即引发异常。

  • How does the search_path influence identifier resolution and the "current schema"

• 始终 引用标识符 正确地防御 SQL 注入和随机错误。架构和表必须单独引用！见：

  • Table name as a PostgreSQL function parameter
  • Truncating all tables in a Postgres database

• 我在table_name ~ _tbl_pattern 中使用正则表达式运算符~ 而不是table_name LIKE ('%' || _tbl_pattern || '%')，这样更简单。无论哪种方式，都要小心模式参数中的特殊字符！见：

  • PostgreSQL Reverse LIKE
  • Escape function for regular expression or LIKE patterns
  • Pattern matching with LIKE, SIMILAR TO or regular expressions in PostgreSQL

• 我在函数调用中为模式名称设置了默认值：_schema text = 'public'。只是为了方便，你可能想要也可能不想要。见：

  • Assigning default value for type

---

寻址your comment：要传递值，请使用USING 子句，例如：

EXECUTE format('SELECT count(*) FROM %I.%I
                WHERE some_column = $1', _schema, _tb_name,column_name)
USING user_def_variable;

相关：

• INSERT with dynamic table name in trigger function

Answer 2

您似乎想要FORMAT 的占位符%I，以便它将您的变量视为标识符。此外，INTO 子句应在准备好的语句之外。

FOR tb_name in select table_name from information_schema.tables where table_schema='some_schema' and table_name like '%1%'
LOOP
  EXECUTE FORMAT('select count(*) from %I', tb_name) INTO tc;
END LOOP