database/postgres
1
0
Fork 0
mirror of https://github.com/postgres/postgres.git synced 2025-08-27 07:42:10 +03:00
Code Activity

Change mechanism to set up source targetlist in MERGE

Browse Source 
We were setting MERGE source subplan's targetlist by expanding the
individual attributes of the source relation completely, early in the
parse analysis phase.  This failed to work when the condition of an
action included a whole-row reference, causing setrefs.c to error out
with
  ERROR:  variable not found in subplan target lists
because at that point there is nothing to resolve the whole-row
reference with.  We can fix this by having preprocess_targetlist expand
the source targetlist for Vars required from the source rel by all
actions.  Moreover, by using this expansion mechanism we can do away
with the targetlist expansion in transformMergeStmt, which is good
because then we no longer pull in columns that aren't needed for
anything.

Add a test case for the problem.

While at it, remove some redundant code in preprocess_targetlist():
MERGE was doing separately what is already being done for UPDATE/DELETE,
so we can just rely on the latter and remove the former.  (The handling
of inherited rels was different for MERGE, but that was a no-longer-
necessary hack.)

Fix outdated, related comments for fix_join_expr also.

Author: Richard Guo <guofenglinux@gmail.com>
Author: Álvaro Herrera <alvherre@alvh.no-ip.org>
Reported-by: Joe Wildish <joe@lateraljoin.com>
Discussion: https://postgr.es/m/fab3b90a-914d-46a9-beb0-df011ee39ee5@www.fastmail.com
This commit is contained in:
Alvaro Herrera
2022-04-12 09:29:39 +02:00
parent a4b57543ac
commit ce4f46fdc8
6 changed files with 80 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/backend/optimizer/plan/setrefs.c
View File

@@ -2748,7 +2748,7 @@ search_indexed_tlist_for_sortgroupref(Expr *node,
* relation target lists. Also perform opcode lookup and add
* regclass OIDs to root->glob->relationOids.
*
* This is used in three different scenarios:
* This is used in four different scenarios:
* 1) a normal join clause, where all the Vars in the clause *must* be
* replaced by OUTER_VAR or INNER_VAR references. In this case
* acceptable_rel should be zero so that any failure to match a Var will be
@@ -2763,6 +2763,11 @@ search_indexed_tlist_for_sortgroupref(Expr *node,
* to-be-updated relation) alone. Correspondingly inner_itlist is to be
* EXCLUDED elements, outer_itlist = NULL and acceptable_rel the target
* relation.
* 4) MERGE. In this case, references to the source relation are to be
* replaced with INNER_VAR references, leaving Vars of the target
* relation (the to-be-modified relation) alone. So inner_itlist is to be
* the source relation elements, outer_itlist = NULL and acceptable_rel
* the target relation.
*
* 'clauses' is the targetlist or list of join clauses
* 'outer_itlist' is the indexed target list of the outer join relation,

63
src/backend/optimizer/prep/preptlist.c
View File

@@ -107,14 +107,15 @@ preprocess_targetlist(PlannerInfo *root)
root->update_colnos = extract_update_targetlist_colnos(tlist);
/*
* For non-inherited UPDATE/DELETE, register any junk column(s) needed to
* allow the executor to identify the rows to be updated or deleted. In
* the inheritance case, we do nothing now, leaving this to be dealt with
* when expand_inherited_rtentry() makes the leaf target relations. (But
* there might not be any leaf target relations, in which case we must do
* this in distribute_row_identity_vars().)
* For non-inherited UPDATE/DELETE/MERGE, register any junk column(s)
* needed to allow the executor to identify the rows to be updated or
* deleted. In the inheritance case, we do nothing now, leaving this to
* be dealt with when expand_inherited_rtentry() makes the leaf target
* relations. (But there might not be any leaf target relations, in which
* case we must do this in distribute_row_identity_vars().)
*/
if ((command_type == CMD_UPDATE || command_type == CMD_DELETE) &&
if ((command_type == CMD_UPDATE || command_type == CMD_DELETE ||
command_type == CMD_MERGE) &&
!target_rte->inh)
{
/* row-identity logic expects to add stuff to processed_tlist */
@@ -125,23 +126,15 @@ preprocess_targetlist(PlannerInfo *root)
}
/*
* For MERGE we need to handle the target list for the target relation,
* and also target list for each action (only INSERT/UPDATE matter).
* For MERGE we also need to handle the target list for each INSERT and
* UPDATE action separately. In addition, we examine the qual of each
* action and add any Vars there (other than those of the target rel) to
* the subplan targetlist.
*/
if (command_type == CMD_MERGE)
{
ListCell *l;
/*
* For MERGE, add any junk column(s) needed to allow the executor to
* identify the rows to be inserted or updated.
*/
root->processed_tlist = tlist;
add_row_identity_columns(root, result_relation,
target_rte, target_relation);
tlist = root->processed_tlist;
/*
* For MERGE, handle targetlist of each MergeAction separately. Give
* the same treatment to MergeAction->targetList as we would have
@@ -151,6 +144,8 @@ preprocess_targetlist(PlannerInfo *root)
foreach(l, parse->mergeActionList)
{
MergeAction *action = (MergeAction *) lfirst(l);
List *vars;
ListCell *l2;
if (action->commandType == CMD_INSERT)
action->targetList = expand_insert_targetlist(action->targetList,
@@ -158,6 +153,36 @@ preprocess_targetlist(PlannerInfo *root)
else if (action->commandType == CMD_UPDATE)
action->updateColnos =
extract_update_targetlist_colnos(action->targetList);
/*
* Add resjunk entries for any Vars used in each action's
* targetlist and WHEN condition that belong to relations other
* than target. Note that aggregates, window functions and
* placeholder vars are not possible anywhere in MERGE's WHEN
* clauses. (PHVs may be added later, but they don't concern us
* here.)
*/
vars = pull_var_clause((Node *)
list_concat_copy((List *) action->qual,
action->targetList),
0);
foreach(l2, vars)
{
Var *var = (Var *) lfirst(l2);
TargetEntry *tle;
if (IsA(var, Var) && var->varno == result_relation)
continue; /* don't need it */
if (tlist_member((Expr *) var, tlist))
continue; /* already got it */
tle = makeTargetEntry((Expr *) var,
list_length(tlist) + 1,
NULL, true);
tlist = lappend(tlist, tle);
}
list_free(vars);
}
}