pg_resetwal: Allow users to change the WAL segment size

This adds a new option --wal-segsize (analogous to initdb) that changes
the WAL segment size in pg_control.

Author: Nathan Bossart <bossartn@amazon.com>

diff --git a/doc/src/sgml/ref/pg_resetwal.sgml b/doc/src/sgml/ref/pg_resetwal.sgml
index 5e2e3086a6eaf87baf6702be774c8598968257a3..3f885bdd620c6a2be3f678852806410b6afecb9b 100644 (file)
--- a/doc/src/sgml/ref/pg_resetwal.sgml
+++ b/doc/src/sgml/ref/pg_resetwal.sgml
@@ -275,6 +275,28 @@ PostgreSQL documentation
     </listitem>
    </varlistentry>
 
+   <varlistentry>
+    <term><option>--wal-segsize=<replaceable class="parameter">wal_segment_size</replaceable></option></term>
+    <listitem>
+     <para>
+      Set the new WAL segment size, in megabytes.  The value must be set to a
+      power of 2 between 1 and 1024 (megabytes).  See the same option of <xref
+      linkend="app-initdb"/> for more information.
+     </para>
+
+     <note>
+      <para>
+       While <command>pg_resetwal</command> will set the WAL starting address
+       beyond the latest existing WAL segment file, some segment size changes
+       can cause previous WAL file names to be reused.  It is recommended to
+       use <option>-l</option> together with this option to manually set the
+       WAL starting address if WAL file name overlap will cause problems with
+       your archiving strategy.
+      </para>
+     </note>
+    </listitem>
+   </varlistentry>
+
    <varlistentry>
     <term><option>-x <replaceable class="parameter">xid</replaceable></option></term>
     <term><option>--next-transaction-id=<replaceable class="parameter">xid</replaceable></option></term>

diff --git a/src/bin/pg_resetwal/pg_resetwal.c b/src/bin/pg_resetwal/pg_resetwal.c
index d5b4a5a0ee19c1290f3b470c63e60463856b0b64..eba7c5fdee0464827b9638fc059af569421d4a16 100644 (file)
--- a/src/bin/pg_resetwal/pg_resetwal.c
+++ b/src/bin/pg_resetwal/pg_resetwal.c
@@ -72,6 +72,7 @@ static MultiXactOffset set_mxoff = (MultiXactOffset) -1;
 static uint32 minXlogTli = 0;
 static XLogSegNo minXlogSegNo = 0;
 static int WalSegSz;
+static int set_wal_segsize;
 
 static void CheckDataVersion(void);
 static bool ReadControlFile(void);
@@ -100,6 +101,7 @@ main(int argc, char *argv[])
        {"next-oid", required_argument, NULL, 'o'},
        {"multixact-offset", required_argument, NULL, 'O'},
        {"next-transaction-id", required_argument, NULL, 'x'},
+       {"wal-segsize", required_argument, NULL, 1},
        {NULL, 0, NULL, 0}
    };
 
@@ -290,6 +292,24 @@ main(int argc, char *argv[])
                log_fname = pg_strdup(optarg);
                break;
 
+           case 1:
+               set_wal_segsize = strtol(optarg, &endptr, 10) * 1024 * 1024;
+               if (endptr == optarg || *endptr != '\0')
+               {
+                   fprintf(stderr,
+                           _("%s: argument of --wal-segsize must be a number\n"),
+                           progname);
+                   exit(1);
+               }
+               if (!IsValidWalSegSize(set_wal_segsize))
+               {
+                   fprintf(stderr,
+                           _("%s: argument of --wal-segsize must be a power of 2 between 1 and 1024\n"),
+                           progname);
+                   exit(1);
+               }
+               break;
+
            default:
                fprintf(stderr, _("Try \"%s --help\" for more information.\n"), progname);
                exit(1);
@@ -372,6 +392,14 @@ main(int argc, char *argv[])
    if (!ReadControlFile())
        GuessControlValues();
 
