Support PGXS.
This commit is contained in:
Takahiro Itagaki
parent
e1011e11fe
commit
574b6dc296
@ -3,16 +3,15 @@
|
|||||||
#
|
#
|
||||||
# Copyright (c) 2008-2009, NIPPON TELEGRAPH AND TELEPHONE CORPORATION
|
# Copyright (c) 2008-2009, NIPPON TELEGRAPH AND TELEPHONE CORPORATION
|
||||||
#
|
#
|
||||||
SRCS = \
|
SRCS = pg_reorg.c pgut/pgut.c
|
||||||
pg_reorg.c
|
OBJS = $(SRCS:.c=.o)
|
||||||
OBJS = $(SRCS:.c=.o) $(top_builddir)/src/bin/scripts/common.o
|
|
||||||
PROGRAM = pg_reorg
|
PROGRAM = pg_reorg
|
||||||
REGRESS = init reorg
|
REGRESS = init reorg
|
||||||
|
|
||||||
ifdef DEBUG_REORG
|
ifdef DEBUG_REORG
|
||||||
PG_CPPFLAGS = -I$(libpq_srcdir) -I$(top_builddir)/src/bin/scripts -DDEBUG_REORG
|
PG_CPPFLAGS = -I$(libpq_srcdir) -DDEBUG_REORG
|
||||||
else
|
else
|
||||||
PG_CPPFLAGS = -I$(libpq_srcdir) -I$(top_builddir)/src/bin/scripts
|
PG_CPPFLAGS = -I$(libpq_srcdir)
|
||||||
endif
|
endif
|
||||||
PG_LIBS = $(libpq)
|
PG_LIBS = $(libpq)
|
||||||
|
|
||||||
|
|||||||
601
bin/pg_reorg.c
601
bin/pg_reorg.c
@ -8,36 +8,43 @@
|
|||||||
* @brief Client Modules
|
* @brief Client Modules
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#include "postgres_fe.h"
|
#define PROGRAM_VERSION "1.0.4"
|
||||||
#include "common.h"
|
#define PROGRAM_URL "http://reorg.projects.postgresql.org/"
|
||||||
#include "libpq/pqsignal.h"
|
#define PROGRAM_EMAIL "reorg-general@lists.pgfoundry.org"
|
||||||
|
|
||||||
|
#include "pgut/pgut.h"
|
||||||
|
#include "pqexpbuffer.h"
|
||||||
|
|
||||||
|
#include <string.h>
|
||||||
|
#include <stdlib.h>
|
||||||
#include <unistd.h>
|
#include <unistd.h>
|
||||||
#include <signal.h>
|
|
||||||
|
|
||||||
|
|
||||||
#define REORG_VERSION "1.0.3"
|
|
||||||
#define REORG_URL "http://reorg.projects.postgresql.org/"
|
|
||||||
#define REORG_EMAIL "reorg-general@lists.pgfoundry.org"
|
|
||||||
|
|
||||||
|
#define EXITCODE_HELP 2
|
||||||
#define APPLY_COUNT 1000
|
#define APPLY_COUNT 1000
|
||||||
|
|
||||||
|
#define SQL_XID_SNAPSHOT_80300 \
|
||||||
#if PG_VERSION_NUM >= 80300
|
|
||||||
#define SQL_XID_SNAPSHOT \
|
|
||||||
"SELECT reorg.array_accum(virtualtransaction) FROM pg_locks"\
|
"SELECT reorg.array_accum(virtualtransaction) FROM pg_locks"\
|
||||||
" WHERE locktype = 'virtualxid' AND pid <> pg_backend_pid()"
|
" WHERE locktype = 'virtualxid' AND pid <> pg_backend_pid()"
|
||||||
#define SQL_XID_ALIVE \
|
#define SQL_XID_SNAPSHOT_80200 \
|
||||||
"SELECT 1 FROM pg_locks WHERE locktype = 'virtualxid'"\
|
|
||||||
" AND pid <> pg_backend_pid() AND virtualtransaction = ANY($1) LIMIT 1"
|
|
||||||
#else
|
|
||||||
#define SQL_XID_SNAPSHOT \
|
|
||||||
"SELECT reorg.array_accum(transactionid) FROM pg_locks"\
|
"SELECT reorg.array_accum(transactionid) FROM pg_locks"\
|
||||||
" WHERE locktype = 'transactionid' AND pid <> pg_backend_pid()"
|
" WHERE locktype = 'transactionid' AND pid <> pg_backend_pid()"
|
||||||
#define SQL_XID_ALIVE \
|
|
||||||
|
#define SQL_XID_ALIVE_80300 \
|
||||||
|
"SELECT 1 FROM pg_locks WHERE locktype = 'virtualxid'"\
|
||||||
|
" AND pid <> pg_backend_pid() AND virtualtransaction = ANY($1) LIMIT 1"
|
||||||
|
#define SQL_XID_ALIVE_80200 \
|
||||||
"SELECT 1 FROM pg_locks WHERE locktype = 'transactionid'"\
|
"SELECT 1 FROM pg_locks WHERE locktype = 'transactionid'"\
|
||||||
" AND pid <> pg_backend_pid() AND transactionid = ANY($1) LIMIT 1"
|
" AND pid <> pg_backend_pid() AND transactionid = ANY($1) LIMIT 1"
|
||||||
#endif
|
|
||||||
|
|
||||||
|
#define SQL_XID_SNAPSHOT \
|
||||||
|
(PQserverVersion(current_conn) >= 80300 \
|
||||||
|
? SQL_XID_SNAPSHOT_80300 \
|
||||||
|
: SQL_XID_SNAPSHOT_80200)
|
||||||
|
|
||||||
|
#define SQL_XID_ALIVE \
|
||||||
|
(PQserverVersion(current_conn) >= 80300 \
|
||||||
|
? SQL_XID_ALIVE_80300 \
|
||||||
|
: SQL_XID_ALIVE_80200)
|
||||||
|
|
||||||
/*
|
/*
|
||||||
* per-table information
|
* per-table information
|
||||||
@ -73,45 +80,23 @@ typedef struct reorg_index
|
|||||||
} reorg_index;
|
} reorg_index;
|
||||||
|
|
||||||
static void reorg_all_databases(const char *orderby);
|
static void reorg_all_databases(const char *orderby);
|
||||||
static bool reorg_one_database(const char *orderby, const char *table);
|
static pqbool reorg_one_database(const char *orderby, const char *table);
|
||||||
static void reorg_one_table(const reorg_table *table, const char *orderby);
|
static void reorg_one_table(const reorg_table *table, const char *orderby);
|
||||||
|
|
||||||
static void reconnect(void);
|
|
||||||
static void disconnect(void);
|
|
||||||
static PGresult *execute_nothrow(const char *query, int nParams, const char **params);
|
|
||||||
static PGresult *execute(const char *query, int nParams, const char **params);
|
|
||||||
static void command(const char *query, int nParams, const char **params);
|
|
||||||
static void cleanup(void);
|
|
||||||
static void exit_with_cleanup(int exitcode);
|
|
||||||
|
|
||||||
static void reorg_setup_cancel_handler(void);
|
|
||||||
static void reorg_command_begin(PGconn *conn);
|
|
||||||
static void reorg_command_end(void);
|
|
||||||
|
|
||||||
static void PrintHelp(const char *progname);
|
|
||||||
static void PrintVersion(void);
|
|
||||||
static char *getstr(PGresult *res, int row, int col);
|
static char *getstr(PGresult *res, int row, int col);
|
||||||
static Oid getoid(PGresult *res, int row, int col);
|
static Oid getoid(PGresult *res, int row, int col);
|
||||||
|
|
||||||
#define SQLSTATE_INVALID_SCHEMA_NAME "3F000"
|
#define SQLSTATE_INVALID_SCHEMA_NAME "3F000"
|
||||||
#define SQLSTATE_LOCK_NOT_AVAILABLE "55P03"
|
#define SQLSTATE_LOCK_NOT_AVAILABLE "55P03"
|
||||||
|
|
||||||
static bool sqlstate_equals(PGresult *res, const char *state)
|
static pqbool sqlstate_equals(PGresult *res, const char *state)
|
||||||
{
|
{
|
||||||
return strcmp(PQresultErrorField(res, PG_DIAG_SQLSTATE), state) == 0;
|
return strcmp(PQresultErrorField(res, PG_DIAG_SQLSTATE), state) == 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
static const char *progname = NULL;
|
static pqbool echo = false;
|
||||||
static bool echo = false;
|
static pqbool verbose = false;
|
||||||
static bool verbose = false;
|
static pqbool quiet = false;
|
||||||
static bool quiet = false;
|
|
||||||
|
|
||||||
/* connectin parameters */
|
|
||||||
static const char *dbname = NULL;
|
|
||||||
static char *host = NULL;
|
|
||||||
static char *port = NULL;
|
|
||||||
static char *username = NULL;
|
|
||||||
static bool password = false;
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
* The table begin re-organized. If not null, we need to cleanup temp
|
* The table begin re-organized. If not null, we need to cleanup temp
|
||||||
@ -119,26 +104,6 @@ static bool password = false;
|
|||||||
*/
|
*/
|
||||||
static const reorg_table *current_table = NULL;
|
static const reorg_table *current_table = NULL;
|
||||||
|
|
||||||
/* Current connection initizlied with coneection parameters. */
|
|
||||||
static PGconn *current_conn = NULL;
|
|
||||||
|
|
||||||
/* Interrupted by SIGINT (Ctrl+C) ? */
|
|
||||||
static bool interrupted = false;
|
|
||||||
|
|
||||||
/* Not null during executing some SQL commands. */
|
|
||||||
static PGcancel *volatile cancelConn = NULL;
|
|
||||||
|
|
||||||
|
|
||||||
#ifdef WIN32
|
|
||||||
static CRITICAL_SECTION cancelConnLock;
|
|
||||||
|
|
||||||
static unsigned int sleep(unsigned int seconds)
|
|
||||||
{
|
|
||||||
Sleep(seconds * 1000);
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
/* buffer should have at least 11 bytes */
|
/* buffer should have at least 11 bytes */
|
||||||
static char *
|
static char *
|
||||||
utoa(unsigned int value, char *buffer)
|
utoa(unsigned int value, char *buffer)
|
||||||
@ -147,134 +112,69 @@ utoa(unsigned int value, char *buffer)
|
|||||||
return buffer;
|
return buffer;
|
||||||
}
|
}
|
||||||
|
|
||||||
/* called by atexit */
|
const char *pgut_optstring = "eqvat:no:";
|
||||||
static void
|
|
||||||
warn_if_unclean(void)
|
const struct option pgut_longopts[] = {
|
||||||
|
{"echo", no_argument, NULL, 'e'},
|
||||||
|
{"quiet", no_argument, NULL, 'q'},
|
||||||
|
{"verbose", no_argument, NULL, 'v'},
|
||||||
|
{"all", no_argument, NULL, 'a'},
|
||||||
|
{"table", required_argument, NULL, 't'},
|
||||||
|
{"no-order", no_argument, NULL, 'n'},
|
||||||
|
{"order-by", required_argument, NULL, 'o'},
|
||||||
|
{NULL, 0, NULL, 0}
|
||||||
|
};
|
||||||
|
|
||||||
|
pqbool alldb = false;
|
||||||
|
const char *table = NULL;
|
||||||
|
const char *orderby = NULL;
|
||||||
|
|
||||||
|
pqbool
|
||||||
|
pgut_argument(int c, const char *arg)
|
||||||
{
|
{
|
||||||
if (current_table)
|
switch (c)
|
||||||
fprintf(stderr, _("!!!FATAL ERROR!!! Please refer to a manual.\n\n"));
|
{
|
||||||
|
case 'e':
|
||||||
|
echo = true;
|
||||||
|
break;
|
||||||
|
case 'q':
|
||||||
|
quiet = true;
|
||||||
|
break;
|
||||||
|
case 'v':
|
||||||
|
verbose = true;
|
||||||
|
break;
|
||||||
|
case 'a':
|
||||||
|
alldb = true;
|
||||||
|
break;
|
||||||
|
case 't':
|
||||||
|
table = arg;
|
||||||
|
break;
|
||||||
|
case 'n':
|
||||||
|
orderby = "";
|
||||||
|
break;
|
||||||
|
case 'o':
|
||||||
|
orderby = arg;
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
int
|
int
|
||||||
main(int argc, char *argv[])
|
main(int argc, char *argv[])
|
||||||
{
|
{
|
||||||
static struct option long_options[] = {
|
int exitcode;
|
||||||
{"host", required_argument, NULL, 'h'},
|
|
||||||
{"port", required_argument, NULL, 'p'},
|
|
||||||
{"username", required_argument, NULL, 'U'},
|
|
||||||
{"password", no_argument, NULL, 'W'},
|
|
||||||
{"echo", no_argument, NULL, 'e'},
|
|
||||||
{"quiet", no_argument, NULL, 'q'},
|
|
||||||
{"verbose", no_argument, NULL, 'v'},
|
|
||||||
{"dbname", required_argument, NULL, 'd'},
|
|
||||||
{"all", no_argument, NULL, 'a'},
|
|
||||||
{"table", required_argument, NULL, 't'},
|
|
||||||
{"no-order", no_argument, NULL, 'n'},
|
|
||||||
{"order-by", required_argument, NULL, 'o'},
|
|
||||||
{NULL, 0, NULL, 0}
|
|
||||||
};
|
|
||||||
|
|
||||||
int optindex;
|
exitcode = pgut_getopt(argc, argv);
|
||||||
int c;
|
if (exitcode)
|
||||||
|
return exitcode;
|
||||||
bool alldb = false;
|
|
||||||
const char *table = NULL;
|
|
||||||
const char *orderby = NULL;
|
|
||||||
|
|
||||||
progname = get_progname(argv[0]);
|
|
||||||
set_pglocale_pgservice(argv[0], "pgscripts");
|
|
||||||
|
|
||||||
/*
|
|
||||||
* Help message and version are handled at first.
|
|
||||||
*/
|
|
||||||
if (argc > 1)
|
|
||||||
{
|
|
||||||
if (strcmp(argv[1], "--help") == 0 || strcmp(argv[1], "-?") == 0)
|
|
||||||
{
|
|
||||||
PrintHelp(progname);
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
if (strcmp(argv[1], "--version") == 0 || strcmp(argv[1], "-V") == 0)
|
|
||||||
{
|
|
||||||
PrintVersion();
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
while ((c = getopt_long(argc, argv, "h:p:U:Weqvd:at:no:", long_options, &optindex)) != -1)
|
|
||||||
{
|
|
||||||
switch (c)
|
|
||||||
{
|
|
||||||
case 'h':
|
|
||||||
host = optarg;
|
|
||||||
break;
|
|
||||||
case 'p':
|
|
||||||
port = optarg;
|
|
||||||
break;
|
|
||||||
case 'U':
|
|
||||||
username = optarg;
|
|
||||||
break;
|
|
||||||
case 'W':
|
|
||||||
password = true;
|
|
||||||
break;
|
|
||||||
case 'e':
|
|
||||||
echo = true;
|
|
||||||
break;
|
|
||||||
case 'q':
|
|
||||||
quiet = true;
|
|
||||||
break;
|
|
||||||
case 'v':
|
|
||||||
verbose = true;
|
|
||||||
break;
|
|
||||||
case 'd':
|
|
||||||
dbname = optarg;
|
|
||||||
break;
|
|
||||||
case 'a':
|
|
||||||
alldb = true;
|
|
||||||
break;
|
|
||||||
case 't':
|
|
||||||
table = optarg;
|
|
||||||
break;
|
|
||||||
case 'n':
|
|
||||||
orderby = "";
|
|
||||||
break;
|
|
||||||
case 'o':
|
|
||||||
orderby = optarg;
|
|
||||||
break;
|
|
||||||
default:
|
|
||||||
fprintf(stderr, _("Try \"%s --help\" for more information.\n"), progname);
|
|
||||||
exit(1);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
switch (argc - optind)
|
|
||||||
{
|
|
||||||
case 0:
|
|
||||||
break;
|
|
||||||
case 1:
|
|
||||||
dbname = argv[optind];
|
|
||||||
break;
|
|
||||||
default:
|
|
||||||
fprintf(stderr, _("%s: too many command-line arguments (first is \"%s\")\n"),
|
|
||||||
progname, argv[optind + 1]);
|
|
||||||
fprintf(stderr, _("Try \"%s --help\" for more information.\n"), progname);
|
|
||||||
exit(1);
|
|
||||||
}
|
|
||||||
|
|
||||||
reorg_setup_cancel_handler();
|
|
||||||
atexit(warn_if_unclean);
|
|
||||||
|
|
||||||
if (alldb)
|
if (alldb)
|
||||||
{
|
{
|
||||||
if (dbname)
|
|
||||||
{
|
|
||||||
fprintf(stderr, _("%s: cannot reorg all databases and a specific one at the same time\n"),
|
|
||||||
progname);
|
|
||||||
exit(1);
|
|
||||||
}
|
|
||||||
if (table)
|
if (table)
|
||||||
{
|
{
|
||||||
fprintf(stderr, _("%s: cannot reorg a specific table in all databases\n"),
|
fprintf(stderr, "%s: cannot reorg a specific table in all databases\n",
|
||||||
progname);
|
progname);
|
||||||
exit(1);
|
exit(1);
|
||||||
}
|
}
|
||||||
@ -283,14 +183,9 @@ main(int argc, char *argv[])
|
|||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
(void) (dbname ||
|
|
||||||
(dbname = getenv("PGDATABASE")) ||
|
|
||||||
(dbname = getenv("PGUSER")) ||
|
|
||||||
(dbname = get_user_name(progname)));
|
|
||||||
|
|
||||||
if (!reorg_one_database(orderby, table))
|
if (!reorg_one_database(orderby, table))
|
||||||
{
|
{
|
||||||
fprintf(stderr, _("ERROR: %s is not installed\n"), progname);
|
fprintf(stderr, "ERROR: %s is not installed\n", progname);
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -314,13 +209,13 @@ reorg_all_databases(const char *orderby)
|
|||||||
|
|
||||||
for (i = 0; i < PQntuples(result); i++)
|
for (i = 0; i < PQntuples(result); i++)
|
||||||
{
|
{
|
||||||
bool ret;
|
pqbool ret;
|
||||||
|
|
||||||
dbname = PQgetvalue(result, i, 0);
|
dbname = PQgetvalue(result, i, 0);
|
||||||
|
|
||||||
if (!quiet)
|
if (!quiet)
|
||||||
{
|
{
|
||||||
printf(_("%s: reorg database \"%s\""), progname, dbname);
|
printf("%s: reorg database \"%s\"", progname, dbname);
|
||||||
fflush(stdout);
|
fflush(stdout);
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -331,7 +226,7 @@ reorg_all_databases(const char *orderby)
|
|||||||
if (ret)
|
if (ret)
|
||||||
printf("\n");
|
printf("\n");
|
||||||
else
|
else
|
||||||
printf(_(" ... skipped\n"));
|
printf(" ... skipped\n");
|
||||||
fflush(stdout);
|
fflush(stdout);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -361,10 +256,10 @@ getoid(PGresult *res, int row, int col)
|
|||||||
/*
|
/*
|
||||||
* Call reorg_one_table for the target table or each table in a database.
|
* Call reorg_one_table for the target table or each table in a database.
|
||||||
*/
|
*/
|
||||||
static bool
|
static pqbool
|
||||||
reorg_one_database(const char *orderby, const char *table)
|
reorg_one_database(const char *orderby, const char *table)
|
||||||
{
|
{
|
||||||
bool ret = true;
|
pqbool ret = true;
|
||||||
PGresult *res;
|
PGresult *res;
|
||||||
int i;
|
int i;
|
||||||
int num;
|
int num;
|
||||||
@ -408,7 +303,7 @@ reorg_one_database(const char *orderby, const char *table)
|
|||||||
/* exit otherwise */
|
/* exit otherwise */
|
||||||
printf("%s", PQerrorMessage(current_conn));
|
printf("%s", PQerrorMessage(current_conn));
|
||||||
PQclear(res);
|
PQclear(res);
|
||||||
exit_with_cleanup(1);
|
exit(1);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -430,8 +325,8 @@ reorg_one_database(const char *orderby, const char *table)
|
|||||||
|
|
||||||
if (table.pkid == 0)
|
if (table.pkid == 0)
|
||||||
{
|
{
|
||||||
fprintf(stderr, _("ERROR: relation \"%s\" has no primary key\n"), table.target_name);
|
fprintf(stderr, "ERROR: relation \"%s\" has no primary key\n", table.target_name);
|
||||||
exit_with_cleanup(1);
|
exit(1);
|
||||||
}
|
}
|
||||||
|
|
||||||
table.create_pktype = getstr(res, i, c++);
|
table.create_pktype = getstr(res, i, c++);
|
||||||
@ -449,8 +344,8 @@ reorg_one_database(const char *orderby, const char *table)
|
|||||||
/* CLUSTER mode */
|
/* CLUSTER mode */
|
||||||
if (ckey == NULL)
|
if (ckey == NULL)
|
||||||
{
|
{
|
||||||
fprintf(stderr, _("ERROR: relation \"%s\" has no cluster key\n"), table.target_name);
|
fprintf(stderr, "ERROR: relation \"%s\" has no cluster key\n", table.target_name);
|
||||||
exit_with_cleanup(1);
|
exit(1);
|
||||||
}
|
}
|
||||||
appendPQExpBuffer(&sql, "%s ORDER BY %s", create_table, ckey);
|
appendPQExpBuffer(&sql, "%s ORDER BY %s", create_table, ckey);
|
||||||
table.create_table = sql.data;
|
table.create_table = sql.data;
|
||||||
@ -561,9 +456,9 @@ reorg_one_table(const reorg_table *table, const char *orderby)
|
|||||||
1, params);
|
1, params);
|
||||||
if (PQntuples(res) > 0)
|
if (PQntuples(res) > 0)
|
||||||
{
|
{
|
||||||
fprintf(stderr, _("%s: trigger conflicted for %s\n"),
|
fprintf(stderr, "%s: trigger conflicted for %s\n",
|
||||||
progname, table->target_name);
|
progname, table->target_name);
|
||||||
exit_with_cleanup(1);
|
exit(1);
|
||||||
}
|
}
|
||||||
|
|
||||||
command(table->create_pktype, 0, NULL);
|
command(table->create_pktype, 0, NULL);
|
||||||
@ -685,7 +580,7 @@ reorg_one_table(const reorg_table *table, const char *orderby)
|
|||||||
/* exit otherwise */
|
/* exit otherwise */
|
||||||
printf("%s", PQerrorMessage(current_conn));
|
printf("%s", PQerrorMessage(current_conn));
|
||||||
PQclear(res);
|
PQclear(res);
|
||||||
exit_with_cleanup(1);
|
exit(1);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -710,279 +605,73 @@ reorg_one_table(const reorg_table *table, const char *orderby)
|
|||||||
free(vxid);
|
free(vxid);
|
||||||
}
|
}
|
||||||
|
|
||||||
static void
|
void
|
||||||
cleanup(void)
|
pgut_cleanup(pqbool fatal)
|
||||||
{
|
{
|
||||||
char buffer[12];
|
if (fatal)
|
||||||
const char *params[1];
|
|
||||||
|
|
||||||
if (!current_table)
|
|
||||||
return;
|
|
||||||
|
|
||||||
params[0] = utoa(current_table->target_oid, buffer);
|
|
||||||
execute("SELECT reorg.reorg_drop($1)", 1, params);
|
|
||||||
current_table = NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
static void
|
|
||||||
reconnect(void)
|
|
||||||
{
|
|
||||||
disconnect();
|
|
||||||
current_conn = connectDatabase(dbname, host, port, username, password, progname);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void
|
|
||||||
disconnect(void)
|
|
||||||
{
|
|
||||||
if (current_conn)
|
|
||||||
{
|
{
|
||||||
PQfinish(current_conn);
|
if (current_table)
|
||||||
current_conn = NULL;
|
fprintf(stderr, "!!!FATAL ERROR!!! Please refer to a manual.\n\n");
|
||||||
}
|
}
|
||||||
}
|
else
|
||||||
|
|
||||||
static void
|
|
||||||
exit_with_cleanup(int exitcode)
|
|
||||||
{
|
|
||||||
if (current_table)
|
|
||||||
{
|
{
|
||||||
|
char buffer[12];
|
||||||
|
const char *params[1];
|
||||||
|
|
||||||
|
if (current_table == NULL)
|
||||||
|
return; /* no needs to cleanup */
|
||||||
|
|
||||||
/* Rollback current transaction */
|
/* Rollback current transaction */
|
||||||
if (current_conn)
|
if (current_conn)
|
||||||
{
|
command("ROLLBACK", 0, NULL);
|
||||||
PGresult *res;
|
|
||||||
res = PQexec(current_conn, "ROLLBACK");
|
|
||||||
if (PQresultStatus(res) != PGRES_COMMAND_OK)
|
|
||||||
exit(1); /* fatal error */
|
|
||||||
PQclear(res);
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Try reconnection if not available. */
|
/* Try reconnection if not available. */
|
||||||
if (PQstatus(current_conn) != CONNECTION_OK)
|
if (PQstatus(current_conn) != CONNECTION_OK)
|
||||||
reconnect();
|
reconnect();
|
||||||
|
|
||||||
cleanup();
|
/* do cleanup */
|
||||||
|
params[0] = utoa(current_table->target_oid, buffer);
|
||||||
|
command("SELECT reorg.reorg_drop($1)", 1, params);
|
||||||
|
current_table = NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
disconnect();
|
|
||||||
exit(exitcode);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
static PGresult *
|
int
|
||||||
execute_nothrow(const char *query, int nParams, const char **params)
|
pgut_help(void)
|
||||||
{
|
{
|
||||||
PGresult *res;
|
fprintf(stderr,
|
||||||
|
"%s re-organizes a PostgreSQL database.\n\n"
|
||||||
if (echo)
|
"Usage:\n"
|
||||||
fprintf(stderr, _("%s: executing %s\n"), progname, query);
|
" %s [OPTION]... [DBNAME]\n"
|
||||||
|
"\nOptions:\n"
|
||||||
#ifdef DEBUG_REORG
|
" -a, --all reorg all databases\n"
|
||||||
fprintf(stderr, "debug: suspend in execute. (sql='%s')\npush enter key: ", query);
|
" -d, --dbname=DBNAME database to reorg\n"
|
||||||
fgetc(stdin);
|
" -t, --table=TABLE reorg specific table only\n"
|
||||||
|
" -n, --no-order do vacuum full instead of cluster\n"
|
||||||
|
" -o, --order-by=columns order by columns instead of cluster keys\n"
|
||||||
|
" -e, --echo show the commands being sent to the server\n"
|
||||||
|
" -q, --quiet don't write any messages\n"
|
||||||
|
" -v, --verbose display detailed information during processing\n"
|
||||||
|
" --help show this help, then exit\n"
|
||||||
|
" --version output version information, then exit\n"
|
||||||
|
"\nConnection options:\n"
|
||||||
|
" -h, --host=HOSTNAME database server host or socket directory\n"
|
||||||
|
" -p, --port=PORT database server port\n"
|
||||||
|
" -U, --username=USERNAME user name to connect as\n"
|
||||||
|
" -W, --password force password prompt\n",
|
||||||
|
progname, progname);
|
||||||
|
#ifdef PROGRAM_URL
|
||||||
|
fprintf(stderr,"\nRead the website for details. <" PROGRAM_URL ">\n");
|
||||||
#endif
|
#endif
|
||||||
|
#ifdef PROGRAM_EMAIL
|
||||||
reorg_command_begin(current_conn);
|
fprintf(stderr,"\nReport bugs to <" PROGRAM_EMAIL ">.\n");
|
||||||
if (nParams == 0)
|
|
||||||
res = PQexec(current_conn, query);
|
|
||||||
else
|
|
||||||
res = PQexecParams(current_conn, query, nParams, NULL, params, NULL, NULL, 0);
|
|
||||||
reorg_command_end();
|
|
||||||
|
|
||||||
return res;
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
|
||||||
* execute - Execute a SQL and return the result, or exit() if failed.
|
|
||||||
*/
|
|
||||||
static PGresult *
|
|
||||||
execute(const char *query, int nParams, const char **params)
|
|
||||||
{
|
|
||||||
if (interrupted)
|
|
||||||
{
|
|
||||||
interrupted = false;
|
|
||||||
fprintf(stderr, _("%s: interrupted\n"), progname);
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
PGresult *res = execute_nothrow(query, nParams, params);
|
|
||||||
|
|
||||||
if (PQresultStatus(res) == PGRES_TUPLES_OK ||
|
|
||||||
PQresultStatus(res) == PGRES_COMMAND_OK)
|
|
||||||
return res;
|
|
||||||
|
|
||||||
fprintf(stderr, _("%s: query failed: %s"),
|
|
||||||
progname, PQerrorMessage(current_conn));
|
|
||||||
fprintf(stderr, _("%s: query was: %s\n"),
|
|
||||||
progname, query);
|
|
||||||
PQclear(res);
|
|
||||||
}
|
|
||||||
|
|
||||||
exit_with_cleanup(1);
|
|
||||||
return NULL; /* keep compiler quiet */
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
|
||||||
* command - Execute a SQL and discard the result, or exit() if failed.
|
|
||||||
*/
|
|
||||||
static void
|
|
||||||
command(const char *query, int nParams, const char **params)
|
|
||||||
{
|
|
||||||
PGresult *res = execute(query, nParams, params);
|
|
||||||
PQclear(res);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void
|
|
||||||
PrintHelp(const char *progname)
|
|
||||||
{
|
|
||||||
printf(_("%s re-organizes a PostgreSQL database.\n\n"), progname);
|
|
||||||
printf(_("Usage:\n"));
|
|
||||||
printf(_(" %s [OPTION]... [DBNAME]\n"), progname);
|
|
||||||
printf(_("\nOptions:\n"));
|
|
||||||
printf(_(" -a, --all reorg all databases\n"));
|
|
||||||
printf(_(" -d, --dbname=DBNAME database to reorg\n"));
|
|
||||||
printf(_(" -t, --table=TABLE reorg specific table only\n"));
|
|
||||||
printf(_(" -n, --no-order do vacuum full instead of cluster\n"));
|
|
||||||
printf(_(" -o, --order-by=columns order by columns instead of cluster keys\n"));
|
|
||||||
printf(_(" -e, --echo show the commands being sent to the server\n"));
|
|
||||||
printf(_(" -q, --quiet don't write any messages\n"));
|
|
||||||
printf(_(" -v, --verbose display detailed information during processing\n"));
|
|
||||||
printf(_(" --help show this help, then exit\n"));
|
|
||||||
printf(_(" --version output version information, 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 user name to connect as\n"));
|
|
||||||
printf(_(" -W, --password force password prompt\n"));
|
|
||||||
#ifdef REORG_URL
|
|
||||||
printf(_("\nRead the website for details. <" REORG_URL ">\n"));
|
|
||||||
#endif
|
|
||||||
#ifdef REORG_EMAIL
|
|
||||||
printf(_("\nReport bugs to <" REORG_EMAIL ">.\n"));
|
|
||||||
#endif
|
#endif
|
||||||
|
return EXITCODE_HELP;
|
||||||
}
|
}
|
||||||
|
|
||||||
static void
|
int
|
||||||
PrintVersion(void)
|
pgut_version(void)
|
||||||
{
|
{
|
||||||
fprintf(stderr, "pg_reorg " REORG_VERSION "\n");
|
fprintf(stderr, "%s %s\n", progname, PROGRAM_VERSION);
|
||||||
|
return EXITCODE_HELP;
|
||||||
}
|
}
|
||||||
|
|
||||||
/*
|
|
||||||
* reorg_command_begin
|
|
||||||
*
|
|
||||||
* Set cancelConn to point to the current database connection.
|
|
||||||
*/
|
|
||||||
static void
|
|
||||||
reorg_command_begin(PGconn *conn)
|
|
||||||
{
|
|
||||||
PGcancel *oldCancelConn;
|
|
||||||
|
|
||||||
#ifdef WIN32
|
|
||||||
EnterCriticalSection(&cancelConnLock);
|
|
||||||
#endif
|
|
||||||
|
|
||||||
/* Free the old one if we have one */
|
|
||||||
oldCancelConn = cancelConn;
|
|
||||||
|
|
||||||
/* be sure handle_sigint doesn't use pointer while freeing */
|
|
||||||
cancelConn = NULL;
|
|
||||||
|
|
||||||
if (oldCancelConn != NULL)
|
|
||||||
PQfreeCancel(oldCancelConn);
|
|
||||||
|
|
||||||
cancelConn = PQgetCancel(conn);
|
|
||||||
|
|
||||||
#ifdef WIN32
|
|
||||||
LeaveCriticalSection(&cancelConnLock);
|
|
||||||
#endif
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
|
||||||
* reorg_command_end
|
|
||||||
*
|
|
||||||
* Free the current cancel connection, if any, and set to NULL.
|
|
||||||
*/
|
|
||||||
static void
|
|
||||||
reorg_command_end(void)
|
|
||||||
{
|
|
||||||
PGcancel *oldCancelConn;
|
|
||||||
|
|
||||||
#ifdef WIN32
|
|
||||||
EnterCriticalSection(&cancelConnLock);
|
|
||||||
#endif
|
|
||||||
|
|
||||||
oldCancelConn = cancelConn;
|
|
||||||
|
|
||||||
/* be sure handle_sigint doesn't use pointer while freeing */
|
|
||||||
cancelConn = NULL;
|
|
||||||
|
|
||||||
if (oldCancelConn != NULL)
|
|
||||||
PQfreeCancel(oldCancelConn);
|
|
||||||
|
|
||||||
#ifdef WIN32
|
|
||||||
LeaveCriticalSection(&cancelConnLock);
|
|
||||||
#endif
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
|
||||||
* Handle interrupt signals by cancelling the current command.
|
|
||||||
*/
|
|
||||||
static void
|
|
||||||
reorg_cancel(void)
|
|
||||||
{
|
|
||||||
int save_errno = errno;
|
|
||||||
char errbuf[256];
|
|
||||||
|
|
||||||
/* Set interruped flag */
|
|
||||||
interrupted = true;
|
|
||||||
|
|
||||||
/* Send QueryCancel if we are processing a database query */
|
|
||||||
if (cancelConn != NULL && PQcancel(cancelConn, errbuf, sizeof(errbuf)))
|
|
||||||
fprintf(stderr, _("Cancel request sent\n"));
|
|
||||||
|
|
||||||
errno = save_errno; /* just in case the write changed it */
|
|
||||||
}
|
|
||||||
|
|
||||||
#ifndef WIN32
|
|
||||||
static void
|
|
||||||
handle_sigint(SIGNAL_ARGS)
|
|
||||||
{
|
|
||||||
reorg_cancel();
|
|
||||||
}
|
|
||||||
|
|
||||||
static void
|
|
||||||
reorg_setup_cancel_handler(void)
|
|
||||||
{
|
|
||||||
pqsignal(SIGINT, handle_sigint);
|
|
||||||
}
|
|
||||||
#else /* WIN32 */
|
|
||||||
|
|
||||||
/*
|
|
||||||
* Console control handler for Win32. Note that the control handler will
|
|
||||||
* execute on a *different thread* than the main one, so we need to do
|
|
||||||
* proper locking around those structures.
|
|
||||||
*/
|
|
||||||
static BOOL WINAPI
|
|
||||||
consoleHandler(DWORD dwCtrlType)
|
|
||||||
{
|
|
||||||
if (dwCtrlType == CTRL_C_EVENT ||
|
|
||||||
dwCtrlType == CTRL_BREAK_EVENT)
|
|
||||||
{
|
|
||||||
EnterCriticalSection(&cancelConnLock);
|
|
||||||
reorg_cancel();
|
|
||||||
LeaveCriticalSection(&cancelConnLock);
|
|
||||||
return TRUE;
|
|
||||||
}
|
|
||||||
else
|
|
||||||
/* Return FALSE for any signals not being handled */
|
|
||||||
return FALSE;
|
|
||||||
}
|
|
||||||
|
|
||||||
static void
|
|
||||||
reorg_setup_cancel_handler(void)
|
|
||||||
{
|
|
||||||
InitializeCriticalSection(&cancelConnLock);
|
|
||||||
|
|
||||||
SetConsoleCtrlHandler(consoleHandler, TRUE);
|
|
||||||
}
|
|
||||||
|
|
||||||
#endif /* WIN32 */
|
|
||||||
|
|||||||
467
bin/pgut/pgut.c
Executable file
467
bin/pgut/pgut.c
Executable file
@ -0,0 +1,467 @@
|
|||||||
|
/*
|
||||||
|
* pgut.c
|
||||||
|
*
|
||||||
|
* Copyright (c) 2009, NIPPON TELEGRAPH AND TELEPHONE CORPORATION
|
||||||
|
*/
|
||||||
|
#include "pgut.h"
|
||||||
|
|
||||||
|
#include "postgres_fe.h"
|
||||||
|
#include "libpq/pqsignal.h"
|
||||||
|
|
||||||
|
#include <unistd.h>
|
||||||
|
|
||||||
|
const char *progname = NULL;
|
||||||
|
const char *dbname = NULL;
|
||||||
|
char *host = NULL;
|
||||||
|
char *port = NULL;
|
||||||
|
char *username = NULL;
|
||||||
|
pqbool password = false;
|
||||||
|
|
||||||
|
/* Interrupted by SIGINT (Ctrl+C) ? */
|
||||||
|
pqbool interrupted = false;
|
||||||
|
|
||||||
|
/* Database connections */
|
||||||
|
PGconn *current_conn = NULL;
|
||||||
|
static PGcancel *volatile cancel_conn = NULL;
|
||||||
|
|
||||||
|
/* Connection routines */
|
||||||
|
static void init_cancel_handler(void);
|
||||||
|
static void on_before_exec(PGconn *conn);
|
||||||
|
static void on_after_exec(void);
|
||||||
|
static void on_interrupt(void);
|
||||||
|
static void on_exit(void);
|
||||||
|
static void exit_or_abort(int exitcode);
|
||||||
|
const char *get_user_name(const char *progname);
|
||||||
|
|
||||||
|
const char default_optstring[] = "d:h:p:U:W";
|
||||||
|
|
||||||
|
const struct option default_longopts[] =
|
||||||
|
{
|
||||||
|
{"dbname", required_argument, NULL, 'd'},
|
||||||
|
{"host", required_argument, NULL, 'h'},
|
||||||
|
{"port", required_argument, NULL, 'p'},
|
||||||
|
{"username", required_argument, NULL, 'U'},
|
||||||
|
{"password", no_argument, NULL, 'W'},
|
||||||
|
{NULL, 0, NULL, 0}
|
||||||
|
};
|
||||||
|
|
||||||
|
static const char *
|
||||||
|
merge_optstring(const char *opts)
|
||||||
|
{
|
||||||
|
size_t len;
|
||||||
|
char *result;
|
||||||
|
|
||||||
|
if (opts == NULL)
|
||||||
|
return default_optstring;
|
||||||
|
|
||||||
|
len = strlen(opts);
|
||||||
|
if (len == 0)
|
||||||
|
return default_optstring;
|
||||||
|
|
||||||
|
result = malloc(len + lengthof(default_optstring));
|
||||||
|
memcpy(&result[0], opts, len);
|
||||||
|
memcpy(&result[len], default_optstring, lengthof(default_optstring));
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
static const struct option *
|
||||||
|
merge_longopts(const struct option *opts)
|
||||||
|
{
|
||||||
|
size_t len;
|
||||||
|
struct option *result;
|
||||||
|
|
||||||
|
if (opts == NULL)
|
||||||
|
return default_longopts;
|
||||||
|
|
||||||
|
for (len = 0; opts[len].name; len++) { }
|
||||||
|
if (len == 0)
|
||||||
|
return default_longopts;
|
||||||
|
|
||||||
|
result = (struct option *) malloc((len + lengthof(default_longopts)) * sizeof(struct option));
|
||||||
|
memcpy(&result[0], opts, len * sizeof(struct option));
|
||||||
|
memcpy(&result[len], default_longopts, lengthof(default_longopts) * sizeof(struct option));
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
int
|
||||||
|
pgut_getopt(int argc, char **argv)
|
||||||
|
{
|
||||||
|
const char *optstring;
|
||||||
|
const struct option *longopts;
|
||||||
|
|
||||||
|
int c;
|
||||||
|
int optindex = 0;
|
||||||
|
|
||||||
|
progname = get_progname(argv[0]);
|
||||||
|
set_pglocale_pgservice(argv[0], "pgscripts");
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Help message and version are handled at first.
|
||||||
|
*/
|
||||||
|
if (argc > 1)
|
||||||
|
{
|
||||||
|
if (strcmp(argv[1], "--help") == 0 || strcmp(argv[1], "-?") == 0)
|
||||||
|
return pgut_help();
|
||||||
|
if (strcmp(argv[1], "--version") == 0 || strcmp(argv[1], "-V") == 0)
|
||||||
|
return pgut_version();
|
||||||
|
}
|
||||||
|
|
||||||
|
optstring = merge_optstring(pgut_optstring);
|
||||||
|
longopts = merge_longopts(pgut_longopts);
|
||||||
|
|
||||||
|
while ((c = getopt_long(argc, argv, optstring, longopts, &optindex)) != -1)
|
||||||
|
{
|
||||||
|
switch (c)
|
||||||
|
{
|
||||||
|
case 'h':
|
||||||
|
host = optarg;
|
||||||
|
break;
|
||||||
|
case 'p':
|
||||||
|
port = optarg;
|
||||||
|
break;
|
||||||
|
case 'U':
|
||||||
|
username = optarg;
|
||||||
|
break;
|
||||||
|
case 'W':
|
||||||
|
password = true;
|
||||||
|
break;
|
||||||
|
case 'd':
|
||||||
|
dbname = optarg;
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
if (!pgut_argument(c, optarg))
|
||||||
|
{
|
||||||
|
fprintf(stderr, _("Try \"%s --help\" for more information.\n"), progname);
|
||||||
|
exit_or_abort(EXITCODE_ERROR);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for (; optind < argc; optind++)
|
||||||
|
{
|
||||||
|
if (!pgut_argument(0, argv[optind]))
|
||||||
|
{
|
||||||
|
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);
|
||||||
|
exit_or_abort(EXITCODE_ERROR);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
init_cancel_handler();
|
||||||
|
atexit(on_exit);
|
||||||
|
|
||||||
|
(void) (dbname ||
|
||||||
|
(dbname = getenv("PGDATABASE")) ||
|
||||||
|
(dbname = getenv("PGUSER")) ||
|
||||||
|
(dbname = get_user_name(progname)));
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
void
|
||||||
|
reconnect(void)
|
||||||
|
{
|
||||||
|
PGconn *conn;
|
||||||
|
char *pwd = NULL;
|
||||||
|
bool new_pass;
|
||||||
|
|
||||||
|
disconnect();
|
||||||
|
|
||||||
|
if (password)
|
||||||
|
pwd = simple_prompt("Password: ", 100, false);
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Start the connection. Loop until we have a password if requested by
|
||||||
|
* backend.
|
||||||
|
*/
|
||||||
|
do
|
||||||
|
{
|
||||||
|
new_pass = false;
|
||||||
|
conn = PQsetdbLogin(host, port, NULL, NULL, dbname, username, pwd);
|
||||||
|
|
||||||
|
if (!conn)
|
||||||
|
{
|
||||||
|
fprintf(stderr, _("%s: could not connect to database %s\n"),
|
||||||
|
progname, dbname);
|
||||||
|
exit_or_abort(EXITCODE_ERROR);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (PQstatus(conn) == CONNECTION_BAD &&
|
||||||
|
PQconnectionNeedsPassword(conn) &&
|
||||||
|
pwd == NULL)
|
||||||
|
{
|
||||||
|
PQfinish(conn);
|
||||||
|
pwd = simple_prompt("Password: ", 100, false);
|
||||||
|
new_pass = true;
|
||||||
|
}
|
||||||
|
} while (new_pass);
|
||||||
|
|
||||||
|
free(pwd);
|
||||||
|
|
||||||
|
/* check to see that the backend connection was successfully made */
|
||||||
|
if (PQstatus(conn) == CONNECTION_BAD)
|
||||||
|
{
|
||||||
|
fprintf(stderr, _("%s: could not connect to database %s: %s"),
|
||||||
|
progname, dbname, PQerrorMessage(conn));
|
||||||
|
exit_or_abort(EXITCODE_ERROR);
|
||||||
|
}
|
||||||
|
|
||||||
|
current_conn = conn;
|
||||||
|
}
|
||||||
|
|
||||||
|
void
|
||||||
|
disconnect(void)
|
||||||
|
{
|
||||||
|
if (current_conn)
|
||||||
|
{
|
||||||
|
PQfinish(current_conn);
|
||||||
|
current_conn = NULL;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
PGresult *
|
||||||
|
execute_nothrow(const char *query, int nParams, const char **params)
|
||||||
|
{
|
||||||
|
PGresult *res;
|
||||||
|
|
||||||
|
on_before_exec(current_conn);
|
||||||
|
if (nParams == 0)
|
||||||
|
res = PQexec(current_conn, query);
|
||||||
|
else
|
||||||
|
res = PQexecParams(current_conn, query, nParams, NULL, params, NULL, NULL, 0);
|
||||||
|
on_after_exec();
|
||||||
|
|
||||||
|
return res;
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* execute - Execute a SQL and return the result, or exit_or_abort() if failed.
|
||||||
|
*/
|
||||||
|
PGresult *
|
||||||
|
execute(const char *query, int nParams, const char **params)
|
||||||
|
{
|
||||||
|
if (interrupted)
|
||||||
|
{
|
||||||
|
interrupted = false;
|
||||||
|
fprintf(stderr, _("%s: interrupted\n"), progname);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
PGresult *res = execute_nothrow(query, nParams, params);
|
||||||
|
|
||||||
|
if (PQresultStatus(res) == PGRES_TUPLES_OK ||
|
||||||
|
PQresultStatus(res) == PGRES_COMMAND_OK)
|
||||||
|
return res;
|
||||||
|
|
||||||
|
fprintf(stderr, _("%s: query failed: %s"),
|
||||||
|
progname, PQerrorMessage(current_conn));
|
||||||
|
fprintf(stderr, _("%s: query was: %s\n"),
|
||||||
|
progname, query);
|
||||||
|
PQclear(res);
|
||||||
|
}
|
||||||
|
|
||||||
|
exit_or_abort(EXITCODE_ERROR);
|
||||||
|
return NULL; /* keep compiler quiet */
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* command - Execute a SQL and discard the result, or exit_or_abort() if failed.
|
||||||
|
*/
|
||||||
|
void
|
||||||
|
command(const char *query, int nParams, const char **params)
|
||||||
|
{
|
||||||
|
PGresult *res = execute(query, nParams, params);
|
||||||
|
PQclear(res);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
#ifdef WIN32
|
||||||
|
static CRITICAL_SECTION cancelConnLock;
|
||||||
|
#endif
|
||||||
|
|
||||||
|
/*
|
||||||
|
* on_before_exec
|
||||||
|
*
|
||||||
|
* Set cancel_conn to point to the current database connection.
|
||||||
|
*/
|
||||||
|
static void
|
||||||
|
on_before_exec(PGconn *conn)
|
||||||
|
{
|
||||||
|
PGcancel *old;
|
||||||
|
|
||||||
|
#ifdef WIN32
|
||||||
|
EnterCriticalSection(&cancelConnLock);
|
||||||
|
#endif
|
||||||
|
|
||||||
|
/* Free the old one if we have one */
|
||||||
|
old = cancel_conn;
|
||||||
|
|
||||||
|
/* be sure handle_sigint doesn't use pointer while freeing */
|
||||||
|
cancel_conn = NULL;
|
||||||
|
|
||||||
|
if (old != NULL)
|
||||||
|
PQfreeCancel(old);
|
||||||
|
|
||||||
|
cancel_conn = PQgetCancel(conn);
|
||||||
|
|
||||||
|
#ifdef WIN32
|
||||||
|
LeaveCriticalSection(&cancelConnLock);
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* on_after_exec
|
||||||
|
*
|
||||||
|
* Free the current cancel connection, if any, and set to NULL.
|
||||||
|
*/
|
||||||
|
static void
|
||||||
|
on_after_exec(void)
|
||||||
|
{
|
||||||
|
PGcancel *old;
|
||||||
|
|
||||||
|
#ifdef WIN32
|
||||||
|
EnterCriticalSection(&cancelConnLock);
|
||||||
|
#endif
|
||||||
|
|
||||||
|
old = cancel_conn;
|
||||||
|
|
||||||
|
/* be sure handle_sigint doesn't use pointer while freeing */
|
||||||
|
cancel_conn = NULL;
|
||||||
|
|
||||||
|
if (old != NULL)
|
||||||
|
PQfreeCancel(old);
|
||||||
|
|
||||||
|
#ifdef WIN32
|
||||||
|
LeaveCriticalSection(&cancelConnLock);
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Handle interrupt signals by cancelling the current command.
|
||||||
|
*/
|
||||||
|
static void
|
||||||
|
on_interrupt(void)
|
||||||
|
{
|
||||||
|
int save_errno = errno;
|
||||||
|
char errbuf[256];
|
||||||
|
|
||||||
|
/* Set interruped flag */
|
||||||
|
interrupted = true;
|
||||||
|
|
||||||
|
/* Send QueryCancel if we are processing a database query */
|
||||||
|
if (cancel_conn != NULL && PQcancel(cancel_conn, errbuf, sizeof(errbuf)))
|
||||||
|
fprintf(stderr, _("Cancel request sent\n"));
|
||||||
|
|
||||||
|
errno = save_errno; /* just in case the write changed it */
|
||||||
|
}
|
||||||
|
|
||||||
|
static pqbool in_cleanup = false;
|
||||||
|
|
||||||
|
static void
|
||||||
|
on_exit(void)
|
||||||
|
{
|
||||||
|
in_cleanup = true;
|
||||||
|
pgut_cleanup(false);
|
||||||
|
disconnect();
|
||||||
|
}
|
||||||
|
|
||||||
|
static void
|
||||||
|
exit_or_abort(int exitcode)
|
||||||
|
{
|
||||||
|
if (in_cleanup)
|
||||||
|
{
|
||||||
|
/* oops, error in cleanup*/
|
||||||
|
pgut_cleanup(true);
|
||||||
|
abort();
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
/* normal exit */
|
||||||
|
exit(exitcode);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Returns the current user name.
|
||||||
|
*/
|
||||||
|
const char *
|
||||||
|
get_user_name(const char *progname)
|
||||||
|
{
|
||||||
|
#ifndef WIN32
|
||||||
|
struct passwd *pw;
|
||||||
|
|
||||||
|
pw = getpwuid(geteuid());
|
||||||
|
if (!pw)
|
||||||
|
{
|
||||||
|
fprintf(stderr, _("%s: could not obtain information about current user: %s\n"),
|
||||||
|
progname, strerror(errno));
|
||||||
|
exit_or_abort(EXITCODE_ERROR);
|
||||||
|
}
|
||||||
|
return pw->pw_name;
|
||||||
|
#else
|
||||||
|
static char username[128]; /* remains after function exit */
|
||||||
|
DWORD len = sizeof(username) - 1;
|
||||||
|
|
||||||
|
if (!GetUserName(username, &len))
|
||||||
|
{
|
||||||
|
fprintf(stderr, _("%s: could not get current user name: %s\n"),
|
||||||
|
progname, strerror(errno));
|
||||||
|
exit_or_abort(EXITCODE_ERROR);
|
||||||
|
}
|
||||||
|
return username;
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
#ifndef WIN32
|
||||||
|
static void
|
||||||
|
handle_sigint(SIGNAL_ARGS)
|
||||||
|
{
|
||||||
|
on_interrupt();
|
||||||
|
}
|
||||||
|
|
||||||
|
static void
|
||||||
|
init_cancel_handler(void)
|
||||||
|
{
|
||||||
|
pqsignal(SIGINT, handle_sigint);
|
||||||
|
}
|
||||||
|
#else /* WIN32 */
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Console control handler for Win32. Note that the control handler will
|
||||||
|
* execute on a *different thread* than the main one, so we need to do
|
||||||
|
* proper locking around those structures.
|
||||||
|
*/
|
||||||
|
static BOOL WINAPI
|
||||||
|
consoleHandler(DWORD dwCtrlType)
|
||||||
|
{
|
||||||
|
if (dwCtrlType == CTRL_C_EVENT ||
|
||||||
|
dwCtrlType == CTRL_BREAK_EVENT)
|
||||||
|
{
|
||||||
|
EnterCriticalSection(&cancelConnLock);
|
||||||
|
on_interrupt();
|
||||||
|
LeaveCriticalSection(&cancelConnLock);
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
/* Return FALSE for any signals not being handled */
|
||||||
|
return FALSE;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void
|
||||||
|
init_cancel_handler(void)
|
||||||
|
{
|
||||||
|
InitializeCriticalSection(&cancelConnLock);
|
||||||
|
|
||||||
|
SetConsoleCtrlHandler(consoleHandler, TRUE);
|
||||||
|
}
|
||||||
|
|
||||||
|
unsigned int
|
||||||
|
sleep(unsigned int seconds)
|
||||||
|
{
|
||||||
|
Sleep(seconds * 1000);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
#endif /* WIN32 */
|
||||||
65
bin/pgut/pgut.h
Executable file
65
bin/pgut/pgut.h
Executable file
@ -0,0 +1,65 @@
|
|||||||
|
/*
|
||||||
|
* pgut.h
|
||||||
|
*
|
||||||
|
* Copyright (c) 2009, NIPPON TELEGRAPH AND TELEPHONE CORPORATION
|
||||||
|
*/
|
||||||
|
|
||||||
|
#ifndef PGUT_H
|
||||||
|
#define PGUT_H
|
||||||
|
|
||||||
|
#include "libpq-fe.h"
|
||||||
|
#include <getopt.h>
|
||||||
|
|
||||||
|
/*
|
||||||
|
* pgut client variables and functions
|
||||||
|
*/
|
||||||
|
extern const char *pgut_optstring;
|
||||||
|
extern const struct option pgut_longopts[];
|
||||||
|
|
||||||
|
extern pqbool pgut_argument(int c, const char *arg);
|
||||||
|
extern int pgut_help(void);
|
||||||
|
extern int pgut_version(void);
|
||||||
|
extern void pgut_cleanup(pqbool fatal);
|
||||||
|
|
||||||
|
/*
|
||||||
|
* exit codes
|
||||||
|
*/
|
||||||
|
|
||||||
|
#define EXITCODE_OK 0 /**< normal exit */
|
||||||
|
#define EXITCODE_ERROR 1 /**< normal error */
|
||||||
|
#define EXITCODE_HELP 2 /**< help and version mode */
|
||||||
|
#define EXITCODE_FATAL 3 /**< fatal error */
|
||||||
|
|
||||||
|
/*
|
||||||
|
* pgut framework variables and functions
|
||||||
|
*/
|
||||||
|
|
||||||
|
#ifndef true
|
||||||
|
#define true 1
|
||||||
|
#endif
|
||||||
|
#ifndef false
|
||||||
|
#define false 0
|
||||||
|
#endif
|
||||||
|
|
||||||
|
extern const char *progname;
|
||||||
|
extern const char *dbname;
|
||||||
|
extern char *host;
|
||||||
|
extern char *port;
|
||||||
|
extern char *username;
|
||||||
|
extern pqbool password;
|
||||||
|
extern pqbool interrupted;
|
||||||
|
extern PGconn *current_conn;
|
||||||
|
|
||||||
|
extern int pgut_getopt(int argc, char **argv);
|
||||||
|
|
||||||
|
extern void reconnect(void);
|
||||||
|
extern void disconnect(void);
|
||||||
|
extern PGresult *execute_nothrow(const char *query, int nParams, const char **params);
|
||||||
|
extern PGresult *execute(const char *query, int nParams, const char **params);
|
||||||
|
extern void command(const char *query, int nParams, const char **params);
|
||||||
|
|
||||||
|
#ifdef WIN32
|
||||||
|
extern unsigned int sleep(unsigned int seconds);
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#endif /* PGUT_H */
|
||||||
Loading…
x
Reference in New Issue
Block a user