From 18000dc65214587fd05989de99b576c353ddba6f Mon Sep 17 00:00:00 2001 From: Paul Guo Date: Thu, 18 Apr 2019 19:37:36 +0800 Subject: [PATCH v10] Add option to write recovery configuration information in pg_rewind. --- doc/src/sgml/ref/pg_rewind.sgml | 12 ++++++++++++ src/bin/pg_rewind/Makefile | 2 +- src/bin/pg_rewind/libpq_fetch.c | 3 +-- src/bin/pg_rewind/pg_rewind.c | 33 ++++++++++++++++++++++++++++++++- src/bin/pg_rewind/pg_rewind.h | 4 +++- 5 files changed, 49 insertions(+), 5 deletions(-) diff --git a/doc/src/sgml/ref/pg_rewind.sgml b/doc/src/sgml/ref/pg_rewind.sgml index a06e5ac5e1..a5fdbcd4d5 100644 --- a/doc/src/sgml/ref/pg_rewind.sgml +++ b/doc/src/sgml/ref/pg_rewind.sgml @@ -180,6 +180,18 @@ PostgreSQL documentation + + + + + + Create standby.signal and append connection settings + to postgresql.auto.conf in the output directory. + It is only used by the case with source server. + + + + diff --git a/src/bin/pg_rewind/Makefile b/src/bin/pg_rewind/Makefile index 859d3abc41..98fb381c62 100644 --- a/src/bin/pg_rewind/Makefile +++ b/src/bin/pg_rewind/Makefile @@ -16,7 +16,7 @@ top_builddir = ../../.. include $(top_builddir)/src/Makefile.global override CPPFLAGS := -I$(libpq_srcdir) -DFRONTEND $(CPPFLAGS) -LDFLAGS_INTERNAL += $(libpq_pgport) +LDFLAGS_INTERNAL += $(libpq_pgport) -L$(top_builddir)/src/fe_utils -lpgfeutils OBJS = pg_rewind.o parsexlog.o xlogreader.o datapagemap.o timeline.o \ fetch.o file_ops.o copy_fetch.o libpq_fetch.o filemap.o \ diff --git a/src/bin/pg_rewind/libpq_fetch.c b/src/bin/pg_rewind/libpq_fetch.c index 002776f696..f4ebf7d842 100644 --- a/src/bin/pg_rewind/libpq_fetch.c +++ b/src/bin/pg_rewind/libpq_fetch.c @@ -20,12 +20,11 @@ #include "file_ops.h" #include "filemap.h" -#include "libpq-fe.h" #include "catalog/pg_type_d.h" #include "fe_utils/connect.h" #include "port/pg_bswap.h" -static PGconn *conn = NULL; +PGconn *conn = NULL; /* * Files are fetched max CHUNKSIZE bytes at a time. diff --git a/src/bin/pg_rewind/pg_rewind.c b/src/bin/pg_rewind/pg_rewind.c index 8cb0d726cf..63a2059021 100644 --- a/src/bin/pg_rewind/pg_rewind.c +++ b/src/bin/pg_rewind/pg_rewind.c @@ -27,6 +27,7 @@ #include "common/file_perm.h" #include "common/file_utils.h" #include "common/restricted_token.h" +#include "fe_utils/recovery_gen.h" #include "getopt_long.h" #include "storage/bufpage.h" @@ -66,6 +67,12 @@ int targetNentries; uint64 fetch_size; uint64 fetch_done; +static void +disconnect_atexit(void) +{ + if (conn != NULL) + PQfinish(conn); +} static void usage(const char *progname) @@ -76,6 +83,7 @@ usage(const char *progname) printf(_(" -D, --target-pgdata=DIRECTORY existing data directory to modify\n")); printf(_(" --source-pgdata=DIRECTORY source data directory to synchronize with\n")); printf(_(" --source-server=CONNSTR source server to synchronize with\n")); + printf(_(" -R, --write-recovery-conf write configuration for replication\n")); printf(_(" -n, --dry-run stop before modifying anything\n")); printf(_(" -N, --no-sync do not wait for changes to be written\n" " safely to disk\n")); @@ -94,6 +102,7 @@ main(int argc, char **argv) static struct option long_options[] = { {"help", no_argument, NULL, '?'}, {"target-pgdata", required_argument, NULL, 'D'}, + {"write-recovery-conf", no_argument, NULL, 'R'}, {"source-pgdata", required_argument, NULL, 1}, {"source-server", required_argument, NULL, 2}, {"no-ensure-shutdown", no_argument, NULL, 44}, @@ -118,6 +127,7 @@ main(int argc, char **argv) XLogRecPtr endrec; TimeLineID endtli; ControlFileData ControlFile_new; + bool writerecoveryconf = false; pg_logging_init(argv[0]); set_pglocale_pgservice(argv[0], PG_TEXTDOMAIN("pg_rewind")); @@ -138,7 +148,7 @@ main(int argc, char **argv) } } - while ((c = getopt_long(argc, argv, "D:nNP", long_options, &option_index)) != -1) + while ((c = getopt_long(argc, argv, "D:nNPR", long_options, &option_index)) != -1) { switch (c) { @@ -158,6 +168,10 @@ main(int argc, char **argv) do_sync = false; break; + case 'R': + writerecoveryconf = true; + break; + case 3: debug = true; pg_logging_set_level(PG_LOG_DEBUG); @@ -200,6 +214,14 @@ main(int argc, char **argv) exit(1); } + if (writerecoveryconf && connstr_source == NULL) + { + pg_log_error("no source server information is specified " + "(--source--server) for -R or --write-recovery-conf"); + fprintf(stderr, _("Try \"%s --help\" for more information.\n"), progname); + exit(1); + } + if (optind < argc) { pg_log_error("too many command-line arguments (first is \"%s\")", @@ -236,6 +258,8 @@ main(int argc, char **argv) umask(pg_mode_mask); + atexit(disconnect_atexit); + /* Connect to remote server */ if (connstr_source) libpqConnect(connstr_source); @@ -322,6 +346,9 @@ main(int argc, char **argv) if (!rewind_needed) { pg_log_info("no rewind required"); + if (writerecoveryconf) + WriteRecoveryConfig(conn, datadir_target, + GenerateRecoveryConfig(conn, NULL)); exit(0); } @@ -419,6 +446,10 @@ main(int argc, char **argv) pg_log_info("syncing target data directory"); syncTargetDirectory(); + if (writerecoveryconf) + WriteRecoveryConfig(conn, datadir_target, + GenerateRecoveryConfig(conn, NULL)); + pg_log_info("Done!"); return 0; diff --git a/src/bin/pg_rewind/pg_rewind.h b/src/bin/pg_rewind/pg_rewind.h index 1125c7e60f..46aa6df4ba 100644 --- a/src/bin/pg_rewind/pg_rewind.h +++ b/src/bin/pg_rewind/pg_rewind.h @@ -16,9 +16,10 @@ #include "access/timeline.h" #include "storage/block.h" #include "storage/relfilenode.h" - #include "common/logging.h" +#include "libpq-fe.h" + /* Configuration options */ extern char *datadir_target; extern char *datadir_source; @@ -26,6 +27,7 @@ extern char *connstr_source; extern bool showprogress; extern bool dry_run; extern int WalSegSz; +extern PGconn *conn; /* Target history */ extern TimeLineHistoryEntry *targetHistory; -- 2.17.2