granicus.if.org Git - postgresql/blob - contrib/pg_upgrade/pg_upgrade.c

   1 /*
   2  *      pg_upgrade.c
   3  *
   4  *      main source file
   5  *
   6  *      Copyright (c) 2010-2011, PostgreSQL Global Development Group
   7  *      contrib/pg_upgrade/pg_upgrade.c
   8  */
   9
  10 /*
  11  *      To simplify the upgrade process, we force certain system values to be
  12  *      identical between old and new clusters:
  13  *
  14  *      We control all assignments of pg_class.oid (and relfilenode) so toast
  15  *      oids are the same between old and new clusters.  This is important
  16  *      because toast oids are stored as toast pointers in user tables.
  17  *
  18  *      FYI, while pg_class.oid and pg_class.relfilenode are intially the same
  19  *      in a cluster, but they can diverge due to CLUSTER, REINDEX, or VACUUM
  20  *      FULL.  The new cluster will have matching pg_class.oid and
  21  *      pg_class.relfilenode values and be based on the old oid value.  This can
  22  *      cause the old and new pg_class.relfilenode values to differ.  In summary,
  23  *      old and new pg_class.oid and new pg_class.relfilenode will have the
  24  *      same value, and old pg_class.relfilenode might differ.
  25  *
  26  *      We control all assignments of pg_type.oid because these oids are stored
  27  *      in user composite type values.
  28  *
  29  *      We control all assignments of pg_enum.oid because these oids are stored
  30  *      in user tables as enum values.
  31  *
  32  *      We control all assignments of pg_auth.oid because these oids are stored
  33  *      in pg_largeobject_metadata.
  34  */
  35
  36
  37
  38 #include "pg_upgrade.h"
  39
  40 #ifdef HAVE_LANGINFO_H
  41 #include <langinfo.h>
  42 #endif
  43
  44 static void disable_old_cluster(void);
  45 static void prepare_new_cluster(void);
  46 static void prepare_new_databases(void);
  47 static void create_new_objects(void);
  48 static void copy_clog_xlog_xid(void);
  49 static void set_frozenxids(void);
  50 static void setup(char *argv0, bool live_check);
  51 static void cleanup(void);
  52
  53 /* This is the database used by pg_dumpall to restore global tables */
  54 #define GLOBAL_DUMP_DB  "postgres"
  55
  56 ClusterInfo old_cluster,
  57                         new_cluster;
  58 OSInfo          os_info;
  59
  60 int
  61 main(int argc, char **argv)
  62 {
  63         char       *sequence_script_file_name = NULL;
  64         char       *deletion_script_file_name = NULL;
  65         bool            live_check = false;
  66
  67         parseCommandLine(argc, argv);
  68
  69         output_check_banner(&live_check);
  70
  71         setup(argv[0], live_check);
  72
  73         check_cluster_versions();
  74         check_cluster_compatibility(live_check);
  75
  76         check_old_cluster(live_check, &sequence_script_file_name);
  77
  78
  79         /* -- NEW -- */
  80         start_postmaster(&new_cluster);
  81
  82         check_new_cluster();
  83         report_clusters_compatible();
  84
  85         pg_log(PG_REPORT, "\nPerforming Upgrade\n");
  86         pg_log(PG_REPORT, "------------------\n");
  87
  88         disable_old_cluster();
  89         prepare_new_cluster();
  90
  91         stop_postmaster(false);
  92
  93         /*
  94          * Destructive Changes to New Cluster
  95          */
  96
  97         copy_clog_xlog_xid();
  98
  99         /* New now using xids of the old system */
 100
 101         /* -- NEW -- */
 102         start_postmaster(&new_cluster);
 103
 104         prepare_new_databases();
 105
 106         create_new_objects();
 107
 108         stop_postmaster(false);
 109
 110         transfer_all_new_dbs(&old_cluster.dbarr, &new_cluster.dbarr,
 111                                                  old_cluster.pgdata, new_cluster.pgdata);
 112
 113         /*
 114          * Assuming OIDs are only used in system tables, there is no need to
 115          * restore the OID counter because we have not transferred any OIDs from
 116          * the old system, but we do it anyway just in case.  We do it late here
 117          * because there is no need to have the schema load use new oids.
 118          */
 119         prep_status("Setting next oid for new cluster");
 120         exec_prog(true, SYSTEMQUOTE "\"%s/pg_resetxlog\" -o %u \"%s\" > "
 121                           DEVNULL SYSTEMQUOTE,
 122                           new_cluster.bindir, old_cluster.controldata.chkpnt_nxtoid, new_cluster.pgdata);
 123         check_ok();
 124
 125         create_script_for_old_cluster_deletion(&deletion_script_file_name);
 126
 127         issue_warnings(sequence_script_file_name);
 128
 129         pg_log(PG_REPORT, "\nUpgrade complete\n");
 130         pg_log(PG_REPORT, "----------------\n");
 131
 132         output_completion_banner(deletion_script_file_name);
 133
 134         pg_free(deletion_script_file_name);
 135         pg_free(sequence_script_file_name);
 136
 137         cleanup();
 138
 139         return 0;
 140 }
 141
 142
 143 static void
 144 setup(char *argv0, bool live_check)
 145 {
 146         char            exec_path[MAXPGPATH];   /* full path to my executable */
 147
 148         /*
 149          * make sure the user has a clean environment, otherwise, we may confuse
 150          * libpq when we connect to one (or both) of the servers.
 151          */
 152         check_for_libpq_envvars();
 153
 154         verify_directories();
 155
 156         /* no postmasters should be running */
 157         if (!live_check && is_server_running(old_cluster.pgdata))
 158         {
 159                 pg_log(PG_FATAL, "There seems to be a postmaster servicing the old cluster.\n"
 160                            "Please shutdown that postmaster and try again.\n");
 161         }
 162
 163         /* same goes for the new postmaster */
 164         if (is_server_running(new_cluster.pgdata))
 165         {
 166                 pg_log(PG_FATAL, "There seems to be a postmaster servicing the new cluster.\n"
 167                            "Please shutdown that postmaster and try again.\n");
 168         }
 169
 170         /* get path to pg_upgrade executable */
 171         if (find_my_exec(argv0, exec_path) < 0)
 172                 pg_log(PG_FATAL, "Could not get pathname to pg_upgrade: %s\n", getErrorText(errno));
 173
 174         /* Trim off program name and keep just path */
 175         *last_dir_separator(exec_path) = '\0';
 176         canonicalize_path(exec_path);
 177         os_info.exec_path = pg_strdup(exec_path);
 178 }
 179
 180
 181 static void
 182 disable_old_cluster(void)
 183 {
 184         /* rename pg_control so old server cannot be accidentally started */
 185         rename_old_pg_control();
 186 }
 187
 188
 189 static void
 190 prepare_new_cluster(void)
 191 {
 192         /*
 193          * It would make more sense to freeze after loading the schema, but that
 194          * would cause us to lose the frozenids restored by the load. We use
 195          * --analyze so autovacuum doesn't update statistics later
 196          */
 197         prep_status("Analyzing all rows in the new cluster");
 198         exec_prog(true,
 199                           SYSTEMQUOTE "\"%s/vacuumdb\" --port %d --username \"%s\" "
 200                           "--all --analyze >> %s 2>&1" SYSTEMQUOTE,
 201           new_cluster.bindir, new_cluster.port, os_info.user, log_opts.filename);
 202         check_ok();
 203
 204         /*
 205          * We do freeze after analyze so pg_statistic is also frozen. template0 is
 206          * not frozen here, but data rows were frozen by initdb, and we set its
 207          * datfrozenxid and relfrozenxids later to match the new xid counter
 208          * later.
 209          */
 210         prep_status("Freezing all rows on the new cluster");
 211         exec_prog(true,
 212                           SYSTEMQUOTE "\"%s/vacuumdb\" --port %d --username \"%s\" "
 213                           "--all --freeze >> %s 2>&1" SYSTEMQUOTE,
 214           new_cluster.bindir, new_cluster.port, os_info.user, log_opts.filename);
 215         check_ok();
 216
 217         get_pg_database_relfilenode(&new_cluster);
 218 }
 219
 220
 221 static void
 222 prepare_new_databases(void)
 223 {
 224         /*
 225          * We set autovacuum_freeze_max_age to its maximum value so autovacuum
 226          * does not launch here and delete clog files, before the frozen xids are
 227          * set.
 228          */
 229
 230         set_frozenxids();
 231
 232         prep_status("Creating databases in the new cluster");
 233
 234         /*
 235          * Install support functions in the global-restore database to preserve
 236          * pg_authid.oid.
 237          */
 238         install_support_functions_in_new_db(GLOBAL_DUMP_DB);
 239
 240         /*
 241          * We have to create the databases first so we can install support
 242          * functions in all the other databases.  Ideally we could create the
 243          * support functions in template1 but pg_dumpall creates database using
 244          * the template0 template.
 245          */
 246         exec_prog(true,
 247                           SYSTEMQUOTE "\"%s/psql\" --set ON_ERROR_STOP=on "
 248         /* --no-psqlrc prevents AUTOCOMMIT=off */
 249                           "--no-psqlrc --port %d --username \"%s\" "
 250                           "-f \"%s/%s\" --dbname template1 >> \"%s\"" SYSTEMQUOTE,
 251                           new_cluster.bindir, new_cluster.port, os_info.user, os_info.cwd,
 252                           GLOBALS_DUMP_FILE, log_opts.filename);
 253         check_ok();
 254
 255         /* we load this to get a current list of databases */
 256         get_db_and_rel_infos(&new_cluster);
 257 }
 258
 259
 260 static void
 261 create_new_objects(void)
 262 {
 263         int                     dbnum;
 264
 265         prep_status("Adding support functions to new cluster");
 266
 267         for (dbnum = 0; dbnum < new_cluster.dbarr.ndbs; dbnum++)
 268         {
 269                 DbInfo     *new_db = &new_cluster.dbarr.dbs[dbnum];
 270
 271                 /* skip db we already installed */
 272                 if (strcmp(new_db->db_name, GLOBAL_DUMP_DB) != 0)
 273                         install_support_functions_in_new_db(new_db->db_name);
 274         }
 275         check_ok();
 276
 277         prep_status("Restoring database schema to new cluster");
 278         exec_prog(true,
 279                           SYSTEMQUOTE "\"%s/psql\" --set ON_ERROR_STOP=on "
 280                           "--no-psqlrc --port %d --username \"%s\" "
 281                           "-f \"%s/%s\" --dbname template1 >> \"%s\"" SYSTEMQUOTE,
 282                           new_cluster.bindir, new_cluster.port, os_info.user, os_info.cwd,
 283                           DB_DUMP_FILE, log_opts.filename);
 284         check_ok();
 285
 286         /* regenerate now that we have objects in the databases */
 287         get_db_and_rel_infos(&new_cluster);
 288
 289         uninstall_support_functions_from_new_cluster();
 290 }
 291
 292
 293 static void
 294 copy_clog_xlog_xid(void)
 295 {
 296         char            old_clog_path[MAXPGPATH];
 297         char            new_clog_path[MAXPGPATH];
 298
 299         /* copy old commit logs to new data dir */
 300         prep_status("Deleting new commit clogs");
 301
 302         snprintf(old_clog_path, sizeof(old_clog_path), "%s/pg_clog", old_cluster.pgdata);
 303         snprintf(new_clog_path, sizeof(new_clog_path), "%s/pg_clog", new_cluster.pgdata);
 304         if (!rmtree(new_clog_path, true))
 305                 pg_log(PG_FATAL, "Unable to delete directory %s\n", new_clog_path);
 306         check_ok();
 307
 308         prep_status("Copying old commit clogs to new server");
 309 #ifndef WIN32
 310         exec_prog(true, SYSTEMQUOTE "%s \"%s\" \"%s\"" SYSTEMQUOTE,
 311                           "cp -Rf",
 312 #else
 313         /* flags: everything, no confirm, quiet, overwrite read-only */
 314         exec_prog(true, SYSTEMQUOTE "%s \"%s\" \"%s\\\"" SYSTEMQUOTE,
 315                           "xcopy /e /y /q /r",
 316 #endif
 317                           old_clog_path, new_clog_path);
 318         check_ok();
 319
 320         /* set the next transaction id of the new cluster */
 321         prep_status("Setting next transaction id for new cluster");
 322         exec_prog(true, SYSTEMQUOTE "\"%s/pg_resetxlog\" -f -x %u \"%s\" > " DEVNULL SYSTEMQUOTE,
 323                           new_cluster.bindir, old_cluster.controldata.chkpnt_nxtxid, new_cluster.pgdata);
 324         check_ok();
 325
 326         /* now reset the wal archives in the new cluster */
 327         prep_status("Resetting WAL archives");
 328         exec_prog(true, SYSTEMQUOTE "\"%s/pg_resetxlog\" -l %u,%u,%u \"%s\" >> \"%s\" 2>&1" SYSTEMQUOTE,
 329                           new_cluster.bindir, old_cluster.controldata.chkpnt_tli,
 330                         old_cluster.controldata.logid, old_cluster.controldata.nxtlogseg,
 331                           new_cluster.pgdata, log_opts.filename);
 332         check_ok();
 333 }
 334
 335
 336 /*
 337  *      set_frozenxids()
 338  *
 339  *      We have frozen all xids, so set relfrozenxid and datfrozenxid
 340  *      to be the old cluster's xid counter, which we just set in the new
 341  *      cluster.  User-table frozenxid values will be set by pg_dumpall
 342  *      --binary-upgrade, but objects not set by the pg_dump must have
 343  *      proper frozen counters.
 344  */
 345 static
 346 void
 347 set_frozenxids(void)
 348 {
 349         int                     dbnum;
 350         PGconn     *conn,
 351                            *conn_template1;
 352         PGresult   *dbres;
 353         int                     ntups;
 354         int                     i_datname;
 355         int                     i_datallowconn;
 356
 357         prep_status("Setting frozenxid counters in new cluster");
 358
 359         conn_template1 = connectToServer(&new_cluster, "template1");
 360
 361         /* set pg_database.datfrozenxid */
 362         PQclear(executeQueryOrDie(conn_template1,
 363                                                           "UPDATE pg_catalog.pg_database "
 364                                                           "SET  datfrozenxid = '%u'",
 365                                                           old_cluster.controldata.chkpnt_nxtxid));
 366
 367         /* get database names */
 368         dbres = executeQueryOrDie(conn_template1,
 369                                                           "SELECT       datname, datallowconn "
 370                                                           "FROM pg_catalog.pg_database");
 371
 372         i_datname = PQfnumber(dbres, "datname");
 373         i_datallowconn = PQfnumber(dbres, "datallowconn");
 374
 375         ntups = PQntuples(dbres);
 376         for (dbnum = 0; dbnum < ntups; dbnum++)
 377         {
 378                 char       *datname = PQgetvalue(dbres, dbnum, i_datname);
 379                 char       *datallowconn = PQgetvalue(dbres, dbnum, i_datallowconn);
 380
 381                 /*
 382                  * We must update databases where datallowconn = false, e.g.
 383                  * template0, because autovacuum increments their datfrozenxids and
 384                  * relfrozenxids even if autovacuum is turned off, and even though all
 385                  * the data rows are already frozen  To enable this, we temporarily
 386                  * change datallowconn.
 387                  */
 388                 if (strcmp(datallowconn, "f") == 0)
 389                         PQclear(executeQueryOrDie(conn_template1,
 390                                                                           "UPDATE pg_catalog.pg_database "
 391                                                                           "SET  datallowconn = true "
 392                                                                           "WHERE datname = '%s'", datname));
 393
 394                 conn = connectToServer(&new_cluster, datname);
 395
 396                 /* set pg_class.relfrozenxid */
 397                 PQclear(executeQueryOrDie(conn,
 398                                                                   "UPDATE       pg_catalog.pg_class "
 399                                                                   "SET  relfrozenxid = '%u' "
 400                 /* only heap and TOAST are vacuumed */
 401                                                                   "WHERE        relkind IN ('r', 't')",
 402                                                                   old_cluster.controldata.chkpnt_nxtxid));
 403                 PQfinish(conn);
 404
 405                 /* Reset datallowconn flag */
 406                 if (strcmp(datallowconn, "f") == 0)
 407                         PQclear(executeQueryOrDie(conn_template1,
 408                                                                           "UPDATE pg_catalog.pg_database "
 409                                                                           "SET  datallowconn = false "
 410                                                                           "WHERE datname = '%s'", datname));
 411         }
 412
 413         PQclear(dbres);
 414
 415         PQfinish(conn_template1);
 416
 417         check_ok();
 418 }
 419
 420
 421 static void
 422 cleanup(void)
 423 {
 424         char            filename[MAXPGPATH];
 425
 426         if (log_opts.fd)
 427                 fclose(log_opts.fd);
 428
 429         if (log_opts.debug_fd)
 430                 fclose(log_opts.debug_fd);
 431
 432         snprintf(filename, sizeof(filename), "%s/%s", os_info.cwd, ALL_DUMP_FILE);
 433         unlink(filename);
 434         snprintf(filename, sizeof(filename), "%s/%s", os_info.cwd, GLOBALS_DUMP_FILE);
 435         unlink(filename);
 436         snprintf(filename, sizeof(filename), "%s/%s", os_info.cwd, DB_DUMP_FILE);
 437         unlink(filename);
 438 }