Location via proxy:   [ UP ]  
[Report a bug]   [Manage cookies]                
Skip to content

Commit 88ceac5

Browse files
committed
Fix parallel-safety marking when moving initplans to another node.
Our policy since commit ab77a5a has been that a plan node having any initplans is automatically not parallel-safe. (This could be relaxed, but not today.) clean_up_removed_plan_level neglected this, and could attach initplans to a parallel-safe child plan node without clearing the plan's parallel-safe flag. That could lead to "subplan was not initialized" errors at runtime, in case an initplan referenced another one and only the referencing one got transmitted to parallel workers. The fix in clean_up_removed_plan_level is trivial enough. materialize_finished_plan also moves initplans from one node to another, but it's okay because it already copies the source node's parallel_safe flag. The other place that does this kind of thing is standard_planner's hack to inject a top-level Gather when debug_parallel_query is active. But that's actually dead code given that we're correctly enforcing the "initplans aren't parallel safe" rule, so just replace it with an Assert that there are no initplans. Also improve some related comments. Normally we'd add a regression test case for this sort of bug. The mistake itself is already reached by existing tests, but there is accidentally no visible problem. The only known test case that creates an actual failure seems too indirect and fragile to justify keeping it as a regression test (not least because it fails to fail in v11, though the bug is clearly present there too). Per report from Justin Pryzby. Back-patch to all supported branches. Discussion: https://postgr.es/m/ZDVt6MaNWkRDO1LQ@telsasoft.com
1 parent 91199dd commit 88ceac5

File tree

4 files changed

+22
-10
lines changed

4 files changed

+22
-10
lines changed

src/backend/optimizer/plan/planner.c

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -436,12 +436,10 @@ standard_planner(Query *parse, const char *query_string, int cursorOptions,
436436
Gather *gather = makeNode(Gather);
437437

438438
/*
439-
* If there are any initPlans attached to the formerly-top plan node,
440-
* move them up to the Gather node; same as we do for Material node in
441-
* materialize_finished_plan.
439+
* Top plan must not have any initPlans, else it shouldn't have been
440+
* marked parallel-safe.
442441
*/
443-
gather->plan.initPlan = top_plan->initPlan;
444-
top_plan->initPlan = NIL;
442+
Assert(top_plan->initPlan == NIL);
445443

446444
gather->plan.targetlist = top_plan->targetlist;
447445
gather->plan.qual = NIL;

src/backend/optimizer/plan/setrefs.c

Lines changed: 13 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1542,7 +1542,19 @@ trivial_subqueryscan(SubqueryScan *plan)
15421542
static Plan *
15431543
clean_up_removed_plan_level(Plan *parent, Plan *child)
15441544
{
1545-
/* We have to be sure we don't lose any initplans */
1545+
/*
1546+
* We have to be sure we don't lose any initplans, so move any that were
1547+
* attached to the parent plan to the child. If we do move any, the child
1548+
* is no longer parallel-safe.
1549+
*/
1550+
if (parent->initPlan)
1551+
child->parallel_safe = false;
1552+
1553+
/*
1554+
* Attach plans this way so that parent's initplans are processed before
1555+
* any pre-existing initplans of the child. Probably doesn't matter, but
1556+
* let's preserve the ordering just in case.
1557+
*/
15461558
child->initPlan = list_concat(parent->initPlan,
15471559
child->initPlan);
15481560

src/backend/optimizer/plan/subselect.c

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2126,7 +2126,7 @@ SS_identify_outer_params(PlannerInfo *root)
21262126
* This is separate from SS_attach_initplans because we might conditionally
21272127
* create more initPlans during create_plan(), depending on which Path we
21282128
* select. However, Paths that would generate such initPlans are expected
2129-
* to have included their cost already.
2129+
* to have included their cost and parallel-safety effects already.
21302130
*/
21312131
void
21322132
SS_charge_for_initplans(PlannerInfo *root, RelOptInfo *final_rel)
@@ -2182,8 +2182,10 @@ SS_charge_for_initplans(PlannerInfo *root, RelOptInfo *final_rel)
21822182
* (In principle the initPlans could go in any node at or above where they're
21832183
* referenced; but there seems no reason to put them any lower than the
21842184
* topmost node, so we don't bother to track exactly where they came from.)
2185-
* We do not touch the plan node's cost; the initplans should have been
2186-
* accounted for in path costing.
2185+
*
2186+
* We do not touch the plan node's cost or parallel_safe flag. The initplans
2187+
* must have been accounted for in SS_charge_for_initplans, or by any later
2188+
* code that adds initplans via SS_make_initplan_from_plan.
21872189
*/
21882190
void
21892191
SS_attach_initplans(PlannerInfo *root, Plan *plan)

src/backend/optimizer/util/pathnode.c

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3348,7 +3348,7 @@ create_minmaxagg_path(PlannerInfo *root,
33483348
/* For now, assume we are above any joins, so no parameterization */
33493349
pathnode->path.param_info = NULL;
33503350
pathnode->path.parallel_aware = false;
3351-
/* A MinMaxAggPath implies use of subplans, so cannot be parallel-safe */
3351+
/* A MinMaxAggPath implies use of initplans, so cannot be parallel-safe */
33523352
pathnode->path.parallel_safe = false;
33533353
pathnode->path.parallel_workers = 0;
33543354
/* Result is one unordered row */

0 commit comments

Comments
 (0)