diff --git a/src/bin/scripts/.gitignore b/src/bin/scripts/.gitignore new file mode 100644 index e62f4b0..0b9b786 *** a/src/bin/scripts/.gitignore --- b/src/bin/scripts/.gitignore *************** *** 7,12 **** --- 7,13 ---- /dropuser /reindexdb /vacuumdb + /pg_isready /dumputils.c /keywords.c diff --git a/src/bin/scripts/Makefile b/src/bin/scripts/Makefile new file mode 100644 index ea48411..26f6f8b *** a/src/bin/scripts/Makefile --- b/src/bin/scripts/Makefile *************** subdir = src/bin/scripts *** 16,22 **** top_builddir = ../../.. include $(top_builddir)/src/Makefile.global ! PROGRAMS = createdb createlang createuser dropdb droplang dropuser clusterdb vacuumdb reindexdb override CPPFLAGS := -I$(top_srcdir)/src/bin/pg_dump -I$(top_srcdir)/src/bin/psql -I$(libpq_srcdir) $(CPPFLAGS) --- 16,22 ---- top_builddir = ../../.. include $(top_builddir)/src/Makefile.global ! PROGRAMS = createdb createlang createuser dropdb droplang dropuser clusterdb vacuumdb reindexdb pg_isready override CPPFLAGS := -I$(top_srcdir)/src/bin/pg_dump -I$(top_srcdir)/src/bin/psql -I$(libpq_srcdir) $(CPPFLAGS) *************** dropuser: dropuser.o common.o dumputils. *** 34,39 **** --- 34,40 ---- clusterdb: clusterdb.o common.o dumputils.o kwlookup.o keywords.o | submake-libpq vacuumdb: vacuumdb.o common.o dumputils.o kwlookup.o keywords.o | submake-libpq reindexdb: reindexdb.o common.o dumputils.o kwlookup.o keywords.o | submake-libpq + pg_isready: pg_isready.o common.o | submake-libpq submake-libpgport dumputils.c keywords.c: % : $(top_srcdir)/src/bin/pg_dump/% rm -f $@ && $(LN_S) $< . *************** install: all installdirs *** 54,59 **** --- 55,61 ---- $(INSTALL_PROGRAM) clusterdb$(X) '$(DESTDIR)$(bindir)'/clusterdb$(X) $(INSTALL_PROGRAM) vacuumdb$(X) '$(DESTDIR)$(bindir)'/vacuumdb$(X) $(INSTALL_PROGRAM) reindexdb$(X) '$(DESTDIR)$(bindir)'/reindexdb$(X) + $(INSTALL_PROGRAM) pg_isready$(X) '$(DESTDIR)$(bindir)'/pg_isready$(X) installdirs: $(MKDIR_P) '$(DESTDIR)$(bindir)' diff --git a/src/bin/scripts/pg_isready.c b/src/bin/scripts/pg_isready.c new file mode 100644 index ...feee1a7 *** a/src/bin/scripts/pg_isready.c --- b/src/bin/scripts/pg_isready.c *************** *** 0 **** --- 1,202 ---- + /*------------------------------------------------------------------------- + * + * pg_isready --- checks the status of the PostgreSQL server + * + * Copyright (c) 2013, PostgreSQL Global Development Group + * + * src/bin/scripts/pg_isready.c + * + *------------------------------------------------------------------------- + */ + + #include "postgres_fe.h" + #include "common.h" + + static void + help(const char *progname); + + int + main(int argc, char **argv) + { + int c,optindex,opt_index = 0; + + const char *progname; + + const char *pghost = NULL; + const char *pgport = NULL; + const char *pguser = NULL; + const char *pgdbname = NULL; + + const char *keywords[4], *values[4]; + + bool quiet = false; + + PGPing rv; + PQconninfoOption *connect_options, *conn_opt_ptr; + + /* + * We accept user and database as options to avoid + * useless errors from connecting with invalid params + */ + + static struct option long_options[] = { + {"dbname", required_argument, NULL, 'd'}, + {"host", required_argument, NULL, 'h'}, + {"port", required_argument, NULL, 'p'}, + {"quiet", no_argument, NULL, 'q'}, + {"username", required_argument, NULL, 'U'}, + {NULL, 0, NULL, 0} + }; + + progname = get_progname(argv[0]); + handle_help_version_opts(argc, argv, progname, help); + + while ((c = getopt_long(argc, argv, "d:h:p:qU:V", long_options, &optindex)) != -1) + { + switch (c) + { + case 'd': + pgdbname = pg_strdup(optarg); + break; + case 'h': + pghost = pg_strdup(optarg); + break; + case 'p': + pgport = pg_strdup(optarg); + break; + case 'q': + quiet = true; + break; + case 'U': + pguser = pg_strdup(optarg); + break; + default: + /* + * We need to make sure we don't return 1 here because someone + * checking the return code might infer unintended meaning + */ + exit(PQPING_NO_ATTEMPT); + } + } + + if (optind < argc) + { + fprintf(stderr, _("%s: too many command-line arguments (first is \"%s\")\n"), + progname, argv[optind]); + fprintf(stderr, _("Try \"%s --help\" for more information.\n"), progname); + /* + * We need to make sure we don't return 1 here because someone + * checking the return code might infer unintended meaning + */ + exit(PQPING_NO_ATTEMPT); + } + + /* + * Get the default options so we can display them in our output + */ + + connect_options = PQconndefaults(); + conn_opt_ptr = connect_options; + + while (conn_opt_ptr->keyword) + { + if (strncmp(conn_opt_ptr->keyword, "host", 5) == 0) + { + if (pghost) + { + keywords[opt_index] = conn_opt_ptr->keyword; + values[opt_index] = pghost; + opt_index++; + } + else if (conn_opt_ptr->val) + pghost = conn_opt_ptr->val; + else + pghost = DEFAULT_PGSOCKET_DIR; + } + else if (strncmp(conn_opt_ptr->keyword, "port", 5) == 0) + { + if (pgport) + { + keywords[opt_index] = conn_opt_ptr->keyword; + values[opt_index] = pgport; + opt_index++; + } + else if (conn_opt_ptr->val) + pgport = conn_opt_ptr->val; + } + else if (strncmp(conn_opt_ptr->keyword, "user", 5) == 0) + { + if (pguser) + { + keywords[opt_index] = conn_opt_ptr->keyword; + values[opt_index] = pguser; + opt_index++; + } + else if (conn_opt_ptr->val) + pguser = conn_opt_ptr->val; + } + else if (strncmp(conn_opt_ptr->keyword, "dbname", 7) == 0) + { + if (pgdbname) + { + keywords[opt_index] = conn_opt_ptr->keyword; + values[opt_index] = pgdbname; + opt_index++; + } + else if (conn_opt_ptr->val) + pgdbname = conn_opt_ptr->val; + } + conn_opt_ptr++; + } + + keywords[opt_index] = NULL; + values[opt_index] = NULL; + + rv = PQpingParams(keywords, values, 1); + + if (!quiet) + { + printf("%s:%s - ", pghost, pgport); + + switch (rv) + { + case PQPING_OK: + printf("accepting connections\n"); + break; + case PQPING_REJECT: + printf("rejecting connections\n"); + break; + case PQPING_NO_RESPONSE: + printf("no response\n"); + break; + case PQPING_NO_ATTEMPT: + printf("no attempt\n"); + break; + default: + printf("unknown\n"); + } + } + + PQconninfoFree(connect_options); + + exit(rv); + } + + static void + help(const char *progname) + { + printf(_("%s issues a connection check to a PostgreSQL database.\n\n"), progname); + printf(_("Usage:\n")); + printf(_(" %s [OPTION]...\n"), progname); + + printf(_("\nOptions:\n")); + printf(_(" -d, --dbname=DBNAME database name\n")); + printf(_(" -q, --quiet run quietly\n")); + printf(_(" -V, --version output version information, then exit\n")); + printf(_(" -?, --help show this help, then exit\n")); + + printf(_("\nConnection options:\n")); + printf(_(" -h, --host=HOSTNAME database server host or socket directory\n")); + printf(_(" -p, --port=PORT database server port\n")); + printf(_(" -U, --username=USERNAME database username\n")); + } diff --git a/src/interfaces/libpq/libpq-fe.h b/src/interfaces/libpq/libpq-fe.h new file mode 100644 index 895afbc..e0f4bc7 *** a/src/interfaces/libpq/libpq-fe.h --- b/src/interfaces/libpq/libpq-fe.h *************** typedef enum *** 110,115 **** --- 110,120 ---- PQERRORS_VERBOSE /* all the facts, ma'am */ } PGVerbosity; + /* + * PGPing - The ordering of this enum should not be altered because the + * values are exposed externally via pg_isready. + */ + typedef enum { PQPING_OK, /* server is accepting connections */