When searching for backend status iterate over the index in the LocalPgBackendStatus array, not over BackendId for v16 or ProcNumber for 16+.

antamel · antamel · commit aeabc16b73b6 · 2024-04-16T15:20:08.000+03:00
BackendId may not be equal  to the index +1 in the LocalPgBackendStatus array as well as ProcNumber may not be equal  the index.
diff --git a/pg_query_state.c b/pg_query_state.c
@@ -365,10 +365,16 @@ search_be_status(int pid)
 
 	for (beid = 1; beid <= pgstat_fetch_stat_numbackends(); beid++)
 	{
-#if PG_VERSION_NUM >= 170000
-		PgBackendStatus *be_status = pgstat_get_beentry_by_proc_number(beid);
-#elif PG_VERSION_NUM >= 160000
-		PgBackendStatus *be_status = pgstat_get_beentry_by_backend_id(beid);
+#if PG_VERSION_NUM >= 160000
+		LocalPgBackendStatus *lbe_status = pgstat_get_local_beentry_by_index(beid);
+		PgBackendStatus *be_status;
+
+		Assert(lbe_status);
+	#ifndef PGPRO_STD
+		be_status = &lbe_status->backendStatus;
+	#else
+		be_status = lbe_status->backendStatus;
+	#endif
 #else
 		PgBackendStatus *be_status = pgstat_fetch_stat_beentry(beid);
 #endif
