diff --git a/src/backend/access/heap/heapam.c b/src/backend/access/heap/heapam.c index f7b08ffdd13..9afbc8228de 100644 --- a/src/backend/access/heap/heapam.c +++ b/src/backend/access/heap/heapam.c @@ -1615,8 +1615,14 @@ heap_endscan(HeapScanDesc scan) Size heap_parallelscan_estimate(Snapshot snapshot) { - return add_size(offsetof(ParallelHeapScanDescData, phs_snapshot_data), - EstimateSnapshotSpace(snapshot)); + Size sz = offsetof(ParallelHeapScanDescData, phs_snapshot_data); + + if (IsMVCCSnapshot(snapshot)) + sz = add_size(sz, EstimateSnapshotSpace(snapshot)); + else + Assert(snapshot == SnapshotAny); + + return sz; } /* ---------------- diff --git a/src/backend/access/nbtree/nbtsort.c b/src/backend/access/nbtree/nbtsort.c index 83966dc4d41..5cc3cf57e22 100644 --- a/src/backend/access/nbtree/nbtsort.c +++ b/src/backend/access/nbtree/nbtsort.c @@ -158,7 +158,7 @@ typedef struct BTShared /* * This variable-sized field must come last. * - * See _bt_parallel_estimate_shared(). + * See _bt_parallel_estimate_shared() and heap_parallelscan_estimate(). */ ParallelHeapScanDescData heapdesc; } BTShared; @@ -1405,15 +1405,8 @@ _bt_end_parallel(BTLeader *btleader) static Size _bt_parallel_estimate_shared(Snapshot snapshot) { - if (!IsMVCCSnapshot(snapshot)) - { - Assert(snapshot == SnapshotAny); - return sizeof(BTShared); - } - - return add_size(offsetof(BTShared, heapdesc) + - offsetof(ParallelHeapScanDescData, phs_snapshot_data), - EstimateSnapshotSpace(snapshot)); + return add_size(offsetof(BTShared, heapdesc), + heap_parallelscan_estimate(snapshot)); } /*