如何在PostgreSQL的\copy命令中使用变量？

方法1：通过临时表中转 #

先将符合条件的数据存入临时表，再导出临时表的数据：

\set date '2023-11-01'
-- 创建临时表存储查询结果
CREATE TEMP TABLE temp_quote AS SELECT * FROM quote WHERE date = :'date';
-- 导出临时表到文件
\copy temp_quote TO '/tmp/result.csv' WITH (FORMAT csv);
-- 临时表会在会话结束后自动删除，手动删除可选
DROP TABLE temp_quote;

方法2：拼接完整命令后执行 #

利用psql的变量拼接功能，先构造完整的\copy命令，再执行：

\set date '2023-11-01'
-- 拼接\copy命令，注意转义单引号
\set copy_cmd '\\copy (SELECT * FROM quote WHERE date = ''' :date ''' ) TO ''/tmp/result.csv'' WITH (FORMAT csv)'
-- 可选：验证拼接后的命令是否正确
\echo :copy_cmd
-- 执行拼接好的命令
\gexec

方法3：使用SELECT结合\g重定向 #

如果只需客户端导出结果，可直接用\g将查询结果重定向到文件：

\set date '2023-11-01'
SELECT * FROM quote WHERE date = :'date' \g /tmp/result.csv

报错原因 #

psql的变量替换（:变量语法）仅在psql客户端层面解析，而\copy (...)会将括号内的SQL语句直接发送给PostgreSQL服务器执行，服务器无法识别psql的变量语法，因此会报语法错误。上述方法均通过绕开服务器直接处理变量的方式解决问题。

内容的提问来源于stack exchange，提问作者showkey