Fix LOCK_TIMEOUT handling in slotsync worker.

Previously, the slotsync worker relied on SIGINT for graceful shutdown
during promotion. However, SIGINT is also used by the LOCK_TIMEOUT handler
to cancel queries. Since the slotsync worker can lock catalog tables while
parsing libpq tuples, this overlap caused it to ignore LOCK_TIMEOUT
signals and potentially wait indefinitely on locks.

This patch replaces the slotsync worker's SIGINT handler with
StatementCancelHandler to correctly process query-cancel interrupts.
Additionally, the startup process now uses SIGUSR1 to signal the slotsync
worker to stop during promotion. The worker exits after detecting that the
shared memory flag stopSignaled is set.

Author: Hou Zhijie <[email protected]>
Reviewed-by: shveta malik <[email protected]>
Reviewed-by: Chao Li <[email protected]>
Reviewed-by: Amit Kapila <[email protected]>
Backpatch-through: 17, here it was introduced
Discussion: https://postgr.es/m/TY4PR01MB169078F33846E9568412D878C94A2A@TY4PR01MB16907.jpnprd01.prod.outlook.com

diff --git a/src/backend/replication/logical/slotsync.c b/src/backend/replication/logical/slotsync.c
index 61b2e9396aa697010b191f26cf12b6fc247bfcf3..35be111aa73e0204c139044eba633120d0e4a425 100644 (file)
--- a/src/backend/replication/logical/slotsync.c
+++ b/src/backend/replication/logical/slotsync.c
@@ -1175,10 +1175,10 @@ ProcessSlotSyncInterrupts(WalReceiverConn *wrconn)
 {
    CHECK_FOR_INTERRUPTS();
 
-   if (ShutdownRequestPending)
+   if (SlotSyncCtx->stopSignaled)
    {
        ereport(LOG,
-               errmsg("replication slot synchronization worker is shutting down on receiving SIGINT"));
+               errmsg("replication slot synchronization worker is shutting down because promotion is triggered"));
 
        proc_exit(0);
    }
@@ -1409,7 +1409,7 @@ ReplSlotSyncWorkerMain(const void *startup_data, size_t startup_data_len)
 
    /* Setup signal handling */
    pqsignal(SIGHUP, SignalHandlerForConfigReload);
-   pqsignal(SIGINT, SignalHandlerForShutdownRequest);
+   pqsignal(SIGINT, StatementCancelHandler);
    pqsignal(SIGTERM, die);
    pqsignal(SIGFPE, FloatExceptionHandler);
    pqsignal(SIGUSR1, procsignal_sigusr1_handler);
@@ -1516,7 +1516,8 @@ ReplSlotSyncWorkerMain(const void *startup_data, size_t startup_data_len)
 
    /*
     * The slot sync worker can't get here because it will only stop when it
-    * receives a SIGINT from the startup process, or when there is an error.
+    * receives a stop request from the startup process, or when there is an
+    * error.
     */
    Assert(false);
 }
@@ -1601,8 +1602,12 @@ ShutDownSlotSync(void)
 
    SpinLockRelease(&SlotSyncCtx->mutex);
 
+   /*
+    * Signal slotsync worker if it was still running. The worker will stop
+    * upon detecting that the stopSignaled flag is set to true.
+    */
    if (worker_pid != InvalidPid)
-       kill(worker_pid, SIGINT);
+       kill(worker_pid, SIGUSR1);
 
    /* Wait for slot sync to end */
    for (;;)