sql文件中的Postgresql命令行變量用法

Question

我正在使用 psql 命令將文件繞過參數調用為-v argument=value 。 在 SQL 文件中，參數需要用作字符串的一部分。 例如prefix_:argument_suffix 。 這是行不通的。 如果我將它用作prefix_:argument它會起作用，但是當涉及到后綴時它不會。

這是用於在create database語句中構建數據庫名稱。

有人可以幫忙嗎？

Answer 1

使用另一個變量，如：

nd@postgres=# \set argument 'aaa'
nd@postgres=# \set dbname 'foo_':argument'_bar'
nd@postgres=# select :'dbname';
┌─────────────┐
│  ?column?   │
├─────────────┤
│ foo_aaa_bar │
└─────────────┘

nd@postgres=# create database :dbname;
CREATE DATABASE

nd@postgres=# \l
                                    List of databases
┌─────────────┬──────────┬──────────┬─────────────┬─────────────┬───────────────────────┐
│    Name     │  Owner   │ Encoding │   Collate   │    Ctype    │   Access privileges   │
├─────────────┼──────────┼──────────┼─────────────┼─────────────┼───────────────────────┤
│ foo_aaa_bar │ nd       │ UTF8     │ en_US.UTF-8 │ en_US.UTF-8 │                       │
│ postgres    │ postgres │ UTF8     │ en_US.UTF-8 │ en_US.UTF-8 │                       │
│ template0   │ postgres │ UTF8     │ en_US.UTF-8 │ en_US.UTF-8 │ =c/postgres          ↵│
│             │          │          │             │             │ postgres=CTc/postgres │
│ template1   │ postgres │ UTF8     │ en_US.UTF-8 │ en_US.UTF-8 │ =c/postgres          ↵│
│             │          │          │             │             │ postgres=CTc/postgres │
└─────────────┴──────────┴──────────┴─────────────┴─────────────┴───────────────────────┘

Answer 2

該字符串可以構建為

'prefix_' || :'variable' || '_suffix'

要在CREATE DATABASE語句中使用它，您必須在函數中使用動態 SQL：

CREATE FUNCTION pg_temp.mkdb(var text) RETURNS void
   LANGUAGE plpgsql STRICT AS
$$BEGIN
   EXECUTE format('CREATE DATABASE %I;', 'prefix_' || var || '_suffix');
END;$$;

SELECT pg_temp.mkdb(:'variable');

由於該函數是在臨時模式中創建的，因此它將僅存在於該數據庫會話中。