PostgreSQL COPY 命令中动态生成的表名

发布于 2024-10-16 00:45:27 字数 122 浏览 5 评论 0原文

这个 PostgreSQL COPY 命令有效：

copy tablename from E'c:\\abc\\a.txt';

但我希望动态生成表名。我该怎么做？

原文

This PostgreSQL COPY command works:

copy tablename from E'c:\\abc\\a.txt';

but I want the tablename to be dynamically generated. How can I do this?

分享到QQ

分享到微博

如果你对这篇内容有疑问，欢迎到本站社区发帖提问参与讨论，获取更多帮助，或者扫码二维码加入 Web 技术交流群。

发布评论

需要登录才能够评论，你可以免费注册一个本站的账号。

橘亓 2024-10-23 00:45:27

您需要构建一个字符串，连接动态表名称，然后使用execute。请注意，您对 ' by '' 进行了转义。这还包括保存文件的动态名称。您需要将 savedir 替换为您正在使用的实际目录。

CREATE OR REPLACE FUNCTION dynamicCopy(tablename text, outname text) RETURNS VOID AS $

DECLARE STATEMENT TEXT;

BEGIN

  STATEMENT := 'COPY (select * from ' || quote_ident(tablename) || ') to ''savedir' || outname ||'.txt''';

  EXECUTE STATEMENT;

END;

$ LANGUAGE 'plpgsql';

编辑：

自从我第一次写这篇文章以来，我发现了格式函数，我认为它通常比使用连接运算符 || 生成的 SQL 更容易阅读。并且更加灵活。

CREATE OR REPLACE FUNCTION dynamicCopy(tablename text, outname text) RETURNS VOID AS 
$BODY$
BEGIN
  EXECUTE FORMAT('COPY (SELECT * FROM %s) TO ''savedir%s.csv''',
                  tablename, 
                  outname);
END
$BODY$ 
LANGUAGE plpgsql;

有关完整讨论，请参阅官方文档： https://www.postgresql.org/docs/current/static/plpgsql-statements.html#PLPGSQL-STATEMENTS-EXECUTING-DYN

You need to build a string, concatenating in the dynamic table name, and then use execute. Note that you escape the ' by ''. This also includes a dynamic name to save the file too. You need to replace savedir with the actual directory you are using.

CREATE OR REPLACE FUNCTION dynamicCopy(tablename text, outname text) RETURNS VOID AS $

DECLARE STATEMENT TEXT;

BEGIN

  STATEMENT := 'COPY (select * from ' || quote_ident(tablename) || ') to ''savedir' || outname ||'.txt''';

  EXECUTE STATEMENT;

END;

$ LANGUAGE 'plpgsql';

EDIT:

Since I first wrote this, I have discovered the format function, which I think is generally easier to read than SQL generated with the concatenation operator || and more flexible.

CREATE OR REPLACE FUNCTION dynamicCopy(tablename text, outname text) RETURNS VOID AS 
$BODY$
BEGIN
  EXECUTE FORMAT('COPY (SELECT * FROM %s) TO ''savedir%s.csv''',
                  tablename, 
                  outname);
END
$BODY$ 
LANGUAGE plpgsql;

See the official docs for a full discussion: https://www.postgresql.org/docs/current/static/plpgsql-statements.html#PLPGSQL-STATEMENTS-EXECUTING-DYN

回复收藏 0 原文

浪漫人生路 2024-10-23 00:45:27

您可以通过 shell 脚本将多个 csv 文件复制到表中。

制作脚本文件：vim csvtotable
编写csvtotable脚本。这是我的例子：
#!/bin/sh DBNAME=postgres files=$1 表示 ${files} 中的文件；执行 psql -d ${DBNAME} -c "\copy Parent_tree(parent_id, some_text) FROM '${file}' 分隔符 ',' csv header" 完成
执行脚本。
./csv2table“$(ls *.out.csv)”

显然，本地 CSV 文件应该与表匹配。然后，如果 csv 文件名以 .out.csv 结尾，它将从 csv 导入到数据库表。
我不确定 csv 名称匹配是全局匹配还是仅当前目录匹配。
参考链接：
-c 命令参考 https ://www.postgresql.org/docs/current/app-psql.html#APP-PSQL-PATTERNS
chomd 命令： https://linuxize.com/post/chmod-command-in- linux/
bash shanebang： https://linuxize.com/post/bash-shebang/