POSTGRESQL PSQL 命令中如何使用变量带入查询和函数

最近有人问，想通过SHELL 来传入变量到 PSQL的SQL 语句中，如何去撰写，因为他写的程序老是有问题。PSQL 命令中被经常DISS的问题除了不能带有密码外，就是这个问题了，变量。

怎么在PSQL 外部将变量设置，并传入到POSTGRESQL命令行内，我们做一个例子：

psql -X -v a=b

\echo THE VALUE OF VAR a IS :a

psql -X --set=a=c

举例：我们想将多个字段传入到PG内，可以将多个值进行引号设置即可

[postgres@pg_qixun ~]$ psql -X -v a='postgresql EDB enterprise database'

psql (14.7)

Type "help" for help.

postgres=# \echo The variable a is :a

The variable a is postgresql EDB enterprise database

下面我们举一个复杂的例子

我们的变量在一个文本中，而我们要执行的脚本在另一个文件中

psql -x -v a="$( cat file.txt )" -f show.sql

而如果你有一个更复杂的执行方式，如同下面的这个例子

[postgres@pg_qixun ~]$ cat show.sql

select * from :a limit 1;

:b

[postgres@pg_qixun ~]$ psql -X -v a="$( cat file.txt )" -v b='select version();' -f show.sql

oid | datname | datdba | encoding | datcollate | datctype | datistemplate | datallowconn | datconnlimit | datlastsysoid | dat

frozenxid | datminmxid | dattablespace | datacl

-------+----------+--------+----------+-------------+-------------+---------------+--------------+--------------+---------------+----

----------+------------+---------------+--------

13892 | postgres | 10 | 6 | en_US.UTF-8 | en_US.UTF-8 | f | t | -1 | 13891 |

1201 | 1 | 1663 |

(1 row)

version

---------------------------------------------------------------------------------------------------------

PostgreSQL 14.7 on x86_64-pc-linux-gnu, compiled by gcc (GCC) 4.8.5 20150623 (Red Hat 4.8.5-44), 64-bit

(1 row)

[postgres@pg_qixun ~]$ psql -X -v a=1 -v b=2 -v c=3 -v d=4 -f show.sql

datname

----------

postgres

(1 row)

datname

-----------

postgres

dvdrental

(2 rows)

datname

-----------

postgres

dvdrental

template1

(3 rows)

datname

-----------

postgres

dvdrental

template1

template0

(4 rows)

[postgres@pg_qixun ~]$ cat show.sql

select datname from pg_database limit :a;

select datname from pg_database limit :b;

select datname from pg_database limit :c;

select datname from pg_database limit :d;

以上为将变量带入查询中的一些简单的操作，而在POSTGRESQL 有一部分情况是通过将变量带入到函数中的，我们下面举一个例子来看看如何将变量带入到函数，我们简单的写一个函数，来进行当前PG实例中有多少数据库的一个计算，但是我们查询的是符合我们要求的PG的数据库的数量，或者判断符合我们名字要求的PG是否存在于

CREATE OR REPLACE FUNCTION count_db(IN var_b varchar(20)) RETURNS INT8 AS $$

DECLARE

v_int INT4;

BEGIN

SELECT COUNT(*) INTO v_int FROM pg_database WHERE datname like var_b;

RETURN v_int;

END;

$$ LANGUAGE plpgsql;

然后我们从外部来调用这个函数并赋予变量

方法1 失败

[postgres@pg_qixun ~]$ psql -X -v a=postgres% -c 'select count_db(:a)'

ERROR: syntax error at or near ":"

LINE 1: select count_db(:a)

方法2 失败

psql -X -v a="postgres%" -c 'select count_db(:a)'

ERROR: syntax error at or near ":"

LINE 1: select count_db(:a)

方法3 失败

[postgres@pg_qixun ~]$ psql -X -v a="postgres%" -f show.sql

psql:show.sql:1: error: invalid command \SET

psql:show.sql:2: ERROR: syntax error at or near ")"

LINE 1: SELECT count_db(postgres%);

[postgres@pg_qixun ~]$ cat show.sql

\SET a postgres%

SELECT count_db(:a);

后面又试过 4 5 6 7 8 种方案，均有问题。

最后经过查阅，如果要在外部调用函数，给出变量是不能单独写语句的，而是要用其他的方式来代替 -c 或 -f 调用命令的方式 ，具体的写法如下：

psql -X -v a="'postgres'" <<< 'select count_db(:a);'