postgresql - 将参数传递给 psql 程序脚本

Question

我在 psql 上有一个循环命令脚本，如下所示:

脚本.sql

DO $$DECLARE
    data_rec RECORD;
    r RECORD;
    r2 RECORD;
BEGIN
select mytables.data_id into data_rec from mytables where id = :arg1;

    FOR r IN select * from
    (select * from ...)
    LOOP
        FOR r2 IN select * from
        (...)
              LOOP
            ......
          END LOOP;
    END LOOP;
END$$;

我想将 arg1 作为命令行参数传递:

psql -h db.server.com -f script.sql -v arg1=1234 > foo.out

但是我一直在 where id = :arg1 处收到语法错误，所以我不知道如何将一个简单的参数传递给这个脚本。欢迎提出建议

Answer 1

可能性很小，怎么办。匿名 block 与客户端环境相对隔离，因此参数化不是直观的 - 但它不是艰苦的工作。

您可以使用服务器端 session 变量。这些变量可从两种环境访问。

postgres=# \set txt Ahoj
postgres=# set myvars.txt to :'txt'; -- fill server side variable
SET
postgres=# do $$ begin raise notice '%', current_setting('myvars.txt'); end;$$;
NOTICE:  Ahoj
DO

稍微复杂一点的例子

bash-4.1$ cat test.sh
echo "
set myvars.msgcount TO :'msgcount'; 
DO \$\$ 
BEGIN 
  FOR i IN 1..current_setting('myvars.msgcount')::int LOOP 
    RAISE NOTICE 'Hello';
  END LOOP; 
END \$\$" | psql postgres -v msgcount=$1


bash-4.1$ sh test.sh 10
SET
Time: 0.935 ms
NOTICE:  Hello
NOTICE:  Hello
NOTICE:  Hello
NOTICE:  Hello
NOTICE:  Hello
NOTICE:  Hello
NOTICE:  Hello
NOTICE:  Hello
NOTICE:  Hello
NOTICE:  Hello
DO
Time: 1.709 ms