Age Owner Branch data TLA Line data Source code
1 : : /*-------------------------------------------------------------------------
2 : : *
3 : : * appendinfo.c
4 : : * Routines for mapping between append parent(s) and children
5 : : *
6 : : * Portions Copyright (c) 1996-2026, PostgreSQL Global Development Group
7 : : * Portions Copyright (c) 1994, Regents of the University of California
8 : : *
9 : : *
10 : : * IDENTIFICATION
11 : : * src/backend/optimizer/util/appendinfo.c
12 : : *
13 : : *-------------------------------------------------------------------------
14 : : */
15 : : #include "postgres.h"
16 : :
17 : : #include "access/htup_details.h"
18 : : #include "access/sysattr.h"
19 : : #include "access/table.h"
20 : : #include "foreign/fdwapi.h"
21 : : #include "nodes/makefuncs.h"
22 : : #include "nodes/nodeFuncs.h"
23 : : #include "optimizer/appendinfo.h"
24 : : #include "optimizer/pathnode.h"
25 : : #include "optimizer/planmain.h"
26 : : #include "parser/parsetree.h"
27 : : #include "utils/lsyscache.h"
28 : : #include "utils/rel.h"
29 : : #include "utils/syscache.h"
30 : :
31 : :
32 : : typedef struct
33 : : {
34 : : PlannerInfo *root;
35 : : int nappinfos;
36 : : AppendRelInfo **appinfos;
37 : : } adjust_appendrel_attrs_context;
38 : :
39 : : static void make_inh_translation_list(Relation oldrelation,
40 : : Relation newrelation,
41 : : Index newvarno,
42 : : AppendRelInfo *appinfo);
43 : : static Node *adjust_appendrel_attrs_mutator(Node *node,
44 : : adjust_appendrel_attrs_context *context);
45 : :
46 : :
47 : : /*
48 : : * make_append_rel_info
49 : : * Build an AppendRelInfo for the parent-child pair
50 : : */
51 : : AppendRelInfo *
2786 alvherre@alvh.no-ip. 52 :CBC 35361 : make_append_rel_info(Relation parentrel, Relation childrel,
53 : : Index parentRTindex, Index childRTindex)
54 : : {
55 : 35361 : AppendRelInfo *appinfo = makeNode(AppendRelInfo);
56 : :
57 : 35361 : appinfo->parent_relid = parentRTindex;
58 : 35361 : appinfo->child_relid = childRTindex;
59 : 35361 : appinfo->parent_reltype = parentrel->rd_rel->reltype;
60 : 35361 : appinfo->child_reltype = childrel->rd_rel->reltype;
2460 tgl@sss.pgh.pa.us 61 : 35361 : make_inh_translation_list(parentrel, childrel, childRTindex, appinfo);
2786 alvherre@alvh.no-ip. 62 : 35360 : appinfo->parent_reloid = RelationGetRelid(parentrel);
63 : :
64 : 35360 : return appinfo;
65 : : }
66 : :
67 : : /*
68 : : * make_inh_translation_list
69 : : * Build the list of translations from parent Vars to child Vars for
70 : : * an inheritance child, as well as a reverse-translation array.
71 : : *
72 : : * The reverse-translation array has an entry for each child relation
73 : : * column, which is either the 1-based index of the corresponding parent
74 : : * column, or 0 if there's no match (that happens for dropped child columns,
75 : : * as well as child columns beyond those of the parent, which are allowed in
76 : : * traditional inheritance though not partitioning).
77 : : *
78 : : * For paranoia's sake, we match type/collation as well as attribute name.
79 : : */
80 : : static void
81 : 35361 : make_inh_translation_list(Relation oldrelation, Relation newrelation,
82 : : Index newvarno,
83 : : AppendRelInfo *appinfo)
84 : : {
85 : 35361 : List *vars = NIL;
86 : : AttrNumber *pcolnos;
87 : 35361 : TupleDesc old_tupdesc = RelationGetDescr(oldrelation);
88 : 35361 : TupleDesc new_tupdesc = RelationGetDescr(newrelation);
89 : 35361 : Oid new_relid = RelationGetRelid(newrelation);
90 : 35361 : int oldnatts = old_tupdesc->natts;
91 : 35361 : int newnatts = new_tupdesc->natts;
92 : : int old_attno;
93 : 35361 : int new_attno = 0;
94 : :
95 : : /* Initialize reverse-translation array with all entries zero */
2460 tgl@sss.pgh.pa.us 96 : 35361 : appinfo->num_child_cols = newnatts;
10 michael@paquier.xyz 97 :GNC 35361 : appinfo->parent_colnos = pcolnos = palloc0_array(AttrNumber, newnatts);
98 : :
2786 alvherre@alvh.no-ip. 99 [ + + ]:CBC 128859 : for (old_attno = 0; old_attno < oldnatts; old_attno++)
100 : : {
101 : : Form_pg_attribute att;
102 : : char *attname;
103 : : Oid atttypid;
104 : : int32 atttypmod;
105 : : Oid attcollation;
106 : :
107 : 93499 : att = TupleDescAttr(old_tupdesc, old_attno);
108 [ + + ]: 93499 : if (att->attisdropped)
109 : : {
110 : : /* Just put NULL into this list entry */
111 : 2643 : vars = lappend(vars, NULL);
112 : 2643 : continue;
113 : : }
114 : 90856 : attname = NameStr(att->attname);
115 : 90856 : atttypid = att->atttypid;
116 : 90856 : atttypmod = att->atttypmod;
117 : 90856 : attcollation = att->attcollation;
118 : :
119 : : /*
120 : : * When we are generating the "translation list" for the parent table
121 : : * of an inheritance set, no need to search for matches.
122 : : */
123 [ + + ]: 90856 : if (oldrelation == newrelation)
124 : : {
125 : 6059 : vars = lappend(vars, makeVar(newvarno,
126 : 6059 : (AttrNumber) (old_attno + 1),
127 : : atttypid,
128 : : atttypmod,
129 : : attcollation,
130 : : 0));
2460 tgl@sss.pgh.pa.us 131 : 6059 : pcolnos[old_attno] = old_attno + 1;
2786 alvherre@alvh.no-ip. 132 : 6059 : continue;
133 : : }
134 : :
135 : : /*
136 : : * Otherwise we have to search for the matching column by name.
137 : : * There's no guarantee it'll have the same column position, because
138 : : * of cases like ALTER TABLE ADD COLUMN and multiple inheritance.
139 : : * However, in simple cases, the relative order of columns is mostly
140 : : * the same in both relations, so try the column of newrelation that
141 : : * follows immediately after the one that we just found, and if that
142 : : * fails, let syscache handle it.
143 : : */
144 [ + + ]: 84797 : if (new_attno >= newnatts ||
145 [ + + ]: 82961 : (att = TupleDescAttr(new_tupdesc, new_attno))->attisdropped ||
146 [ + + ]: 82101 : strcmp(attname, NameStr(att->attname)) != 0)
147 : : {
148 : : HeapTuple newtup;
149 : :
150 : 7047 : newtup = SearchSysCacheAttName(new_relid, attname);
2671 tgl@sss.pgh.pa.us 151 [ - + ]: 7047 : if (!HeapTupleIsValid(newtup))
2786 alvherre@alvh.no-ip. 152 [ # # ]:UBC 0 : elog(ERROR, "could not find inherited attribute \"%s\" of relation \"%s\"",
153 : : attname, RelationGetRelationName(newrelation));
2786 alvherre@alvh.no-ip. 154 :CBC 7047 : new_attno = ((Form_pg_attribute) GETSTRUCT(newtup))->attnum - 1;
2460 tgl@sss.pgh.pa.us 155 [ + - - + ]: 7047 : Assert(new_attno >= 0 && new_attno < newnatts);
2786 alvherre@alvh.no-ip. 156 : 7047 : ReleaseSysCache(newtup);
157 : :
158 : 7047 : att = TupleDescAttr(new_tupdesc, new_attno);
159 : : }
160 : :
161 : : /* Found it, check type and collation match */
162 [ + + - + ]: 84797 : if (atttypid != att->atttypid || atttypmod != att->atttypmod)
784 michael@paquier.xyz 163 [ + - ]: 1 : ereport(ERROR,
164 : : (errcode(ERRCODE_INVALID_COLUMN_DEFINITION),
165 : : errmsg("attribute \"%s\" of relation \"%s\" does not match parent's type",
166 : : attname, RelationGetRelationName(newrelation))));
2786 alvherre@alvh.no-ip. 167 [ - + ]: 84796 : if (attcollation != att->attcollation)
784 michael@paquier.xyz 168 [ # # ]:UBC 0 : ereport(ERROR,
169 : : (errcode(ERRCODE_INVALID_COLUMN_DEFINITION),
170 : : errmsg("attribute \"%s\" of relation \"%s\" does not match parent's collation",
171 : : attname, RelationGetRelationName(newrelation))));
172 : :
2786 alvherre@alvh.no-ip. 173 :CBC 84796 : vars = lappend(vars, makeVar(newvarno,
174 : 84796 : (AttrNumber) (new_attno + 1),
175 : : atttypid,
176 : : atttypmod,
177 : : attcollation,
178 : : 0));
2460 tgl@sss.pgh.pa.us 179 : 84796 : pcolnos[new_attno] = old_attno + 1;
2786 alvherre@alvh.no-ip. 180 : 84796 : new_attno++;
181 : : }
182 : :
2460 tgl@sss.pgh.pa.us 183 : 35360 : appinfo->translated_vars = vars;
2786 alvherre@alvh.no-ip. 184 : 35360 : }
185 : :
186 : : /*
187 : : * adjust_appendrel_attrs
188 : : * Copy the specified query or expression and translate Vars referring to a
189 : : * parent rel to refer to the corresponding child rel instead. We also
190 : : * update rtindexes appearing outside Vars, such as resultRelation and
191 : : * jointree relids.
192 : : *
193 : : * Note: this is only applied after conversion of sublinks to subplans,
194 : : * so we don't need to cope with recursion into sub-queries.
195 : : *
196 : : * Note: this is not hugely different from what pullup_replace_vars() does;
197 : : * maybe we should try to fold the two routines together.
198 : : */
199 : : Node *
200 : 240659 : adjust_appendrel_attrs(PlannerInfo *root, Node *node, int nappinfos,
201 : : AppendRelInfo **appinfos)
202 : : {
203 : : adjust_appendrel_attrs_context context;
204 : :
205 : 240659 : context.root = root;
206 : 240659 : context.nappinfos = nappinfos;
207 : 240659 : context.appinfos = appinfos;
208 : :
209 : : /* If there's nothing to adjust, don't call this function. */
210 [ + - - + ]: 240659 : Assert(nappinfos >= 1 && appinfos != NULL);
211 : :
212 : : /* Should never be translating a Query tree. */
1975 tgl@sss.pgh.pa.us 213 [ + + - + ]: 240659 : Assert(node == NULL || !IsA(node, Query));
214 : :
215 : 240659 : return adjust_appendrel_attrs_mutator(node, &context);
216 : : }
217 : :
218 : : static Node *
2786 alvherre@alvh.no-ip. 219 : 1260014 : adjust_appendrel_attrs_mutator(Node *node,
220 : : adjust_appendrel_attrs_context *context)
221 : : {
222 : 1260014 : AppendRelInfo **appinfos = context->appinfos;
223 : 1260014 : int nappinfos = context->nappinfos;
224 : : int cnt;
225 : :
226 [ + + ]: 1260014 : if (node == NULL)
227 : 238298 : return NULL;
228 [ + + ]: 1021716 : if (IsA(node, Var))
229 : : {
230 : 453081 : Var *var = (Var *) copyObject(node);
231 : 453081 : AppendRelInfo *appinfo = NULL;
232 : :
2422 tgl@sss.pgh.pa.us 233 [ - + ]: 453081 : if (var->varlevelsup != 0)
2422 tgl@sss.pgh.pa.us 234 :UBC 0 : return (Node *) var; /* no changes needed */
235 : :
236 : : /*
237 : : * You might think we need to adjust var->varnullingrels, but that
238 : : * shouldn't need any changes. It will contain outer-join relids,
239 : : * while the transformation we are making affects only baserels.
240 : : * Below, we just merge var->varnullingrels into the translated Var.
241 : : * (We must merge not just copy: the child Var could have some
242 : : * nullingrel bits set already, and we mustn't drop those.)
243 : : *
244 : : * If var->varnullingrels isn't empty, and the translation wouldn't be
245 : : * a Var, we have to fail. One could imagine wrapping the translated
246 : : * expression in a PlaceHolderVar, but that won't work because this is
247 : : * typically used after freezing placeholders. Fortunately, the case
248 : : * appears unreachable at the moment. We can see nonempty
249 : : * var->varnullingrels here, but only in cases involving partitionwise
250 : : * joining, and in such cases the translations will always be Vars.
251 : : * (Non-Var translations occur only for appendrels made by flattening
252 : : * UNION ALL subqueries.) Should we need to make this work in future,
253 : : * a possible fix is to mandate that prepjointree.c create PHVs for
254 : : * all non-Var outputs of such subqueries, and then we could look up
255 : : * the pre-existing PHV here. Or perhaps just wrap the translations
256 : : * that way to begin with?
257 : : *
258 : : * If var->varreturningtype is not VAR_RETURNING_DEFAULT, then that
259 : : * also needs to be copied to the translated Var. That too would fail
260 : : * if the translation wasn't a Var, but that should never happen since
261 : : * a non-default var->varreturningtype is only used for Vars referring
262 : : * to the result relation, which should never be a flattened UNION ALL
263 : : * subquery.
264 : : */
265 : :
2786 alvherre@alvh.no-ip. 266 [ + + ]:CBC 569593 : for (cnt = 0; cnt < nappinfos; cnt++)
267 : : {
268 [ + + ]: 535059 : if (var->varno == appinfos[cnt]->parent_relid)
269 : : {
270 : 418547 : appinfo = appinfos[cnt];
271 : 418547 : break;
272 : : }
273 : : }
274 : :
2422 tgl@sss.pgh.pa.us 275 [ + + ]: 453081 : if (appinfo)
276 : : {
2786 alvherre@alvh.no-ip. 277 : 418547 : var->varno = appinfo->child_relid;
278 : : /* it's now a generated Var, so drop any syntactic labeling */
2422 tgl@sss.pgh.pa.us 279 : 418547 : var->varnosyn = 0;
280 : 418547 : var->varattnosyn = 0;
2786 alvherre@alvh.no-ip. 281 [ + + ]: 418547 : if (var->varattno > 0)
282 : : {
283 : : Node *newnode;
284 : :
285 [ - + ]: 403881 : if (var->varattno > list_length(appinfo->translated_vars))
2786 alvherre@alvh.no-ip. 286 [ # # ]:UBC 0 : elog(ERROR, "attribute %d of relation \"%s\" does not exist",
287 : : var->varattno, get_rel_name(appinfo->parent_reloid));
2786 alvherre@alvh.no-ip. 288 :CBC 403881 : newnode = copyObject(list_nth(appinfo->translated_vars,
289 : : var->varattno - 1));
290 [ - + ]: 403881 : if (newnode == NULL)
2786 alvherre@alvh.no-ip. 291 [ # # ]:UBC 0 : elog(ERROR, "attribute %d of relation \"%s\" does not exist",
292 : : var->varattno, get_rel_name(appinfo->parent_reloid));
1305 tgl@sss.pgh.pa.us 293 [ + + ]:CBC 403881 : if (IsA(newnode, Var))
294 : : {
188 rguo@postgresql.org 295 : 396135 : Var *newvar = (Var *) newnode;
296 : :
297 : 396135 : newvar->varreturningtype = var->varreturningtype;
298 : 396135 : newvar->varnullingrels = bms_add_members(newvar->varnullingrels,
299 : 396135 : var->varnullingrels);
300 : : }
301 : : else
302 : : {
588 dean.a.rasheed@gmail 303 [ - + ]: 7746 : if (var->varreturningtype != VAR_RETURNING_DEFAULT)
588 dean.a.rasheed@gmail 304 [ # # ]:UBC 0 : elog(ERROR, "failed to apply returningtype to a non-Var");
588 dean.a.rasheed@gmail 305 [ - + ]:CBC 7746 : if (var->varnullingrels != NULL)
588 dean.a.rasheed@gmail 306 [ # # ]:UBC 0 : elog(ERROR, "failed to apply nullingrels to a non-Var");
307 : : }
2786 alvherre@alvh.no-ip. 308 :CBC 403881 : return newnode;
309 : : }
310 [ + + ]: 14666 : else if (var->varattno == 0)
311 : : {
312 : : /*
313 : : * Whole-row Var: if we are dealing with named rowtypes, we
314 : : * can use a whole-row Var for the child table plus a coercion
315 : : * step to convert the tuple layout to the parent's rowtype.
316 : : * Otherwise we have to generate a RowExpr.
317 : : */
318 [ + + ]: 870 : if (OidIsValid(appinfo->child_reltype))
319 : : {
320 [ - + ]: 826 : Assert(var->vartype == appinfo->parent_reltype);
321 [ + + ]: 826 : if (appinfo->parent_reltype != appinfo->child_reltype)
322 : : {
323 : 660 : ConvertRowtypeExpr *r = makeNode(ConvertRowtypeExpr);
324 : :
325 : 660 : r->arg = (Expr *) var;
326 : 660 : r->resulttype = appinfo->parent_reltype;
327 : 660 : r->convertformat = COERCE_IMPLICIT_CAST;
328 : 660 : r->location = -1;
329 : : /* Make sure the Var node has the right type ID, too */
330 : 660 : var->vartype = appinfo->child_reltype;
331 : 660 : return (Node *) r;
332 : : }
333 : : }
334 : : else
335 : : {
336 : : /*
337 : : * Build a RowExpr containing the translated variables.
338 : : *
339 : : * In practice var->vartype will always be RECORDOID here,
340 : : * so we need to come up with some suitable column names.
341 : : * We use the parent RTE's column names.
342 : : *
343 : : * Note: we can't get here for inheritance cases, so there
344 : : * is no need to worry that translated_vars might contain
345 : : * some dummy NULLs.
346 : : */
347 : : RowExpr *rowexpr;
348 : : List *fields;
349 : : RangeTblEntry *rte;
350 : :
351 : 44 : rte = rt_fetch(appinfo->parent_relid,
352 : : context->root->parse->rtable);
353 : 44 : fields = copyObject(appinfo->translated_vars);
354 : 44 : rowexpr = makeNode(RowExpr);
355 : 44 : rowexpr->args = fields;
356 : 44 : rowexpr->row_typeid = var->vartype;
357 : 44 : rowexpr->row_format = COERCE_IMPLICIT_CAST;
358 : 44 : rowexpr->colnames = copyObject(rte->eref->colnames);
359 : 44 : rowexpr->location = -1;
360 : :
588 dean.a.rasheed@gmail 361 [ - + ]: 44 : if (var->varreturningtype != VAR_RETURNING_DEFAULT)
588 dean.a.rasheed@gmail 362 [ # # ]:UBC 0 : elog(ERROR, "failed to apply returningtype to a non-Var");
1305 tgl@sss.pgh.pa.us 363 [ - + ]:CBC 44 : if (var->varnullingrels != NULL)
1305 tgl@sss.pgh.pa.us 364 [ # # ]:UBC 0 : elog(ERROR, "failed to apply nullingrels to a non-Var");
365 : :
2786 alvherre@alvh.no-ip. 366 :CBC 44 : return (Node *) rowexpr;
367 : : }
368 : : }
369 : : /* system attributes don't need any other translation */
370 : : }
1975 tgl@sss.pgh.pa.us 371 [ + + ]: 34534 : else if (var->varno == ROWID_VAR)
372 : : {
373 : : /*
374 : : * If it's a ROWID_VAR placeholder, see if we've reached a leaf
375 : : * target rel, for which we can translate the Var to a specific
376 : : * instantiation. We should never be asked to translate to a set
377 : : * of relids containing more than one leaf target rel, so the
378 : : * answer will be unique. If we're still considering non-leaf
379 : : * inheritance levels, return the ROWID_VAR Var as-is.
380 : : */
381 : 15670 : Relids leaf_result_relids = context->root->leaf_result_relids;
382 : 15670 : Index leaf_relid = 0;
383 : :
384 [ + + ]: 31340 : for (cnt = 0; cnt < nappinfos; cnt++)
385 : : {
386 [ + + ]: 15670 : if (bms_is_member(appinfos[cnt]->child_relid,
387 : : leaf_result_relids))
388 : : {
389 [ - + ]: 13804 : if (leaf_relid)
1975 tgl@sss.pgh.pa.us 390 [ # # ]:UBC 0 : elog(ERROR, "cannot translate to multiple leaf relids");
1975 tgl@sss.pgh.pa.us 391 :CBC 13804 : leaf_relid = appinfos[cnt]->child_relid;
392 : : }
393 : : }
394 : :
395 [ + + ]: 15670 : if (leaf_relid)
396 : : {
397 : : RowIdentityVarInfo *ridinfo = (RowIdentityVarInfo *)
1196 398 : 13804 : list_nth(context->root->row_identity_vars, var->varattno - 1);
399 : :
1975 400 [ + + ]: 13804 : if (bms_is_member(leaf_relid, ridinfo->rowidrels))
401 : : {
402 : : /* Substitute the Var given in the RowIdentityVarInfo */
403 : 13754 : var = copyObject(ridinfo->rowidvar);
404 : : /* ... but use the correct relid */
405 : 13754 : var->varno = leaf_relid;
406 : : /* identity vars shouldn't have nulling rels */
1305 407 [ - + ]: 13754 : Assert(var->varnullingrels == NULL);
408 : : /* varnosyn in the RowIdentityVarInfo is probably wrong */
1975 409 : 13754 : var->varnosyn = 0;
410 : 13754 : var->varattnosyn = 0;
411 : : }
412 : : else
413 : : {
414 : : /*
415 : : * This leaf rel can't return the desired value, so
416 : : * substitute a NULL of the correct type.
417 : : */
418 : 50 : return (Node *) makeNullConst(var->vartype,
419 : : var->vartypmod,
420 : : var->varcollid);
421 : : }
422 : : }
423 : : }
2786 alvherre@alvh.no-ip. 424 : 48446 : return (Node *) var;
425 : : }
426 [ + + ]: 568635 : if (IsA(node, CurrentOfExpr))
427 : : {
428 : 152 : CurrentOfExpr *cexpr = (CurrentOfExpr *) copyObject(node);
429 : :
430 [ + - ]: 152 : for (cnt = 0; cnt < nappinfos; cnt++)
431 : : {
432 : 152 : AppendRelInfo *appinfo = appinfos[cnt];
433 : :
434 [ + - ]: 152 : if (cexpr->cvarno == appinfo->parent_relid)
435 : : {
436 : 152 : cexpr->cvarno = appinfo->child_relid;
437 : 152 : break;
438 : : }
439 : : }
440 : 152 : return (Node *) cexpr;
441 : : }
442 [ + + ]: 568483 : if (IsA(node, PlaceHolderVar))
443 : : {
444 : : /* Copy the PlaceHolderVar node with correct mutation of subnodes */
445 : : PlaceHolderVar *phv;
446 : :
447 : 2692 : phv = (PlaceHolderVar *) expression_tree_mutator(node,
448 : : adjust_appendrel_attrs_mutator,
449 : : context);
450 : : /* now fix PlaceHolderVar's relid sets */
451 [ + - ]: 2692 : if (phv->phlevelsup == 0)
452 : : {
1305 tgl@sss.pgh.pa.us 453 : 2692 : phv->phrels = adjust_child_relids(phv->phrels,
454 : : nappinfos, appinfos);
455 : : /* as above, we needn't touch phnullingrels */
456 : : }
2786 alvherre@alvh.no-ip. 457 : 2692 : return (Node *) phv;
458 : : }
459 : : /* Shouldn't need to handle planner auxiliary nodes here */
460 [ - + ]: 565791 : Assert(!IsA(node, SpecialJoinInfo));
461 [ - + ]: 565791 : Assert(!IsA(node, AppendRelInfo));
462 [ - + ]: 565791 : Assert(!IsA(node, PlaceHolderInfo));
463 [ - + ]: 565791 : Assert(!IsA(node, MinMaxAggInfo));
464 : :
465 : : /*
466 : : * We have to process RestrictInfo nodes specially. (Note: although
467 : : * set_append_rel_pathlist will hide RestrictInfos in the parent's
468 : : * baserestrictinfo list from us, it doesn't hide those in joininfo.)
469 : : */
470 [ + + ]: 565791 : if (IsA(node, RestrictInfo))
471 : : {
472 : 33582 : RestrictInfo *oldinfo = (RestrictInfo *) node;
473 : 33582 : RestrictInfo *newinfo = makeNode(RestrictInfo);
474 : :
475 : : /* Copy all flat-copiable fields, notably including rinfo_serial */
476 : 33582 : memcpy(newinfo, oldinfo, sizeof(RestrictInfo));
477 : :
478 : : /* Recursively fix the clause itself */
479 : 33582 : newinfo->clause = (Expr *)
480 : 33582 : adjust_appendrel_attrs_mutator((Node *) oldinfo->clause, context);
481 : :
482 : : /* and the modified version, if an OR clause */
483 : 33582 : newinfo->orclause = (Expr *)
484 : 33582 : adjust_appendrel_attrs_mutator((Node *) oldinfo->orclause, context);
485 : :
486 : : /* adjust relid sets too */
487 : 33582 : newinfo->clause_relids = adjust_child_relids(oldinfo->clause_relids,
488 : : context->nappinfos,
489 : : context->appinfos);
490 : 33582 : newinfo->required_relids = adjust_child_relids(oldinfo->required_relids,
491 : : context->nappinfos,
492 : : context->appinfos);
493 : 33582 : newinfo->outer_relids = adjust_child_relids(oldinfo->outer_relids,
494 : : context->nappinfos,
495 : : context->appinfos);
496 : 33582 : newinfo->left_relids = adjust_child_relids(oldinfo->left_relids,
497 : : context->nappinfos,
498 : : context->appinfos);
499 : 33582 : newinfo->right_relids = adjust_child_relids(oldinfo->right_relids,
500 : : context->nappinfos,
501 : : context->appinfos);
502 : :
503 : : /*
504 : : * Reset cached derivative fields, since these might need to have
505 : : * different values when considering the child relation. Note we
506 : : * don't reset left_ec/right_ec: each child variable is implicitly
507 : : * equivalent to its parent, so still a member of the same EC if any.
508 : : */
509 : 33582 : newinfo->eval_cost.startup = -1;
510 : 33582 : newinfo->norm_selec = -1;
511 : 33582 : newinfo->outer_selec = -1;
512 : 33582 : newinfo->left_em = NULL;
513 : 33582 : newinfo->right_em = NULL;
514 : 33582 : newinfo->scansel_cache = NIL;
515 : 33582 : newinfo->left_bucketsize = -1;
516 : 33582 : newinfo->right_bucketsize = -1;
517 : 33582 : newinfo->left_mcvfreq = -1;
518 : 33582 : newinfo->right_mcvfreq = -1;
519 : :
520 : 33582 : return (Node *) newinfo;
521 : : }
522 : :
523 : : /*
524 : : * We have to process RelAggInfo nodes specially.
525 : : */
323 rguo@postgresql.org 526 [ + + ]: 532209 : if (IsA(node, RelAggInfo))
527 : : {
528 : 15460 : RelAggInfo *oldinfo = (RelAggInfo *) node;
529 : 15460 : RelAggInfo *newinfo = makeNode(RelAggInfo);
530 : :
531 : 15460 : newinfo->target = (PathTarget *)
532 : 15460 : adjust_appendrel_attrs_mutator((Node *) oldinfo->target,
533 : : context);
534 : :
535 : 15460 : newinfo->agg_input = (PathTarget *)
536 : 15460 : adjust_appendrel_attrs_mutator((Node *) oldinfo->agg_input,
537 : : context);
538 : :
539 : 15460 : newinfo->group_clauses = oldinfo->group_clauses;
540 : :
541 : 15460 : newinfo->group_exprs = (List *)
542 : 15460 : adjust_appendrel_attrs_mutator((Node *) oldinfo->group_exprs,
543 : : context);
544 : :
545 : 15460 : return (Node *) newinfo;
546 : : }
547 : :
548 : : /*
549 : : * We have to process PathTarget nodes specially.
550 : : */
551 [ + + ]: 516749 : if (IsA(node, PathTarget))
552 : : {
553 : 30920 : PathTarget *oldtarget = (PathTarget *) node;
554 : 30920 : PathTarget *newtarget = makeNode(PathTarget);
555 : :
556 : : /* Copy all flat-copiable fields */
557 : 30920 : memcpy(newtarget, oldtarget, sizeof(PathTarget));
558 : :
559 : 30920 : newtarget->exprs = (List *)
560 : 30920 : adjust_appendrel_attrs_mutator((Node *) oldtarget->exprs,
561 : : context);
562 : :
563 [ + - ]: 30920 : if (oldtarget->sortgrouprefs)
564 : : {
565 : 30920 : Size nbytes = list_length(oldtarget->exprs) * sizeof(Index);
566 : :
567 : 30920 : newtarget->sortgrouprefs = (Index *) palloc(nbytes);
568 : 30920 : memcpy(newtarget->sortgrouprefs, oldtarget->sortgrouprefs, nbytes);
569 : : }
570 : :
571 : 30920 : return (Node *) newtarget;
572 : : }
573 : :
574 : : /*
575 : : * NOTE: we do not need to recurse into sublinks, because they should
576 : : * already have been converted to subplans before we see them.
577 : : */
2786 alvherre@alvh.no-ip. 578 [ - + ]: 485829 : Assert(!IsA(node, SubLink));
579 [ - + ]: 485829 : Assert(!IsA(node, Query));
580 : : /* We should never see these Query substructures, either. */
1975 tgl@sss.pgh.pa.us 581 [ - + ]: 485829 : Assert(!IsA(node, RangeTblRef));
582 [ - + ]: 485829 : Assert(!IsA(node, JoinExpr));
583 : :
637 peter@eisentraut.org 584 : 485829 : return expression_tree_mutator(node, adjust_appendrel_attrs_mutator, context);
585 : : }
586 : :
587 : : /*
588 : : * adjust_appendrel_attrs_multilevel
589 : : * Apply Var translations from an appendrel parent down to a child.
590 : : *
591 : : * Replace Vars in the "node" expression that reference "parentrel" with
592 : : * the appropriate Vars for "childrel". childrel can be more than one
593 : : * inheritance level removed from parentrel.
594 : : */
595 : : Node *
2780 alvherre@alvh.no-ip. 596 : 39528 : adjust_appendrel_attrs_multilevel(PlannerInfo *root, Node *node,
597 : : RelOptInfo *childrel,
598 : : RelOptInfo *parentrel)
599 : : {
600 : : AppendRelInfo **appinfos;
601 : : int nappinfos;
602 : :
603 : : /* Recurse if immediate parent is not the top parent. */
1470 tgl@sss.pgh.pa.us 604 [ + + ]: 39528 : if (childrel->parent != parentrel)
605 : : {
606 [ + - ]: 13584 : if (childrel->parent)
607 : 13584 : node = adjust_appendrel_attrs_multilevel(root, node,
608 : 13584 : childrel->parent,
609 : : parentrel);
610 : : else
1470 tgl@sss.pgh.pa.us 611 [ # # ]:UBC 0 : elog(ERROR, "childrel is not a child of parentrel");
612 : : }
613 : :
614 : : /* Now translate for this child. */
1470 tgl@sss.pgh.pa.us 615 :CBC 39528 : appinfos = find_appinfos_by_relids(root, childrel->relids, &nappinfos);
616 : :
2780 alvherre@alvh.no-ip. 617 : 39528 : node = adjust_appendrel_attrs(root, node, nappinfos, appinfos);
618 : :
619 : 39528 : pfree(appinfos);
620 : :
621 : 39528 : return node;
622 : : }
623 : :
624 : : /*
625 : : * Substitute child relids for parent relids in a Relid set. The array of
626 : : * appinfos specifies the substitutions to be performed.
627 : : */
628 : : Relids
2786 629 : 213767 : adjust_child_relids(Relids relids, int nappinfos, AppendRelInfo **appinfos)
630 : : {
631 : 213767 : Bitmapset *result = NULL;
632 : : int cnt;
633 : :
634 [ + + ]: 585109 : for (cnt = 0; cnt < nappinfos; cnt++)
635 : : {
636 : 371342 : AppendRelInfo *appinfo = appinfos[cnt];
637 : :
638 : : /* Remove parent, add child */
639 [ + + ]: 371342 : if (bms_is_member(appinfo->parent_relid, relids))
640 : : {
641 : : /* Make a copy if we are changing the set. */
642 [ + + ]: 249118 : if (!result)
643 : 170453 : result = bms_copy(relids);
644 : :
645 : 249118 : result = bms_del_member(result, appinfo->parent_relid);
646 : 249118 : result = bms_add_member(result, appinfo->child_relid);
647 : : }
648 : : }
649 : :
650 : : /* If we made any changes, return the modified copy. */
651 [ + + ]: 213767 : if (result)
652 : 170453 : return result;
653 : :
654 : : /* Otherwise, return the original set without modification. */
655 : 43314 : return relids;
656 : : }
657 : :
658 : : /*
659 : : * Substitute child's relids for parent's relids in a Relid set.
660 : : * The childrel can be multiple inheritance levels below the parent.
661 : : */
662 : : Relids
663 : 967 : adjust_child_relids_multilevel(PlannerInfo *root, Relids relids,
664 : : RelOptInfo *childrel,
665 : : RelOptInfo *parentrel)
666 : : {
667 : : AppendRelInfo **appinfos;
668 : : int nappinfos;
669 : :
670 : : /*
671 : : * If the given relids set doesn't contain any of the parent relids, it
672 : : * will remain unchanged.
673 : : */
1470 tgl@sss.pgh.pa.us 674 [ - + ]: 967 : if (!bms_overlap(relids, parentrel->relids))
2786 alvherre@alvh.no-ip. 675 :UBC 0 : return relids;
676 : :
677 : : /* Recurse if immediate parent is not the top parent. */
1470 tgl@sss.pgh.pa.us 678 [ + + ]:CBC 967 : if (childrel->parent != parentrel)
679 : : {
680 [ + - ]: 120 : if (childrel->parent)
681 : 120 : relids = adjust_child_relids_multilevel(root, relids,
682 : 120 : childrel->parent,
683 : : parentrel);
684 : : else
1470 tgl@sss.pgh.pa.us 685 [ # # ]:UBC 0 : elog(ERROR, "childrel is not a child of parentrel");
686 : : }
687 : :
688 : : /* Now translate for this child. */
1470 tgl@sss.pgh.pa.us 689 :CBC 967 : appinfos = find_appinfos_by_relids(root, childrel->relids, &nappinfos);
690 : :
691 : 967 : relids = adjust_child_relids(relids, nappinfos, appinfos);
692 : :
2786 alvherre@alvh.no-ip. 693 : 967 : pfree(appinfos);
694 : :
1470 tgl@sss.pgh.pa.us 695 : 967 : return relids;
696 : : }
697 : :
698 : : /*
699 : : * adjust_inherited_attnums
700 : : * Translate an integer list of attribute numbers from parent to child.
701 : : */
702 : : List *
1975 703 : 4149 : adjust_inherited_attnums(List *attnums, AppendRelInfo *context)
704 : : {
705 : 4149 : List *result = NIL;
706 : : ListCell *lc;
707 : :
708 : : /* This should only happen for an inheritance case, not UNION ALL */
2786 alvherre@alvh.no-ip. 709 [ - + ]: 4149 : Assert(OidIsValid(context->parent_reloid));
710 : :
711 : : /* Look up each attribute in the AppendRelInfo's translated_vars list */
1975 tgl@sss.pgh.pa.us 712 [ + - + + : 9369 : foreach(lc, attnums)
+ + ]
713 : : {
714 : 5220 : AttrNumber parentattno = lfirst_int(lc);
715 : : Var *childvar;
716 : :
717 : : /* Look up the translation of this column: it must be a Var */
718 [ + - - + ]: 10440 : if (parentattno <= 0 ||
719 : 5220 : parentattno > list_length(context->translated_vars))
2786 alvherre@alvh.no-ip. 720 [ # # ]:UBC 0 : elog(ERROR, "attribute %d of relation \"%s\" does not exist",
721 : : parentattno, get_rel_name(context->parent_reloid));
1975 tgl@sss.pgh.pa.us 722 :CBC 5220 : childvar = (Var *) list_nth(context->translated_vars, parentattno - 1);
2786 alvherre@alvh.no-ip. 723 [ + - - + ]: 5220 : if (childvar == NULL || !IsA(childvar, Var))
2786 alvherre@alvh.no-ip. 724 [ # # ]:UBC 0 : elog(ERROR, "attribute %d of relation \"%s\" does not exist",
725 : : parentattno, get_rel_name(context->parent_reloid));
726 : :
1975 tgl@sss.pgh.pa.us 727 :CBC 5220 : result = lappend_int(result, childvar->varattno);
728 : : }
729 : 4149 : return result;
730 : : }
731 : :
732 : : /*
733 : : * adjust_inherited_attnums_multilevel
734 : : * As above, but traverse multiple inheritance levels as needed.
735 : : */
736 : : List *
737 : 4149 : adjust_inherited_attnums_multilevel(PlannerInfo *root, List *attnums,
738 : : Index child_relid, Index top_parent_relid)
739 : : {
740 : 4149 : AppendRelInfo *appinfo = root->append_rel_array[child_relid];
741 : :
742 [ - + ]: 4149 : if (!appinfo)
1975 tgl@sss.pgh.pa.us 743 [ # # ]:UBC 0 : elog(ERROR, "child rel %d not found in append_rel_array", child_relid);
744 : :
745 : : /* Recurse if immediate parent is not the top parent. */
1975 tgl@sss.pgh.pa.us 746 [ + + ]:CBC 4149 : if (appinfo->parent_relid != top_parent_relid)
747 : 677 : attnums = adjust_inherited_attnums_multilevel(root, attnums,
748 : : appinfo->parent_relid,
749 : : top_parent_relid);
750 : :
751 : : /* Now translate for this child */
752 : 4149 : return adjust_inherited_attnums(attnums, appinfo);
753 : : }
754 : :
755 : : /*
756 : : * get_translated_update_targetlist
757 : : * Get the processed_tlist of an UPDATE query, translated as needed to
758 : : * match a child target relation.
759 : : *
760 : : * Optionally also return the list of target column numbers translated
761 : : * to this target relation. (The resnos in processed_tlist MUST NOT be
762 : : * relied on for this purpose.)
763 : : */
764 : : void
765 : 57 : get_translated_update_targetlist(PlannerInfo *root, Index relid,
766 : : List **processed_tlist, List **update_colnos)
767 : : {
768 : : /* This is pretty meaningless for commands other than UPDATE. */
769 [ - + ]: 57 : Assert(root->parse->commandType == CMD_UPDATE);
770 [ + + ]: 57 : if (relid == root->parse->resultRelation)
771 : : {
772 : : /*
773 : : * Non-inheritance case, so it's easy. The caller might be expecting
774 : : * a tree it can scribble on, though, so copy.
775 : : */
776 : 37 : *processed_tlist = copyObject(root->processed_tlist);
777 [ + - ]: 37 : if (update_colnos)
778 : 37 : *update_colnos = copyObject(root->update_colnos);
779 : : }
780 : : else
781 : : {
782 [ - + ]: 20 : Assert(bms_is_member(relid, root->all_result_relids));
783 : 20 : *processed_tlist = (List *)
784 : 20 : adjust_appendrel_attrs_multilevel(root,
785 : 20 : (Node *) root->processed_tlist,
786 : : find_base_rel(root, relid),
1470 787 : 20 : find_base_rel(root, root->parse->resultRelation));
1975 788 [ + - ]: 20 : if (update_colnos)
789 : 20 : *update_colnos =
790 : 20 : adjust_inherited_attnums_multilevel(root, root->update_colnos,
791 : : relid,
792 : 20 : root->parse->resultRelation);
793 : : }
2786 alvherre@alvh.no-ip. 794 : 57 : }
795 : :
796 : : /*
797 : : * find_appinfos_by_relids
798 : : * Find AppendRelInfo structures for base relations listed in relids.
799 : : *
800 : : * The relids argument is typically a join relation's relids, which can
801 : : * include outer-join RT indexes in addition to baserels. We silently
802 : : * ignore the outer joins.
803 : : *
804 : : * The AppendRelInfos are returned in an array, which can be pfree'd by the
805 : : * caller. *nappinfos is set to the number of entries in the array.
806 : : */
807 : : AppendRelInfo **
808 : 83234 : find_appinfos_by_relids(PlannerInfo *root, Relids relids, int *nappinfos)
809 : : {
810 : : AppendRelInfo **appinfos;
811 : 83234 : int cnt = 0;
812 : : int i;
813 : :
814 : : /* Allocate an array that's certainly big enough */
260 michael@paquier.xyz 815 : 83234 : appinfos = palloc_array(AppendRelInfo *, bms_num_members(relids));
816 : :
2786 alvherre@alvh.no-ip. 817 : 83234 : i = -1;
818 [ + + ]: 209653 : while ((i = bms_next_member(relids, i)) >= 0)
819 : : {
820 : 126419 : AppendRelInfo *appinfo = root->append_rel_array[i];
821 : :
822 [ + + ]: 126419 : if (!appinfo)
823 : : {
824 : : /* Probably i is an OJ index, but let's check */
1305 tgl@sss.pgh.pa.us 825 [ + - ]: 2976 : if (find_base_rel_ignore_join(root, i) == NULL)
826 : 2976 : continue;
827 : : /* It's a base rel, but we lack an append_rel_array entry */
2786 alvherre@alvh.no-ip. 828 [ # # ]:UBC 0 : elog(ERROR, "child rel %d not found in append_rel_array", i);
829 : : }
830 : :
2786 alvherre@alvh.no-ip. 831 :CBC 123443 : appinfos[cnt++] = appinfo;
832 : : }
1305 tgl@sss.pgh.pa.us 833 : 83234 : *nappinfos = cnt;
2786 alvherre@alvh.no-ip. 834 : 83234 : return appinfos;
835 : : }
836 : :
837 : :
838 : : /*****************************************************************************
839 : : *
840 : : * ROW-IDENTITY VARIABLE MANAGEMENT
841 : : *
842 : : * This code lacks a good home, perhaps. We choose to keep it here because
843 : : * adjust_appendrel_attrs_mutator() is its principal co-conspirator. That
844 : : * function does most of what is needed to expand ROWID_VAR Vars into the
845 : : * right things.
846 : : *
847 : : *****************************************************************************/
848 : :
849 : : /*
850 : : * add_row_identity_var
851 : : * Register a row-identity column to be used in UPDATE/DELETE/MERGE.
852 : : *
853 : : * The Var must be equal(), aside from varno, to any other row-identity
854 : : * column with the same rowid_name. Thus, for example, "wholerow"
855 : : * row identities had better use vartype == RECORDOID.
856 : : *
857 : : * rtindex is currently redundant with rowid_var->varno, but we specify
858 : : * it as a separate parameter in case this is ever generalized to support
859 : : * non-Var expressions. (We could reasonably handle expressions over
860 : : * Vars of the specified rtindex, but for now that seems unnecessary.)
861 : : */
862 : : void
1975 tgl@sss.pgh.pa.us 863 : 22287 : add_row_identity_var(PlannerInfo *root, Var *orig_var,
864 : : Index rtindex, const char *rowid_name)
865 : : {
866 : : TargetEntry *tle;
867 : : Var *rowid_var;
868 : : RowIdentityVarInfo *ridinfo;
869 : : ListCell *lc;
870 : :
871 : : /* For now, the argument must be just a Var of the given rtindex */
872 [ - + ]: 22287 : Assert(IsA(orig_var, Var));
873 [ - + ]: 22287 : Assert(orig_var->varno == rtindex);
874 [ - + ]: 22287 : Assert(orig_var->varlevelsup == 0);
1305 875 [ - + ]: 22287 : Assert(orig_var->varnullingrels == NULL);
876 : :
877 : : /*
878 : : * If we're doing non-inherited UPDATE/DELETE/MERGE, there's little need
879 : : * for ROWID_VAR shenanigans. Just shove the presented Var into the
880 : : * processed_tlist, and we're done.
881 : : */
1975 882 [ + + ]: 22287 : if (rtindex == root->parse->resultRelation)
883 : : {
884 : 13283 : tle = makeTargetEntry((Expr *) orig_var,
885 : 13283 : list_length(root->processed_tlist) + 1,
886 : : pstrdup(rowid_name),
887 : : true);
888 : 13283 : root->processed_tlist = lappend(root->processed_tlist, tle);
889 : 13283 : return;
890 : : }
891 : :
892 : : /*
893 : : * Otherwise, rtindex should reference a leaf target relation that's being
894 : : * added to the query during expand_inherited_rtentry().
895 : : */
896 [ - + ]: 9004 : Assert(bms_is_member(rtindex, root->leaf_result_relids));
897 [ - + ]: 9004 : Assert(root->append_rel_array[rtindex] != NULL);
898 : :
899 : : /*
900 : : * We have to find a matching RowIdentityVarInfo, or make one if there is
901 : : * none. To allow using equal() to match the vars, change the varno to
902 : : * ROWID_VAR, leaving all else alone.
903 : : */
904 : 9004 : rowid_var = copyObject(orig_var);
905 : : /* This could eventually become ChangeVarNodes() */
906 : 9004 : rowid_var->varno = ROWID_VAR;
907 : :
908 : : /* Look for an existing row-id column of the same name */
909 [ + + + + : 13573 : foreach(lc, root->row_identity_vars)
+ + ]
910 : : {
911 : 8903 : ridinfo = (RowIdentityVarInfo *) lfirst(lc);
912 [ + + ]: 8903 : if (strcmp(rowid_name, ridinfo->rowidname) != 0)
913 : 4569 : continue;
914 [ + - ]: 4334 : if (equal(rowid_var, ridinfo->rowidvar))
915 : : {
916 : : /* Found a match; we need only record that rtindex needs it too */
917 : 4334 : ridinfo->rowidrels = bms_add_member(ridinfo->rowidrels, rtindex);
918 : 4334 : return;
919 : : }
920 : : else
921 : : {
922 : : /* Ooops, can't handle this */
1975 tgl@sss.pgh.pa.us 923 [ # # ]:UBC 0 : elog(ERROR, "conflicting uses of row-identity name \"%s\"",
924 : : rowid_name);
925 : : }
926 : : }
927 : :
928 : : /* No request yet, so add a new RowIdentityVarInfo */
1975 tgl@sss.pgh.pa.us 929 :CBC 4670 : ridinfo = makeNode(RowIdentityVarInfo);
930 : 4670 : ridinfo->rowidvar = copyObject(rowid_var);
931 : : /* for the moment, estimate width using just the datatype info */
932 : 4670 : ridinfo->rowidwidth = get_typavgwidth(exprType((Node *) rowid_var),
933 : : exprTypmod((Node *) rowid_var));
934 : 4670 : ridinfo->rowidname = pstrdup(rowid_name);
935 : 4670 : ridinfo->rowidrels = bms_make_singleton(rtindex);
936 : :
937 : 4670 : root->row_identity_vars = lappend(root->row_identity_vars, ridinfo);
938 : :
939 : : /* Change rowid_var into a reference to this row_identity_vars entry */
940 : 4670 : rowid_var->varattno = list_length(root->row_identity_vars);
941 : :
942 : : /* Push the ROWID_VAR reference variable into processed_tlist */
943 : 4670 : tle = makeTargetEntry((Expr *) rowid_var,
944 : 4670 : list_length(root->processed_tlist) + 1,
945 : : pstrdup(rowid_name),
946 : : true);
947 : 4670 : root->processed_tlist = lappend(root->processed_tlist, tle);
948 : : }
949 : :
950 : : /*
951 : : * add_row_identity_columns
952 : : *
953 : : * This function adds the row identity columns needed by the core code.
954 : : * FDWs might call add_row_identity_var() for themselves to add nonstandard
955 : : * columns. (Duplicate requests are fine.)
956 : : */
957 : : void
958 : 17931 : add_row_identity_columns(PlannerInfo *root, Index rtindex,
959 : : RangeTblEntry *target_rte,
960 : : Relation target_relation)
961 : : {
962 : 17931 : CmdType commandType = root->parse->commandType;
963 : 17931 : char relkind = target_relation->rd_rel->relkind;
964 : : Var *var;
965 : :
1613 alvherre@alvh.no-ip. 966 [ + + + + : 17931 : Assert(commandType == CMD_UPDATE || commandType == CMD_DELETE || commandType == CMD_MERGE);
- + ]
967 : :
910 dean.a.rasheed@gmail 968 [ + + + + ]: 17931 : if (relkind == RELKIND_RELATION ||
1975 tgl@sss.pgh.pa.us 969 [ + + ]: 479 : relkind == RELKIND_MATVIEW ||
970 : : relkind == RELKIND_PARTITIONED_TABLE)
971 : : {
972 : : /*
973 : : * Emit CTID so that executor can find the row to merge, update or
974 : : * delete.
975 : : */
976 : 17485 : var = makeVar(rtindex,
977 : : SelfItemPointerAttributeNumber,
978 : : TIDOID,
979 : : -1,
980 : : InvalidOid,
981 : : 0);
982 : 17485 : add_row_identity_var(root, var, rtindex, "ctid");
983 : : }
984 [ + + ]: 446 : else if (relkind == RELKIND_FOREIGN_TABLE)
985 : : {
986 : : /*
987 : : * Let the foreign table's FDW add whatever junk TLEs it wants.
988 : : */
989 : : FdwRoutine *fdwroutine;
990 : :
991 : 205 : fdwroutine = GetFdwRoutineForRelation(target_relation, false);
992 : :
993 [ + + ]: 205 : if (fdwroutine->AddForeignUpdateTargets != NULL)
994 : 198 : fdwroutine->AddForeignUpdateTargets(root, rtindex,
995 : : target_rte, target_relation);
996 : :
997 : : /*
998 : : * For UPDATE, we need to make the FDW fetch unchanged columns by
999 : : * asking it to fetch a whole-row Var. That's because the top-level
1000 : : * targetlist only contains entries for changed columns, but
1001 : : * ExecUpdate will need to build the complete new tuple. (Actually,
1002 : : * we only really need this in UPDATEs that are not pushed to the
1003 : : * remote side, but it's hard to tell if that will be the case at the
1004 : : * point when this function is called.)
1005 : : *
1006 : : * We will also need the whole row if there are any row triggers, so
1007 : : * that the executor will have the "old" row to pass to the trigger.
1008 : : * Alas, this misses system columns.
1009 : : */
1010 [ + + ]: 205 : if (commandType == CMD_UPDATE ||
1011 [ + + ]: 90 : (target_relation->trigdesc &&
1012 [ + + ]: 15 : (target_relation->trigdesc->trig_delete_after_row ||
1013 [ + + ]: 9 : target_relation->trigdesc->trig_delete_before_row)))
1014 : : {
1015 : 123 : var = makeVar(rtindex,
1016 : : InvalidAttrNumber,
1017 : : RECORDOID,
1018 : : -1,
1019 : : InvalidOid,
1020 : : 0);
1021 : 123 : add_row_identity_var(root, var, rtindex, "wholerow");
1022 : : }
1023 : : }
1024 : 17931 : }
1025 : :
1026 : : /*
1027 : : * distribute_row_identity_vars
1028 : : *
1029 : : * After we have finished identifying all the row identity columns
1030 : : * needed by an inherited UPDATE/DELETE/MERGE query, make sure that
1031 : : * these columns will be generated by all the target relations.
1032 : : *
1033 : : * This is more or less like what build_base_rel_tlists() does,
1034 : : * except that it would not understand what to do with ROWID_VAR Vars.
1035 : : * Since that function runs before inheritance relations are expanded,
1036 : : * it will never see any such Vars anyway.
1037 : : */
1038 : : void
1039 : 247877 : distribute_row_identity_vars(PlannerInfo *root)
1040 : : {
1041 : 247877 : Query *parse = root->parse;
1042 : 247877 : int result_relation = parse->resultRelation;
1043 : : RangeTblEntry *target_rte;
1044 : : RelOptInfo *target_rel;
1045 : : ListCell *lc;
1046 : :
1047 : : /*
1048 : : * There's nothing to do if this isn't an inherited UPDATE/DELETE/MERGE.
1049 : : */
1613 alvherre@alvh.no-ip. 1050 [ + + + + ]: 247877 : if (parse->commandType != CMD_UPDATE && parse->commandType != CMD_DELETE &&
1051 [ + + ]: 233598 : parse->commandType != CMD_MERGE)
1052 : : {
1975 tgl@sss.pgh.pa.us 1053 [ - + ]: 232119 : Assert(root->row_identity_vars == NIL);
1054 : 232119 : return;
1055 : : }
1056 : 15758 : target_rte = rt_fetch(result_relation, parse->rtable);
1057 [ + + ]: 15758 : if (!target_rte->inh)
1058 : : {
1059 [ - + ]: 13416 : Assert(root->row_identity_vars == NIL);
1060 : 13416 : return;
1061 : : }
1062 : :
1063 : : /*
1064 : : * Ordinarily, we expect that leaf result relation(s) will have added some
1065 : : * ROWID_VAR Vars to the query. However, it's possible that constraint
1066 : : * exclusion suppressed every leaf relation. The executor will get upset
1067 : : * if the plan has no row identity columns at all, even though it will
1068 : : * certainly process no rows. Handle this edge case by re-opening the top
1069 : : * result relation and adding the row identity columns it would have used,
1070 : : * as preprocess_targetlist() would have done if it weren't marked "inh".
1071 : : * Then re-run build_base_rel_tlists() to ensure that the added columns
1072 : : * get propagated to the relation's reltarget. (This is a bit ugly, but
1073 : : * it seems better to confine the ugliness and extra cycles to this
1074 : : * unusual corner case.)
1075 : : */
1076 [ + + ]: 2342 : if (root->row_identity_vars == NIL)
1077 : : {
1078 : : Relation target_relation;
1079 : :
1080 : 28 : target_relation = table_open(target_rte->relid, NoLock);
1081 : 28 : add_row_identity_columns(root, result_relation,
1082 : : target_rte, target_relation);
1083 : 28 : table_close(target_relation, NoLock);
1248 1084 : 28 : build_base_rel_tlists(root, root->processed_tlist);
1085 : : /* There are no ROWID_VAR Vars in this case, so we're done. */
1975 1086 : 28 : return;
1087 : : }
1088 : :
1089 : : /*
1090 : : * Dig through the processed_tlist to find the ROWID_VAR reference Vars,
1091 : : * and forcibly copy them into the reltarget list of the topmost target
1092 : : * relation. That's sufficient because they'll be copied to the
1093 : : * individual leaf target rels (with appropriate translation) later,
1094 : : * during appendrel expansion --- see set_append_rel_size().
1095 : : */
1096 : 2314 : target_rel = find_base_rel(root, result_relation);
1097 : :
1098 [ + - + + : 9751 : foreach(lc, root->processed_tlist)
+ + ]
1099 : : {
1100 : 7437 : TargetEntry *tle = lfirst(lc);
1101 : 7437 : Var *var = (Var *) tle->expr;
1102 : :
1103 [ + - + + : 7437 : if (var && IsA(var, Var) && var->varno == ROWID_VAR)
+ + ]
1104 : : {
1105 : 4670 : target_rel->reltarget->exprs =
1106 : 4670 : lappend(target_rel->reltarget->exprs, copyObject(var));
1107 : : /* reltarget cost and width will be computed later */
1108 : : }
1109 : : }
1110 : : }
|
