src/bin/scripts/clusterdb.c

   1 /*-------------------------------------------------------------------------
   2  *
   3  * clusterdb
   4  *
   5  * Portions Copyright (c) 2002-2009, PostgreSQL Global Development Group
   6  *
   7  * $PostgreSQL$
   8  *
   9  *-------------------------------------------------------------------------
  10  */
  11
  12 #include "postgres_fe.h"
  13 #include "common.h"
  14 #include "dumputils.h"
  15
  16
  17 static void cluster_one_database(const char *dbname, bool verbose, const char *table,
  18                                          const char *host, const char *port,
  19                                          const char *username, bool password,
  20                                          const char *progname, bool echo);
  21 static void cluster_all_databases(bool verbose, const char *host, const char *port,
  22                                           const char *username, bool password,
  23                                           const char *progname, bool echo, bool quiet);
  24
  25 static void help(const char *progname);
  26
  27
  28 int
  29 main(int argc, char *argv[])
  30 {
  31         static struct option long_options[] = {
  32                 {"host", required_argument, NULL, 'h'},
  33                 {"port", required_argument, NULL, 'p'},
  34                 {"username", required_argument, NULL, 'U'},
  35                 {"password", no_argument, NULL, 'W'},
  36                 {"echo", no_argument, NULL, 'e'},
  37                 {"quiet", no_argument, NULL, 'q'},
  38                 {"dbname", required_argument, NULL, 'd'},
  39                 {"all", no_argument, NULL, 'a'},
  40                 {"table", required_argument, NULL, 't'},
  41                 {"verbose", no_argument, NULL, 'v'},
  42                 {NULL, 0, NULL, 0}
  43         };
  44
  45         const char *progname;
  46         int                     optindex;
  47         int                     c;
  48
  49         const char *dbname = NULL;
  50         char       *host = NULL;
  51         char       *port = NULL;
  52         char       *username = NULL;
  53         bool            password = false;
  54         bool            echo = false;
  55         bool            quiet = false;
  56         bool            alldb = false;
  57         char       *table = NULL;
  58         bool            verbose = false;
  59
  60         progname = get_progname(argv[0]);
  61         set_pglocale_pgservice(argv[0], PG_TEXTDOMAIN("pgscripts"));
  62
  63         handle_help_version_opts(argc, argv, "clusterdb", help);
  64
  65         while ((c = getopt_long(argc, argv, "h:p:U:Weqd:at:v", long_options, &optindex)) != -1)
  66         {
  67                 switch (c)
  68                 {
  69                         case 'h':
  70                                 host = optarg;
  71                                 break;
  72                         case 'p':
  73                                 port = optarg;
  74                                 break;
  75                         case 'U':
  76                                 username = optarg;
  77                                 break;
  78                         case 'W':
  79                                 password = true;
  80                                 break;
  81                         case 'e':
  82                                 echo = true;
  83                                 break;
  84                         case 'q':
  85                                 quiet = true;
  86                                 break;
  87                         case 'd':
  88                                 dbname = optarg;
  89                                 break;
  90                         case 'a':
  91                                 alldb = true;
  92                                 break;
  93                         case 't':
  94                                 table = optarg;
  95                                 break;
  96                         case 'v':
  97                                 verbose = true;
  98                                 break;
  99                         default:
 100                                 fprintf(stderr, _("Try \"%s --help\" for more information.\n"), progname);
 101                                 exit(1);
 102                 }
 103         }
 104
 105         switch (argc - optind)
 106         {
 107                 case 0:
 108                         break;
 109                 case 1:
 110                         dbname = argv[optind];
 111                         break;
 112                 default:
 113                         fprintf(stderr, _("%s: too many command-line arguments (first is \"%s\")\n"),
 114                                         progname, argv[optind + 1]);
 115                         fprintf(stderr, _("Try \"%s --help\" for more information.\n"), progname);
 116                         exit(1);
 117         }
 118
 119         setup_cancel_handler();
 120
 121         if (alldb)
 122         {
 123                 if (dbname)
 124                 {
 125                         fprintf(stderr, _("%s: cannot cluster all databases and a specific one at the same time\n"),
 126                                         progname);
 127                         exit(1);
 128                 }
 129                 if (table)
 130                 {
 131                         fprintf(stderr, _("%s: cannot cluster a specific table in all databases\n"),
 132                                         progname);
 133                         exit(1);
 134                 }
 135
 136                 cluster_all_databases(verbose, host, port, username, password,
 137                                                           progname, echo, quiet);
 138         }
 139         else
 140         {
 141                 if (dbname == NULL)
 142                 {
 143                         if (getenv("PGDATABASE"))
 144                                 dbname = getenv("PGDATABASE");
 145                         else if (getenv("PGUSER"))
 146                                 dbname = getenv("PGUSER");
 147                         else
 148                                 dbname = get_user_name(progname);
 149                 }
 150
 151                 cluster_one_database(dbname, verbose, table,
 152                                                          host, port, username, password,
 153                                                          progname, echo);
 154         }
 155
 156         exit(0);
 157 }
 158
 159
 160 static void
 161 cluster_one_database(const char *dbname, bool verbose, const char *table,
 162                                          const char *host, const char *port,
 163                                          const char *username, bool password,
 164                                          const char *progname, bool echo)
 165 {
 166         PQExpBufferData sql;
 167
 168         PGconn     *conn;
 169
 170         initPQExpBuffer(&sql);
 171
 172         appendPQExpBuffer(&sql, "CLUSTER");
 173         if (verbose)
 174                 appendPQExpBuffer(&sql, " VERBOSE");
 175         if (table)
 176                 appendPQExpBuffer(&sql, " %s", fmtId(table));
 177         appendPQExpBuffer(&sql, ";\n");
 178
 179         conn = connectDatabase(dbname, host, port, username, password, progname);
 180         if (!executeMaintenanceCommand(conn, sql.data, echo))
 181         {
 182                 if (table)
 183                         fprintf(stderr, _("%s: clustering of table \"%s\" in database \"%s\" failed: %s"),
 184                                         progname, table, dbname, PQerrorMessage(conn));
 185                 else
 186                         fprintf(stderr, _("%s: clustering of database \"%s\" failed: %s"),
 187                                         progname, dbname, PQerrorMessage(conn));
 188                 PQfinish(conn);
 189                 exit(1);
 190         }
 191         PQfinish(conn);
 192         termPQExpBuffer(&sql);
 193 }
 194
 195
 196 static void
 197 cluster_all_databases(bool verbose, const char *host, const char *port,
 198                                           const char *username, bool password,
 199                                           const char *progname, bool echo, bool quiet)
 200 {
 201         PGconn     *conn;
 202         PGresult   *result;
 203         int                     i;
 204
 205         conn = connectDatabase("postgres", host, port, username, password, progname);
 206         result = executeQuery(conn, "SELECT datname FROM pg_database WHERE datallowconn ORDER BY 1;", progname, echo);
 207         PQfinish(conn);
 208
 209         for (i = 0; i < PQntuples(result); i++)
 210         {
 211                 char       *dbname = PQgetvalue(result, i, 0);
 212
 213                 if (!quiet)
 214                 {
 215                         printf(_("%s: clustering database \"%s\"\n"), progname, dbname);
 216                         fflush(stdout);
 217                 }
 218
 219                 cluster_one_database(dbname, verbose, NULL,
 220                                                          host, port, username, password,
 221                                                          progname, echo);
 222         }
 223
 224         PQclear(result);
 225 }
 226
 227
 228 static void
 229 help(const char *progname)
 230 {
 231         printf(_("%s clusters all previously clustered tables in a database.\n\n"), progname);
 232         printf(_("Usage:\n"));
 233         printf(_("  %s [OPTION]... [DBNAME]\n"), progname);
 234         printf(_("\nOptions:\n"));
 235         printf(_("  -a, --all                 cluster all databases\n"));
 236         printf(_("  -d, --dbname=DBNAME       database to cluster\n"));
 237         printf(_("  -e, --echo                show the commands being sent to the server\n"));
 238         printf(_("  -q, --quiet               don't write any messages\n"));
 239         printf(_("  -t, --table=TABLE         cluster specific table only\n"));
 240         printf(_("  -v, --verbose             write a lot of output\n"));
 241         printf(_("  --help                    show this help, then exit\n"));
 242         printf(_("  --version                 output version information, then exit\n"));
 243         printf(_("\nConnection options:\n"));
 244         printf(_("  -h, --host=HOSTNAME       database server host or socket directory\n"));
 245         printf(_("  -p, --port=PORT           database server port\n"));
 246         printf(_("  -U, --username=USERNAME   user name to connect as\n"));
 247         printf(_("  -W, --password            force password prompt\n"));
 248         printf(_("\nRead the description of the SQL command CLUSTER for details.\n"));
 249         printf(_("\nReport bugs to <pgsql-bugs@postgresql.org>.\n"));
 250 }