Postgres - 在bash中测试数据库连接

73

我想知道是否有替代psql命令的方法,可以使用bash测试连接到postgresql数据库。

我正在设置一个Core OS集群,并且有一个辅助服务,应该每分钟执行相当于psql 'host=xxx port=xxx dbname=xxx user=xxx' 的操作,以确定服务是否正在运行,更重要的是,是否可以使用给定参数连接到它)。

我不能在Core OS上直接安装postgres。在Core OS中通常使用的命令类似于curl -f ${COREOS_PUBLIC_IPV4}:%i;。但它只告诉您服务本身是否在给定端口上运行,没有任何访问检查。

提前感谢!


2
我无法直接在Core OS上安装Postgres:那么你可以安装什么? - Daniel Vérité
Docker容器...基本上选项似乎只有两个:1.在bash中完成工作或2.运行一个容器(这对于检查其他单元的可用性来说相当不理想) - Raphael
5个回答

125

pg_isready是一个用于检查PostgreSQL数据库服务器连接状态的实用程序。退出状态指定连接检查的结果。

它可以轻松地在bash中使用。PostgresSQL Docs - pg_isready

示例用法:

pg_isready -d <db_name> -h <host_name> -p <port_number> -U <db_user>                      

安装

sudo apt install -y postgresql-client

退出状态

pg_isready将以下内容返回给Shell:

  0 - if the server is accepting connections normally, 
  1 - if the server is rejecting connections (for example during startup), 
  2 - if there was no response to the connection attempt, and 
  3 - if no attempt was made (for example due to invalid parameters).

注意man pg_isready说明:获取服务器状态时不需要提供正确的用户名、密码或数据库名,但如果提供了错误的值,则服务器将记录连接尝试失败。


在我看来,这应该是被接受的答案 @Raphael - Nam G VU
@FreeSoftwareServers 一个例子:pg_isready -d dbname -h localhost -p 5432 -U postgres - Laurent
@Laurent,这个例子已经在4月23日的答案中添加了,不过还是谢谢你。 - FreeSoftwareServers
1
他说:我无法直接在Core OS上安装postgres。因此,他无法使用任何postgres实用程序。 - Gilberto
@NamGVU 这是一个糟糕的回答,根本没有回答问题。 - IC_

18

你可以使用自己喜欢的编程语言编写一个简单的连接脚本。

希望你的核心操作系统已经安装了Perl、PHP、Python、Ruby等语言之一。

这里是一个Python示例:

#!/usr/bin/python2.4
#
import psycopg2
try:
    db = psycopg2.connect("dbname='...' user='...' host='...' password='...'")
except:
    exit(1)

exit(0)

现在你的命令行看起来像这样

python psqltest.py && echo 'OK' || echo 'FAIL'

谢谢大卫。不幸的是,这些语言都没有安装在CoreOs上(因为它是只读的...)。 - Raphael
这并没有回答如何从BASH进行测试的问题。 - Steven K7FAQ
1
同意Steven的观点,其次我的Docker容器中没有psycopg2。 - vogash

3
对于一个纯Bash实现,你可以使用以下代码:
DB_HOST=localhost DB_PORT=5432 bash -c 'printf "" 2>>/dev/null >>/dev/tcp/${DB_HOST}/${DB_PORT}'

如果数据库准备好接受连接,它将返回0,否则返回1

注意:这只会测试数据库的准备就绪状态,但不执行任何身份验证。

我在docker-compose中使用它作为入口点,在启动服务之前等待db准备就绪(depends_on仅检查容器是否存活)。

例如:

    # wait for db to accept connections, then start the process
    entrypoint: |
      bash -c '
        until printf "" 2>>/dev/null >>/dev/tcp/db-host/5432;
          do sleep 1;
        done;
        target/release/app; # execute application
      '

这个命令检查端口$DB_PORT是否在$DB_HOST上打开,并且是否有进程在该端口上监听。但这并不能验证数据库是否正在运行以及我们是否能够连接到它。 - Pierre-Gilles Levallois

2
您可以构建一个简单的容器,它扩展了第一个容器(以节省磁盘空间)来执行检查。例如:
FROM postgres

ENTRYPOINT [ "psql", "-h", "$POSTGRES_PORT_5432_TCP_ADDR",  "-p", "$POSTGRES_PORT_5432_TCP_PORT" ]

如果您使用的是与 postgres 不同的镜像,则当然可以使用该镜像。您可以使用几乎任何命令行,仍然可以从 CoreOS 主机上的 bash 检查退出代码:

#!/bin/sh
if ! docker run --link postgres:postgres psql --command "select * from foo;" ; then
   # Do something
fi

1

要在PSQL Shell (psql)中查看连接信息,请使用\c


postgres=# \c conninfo 致命错误:数据库“conninfo”不存在 - Rafis Ganeev
2
@RafisGaneev,只需要\c而不需要conninfo - Sam Kah Chiin
1
我猜conninfo只是样例数据库的一个名称。 - saulius2
1
@saulius2。现在是修改我的帖子的好时机。 - Gene

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接