Disable timeouts when running pg_rewind with online source cluster

In this case, the transfer uses a libpq connection, which is subject to
the timeout parameters set at system level, and this can make the rewind
operation suddenly canceled which is not good for automation.  One
workaround to such issues would be to use PGOPTIONS to enforce the
wanted timeout parameters, but that's annoying, and for example pg_dump,
which can run potentially long-running queries disables all types of
timeouts.

lock_timeout and statement_timeout are the ones which can cause problems
now.  Note that pg_rewind does not use transactions, so disabling
idle_in_transaction_session_timeout is optional, but it feels safer to
do so for the future.

This is back-patched down to 9.5.  idle_in_transaction_session_timeout
is only present since 9.6.

Author: Alexander Kukushkin
Discussion: https://postgr.es/m/CAFh8B=krcVXksxiwVQh1SoY+ziJ-JC=6FcuoBL3yce_40Es5_g@mail.gmail.com
Backpatch-through: 9.5

diff --git a/src/bin/pg_rewind/libpq_fetch.c b/src/bin/pg_rewind/libpq_fetch.c
index 89d7886fd1733a75cdf0e0cd358457fd060afd94..710eca1c750012753439268fb6a92cffa346a307 100644 (file)
--- a/src/bin/pg_rewind/libpq_fetch.c
+++ b/src/bin/pg_rewind/libpq_fetch.c
@@ -44,6 +44,7 @@ static PGconn *conn = NULL;
 static void receiveFileChunks(const char *sql);
 static void execute_pagemap(datapagemap_t *pagemap, const char *path);
 static char *run_simple_query(const char *sql);
+static void run_simple_command(const char *sql);
 
 void
 libpqConnect(const char *connstr)
@@ -58,6 +59,11 @@ libpqConnect(const char *connstr)
 
        pg_log(PG_PROGRESS, "connected to server\n");
 
+       /* disable all types of timeouts */
+       run_simple_command("SET statement_timeout = 0");
+       run_simple_command("SET lock_timeout = 0");
+       run_simple_command("SET idle_in_transaction_session_timeout = 0");
+
        res = PQexec(conn, ALWAYS_SECURE_SEARCH_PATH_SQL);
        if (PQresultStatus(res) != PGRES_TUPLES_OK)
                pg_fatal("could not clear search_path: %s",
@@ -92,11 +98,7 @@ libpqConnect(const char *connstr)
         * replication, and replication isn't working for some reason, we don't
         * want to get stuck, waiting for it to start working again.
         */
-       res = PQexec(conn, "SET synchronous_commit = off");
-       if (PQresultStatus(res) != PGRES_COMMAND_OK)
-               pg_fatal("could not set up connection context: %s",
-                                PQresultErrorMessage(res));
-       PQclear(res);
+       run_simple_command("SET synchronous_commit = off");
 }
 
 /*
@@ -126,6 +128,24 @@ run_simple_query(const char *sql)
        return result;
 }
 
+/*
+ * Runs a command.
+ * In the event of a failure, exit immediately.
+ */
+static void
+run_simple_command(const char *sql)
+{
+       PGresult   *res;
+
+       res = PQexec(conn, sql);
+
+       if (PQresultStatus(res) != PGRES_COMMAND_OK)
+               pg_fatal("error running query (%s) in source server: %s",
+                                sql, PQresultErrorMessage(res));
+
+       PQclear(res);
+}
+
 /*
  * Calls pg_current_wal_insert_lsn() function
  */
@@ -460,12 +480,7 @@ libpq_executeFileMap(filemap_t *map)
         * need to fetch.
         */
        sql = "CREATE TEMPORARY TABLE fetchchunks(path text, begin int8, len int4);";
-       res = PQexec(conn, sql);
-
-       if (PQresultStatus(res) != PGRES_COMMAND_OK)
-               pg_fatal("could not create temporary table: %s",
-                                PQresultErrorMessage(res));
-       PQclear(res);
+       run_simple_command(sql);
 
        sql = "COPY fetchchunks FROM STDIN";
        res = PQexec(conn, sql);