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

AustinDatabases 2023-06-05

661

开头还是介绍一下群，如果感兴趣polardb ,mongodb ,mysql ,postgresql ,redis 等有问题，有需求都可以加群群内有各大数据库行业大咖，CTO，可以解决你的问题。加群请联系 liuaustin3 ，在新加的朋友会分到2群（共820人左右 1 + 2 + 3）新人会进入3群，进瑞典马工群的单独说。

最近有人问，想通过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;

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

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;