diff --git a/src/backend/access/table/tableam.c b/src/backend/access/table/tableam.c index 68ff0966f1c..8e86915f4f5 100644 --- a/src/backend/access/table/tableam.c +++ b/src/backend/access/table/tableam.c @@ -434,6 +434,17 @@ table_block_parallelscan_reinitialize(Relation rel, ParallelTableScanDesc pscan) { ParallelBlockTableScanDesc bpscan = (ParallelBlockTableScanDesc) pscan; + /* + * Reset the start block and block limit too. These are set by the first + * participant to call table_block_parallelscan_startblock_init(), and a + * rescan may want to scan a different range of blocks, e.g. a TID Range + * Scan with different TID bounds. + */ + SpinLockAcquire(&bpscan->phs_mutex); + bpscan->phs_startblock = InvalidBlockNumber; + bpscan->phs_numblock = InvalidBlockNumber; + SpinLockRelease(&bpscan->phs_mutex); + pg_atomic_write_u64(&bpscan->phs_nallocated, 0); } diff --git a/src/test/regress/expected/tidrangescan.out b/src/test/regress/expected/tidrangescan.out index 41f3ba06c3b..e478dfb32ad 100644 --- a/src/test/regress/expected/tidrangescan.out +++ b/src/test/regress/expected/tidrangescan.out @@ -427,5 +427,36 @@ WHERE t.ctid < '(1,0)'; (0,5) | 5 (5 rows) +-- Parallel rescans with a different TID range on each rescan +EXPLAIN (COSTS OFF) +SELECT r.lo, + (SELECT count(*) FROM parallel_tidrangescan t + WHERE t.ctid >= (SELECT r.lo)) c +FROM (VALUES ('(30,0)'::tid), ('(10,0)'), ('(0,0)')) r(lo); + QUERY PLAN +------------------------------------------------------------------------------ + Values Scan on "*VALUES*" + SubPlan expr_1 + -> Finalize Aggregate + InitPlan expr_2 + -> Result + -> Gather + Workers Planned: 4 + -> Partial Aggregate + -> Parallel Tid Range Scan on parallel_tidrangescan t + TID Cond: (ctid >= (InitPlan expr_2).col1) +(10 rows) + +SELECT r.lo, + (SELECT count(*) FROM parallel_tidrangescan t + WHERE t.ctid >= (SELECT r.lo)) c +FROM (VALUES ('(30,0)'::tid), ('(10,0)'), ('(0,0)')) r(lo); + lo | c +--------+----- + (30,0) | 50 + (10,0) | 150 + (0,0) | 200 +(3 rows) + ROLLBACK; RESET enable_seqscan; diff --git a/src/test/regress/sql/tidrangescan.sql b/src/test/regress/sql/tidrangescan.sql index 79f9e2b32be..f8cee41f92c 100644 --- a/src/test/regress/sql/tidrangescan.sql +++ b/src/test/regress/sql/tidrangescan.sql @@ -153,5 +153,17 @@ SELECT t.ctid,t2.c FROM parallel_tidrangescan t, LATERAL (SELECT count(*) c FROM parallel_tidrangescan t2 WHERE t2.ctid <= t.ctid) t2 WHERE t.ctid < '(1,0)'; +-- Parallel rescans with a different TID range on each rescan +EXPLAIN (COSTS OFF) +SELECT r.lo, + (SELECT count(*) FROM parallel_tidrangescan t + WHERE t.ctid >= (SELECT r.lo)) c +FROM (VALUES ('(30,0)'::tid), ('(10,0)'), ('(0,0)')) r(lo); + +SELECT r.lo, + (SELECT count(*) FROM parallel_tidrangescan t + WHERE t.ctid >= (SELECT r.lo)) c +FROM (VALUES ('(30,0)'::tid), ('(10,0)'), ('(0,0)')) r(lo); + ROLLBACK; RESET enable_seqscan;