postgresql - PL/pgSQL 从表中动态复制数据

Question

我正在尝试编写一些 SQL 以根据 information_schema 中的内容从给定数据库中的所有 PostgreSQL 表复制数据。它应该将数据文件输出到我的本地机器，准备导入到另一台机器。最后，我将对此进行调整，以便我只转储选定的部分表(我转储的一些表有数百万条记录，我只需要一小部分数据用于测试目的)。

这是我目前所拥有的...

--Copy all tables...
DO
$$
DECLARE
    formatstring text;
    rec record;
BEGIN
    RAISE NOTICE 'Copying tables...';
    formatstring = 'COPY (select * from %I) to ''C:\Media\Code\%s.csv'';';
    FOR rec IN 
        select table_name from information_schema.tables where table_schema = 'public' order by table_name
    LOOP
        RAISE NOTICE 'Table: %', rec.table_name;
        RAISE NOTICE format(formatstring, rec.table_name, rec.table_name);
        EXECUTE format(formatstring, rec.table_name, rec.table_name);
    END LOOP;
END;
$$
LANGUAGE plpgsql;

但是，我遇到了这个异常......

ERROR:  unrecognized exception condition "format"
CONTEXT:  compilation of PL/pgSQL function "inline_code_block" near line 12
********** Error **********

ERROR: unrecognized exception condition "format"
SQL state: 42704
Context: compilation of PL/pgSQL function "inline_code_block" near line 12

单引号的转义似乎很好(已经检查过这个问题:Insert text with single quotes in PostgreSQL)。事实上，我可以执行以下操作并且它有效，文本被插入到格式中:

select format('COPY (select * from %I) to ''C:\Media\Code\%s.csv'';', 'system_user', 'system_user');

有人可以协助解决这个问题吗？我可以轻松编写一个脚本或代码来为我生成复制命令，但如果能在一个简单的 SQL 中完成这一切就太好了。

Answer 1

原因是您的第 3 个 RAISE 语句中存在语法错误。有several valid formats ，但您不能将 expression 直接提供给 RAISE。它必须是字符串文字 - 带有字符串插值选项。

同时，简化其他几件事:

DO
$do$
DECLARE
   _sql text;
   _tbl text;
BEGIN
   RAISE NOTICE 'Copying tables...';
   
   FOR _tbl IN 
      SELECT table_name
      FROM   information_schema.tables
      WHERE  table_schema = 'public'
      ORDER  BY table_name
   LOOP
      _sql := format($$COPY %1$I TO 'C:\Media\Code\%1$s.csv'$$, _tbl);
      RAISE NOTICE 'Table: %', _tbl;
      RAISE NOTICE '%', _sql;            -- fixed!
      EXECUTE _sql;
   END LOOP;
END
$do$;

• 使用带有COPY 的普通表名而不是 SELECT * FROM tbl。
• 使用嵌套 dollar-quotes .
• 格式说明符 %1$I and %1$s for the format()函数，所以我们只需要提供一次表名。
• 在 FOR 循环中使用标量变量而不是 record - 无论如何我们只需要一列。