WaitLatch stats

Author: kelvich

arbiter.c

@@ -57,7 +57,8 @@
 #include "replication/slot.h"
 #include "port/atomics.h"
 #include "tcop/utility.h"
-#include "libpq/ip.h"
+#include "common/ip.h"
+#include "pgstat.h"
 
 
 #ifndef USE_EPOLL
@@ -822,7 +823,7 @@ static void MtmMonitor(Datum arg)
 	BackgroundWorkerInitializeConnection(MtmDatabaseName, NULL);
 
 	while (!stop) {
-		int rc = WaitLatch(&MyProc->procLatch, WL_TIMEOUT | WL_POSTMASTER_DEATH, MtmHeartbeatRecvTimeout);
+		int rc = WaitLatch(&MyProc->procLatch, WL_TIMEOUT | WL_POSTMASTER_DEATH, MtmHeartbeatRecvTimeout, PG_WAIT_EXTENSION);
 		if (rc & WL_POSTMASTER_DEATH) { 
 			break;
 		}

multimaster.c

@@ -14,6 +14,8 @@
 #include "fmgr.h"
 #include "miscadmin.h"
 #include "common/pg_socket.h"
+#include "pgstat.h"
+#include "utils/regproc.h"
 
 #include "libpq-fe.h"
 #include "lib/stringinfo.h"
@@ -1290,7 +1292,7 @@ Mtm2PCVoting(MtmCurrentTrans* x, MtmTransState* ts)
 	{
 		MtmUnlock();
 		MTM_TXTRACE(x, "PostPrepareTransaction WaitLatch Start");
-		result = WaitLatch(&MyProc->procLatch, WL_LATCH_SET | WL_TIMEOUT | WL_POSTMASTER_DEATH, MtmHeartbeatSendTimeout);
+		result = WaitLatch(&MyProc->procLatch, WL_LATCH_SET | WL_TIMEOUT | WL_POSTMASTER_DEATH, MtmHeartbeatSendTimeout, PG_WAIT_EXTENSION);
 		MTM_TXTRACE(x, "PostPrepareTransaction WaitLatch Finish");
 		/* Emergency bailout if postmaster has died */
 		if (result & WL_POSTMASTER_DEATH) {

pglogical_receiver.c

@@ -400,7 +400,8 @@ pglogical_receiver_main(Datum main_arg)
 			/* Wait necessary amount of time */
 			rc = WaitLatch(&MyProc->procLatch,
 						   WL_LATCH_SET | WL_TIMEOUT | WL_POSTMASTER_DEATH,
-						   receiver_idle_time * 1L);
+						   receiver_idle_time * 1L,
+						   PG_WAIT_EXTENSION);
 			ResetLatch(&MyProc->procLatch);
 			/* Process signals */
 			if (got_sighup)

referee.c

@@ -26,6 +26,7 @@
 #include "storage/latch.h"
 #include "storage/proc.h"
 #include "storage/ipc.h"
+#include "pgstat.h"
 
 #include "multimaster.h"
 
@@ -96,7 +97,7 @@ static void MtmRefereeLoop(char const** connections, int nConns)
 			}
 		}
 		/* Wait some time */
-		result = WaitLatch(&MyProc->procLatch, WL_TIMEOUT | WL_POSTMASTER_DEATH, MtmHeartbeatRecvTimeout);
+		result = WaitLatch(&MyProc->procLatch, WL_TIMEOUT | WL_POSTMASTER_DEATH, MtmHeartbeatRecvTimeout, PG_WAIT_EXTENSION);
 		if (result & WL_POSTMASTER_DEATH) {
 			proc_exit(1);
 		}