+   /*
+    * If no new WAL segment size was specified, use the control file value.
+    */
+   if (set_wal_segsize != 0)
+       WalSegSz = set_wal_segsize;
+   else
+       WalSegSz = ControlFile.xlog_seg_size;
+
    if (log_fname != NULL)
        XLogFromFileName(log_fname, &minXlogTli, &minXlogSegNo, WalSegSz);
 
@@ -438,6 +466,9 @@ main(int argc, char *argv[])
        ControlFile.checkPointCopy.PrevTimeLineID = minXlogTli;
    }
 
+   if (set_wal_segsize != 0)
+       ControlFile.xlog_seg_size = WalSegSz;
+
    if (minXlogSegNo > newXlogSegNo)
        newXlogSegNo = minXlogSegNo;
 
@@ -608,14 +639,13 @@ ReadControlFile(void)
        }
 
        memcpy(&ControlFile, buffer, sizeof(ControlFile));
-       WalSegSz = ControlFile.xlog_seg_size;
 
-       /* return false if WalSegSz is not valid */
-       if (!IsValidWalSegSize(WalSegSz))
+       /* return false if WAL segment size is not valid */
+       if (!IsValidWalSegSize(ControlFile.xlog_seg_size))
        {
            fprintf(stderr,
                    _("%s: pg_control specifies invalid WAL segment size (%d bytes); proceed with caution \n"),
-                   progname, WalSegSz);
+                   progname, ControlFile.xlog_seg_size);
            return false;
        }
 
@@ -694,7 +724,7 @@ GuessControlValues(void)
    ControlFile.blcksz = BLCKSZ;
    ControlFile.relseg_size = RELSEG_SIZE;
    ControlFile.xlog_blcksz = XLOG_BLCKSZ;
-   WalSegSz = ControlFile.xlog_seg_size = DEFAULT_XLOG_SEG_SIZE;
+   ControlFile.xlog_seg_size = DEFAULT_XLOG_SEG_SIZE;
    ControlFile.nameDataLen = NAMEDATALEN;
    ControlFile.indexMaxKeys = INDEX_MAX_KEYS;
    ControlFile.toast_max_chunk_size = TOAST_MAX_CHUNK_SIZE;
@@ -859,6 +889,12 @@ PrintNewControlValues(void)
        printf(_("newestCommitTsXid:                    %u\n"),
               ControlFile.checkPointCopy.newestCommitTsXid);
    }
+
+   if (set_wal_segsize != 0)
+   {
+       printf(_("Bytes per WAL segment:                %u\n"),
+              ControlFile.xlog_seg_size);
+   }
 }
 
 
@@ -910,9 +946,6 @@ RewriteControlFile(void)
    ControlFile.max_prepared_xacts = 0;
    ControlFile.max_locks_per_xact = 64;
 
-   /* Now we can force the recorded xlog seg size to the right thing. */
-   ControlFile.xlog_seg_size = WalSegSz;
-
    /* Contents are protected with a CRC */
    INIT_CRC32C(ControlFile.crc);
    COMP_CRC32C(ControlFile.crc,
@@ -1048,7 +1081,7 @@ FindEndOfXLOG(void)
     * are in virgin territory.
     */
    xlogbytepos = newXlogSegNo * ControlFile.xlog_seg_size;
-   newXlogSegNo = (xlogbytepos + WalSegSz - 1) / WalSegSz;
+   newXlogSegNo = (xlogbytepos + ControlFile.xlog_seg_size - 1) / WalSegSz;
    newXlogSegNo++;
 }
 
@@ -1279,6 +1312,7 @@ usage(void)
    printf(_("  -O, --multixact-offset=OFFSET  set next multitransaction offset\n"));
    printf(_("  -V, --version                  output version information, then exit\n"));
    printf(_("  -x, --next-transaction-id=XID  set next transaction ID\n"));
+   printf(_("      --wal-segsize=SIZE         size of WAL segments, in megabytes\n"));
    printf(_("  -?, --help                     show this help, then exit\n"));
    printf(_("\nReport bugs to <pgsql-bugs@postgresql.org>.\n"));
 }