30

我想遍历我所有的表来计算每个表中的行数。以下查询给我一个错误：

DO $$
DECLARE
    tables CURSOR FOR
        SELECT tablename FROM pg_tables
        WHERE tablename NOT LIKE 'pg_%'
        ORDER BY tablename;
    tablename varchar(100);
    nbRow int;
BEGIN
    FOR tablename IN tables LOOP
        EXECUTE 'SELECT count(*) FROM ' || tablename INTO nbRow;
        -- Do something with nbRow
    END LOOP;
END$$;

错误：

ERROR:  syntax error at or near ")"
LINE 1: SELECT count(*) FROM (sql_features)
                                          ^
QUERY:  SELECT count(*) FROM (sql_features)
CONTEXT:  PL/pgSQL function inline_code_block line 8 at EXECUTE statement

sql_features是我的数据库中的表名。我已经尝试使用quote_ident()但无济于事。

4

2 回答 2

37

我不记得上次我实际上需要使用显式游标在 plpgsql 中循环。
使用FOR循环的隐式游标，这更干净：

DO
$$
DECLARE
   rec   record;
   nbrow bigint;
BEGIN
   FOR rec IN
      SELECT *
      FROM   pg_tables
      WHERE  tablename NOT LIKE 'pg\_%'
      ORDER  BY tablename
   LOOP
      EXECUTE 'SELECT count(*) FROM '
        || quote_ident(rec.schemaname) || '.'
        || quote_ident(rec.tablename)
      INTO nbrow;
      -- Do something with nbrow
   END LOOP;
END
$$;

您需要包含架构名称以使其适用于所有架构（包括不在您的架构中的架构search_path）。

此外，您实际上需要使用quote_ident()or format()with %Iorregclass变量来防止 SQL 注入。表名几乎可以是双引号内的任何内容。看：

小细节：转义模式_中的下划线 ( )LIKE使其成为文字下划线：tablename NOT LIKE 'pg\_%'

我该怎么做：

DO
$$
DECLARE
    tbl   regclass;
    nbrow bigint;
BEGIN
   FOR tbl IN
      SELECT c.oid
      FROM   pg_class     c
      JOIN   pg_namespace n ON n.oid = c.relnamespace
      WHERE  c.relkind = 'r'
      AND    n.nspname NOT LIKE 'pg\_%'         -- system schema(s)
      AND    n.nspname <> 'information_schema'  -- information schema
      ORDER  BY n.nspname, c.relname
   LOOP
      EXECUTE 'SELECT count(*) FROM ' || tbl INTO nbrow;
      -- raise notice '%: % rows', tbl, nbrow;
   END LOOP;
END
$$;

Querypg_catalog.pg_class而不是tablename，它提供了表的 OID。

对象标识符类型regclass便于简化。特别是，表名在必要时自动被双引号和模式限定（也防止SQL 注入）。

此查询还排除临时表（临时模式在pg_temp%内部命名）。

要仅包含给定模式中的表：

    AND    n.nspname = 'public' -- schema name here, case-sensitive
于 2013-03-07T18:29:35.477 回答
30

游标返回一条记录，而不是标量值，因此“tablename”不是字符串变量。

连接将记录转换为如下所示的字符串(sql_features)。如果您选择了例如带有表名的模式名，则记录的文本表示将是(public,sql_features).

因此，您需要访问记录中的列来创建您的 SQL 语句：

DO $$
DECLARE
    tables CURSOR FOR
        SELECT tablename
        FROM pg_tables
        WHERE tablename NOT LIKE 'pg_%'
        ORDER BY tablename;
    nbRow int;
BEGIN
    FOR table_record IN tables LOOP
        EXECUTE 'SELECT count(*) FROM ' || table_record.tablename INTO nbRow;
        -- Do something with nbRow
    END LOOP;
END$$;

您可能希望使用WHERE schemaname = 'public'而不是not like 'pg_%'排除 Postgres 系统表。

于 2013-03-07T08:46:10.410 回答