在PSQL脚本中使用环境变量

Question

是否可以在.sql文件中使用Linux环境变量？我正在使用copy/select查询写入输出文件,我想将该目录放在变量中.所以我想做一些事情:

COPY (SELECT * FROM a)
TO $outputdir/a.csv

Outputdir将在我的环境中设置.这可能吗？

Answer 1

您可以将shell命令的结果存储在psql变量中,如下所示:

\set afile `echo "$outputdir/a.csv"`
COPY (SELECT * FROM a) TO :'afile';

另一个(在我看来更好)解决方案是仅使用psql变量,请参阅我的关于psql变量的答案,这与您的示例类似.您案例的一个例子是:

\set outputdir '/path/to/output'
\set afile :outputdir '/a.csv'
COPY (SELECT * FROM a) TO :'afile';

请注意,在示例中,您需要在脚本文件中设置变量,但如果在调用时设置它,则可以跳过第一行psql:

psql --set=outputdir="$outputdir" <conn parameters> -f /path/to/yourscript.sql