Fix index in array to accommodate parallel workers

Use poscan->worker_active array for Asserts only.
JoshInnis · Jun 11, 2024 · da149b9 · da149b9
1 parent 6de6b11
commit da149b9
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/include/tableam/handler.h b/include/tableam/handler.h
@@ -209,7 +209,9 @@ typedef struct ParallelOScanDescData
 	dsm_handle	dsmHandle;
 	/* debug only */
 	int			cur_int_pageno;
-	bool		worker_active[10];
+#ifdef USE_ASSERT_CHECKING
+	bool		worker_active[1024];
+#endif
 } ParallelOScanDescData;
 
 typedef ParallelOScanDescData *ParallelOScanDesc;

diff --git a/src/btree/scan.c b/src/btree/scan.c
@@ -1039,13 +1039,18 @@ init_btree_seq_scan(BTreeSeqScan *scan)
 		 * after all workers completed their scans).
 		 */
 		SpinLockAcquire(&poscan->workerStart);
+#ifdef USE_ASSERT_CHECKING
 		for (scan->workerNumber = 0; poscan->worker_active[scan->workerNumber] == true; scan->workerNumber++)
 		{
 		}
 
 		poscan->worker_active[scan->workerNumber] = true;
 		poscan->nworkers = scan->workerNumber + 1;
-		/* leader */
+#else
+		scan->workerNumber = poscan->nworkers;
+		poscan->nworkers++;
+#endif
+		/* Scan leader */
 		if (scan->workerNumber == 0)
 		{
 			Assert(!(poscan->flags & O_PARALLEL_LEADER_STARTED));