Branch data Line data Source code
1 : : /*-------------------------------------------------------------------------
2 : : *
3 : : * nodeFuncs.c
4 : : * Various general-purpose manipulations of Node trees
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/nodes/nodeFuncs.c
12 : : *
13 : : *-------------------------------------------------------------------------
14 : : */
15 : : #include "postgres.h"
16 : :
17 : : #include "catalog/pg_collation.h"
18 : : #include "catalog/pg_type.h"
19 : : #include "miscadmin.h"
20 : : #include "nodes/execnodes.h"
21 : : #include "nodes/nodeFuncs.h"
22 : : #include "nodes/pathnodes.h"
23 : : #include "utils/builtins.h"
24 : : #include "utils/lsyscache.h"
25 : :
26 : : static bool expression_returns_set_walker(Node *node, void *context);
27 : : static int leftmostLoc(int loc1, int loc2);
28 : : static bool fix_opfuncids_walker(Node *node, void *context);
29 : : static bool planstate_walk_subplans(List *plans,
30 : : planstate_tree_walker_callback walker,
31 : : void *context);
32 : : static bool planstate_walk_members(PlanState **planstates, int nplans,
33 : : planstate_tree_walker_callback walker,
34 : : void *context);
35 : :
36 : :
37 : : /*
38 : : * exprType -
39 : : * returns the Oid of the type of the expression's result.
40 : : */
41 : : Oid
42 : 5642793 : exprType(const Node *expr)
43 : : {
44 : 5642793 : Oid type;
45 : :
46 [ + + ]: 5642793 : if (!expr)
47 : 77 : return InvalidOid;
48 : :
49 [ + + + + : 5642716 : switch (nodeTag(expr))
+ + + + +
+ + + + +
+ + + + +
+ + + + +
+ + + + -
+ + + + +
+ + + + +
+ + + + +
+ + + + +
- ]
50 : : {
51 : : case T_Var:
52 : 2742269 : type = ((const Var *) expr)->vartype;
53 : 2742269 : break;
54 : : case T_Const:
55 : 2014726 : type = ((const Const *) expr)->consttype;
56 : 2014726 : break;
57 : : case T_Param:
58 : 272518 : type = ((const Param *) expr)->paramtype;
59 : 272518 : break;
60 : : case T_Aggref:
61 : 39541 : type = ((const Aggref *) expr)->aggtype;
62 : 39541 : break;
63 : : case T_GroupingFunc:
64 : 385 : type = INT4OID;
65 : 385 : break;
66 : : case T_WindowFunc:
67 : 3442 : type = ((const WindowFunc *) expr)->wintype;
68 : 3442 : break;
69 : : case T_MergeSupportFunc:
70 : 160 : type = ((const MergeSupportFunc *) expr)->msftype;
71 : 160 : break;
72 : : case T_SubscriptingRef:
73 : 19951 : type = ((const SubscriptingRef *) expr)->refrestype;
74 : 19951 : break;
75 : : case T_FuncExpr:
76 : 186879 : type = ((const FuncExpr *) expr)->funcresulttype;
77 : 186879 : break;
78 : : case T_NamedArgExpr:
79 : 649 : type = exprType((Node *) ((const NamedArgExpr *) expr)->arg);
80 : 649 : break;
81 : : case T_OpExpr:
82 : 140219 : type = ((const OpExpr *) expr)->opresulttype;
83 : 140219 : break;
84 : : case T_DistinctExpr:
85 : 104 : type = ((const DistinctExpr *) expr)->opresulttype;
86 : 104 : break;
87 : : case T_NullIfExpr:
88 : 240 : type = ((const NullIfExpr *) expr)->opresulttype;
89 : 240 : break;
90 : : case T_ScalarArrayOpExpr:
91 : 12915 : type = BOOLOID;
92 : 12915 : break;
93 : : case T_BoolExpr:
94 : 27037 : type = BOOLOID;
95 : 27037 : break;
96 : : case T_SubLink:
97 : : {
98 : 10739 : const SubLink *sublink = (const SubLink *) expr;
99 : :
100 [ + + + + ]: 10739 : if (sublink->subLinkType == EXPR_SUBLINK ||
101 : 4619 : sublink->subLinkType == ARRAY_SUBLINK)
102 : : {
103 : : /* get the type of the subselect's first target column */
104 : 8163 : Query *qtree = (Query *) sublink->subselect;
105 : 8163 : TargetEntry *tent;
106 : :
107 [ + - ]: 8163 : if (!qtree || !IsA(qtree, Query))
108 [ # # # # ]: 0 : elog(ERROR, "cannot get type for untransformed sublink");
109 : 8163 : tent = linitial_node(TargetEntry, qtree->targetList);
110 [ + - ]: 8163 : Assert(!tent->resjunk);
111 : 8163 : type = exprType((Node *) tent->expr);
112 [ + + ]: 8163 : if (sublink->subLinkType == ARRAY_SUBLINK)
113 : : {
114 : 2043 : type = get_promoted_array_type(type);
115 [ + - ]: 2043 : if (!OidIsValid(type))
116 [ # # # # ]: 0 : ereport(ERROR,
117 : : (errcode(ERRCODE_UNDEFINED_OBJECT),
118 : : errmsg("could not find array type for data type %s",
119 : : format_type_be(exprType((Node *) tent->expr)))));
120 : 2043 : }
121 : 8163 : }
122 [ + + ]: 2576 : else if (sublink->subLinkType == MULTIEXPR_SUBLINK)
123 : : {
124 : : /* MULTIEXPR is always considered to return RECORD */
125 : 22 : type = RECORDOID;
126 : 22 : }
127 : : else
128 : : {
129 : : /* for all other sublink types, result is boolean */
130 : 2554 : type = BOOLOID;
131 : : }
132 : 10739 : }
133 : 10739 : break;
134 : : case T_SubPlan:
135 : : {
136 : 5874 : const SubPlan *subplan = (const SubPlan *) expr;
137 : :
138 [ + + + + ]: 5874 : if (subplan->subLinkType == EXPR_SUBLINK ||
139 : 359 : subplan->subLinkType == ARRAY_SUBLINK)
140 : : {
141 : : /* get the type of the subselect's first target column */
142 : 5571 : type = subplan->firstColType;
143 [ + + ]: 5571 : if (subplan->subLinkType == ARRAY_SUBLINK)
144 : : {
145 : 56 : type = get_promoted_array_type(type);
146 [ + - ]: 56 : if (!OidIsValid(type))
147 [ # # # # ]: 0 : ereport(ERROR,
148 : : (errcode(ERRCODE_UNDEFINED_OBJECT),
149 : : errmsg("could not find array type for data type %s",
150 : : format_type_be(subplan->firstColType))));
151 : 56 : }
152 : 5571 : }
153 [ + + ]: 303 : else if (subplan->subLinkType == MULTIEXPR_SUBLINK)
154 : : {
155 : : /* MULTIEXPR is always considered to return RECORD */
156 : 27 : type = RECORDOID;
157 : 27 : }
158 : : else
159 : : {
160 : : /* for all other subplan types, result is boolean */
161 : 276 : type = BOOLOID;
162 : : }
163 : 5874 : }
164 : 5874 : break;
165 : : case T_AlternativeSubPlan:
166 : : {
167 : 82 : const AlternativeSubPlan *asplan = (const AlternativeSubPlan *) expr;
168 : :
169 : : /* subplans should all return the same thing */
170 : 82 : type = exprType((Node *) linitial(asplan->subplans));
171 : 82 : }
172 : 82 : break;
173 : : case T_FieldSelect:
174 : 1959 : type = ((const FieldSelect *) expr)->resulttype;
175 : 1959 : break;
176 : : case T_FieldStore:
177 : 184 : type = ((const FieldStore *) expr)->resulttype;
178 : 184 : break;
179 : : case T_RelabelType:
180 : 59203 : type = ((const RelabelType *) expr)->resulttype;
181 : 59203 : break;
182 : : case T_CoerceViaIO:
183 : 21320 : type = ((const CoerceViaIO *) expr)->resulttype;
184 : 21320 : break;
185 : : case T_ArrayCoerceExpr:
186 : 1626 : type = ((const ArrayCoerceExpr *) expr)->resulttype;
187 : 1626 : break;
188 : : case T_ConvertRowtypeExpr:
189 : 330 : type = ((const ConvertRowtypeExpr *) expr)->resulttype;
190 : 330 : break;
191 : : case T_CollateExpr:
192 : 1684 : type = exprType((Node *) ((const CollateExpr *) expr)->arg);
193 : 1684 : break;
194 : : case T_CaseExpr:
195 : 29849 : type = ((const CaseExpr *) expr)->casetype;
196 : 29849 : break;
197 : : case T_CaseTestExpr:
198 : 4954 : type = ((const CaseTestExpr *) expr)->typeId;
199 : 4954 : break;
200 : : case T_ArrayExpr:
201 : 7869 : type = ((const ArrayExpr *) expr)->array_typeid;
202 : 7869 : break;
203 : : case T_RowExpr:
204 : 2172 : type = ((const RowExpr *) expr)->row_typeid;
205 : 2172 : break;
206 : : case T_RowCompareExpr:
207 : 74 : type = BOOLOID;
208 : 74 : break;
209 : : case T_CoalesceExpr:
210 : 2862 : type = ((const CoalesceExpr *) expr)->coalescetype;
211 : 2862 : break;
212 : : case T_MinMaxExpr:
213 : 376 : type = ((const MinMaxExpr *) expr)->minmaxtype;
214 : 376 : break;
215 : : case T_SQLValueFunction:
216 : 1190 : type = ((const SQLValueFunction *) expr)->type;
217 : 1190 : break;
218 : : case T_XmlExpr:
219 [ + + ]: 4327 : if (((const XmlExpr *) expr)->op == IS_DOCUMENT)
220 : 17 : type = BOOLOID;
221 [ + + ]: 4310 : else if (((const XmlExpr *) expr)->op == IS_XMLSERIALIZE)
222 : 149 : type = TEXTOID;
223 : : else
224 : 4161 : type = XMLOID;
225 : 4327 : break;
226 : : case T_JsonValueExpr:
227 : : {
228 : 251 : const JsonValueExpr *jve = (const JsonValueExpr *) expr;
229 : :
230 : 251 : type = exprType((Node *) jve->formatted_expr);
231 : 251 : }
232 : 251 : break;
233 : : case T_JsonConstructorExpr:
234 : 1071 : type = ((const JsonConstructorExpr *) expr)->returning->typid;
235 : 1071 : break;
236 : : case T_JsonIsPredicate:
237 : 233 : type = BOOLOID;
238 : 233 : break;
239 : : case T_JsonExpr:
240 : : {
241 : 1655 : const JsonExpr *jexpr = (const JsonExpr *) expr;
242 : :
243 : 1655 : type = jexpr->returning->typid;
244 : : break;
245 : 1655 : }
246 : : case T_JsonBehavior:
247 : : {
248 : 783 : const JsonBehavior *behavior = (const JsonBehavior *) expr;
249 : :
250 : 783 : type = exprType(behavior->expr);
251 : : break;
252 : 783 : }
253 : : case T_NullTest:
254 : 4465 : type = BOOLOID;
255 : 4465 : break;
256 : : case T_BooleanTest:
257 : 171 : type = BOOLOID;
258 : 171 : break;
259 : : case T_CoerceToDomain:
260 : 4396 : type = ((const CoerceToDomain *) expr)->resulttype;
261 : 4396 : break;
262 : : case T_CoerceToDomainValue:
263 : 200 : type = ((const CoerceToDomainValue *) expr)->typeId;
264 : 200 : break;
265 : : case T_SetToDefault:
266 : 8189 : type = ((const SetToDefault *) expr)->typeId;
267 : 8189 : break;
268 : : case T_CurrentOfExpr:
269 : 41 : type = BOOLOID;
270 : 41 : break;
271 : : case T_NextValueExpr:
272 : 267 : type = ((const NextValueExpr *) expr)->typeId;
273 : 267 : break;
274 : : case T_InferenceElem:
275 : : {
276 : 0 : const InferenceElem *n = (const InferenceElem *) expr;
277 : :
278 : 0 : type = exprType((Node *) n->expr);
279 : 0 : }
280 : 0 : break;
281 : : case T_ReturningExpr:
282 : 144 : type = exprType((Node *) ((const ReturningExpr *) expr)->retexpr);
283 : 144 : break;
284 : : case T_PlaceHolderVar:
285 : 3141 : type = exprType((Node *) ((const PlaceHolderVar *) expr)->phexpr);
286 : 3141 : break;
287 : : default:
288 [ # # # # ]: 0 : elog(ERROR, "unrecognized node type: %d", (int) nodeTag(expr));
289 : 0 : type = InvalidOid; /* keep compiler quiet */
290 : 0 : break;
291 : : }
292 : 5642716 : return type;
293 : 5642793 : }
294 : :
295 : : /*
296 : : * exprTypmod -
297 : : * returns the type-specific modifier of the expression's result type,
298 : : * if it can be determined. In many cases, it can't and we return -1.
299 : : */
300 : : int32
301 : 2973428 : exprTypmod(const Node *expr)
302 : : {
303 [ + - ]: 2973428 : if (!expr)
304 : 0 : return -1;
305 : :
306 [ + + + + : 2973428 : switch (nodeTag(expr))
- + + + +
+ + + + +
+ + - + +
+ + + + +
+ + + + +
+ ]
307 : : {
308 : : case T_Var:
309 : 1385051 : return ((const Var *) expr)->vartypmod;
310 : : case T_Const:
311 : 1355960 : return ((const Const *) expr)->consttypmod;
312 : : case T_Param:
313 : 14714 : return ((const Param *) expr)->paramtypmod;
314 : : case T_SubscriptingRef:
315 : 6230 : return ((const SubscriptingRef *) expr)->reftypmod;
316 : : case T_FuncExpr:
317 : : {
318 : 101524 : int32 coercedTypmod;
319 : :
320 : : /* Be smart about length-coercion functions... */
321 [ + + ]: 101524 : if (exprIsLengthCoercion(expr, &coercedTypmod))
322 : 3556 : return coercedTypmod;
323 [ + + ]: 101524 : }
324 : 97968 : break;
325 : : case T_NamedArgExpr:
326 : 0 : return exprTypmod((Node *) ((const NamedArgExpr *) expr)->arg);
327 : : case T_NullIfExpr:
328 : : {
329 : : /*
330 : : * Result is either first argument or NULL, so we can report
331 : : * first argument's typmod if known.
332 : : */
333 : 62 : const NullIfExpr *nexpr = (const NullIfExpr *) expr;
334 : :
335 : 62 : return exprTypmod((Node *) linitial(nexpr->args));
336 : 62 : }
337 : : break;
338 : : case T_SubLink:
339 : : {
340 : 294 : const SubLink *sublink = (const SubLink *) expr;
341 : :
342 [ + + + + ]: 294 : if (sublink->subLinkType == EXPR_SUBLINK ||
343 : 22 : sublink->subLinkType == ARRAY_SUBLINK)
344 : : {
345 : : /* get the typmod of the subselect's first target column */
346 : 283 : Query *qtree = (Query *) sublink->subselect;
347 : 283 : TargetEntry *tent;
348 : :
349 [ + - ]: 283 : if (!qtree || !IsA(qtree, Query))
350 [ # # # # ]: 0 : elog(ERROR, "cannot get type for untransformed sublink");
351 : 283 : tent = linitial_node(TargetEntry, qtree->targetList);
352 [ + - ]: 283 : Assert(!tent->resjunk);
353 : 283 : return exprTypmod((Node *) tent->expr);
354 : : /* note we don't need to care if it's an array */
355 : 283 : }
356 : : /* otherwise, result is RECORD or BOOLEAN, typmod is -1 */
357 [ + + ]: 294 : }
358 : 11 : break;
359 : : case T_SubPlan:
360 : : {
361 : 4141 : const SubPlan *subplan = (const SubPlan *) expr;
362 : :
363 [ + + + + ]: 4141 : if (subplan->subLinkType == EXPR_SUBLINK ||
364 : 205 : subplan->subLinkType == ARRAY_SUBLINK)
365 : : {
366 : : /* get the typmod of the subselect's first target column */
367 : : /* note we don't need to care if it's an array */
368 : 3975 : return subplan->firstColTypmod;
369 : : }
370 : : /* otherwise, result is RECORD or BOOLEAN, typmod is -1 */
371 [ + + ]: 4141 : }
372 : 166 : break;
373 : : case T_AlternativeSubPlan:
374 : : {
375 : 42 : const AlternativeSubPlan *asplan = (const AlternativeSubPlan *) expr;
376 : :
377 : : /* subplans should all return the same thing */
378 : 42 : return exprTypmod((Node *) linitial(asplan->subplans));
379 : 42 : }
380 : : break;
381 : : case T_FieldSelect:
382 : 956 : return ((const FieldSelect *) expr)->resulttypmod;
383 : : case T_RelabelType:
384 : 19436 : return ((const RelabelType *) expr)->resulttypmod;
385 : : case T_ArrayCoerceExpr:
386 : 834 : return ((const ArrayCoerceExpr *) expr)->resulttypmod;
387 : : case T_CollateExpr:
388 : 28 : return exprTypmod((Node *) ((const CollateExpr *) expr)->arg);
389 : : case T_CaseExpr:
390 : : {
391 : : /*
392 : : * If all the alternatives agree on type/typmod, return that
393 : : * typmod, else use -1
394 : : */
395 : 11835 : const CaseExpr *cexpr = (const CaseExpr *) expr;
396 : 11835 : Oid casetype = cexpr->casetype;
397 : 11835 : int32 typmod;
398 : 11835 : ListCell *arg;
399 : :
400 [ + - ]: 11835 : if (!cexpr->defresult)
401 : 0 : return -1;
402 [ - + ]: 11835 : if (exprType((Node *) cexpr->defresult) != casetype)
403 : 0 : return -1;
404 : 11835 : typmod = exprTypmod((Node *) cexpr->defresult);
405 [ - + ]: 11835 : if (typmod < 0)
406 : 11835 : return -1; /* no point in trying harder */
407 [ # # # # : 0 : foreach(arg, cexpr->args)
# # # # ]
408 : : {
409 : 0 : CaseWhen *w = lfirst_node(CaseWhen, arg);
410 : :
411 [ # # ]: 0 : if (exprType((Node *) w->result) != casetype)
412 : 0 : return -1;
413 [ # # ]: 0 : if (exprTypmod((Node *) w->result) != typmod)
414 : 0 : return -1;
415 [ # # ]: 0 : }
416 : 0 : return typmod;
417 : 11835 : }
418 : : break;
419 : : case T_CaseTestExpr:
420 : 1720 : return ((const CaseTestExpr *) expr)->typeMod;
421 : : case T_ArrayExpr:
422 : : {
423 : : /*
424 : : * If all the elements agree on type/typmod, return that
425 : : * typmod, else use -1
426 : : */
427 : 3452 : const ArrayExpr *arrayexpr = (const ArrayExpr *) expr;
428 : 3452 : Oid commontype;
429 : 3452 : int32 typmod;
430 : 3452 : ListCell *elem;
431 : :
432 [ + + ]: 3452 : if (arrayexpr->elements == NIL)
433 : 25 : return -1;
434 : 3427 : typmod = exprTypmod((Node *) linitial(arrayexpr->elements));
435 [ + + ]: 3427 : if (typmod < 0)
436 : 3421 : return -1; /* no point in trying harder */
437 [ - + ]: 6 : if (arrayexpr->multidims)
438 : 0 : commontype = arrayexpr->array_typeid;
439 : : else
440 : 6 : commontype = arrayexpr->element_typeid;
441 [ + - + + : 21 : foreach(elem, arrayexpr->elements)
+ + - + ]
442 : : {
443 : 15 : Node *e = (Node *) lfirst(elem);
444 : :
445 [ - + ]: 15 : if (exprType(e) != commontype)
446 : 0 : return -1;
447 [ - + ]: 15 : if (exprTypmod(e) != typmod)
448 : 0 : return -1;
449 [ - + ]: 15 : }
450 : 6 : return typmod;
451 : 3452 : }
452 : : break;
453 : : case T_CoalesceExpr:
454 : : {
455 : : /*
456 : : * If all the alternatives agree on type/typmod, return that
457 : : * typmod, else use -1
458 : : */
459 : 1043 : const CoalesceExpr *cexpr = (const CoalesceExpr *) expr;
460 : 1043 : Oid coalescetype = cexpr->coalescetype;
461 : 1043 : int32 typmod;
462 : 1043 : ListCell *arg;
463 : :
464 [ - + ]: 1043 : if (exprType((Node *) linitial(cexpr->args)) != coalescetype)
465 : 0 : return -1;
466 : 1043 : typmod = exprTypmod((Node *) linitial(cexpr->args));
467 [ - + ]: 1043 : if (typmod < 0)
468 : 1043 : return -1; /* no point in trying harder */
469 [ # # # # : 0 : for_each_from(arg, cexpr->args, 1)
# # # # ]
470 : : {
471 : 0 : Node *e = (Node *) lfirst(arg);
472 : :
473 [ # # ]: 0 : if (exprType(e) != coalescetype)
474 : 0 : return -1;
475 [ # # ]: 0 : if (exprTypmod(e) != typmod)
476 : 0 : return -1;
477 [ # # ]: 0 : }
478 : 0 : return typmod;
479 : 1043 : }
480 : : break;
481 : : case T_MinMaxExpr:
482 : : {
483 : : /*
484 : : * If all the alternatives agree on type/typmod, return that
485 : : * typmod, else use -1
486 : : */
487 : 197 : const MinMaxExpr *mexpr = (const MinMaxExpr *) expr;
488 : 197 : Oid minmaxtype = mexpr->minmaxtype;
489 : 197 : int32 typmod;
490 : 197 : ListCell *arg;
491 : :
492 [ - + ]: 197 : if (exprType((Node *) linitial(mexpr->args)) != minmaxtype)
493 : 0 : return -1;
494 : 197 : typmod = exprTypmod((Node *) linitial(mexpr->args));
495 [ - + ]: 197 : if (typmod < 0)
496 : 197 : return -1; /* no point in trying harder */
497 [ # # # # : 0 : for_each_from(arg, mexpr->args, 1)
# # # # ]
498 : : {
499 : 0 : Node *e = (Node *) lfirst(arg);
500 : :
501 [ # # ]: 0 : if (exprType(e) != minmaxtype)
502 : 0 : return -1;
503 [ # # ]: 0 : if (exprTypmod(e) != typmod)
504 : 0 : return -1;
505 [ # # ]: 0 : }
506 : 0 : return typmod;
507 : 197 : }
508 : : break;
509 : : case T_SQLValueFunction:
510 : 303 : return ((const SQLValueFunction *) expr)->typmod;
511 : : case T_JsonValueExpr:
512 : 11 : return exprTypmod((Node *) ((const JsonValueExpr *) expr)->formatted_expr);
513 : : case T_JsonConstructorExpr:
514 : 384 : return ((const JsonConstructorExpr *) expr)->returning->typmod;
515 : : case T_JsonExpr:
516 : : {
517 : 606 : const JsonExpr *jexpr = (const JsonExpr *) expr;
518 : :
519 : 606 : return jexpr->returning->typmod;
520 : 606 : }
521 : : break;
522 : : case T_JsonBehavior:
523 : : {
524 : 0 : const JsonBehavior *behavior = (const JsonBehavior *) expr;
525 : :
526 : 0 : return exprTypmod(behavior->expr);
527 : 0 : }
528 : : break;
529 : : case T_CoerceToDomain:
530 : 3377 : return ((const CoerceToDomain *) expr)->resulttypmod;
531 : : case T_CoerceToDomainValue:
532 : 11 : return ((const CoerceToDomainValue *) expr)->typeMod;
533 : : case T_SetToDefault:
534 : 2260 : return ((const SetToDefault *) expr)->typeMod;
535 : : case T_ReturningExpr:
536 : 80 : return exprTypmod((Node *) ((const ReturningExpr *) expr)->retexpr);
537 : : case T_PlaceHolderVar:
538 : 1730 : return exprTypmod((Node *) ((const PlaceHolderVar *) expr)->phexpr);
539 : : default:
540 : 57147 : break;
541 : : }
542 : 155292 : return -1;
543 : 2973428 : }
544 : :
545 : : /*
546 : : * exprIsLengthCoercion
547 : : * Detect whether an expression tree is an application of a datatype's
548 : : * typmod-coercion function. Optionally extract the result's typmod.
549 : : *
550 : : * If coercedTypmod is not NULL, the typmod is stored there if the expression
551 : : * is a length-coercion function, else -1 is stored there.
552 : : *
553 : : * Note that a combined type-and-length coercion will be treated as a
554 : : * length coercion by this routine.
555 : : */
556 : : bool
557 : 101732 : exprIsLengthCoercion(const Node *expr, int32 *coercedTypmod)
558 : : {
559 [ - + ]: 101732 : if (coercedTypmod != NULL)
560 : 101732 : *coercedTypmod = -1; /* default result on failure */
561 : :
562 : : /*
563 : : * Scalar-type length coercions are FuncExprs, array-type length coercions
564 : : * are ArrayCoerceExprs
565 : : */
566 [ + - - + ]: 101732 : if (expr && IsA(expr, FuncExpr))
567 : : {
568 : 101732 : const FuncExpr *func = (const FuncExpr *) expr;
569 : 101732 : int nargs;
570 : 101732 : Const *second_arg;
571 : :
572 : : /*
573 : : * If it didn't come from a coercion context, reject.
574 : : */
575 [ + + + + ]: 101732 : if (func->funcformat != COERCE_EXPLICIT_CAST &&
576 : 98048 : func->funcformat != COERCE_IMPLICIT_CAST)
577 : 85855 : return false;
578 : :
579 : : /*
580 : : * If it's not a two-argument or three-argument function with the
581 : : * second argument being an int4 constant, it can't have been created
582 : : * from a length coercion (it must be a type coercion, instead).
583 : : */
584 : 15877 : nargs = list_length(func->args);
585 [ + + - + ]: 15877 : if (nargs < 2 || nargs > 3)
586 : 12308 : return false;
587 : :
588 : 3569 : second_arg = (Const *) lsecond(func->args);
589 [ + - ]: 3569 : if (!IsA(second_arg, Const) ||
590 [ + - - + ]: 3569 : second_arg->consttype != INT4OID ||
591 : 3569 : second_arg->constisnull)
592 : 0 : return false;
593 : :
594 : : /*
595 : : * OK, it is indeed a length-coercion function.
596 : : */
597 [ - + ]: 3569 : if (coercedTypmod != NULL)
598 : 3569 : *coercedTypmod = DatumGetInt32(second_arg->constvalue);
599 : :
600 : 3569 : return true;
601 : 101732 : }
602 : :
603 [ # # # # ]: 0 : if (expr && IsA(expr, ArrayCoerceExpr))
604 : : {
605 : 0 : const ArrayCoerceExpr *acoerce = (const ArrayCoerceExpr *) expr;
606 : :
607 : : /* It's not a length coercion unless there's a nondefault typmod */
608 [ # # ]: 0 : if (acoerce->resulttypmod < 0)
609 : 0 : return false;
610 : :
611 : : /*
612 : : * OK, it is indeed a length-coercion expression.
613 : : */
614 [ # # ]: 0 : if (coercedTypmod != NULL)
615 : 0 : *coercedTypmod = acoerce->resulttypmod;
616 : :
617 : 0 : return true;
618 : 0 : }
619 : :
620 : 0 : return false;
621 : 101732 : }
622 : :
623 : : /*
624 : : * applyRelabelType
625 : : * Add a RelabelType node if needed to make the expression expose
626 : : * the specified type, typmod, and collation.
627 : : *
628 : : * This is primarily intended to be used during planning. Therefore, it must
629 : : * maintain the post-eval_const_expressions invariants that there are not
630 : : * adjacent RelabelTypes, and that the tree is fully const-folded (hence,
631 : : * we mustn't return a RelabelType atop a Const). If we do find a Const,
632 : : * we'll modify it in-place if "overwrite_ok" is true; that should only be
633 : : * passed as true if caller knows the Const is newly generated.
634 : : */
635 : : Node *
636 : 29159 : applyRelabelType(Node *arg, Oid rtype, int32 rtypmod, Oid rcollid,
637 : : CoercionForm rformat, int rlocation, bool overwrite_ok)
638 : : {
639 : : /*
640 : : * If we find stacked RelabelTypes (eg, from foo::int::oid) we can discard
641 : : * all but the top one, and must do so to ensure that semantically
642 : : * equivalent expressions are equal().
643 : : */
644 [ - + + + ]: 29615 : while (arg && IsA(arg, RelabelType))
645 : 456 : arg = (Node *) ((RelabelType *) arg)->arg;
646 : :
647 [ + - + + ]: 29159 : if (arg && IsA(arg, Const))
648 : : {
649 : : /* Modify the Const directly to preserve const-flatness. */
650 : 8537 : Const *con = (Const *) arg;
651 : :
652 [ + + ]: 8537 : if (!overwrite_ok)
653 : 2053 : con = copyObject(con);
654 : 8537 : con->consttype = rtype;
655 : 8537 : con->consttypmod = rtypmod;
656 : 8537 : con->constcollid = rcollid;
657 : : /* We keep the Const's original location. */
658 : 8537 : return (Node *) con;
659 : 8537 : }
660 [ + + ]: 20622 : else if (exprType(arg) == rtype &&
661 [ + + + + ]: 864 : exprTypmod(arg) == rtypmod &&
662 : 850 : exprCollation(arg) == rcollid)
663 : : {
664 : : /* Sometimes we find a nest of relabels that net out to nothing. */
665 : 465 : return arg;
666 : : }
667 : : else
668 : : {
669 : : /* Nope, gotta have a RelabelType. */
670 : 20157 : RelabelType *newrelabel = makeNode(RelabelType);
671 : :
672 : 20157 : newrelabel->arg = (Expr *) arg;
673 : 20157 : newrelabel->resulttype = rtype;
674 : 20157 : newrelabel->resulttypmod = rtypmod;
675 : 20157 : newrelabel->resultcollid = rcollid;
676 : 20157 : newrelabel->relabelformat = rformat;
677 : 20157 : newrelabel->location = rlocation;
678 : 20157 : return (Node *) newrelabel;
679 : 20157 : }
680 : 29159 : }
681 : :
682 : : /*
683 : : * relabel_to_typmod
684 : : * Add a RelabelType node that changes just the typmod of the expression.
685 : : *
686 : : * Convenience function for a common usage of applyRelabelType.
687 : : */
688 : : Node *
689 : 6 : relabel_to_typmod(Node *expr, int32 typmod)
690 : : {
691 : 6 : return applyRelabelType(expr, exprType(expr), typmod, exprCollation(expr),
692 : : COERCE_EXPLICIT_CAST, -1, false);
693 : : }
694 : :
695 : : /*
696 : : * strip_implicit_coercions: remove implicit coercions at top level of tree
697 : : *
698 : : * This doesn't modify or copy the input expression tree, just return a
699 : : * pointer to a suitable place within it.
700 : : *
701 : : * Note: there isn't any useful thing we can do with a RowExpr here, so
702 : : * just return it unchanged, even if it's marked as an implicit coercion.
703 : : */
704 : : Node *
705 : 82615 : strip_implicit_coercions(Node *node)
706 : : {
707 [ + - ]: 82615 : if (node == NULL)
708 : 0 : return NULL;
709 [ + + ]: 82615 : if (IsA(node, FuncExpr))
710 : : {
711 : 2261 : FuncExpr *f = (FuncExpr *) node;
712 : :
713 [ + + ]: 2261 : if (f->funcformat == COERCE_IMPLICIT_CAST)
714 : 7 : return strip_implicit_coercions(linitial(f->args));
715 [ + + ]: 2261 : }
716 [ + + ]: 80354 : else if (IsA(node, RelabelType))
717 : : {
718 : 1808 : RelabelType *r = (RelabelType *) node;
719 : :
720 [ + + ]: 1808 : if (r->relabelformat == COERCE_IMPLICIT_CAST)
721 : 2 : return strip_implicit_coercions((Node *) r->arg);
722 [ + + ]: 1808 : }
723 [ + + ]: 78546 : else if (IsA(node, CoerceViaIO))
724 : : {
725 : 101 : CoerceViaIO *c = (CoerceViaIO *) node;
726 : :
727 [ - + ]: 101 : if (c->coerceformat == COERCE_IMPLICIT_CAST)
728 : 0 : return strip_implicit_coercions((Node *) c->arg);
729 [ - + ]: 101 : }
730 [ - + ]: 78445 : else if (IsA(node, ArrayCoerceExpr))
731 : : {
732 : 0 : ArrayCoerceExpr *c = (ArrayCoerceExpr *) node;
733 : :
734 [ # # ]: 0 : if (c->coerceformat == COERCE_IMPLICIT_CAST)
735 : 0 : return strip_implicit_coercions((Node *) c->arg);
736 [ # # ]: 0 : }
737 [ - + ]: 78445 : else if (IsA(node, ConvertRowtypeExpr))
738 : : {
739 : 0 : ConvertRowtypeExpr *c = (ConvertRowtypeExpr *) node;
740 : :
741 [ # # ]: 0 : if (c->convertformat == COERCE_IMPLICIT_CAST)
742 : 0 : return strip_implicit_coercions((Node *) c->arg);
743 [ # # ]: 0 : }
744 [ + + ]: 78445 : else if (IsA(node, CoerceToDomain))
745 : : {
746 : 93 : CoerceToDomain *c = (CoerceToDomain *) node;
747 : :
748 [ - + ]: 93 : if (c->coercionformat == COERCE_IMPLICIT_CAST)
749 : 0 : return strip_implicit_coercions((Node *) c->arg);
750 [ - + ]: 93 : }
751 : 82606 : return node;
752 : 82615 : }
753 : :
754 : : /*
755 : : * expression_returns_set
756 : : * Test whether an expression returns a set result.
757 : : *
758 : : * Because we use expression_tree_walker(), this can also be applied to
759 : : * whole targetlists; it'll produce true if any one of the tlist items
760 : : * returns a set.
761 : : */
762 : : bool
763 : 73996 : expression_returns_set(Node *clause)
764 : : {
765 : 73996 : return expression_returns_set_walker(clause, NULL);
766 : : }
767 : :
768 : : static bool
769 : 301148 : expression_returns_set_walker(Node *node, void *context)
770 : : {
771 [ + + ]: 301148 : if (node == NULL)
772 : 3580 : return false;
773 [ + + ]: 297568 : if (IsA(node, FuncExpr))
774 : : {
775 : 9741 : FuncExpr *expr = (FuncExpr *) node;
776 : :
777 [ + + ]: 9741 : if (expr->funcretset)
778 : 1813 : return true;
779 : : /* else fall through to check args */
780 [ + + ]: 9741 : }
781 [ + + ]: 295755 : if (IsA(node, OpExpr))
782 : : {
783 : 69593 : OpExpr *expr = (OpExpr *) node;
784 : :
785 [ + + ]: 69593 : if (expr->opretset)
786 : 1 : return true;
787 : : /* else fall through to check args */
788 [ + + ]: 69593 : }
789 : :
790 : : /*
791 : : * If you add any more cases that return sets, also fix
792 : : * expression_returns_set_rows() in clauses.c and IS_SRF_CALL() in
793 : : * tlist.c.
794 : : */
795 : :
796 : : /* Avoid recursion for some cases that parser checks not to return a set */
797 [ + + ]: 295754 : if (IsA(node, Aggref))
798 : 159 : return false;
799 [ + + ]: 295595 : if (IsA(node, GroupingFunc))
800 : 10 : return false;
801 [ + + ]: 295585 : if (IsA(node, WindowFunc))
802 : 5 : return false;
803 : :
804 : 295580 : return expression_tree_walker(node, expression_returns_set_walker,
805 : : context);
806 : 301148 : }
807 : :
808 : :
809 : : /*
810 : : * exprCollation -
811 : : * returns the Oid of the collation of the expression's result.
812 : : *
813 : : * Note: expression nodes that can invoke functions generally have an
814 : : * "inputcollid" field, which is what the function should use as collation.
815 : : * That is the resolved common collation of the node's inputs. It is often
816 : : * but not always the same as the result collation; in particular, if the
817 : : * function produces a non-collatable result type from collatable inputs
818 : : * or vice versa, the two are different.
819 : : */
820 : : Oid
821 : 3316142 : exprCollation(const Node *expr)
822 : : {
823 : 3316142 : Oid coll;
824 : :
825 [ + - ]: 3316142 : if (!expr)
826 : 0 : return InvalidOid;
827 : :
828 [ + + + + : 3316142 : switch (nodeTag(expr))
+ + + + +
+ - + + +
+ + + + +
+ + + + +
+ + - + +
- + + + +
+ + + + +
+ + + + +
- + + + +
- ]
829 : : {
830 : : case T_Var:
831 : 1798987 : coll = ((const Var *) expr)->varcollid;
832 : 1798987 : break;
833 : : case T_Const:
834 : 1378392 : coll = ((const Const *) expr)->constcollid;
835 : 1378392 : break;
836 : : case T_Param:
837 : 16213 : coll = ((const Param *) expr)->paramcollid;
838 : 16213 : break;
839 : : case T_Aggref:
840 : 13098 : coll = ((const Aggref *) expr)->aggcollid;
841 : 13098 : break;
842 : : case T_GroupingFunc:
843 : 134 : coll = InvalidOid;
844 : 134 : break;
845 : : case T_WindowFunc:
846 : 904 : coll = ((const WindowFunc *) expr)->wincollid;
847 : 904 : break;
848 : : case T_MergeSupportFunc:
849 : 56 : coll = ((const MergeSupportFunc *) expr)->msfcollid;
850 : 56 : break;
851 : : case T_SubscriptingRef:
852 : 1307 : coll = ((const SubscriptingRef *) expr)->refcollid;
853 : 1307 : break;
854 : : case T_FuncExpr:
855 : 43658 : coll = ((const FuncExpr *) expr)->funccollid;
856 : 43658 : break;
857 : : case T_NamedArgExpr:
858 : 0 : coll = exprCollation((Node *) ((const NamedArgExpr *) expr)->arg);
859 : 0 : break;
860 : : case T_OpExpr:
861 : 14031 : coll = ((const OpExpr *) expr)->opcollid;
862 : 14031 : break;
863 : : case T_DistinctExpr:
864 : 13 : coll = ((const DistinctExpr *) expr)->opcollid;
865 : 13 : break;
866 : : case T_NullIfExpr:
867 : 37 : coll = ((const NullIfExpr *) expr)->opcollid;
868 : 37 : break;
869 : : case T_ScalarArrayOpExpr:
870 : : /* ScalarArrayOpExpr's result is boolean ... */
871 : 3159 : coll = InvalidOid; /* ... so it has no collation */
872 : 3159 : break;
873 : : case T_BoolExpr:
874 : : /* BoolExpr's result is boolean ... */
875 : 376 : coll = InvalidOid; /* ... so it has no collation */
876 : 376 : break;
877 : : case T_SubLink:
878 : : {
879 : 204 : const SubLink *sublink = (const SubLink *) expr;
880 : :
881 [ + + + + ]: 204 : if (sublink->subLinkType == EXPR_SUBLINK ||
882 : 15 : sublink->subLinkType == ARRAY_SUBLINK)
883 : : {
884 : : /* get the collation of subselect's first target column */
885 : 195 : Query *qtree = (Query *) sublink->subselect;
886 : 195 : TargetEntry *tent;
887 : :
888 [ + - ]: 195 : if (!qtree || !IsA(qtree, Query))
889 [ # # # # ]: 0 : elog(ERROR, "cannot get collation for untransformed sublink");
890 : 195 : tent = linitial_node(TargetEntry, qtree->targetList);
891 [ + - ]: 195 : Assert(!tent->resjunk);
892 : 195 : coll = exprCollation((Node *) tent->expr);
893 : : /* collation doesn't change if it's converted to array */
894 : 195 : }
895 : : else
896 : : {
897 : : /* otherwise, SubLink's result is RECORD or BOOLEAN */
898 : 9 : coll = InvalidOid; /* ... so it has no collation */
899 : : }
900 : 204 : }
901 : 204 : break;
902 : : case T_SubPlan:
903 : : {
904 : 2617 : const SubPlan *subplan = (const SubPlan *) expr;
905 : :
906 [ + + + + ]: 2617 : if (subplan->subLinkType == EXPR_SUBLINK ||
907 : 50 : subplan->subLinkType == ARRAY_SUBLINK)
908 : : {
909 : : /* get the collation of subselect's first target column */
910 : 2583 : coll = subplan->firstColCollation;
911 : : /* collation doesn't change if it's converted to array */
912 : 2583 : }
913 : : else
914 : : {
915 : : /* otherwise, SubPlan's result is RECORD or BOOLEAN */
916 : 34 : coll = InvalidOid; /* ... so it has no collation */
917 : : }
918 : 2617 : }
919 : 2617 : break;
920 : : case T_AlternativeSubPlan:
921 : : {
922 : 0 : const AlternativeSubPlan *asplan = (const AlternativeSubPlan *) expr;
923 : :
924 : : /* subplans should all return the same thing */
925 : 0 : coll = exprCollation((Node *) linitial(asplan->subplans));
926 : 0 : }
927 : 0 : break;
928 : : case T_FieldSelect:
929 : 529 : coll = ((const FieldSelect *) expr)->resultcollid;
930 : 529 : break;
931 : : case T_FieldStore:
932 : : /* FieldStore's result is composite ... */
933 : 10 : coll = InvalidOid; /* ... so it has no collation */
934 : 10 : break;
935 : : case T_RelabelType:
936 : 10740 : coll = ((const RelabelType *) expr)->resultcollid;
937 : 10740 : break;
938 : : case T_CoerceViaIO:
939 : 6117 : coll = ((const CoerceViaIO *) expr)->resultcollid;
940 : 6117 : break;
941 : : case T_ArrayCoerceExpr:
942 : 196 : coll = ((const ArrayCoerceExpr *) expr)->resultcollid;
943 : 196 : break;
944 : : case T_ConvertRowtypeExpr:
945 : : /* ConvertRowtypeExpr's result is composite ... */
946 : 71 : coll = InvalidOid; /* ... so it has no collation */
947 : 71 : break;
948 : : case T_CollateExpr:
949 : 32 : coll = ((const CollateExpr *) expr)->collOid;
950 : 32 : break;
951 : : case T_CaseExpr:
952 : 10598 : coll = ((const CaseExpr *) expr)->casecollid;
953 : 10598 : break;
954 : : case T_CaseTestExpr:
955 : 3927 : coll = ((const CaseTestExpr *) expr)->collation;
956 : 3927 : break;
957 : : case T_ArrayExpr:
958 : 3017 : coll = ((const ArrayExpr *) expr)->array_collid;
959 : 3017 : break;
960 : : case T_RowExpr:
961 : : /* RowExpr's result is composite ... */
962 : 710 : coll = InvalidOid; /* ... so it has no collation */
963 : 710 : break;
964 : : case T_RowCompareExpr:
965 : : /* RowCompareExpr's result is boolean ... */
966 : 11 : coll = InvalidOid; /* ... so it has no collation */
967 : 11 : break;
968 : : case T_CoalesceExpr:
969 : 512 : coll = ((const CoalesceExpr *) expr)->coalescecollid;
970 : 512 : break;
971 : : case T_MinMaxExpr:
972 : 161 : coll = ((const MinMaxExpr *) expr)->minmaxcollid;
973 : 161 : break;
974 : : case T_SQLValueFunction:
975 : : /* Returns either NAME or a non-collatable type */
976 [ + + ]: 207 : if (((const SQLValueFunction *) expr)->type == NAMEOID)
977 : 183 : coll = C_COLLATION_OID;
978 : : else
979 : 24 : coll = InvalidOid;
980 : 207 : break;
981 : : case T_XmlExpr:
982 : :
983 : : /*
984 : : * XMLSERIALIZE returns text from non-collatable inputs, so its
985 : : * collation is always default. The other cases return boolean or
986 : : * XML, which are non-collatable.
987 : : */
988 [ + + ]: 110 : if (((const XmlExpr *) expr)->op == IS_XMLSERIALIZE)
989 : 27 : coll = DEFAULT_COLLATION_OID;
990 : : else
991 : 83 : coll = InvalidOid;
992 : 110 : break;
993 : : case T_JsonValueExpr:
994 : 2 : coll = exprCollation((Node *) ((const JsonValueExpr *) expr)->formatted_expr);
995 : 2 : break;
996 : : case T_JsonConstructorExpr:
997 : : {
998 : 217 : const JsonConstructorExpr *ctor = (const JsonConstructorExpr *) expr;
999 : :
1000 [ + + ]: 217 : if (ctor->coercion)
1001 : 34 : coll = exprCollation((Node *) ctor->coercion);
1002 : : else
1003 : 183 : coll = InvalidOid;
1004 : 217 : }
1005 : 217 : break;
1006 : : case T_JsonIsPredicate:
1007 : : /* IS JSON's result is boolean ... */
1008 : 41 : coll = InvalidOid; /* ... so it has no collation */
1009 : 41 : break;
1010 : : case T_JsonExpr:
1011 : : {
1012 : 404 : const JsonExpr *jsexpr = (JsonExpr *) expr;
1013 : :
1014 : 404 : coll = jsexpr->collation;
1015 : 404 : }
1016 : 404 : break;
1017 : : case T_JsonBehavior:
1018 : : {
1019 : 0 : const JsonBehavior *behavior = (JsonBehavior *) expr;
1020 : :
1021 [ # # ]: 0 : if (behavior->expr)
1022 : 0 : coll = exprCollation(behavior->expr);
1023 : : else
1024 : 0 : coll = InvalidOid;
1025 : 0 : }
1026 : 0 : break;
1027 : : case T_NullTest:
1028 : : /* NullTest's result is boolean ... */
1029 : 210 : coll = InvalidOid; /* ... so it has no collation */
1030 : 210 : break;
1031 : : case T_BooleanTest:
1032 : : /* BooleanTest's result is boolean ... */
1033 : 13 : coll = InvalidOid; /* ... so it has no collation */
1034 : 13 : break;
1035 : : case T_CoerceToDomain:
1036 : 1153 : coll = ((const CoerceToDomain *) expr)->resultcollid;
1037 : 1153 : break;
1038 : : case T_CoerceToDomainValue:
1039 : 83 : coll = ((const CoerceToDomainValue *) expr)->collation;
1040 : 83 : break;
1041 : : case T_SetToDefault:
1042 : 2202 : coll = ((const SetToDefault *) expr)->collation;
1043 : 2202 : break;
1044 : : case T_CurrentOfExpr:
1045 : : /* CurrentOfExpr's result is boolean ... */
1046 : 41 : coll = InvalidOid; /* ... so it has no collation */
1047 : 41 : break;
1048 : : case T_NextValueExpr:
1049 : : /* NextValueExpr's result is an integer type ... */
1050 : 68 : coll = InvalidOid; /* ... so it has no collation */
1051 : 68 : break;
1052 : : case T_InferenceElem:
1053 : 0 : coll = exprCollation((Node *) ((const InferenceElem *) expr)->expr);
1054 : 0 : break;
1055 : : case T_ReturningExpr:
1056 : 80 : coll = exprCollation((Node *) ((const ReturningExpr *) expr)->retexpr);
1057 : 80 : break;
1058 : : case T_PlaceHolderVar:
1059 : 1494 : coll = exprCollation((Node *) ((const PlaceHolderVar *) expr)->phexpr);
1060 : 1494 : break;
1061 : : default:
1062 [ # # # # ]: 0 : elog(ERROR, "unrecognized node type: %d", (int) nodeTag(expr));
1063 : 0 : coll = InvalidOid; /* keep compiler quiet */
1064 : 0 : break;
1065 : : }
1066 : 3316142 : return coll;
1067 : 3316142 : }
1068 : :
1069 : : /*
1070 : : * exprInputCollation -
1071 : : * returns the Oid of the collation a function should use, if available.
1072 : : *
1073 : : * Result is InvalidOid if the node type doesn't store this information.
1074 : : */
1075 : : Oid
1076 : 200 : exprInputCollation(const Node *expr)
1077 : : {
1078 : 200 : Oid coll;
1079 : :
1080 [ + - ]: 200 : if (!expr)
1081 : 0 : return InvalidOid;
1082 : :
1083 [ - - + + : 200 : switch (nodeTag(expr))
+ + + +
+ ]
1084 : : {
1085 : : case T_Aggref:
1086 : 0 : coll = ((const Aggref *) expr)->inputcollid;
1087 : 0 : break;
1088 : : case T_WindowFunc:
1089 : 0 : coll = ((const WindowFunc *) expr)->inputcollid;
1090 : 0 : break;
1091 : : case T_FuncExpr:
1092 : 16 : coll = ((const FuncExpr *) expr)->inputcollid;
1093 : 16 : break;
1094 : : case T_OpExpr:
1095 : 49 : coll = ((const OpExpr *) expr)->inputcollid;
1096 : 49 : break;
1097 : : case T_DistinctExpr:
1098 : 1 : coll = ((const DistinctExpr *) expr)->inputcollid;
1099 : 1 : break;
1100 : : case T_NullIfExpr:
1101 : 2 : coll = ((const NullIfExpr *) expr)->inputcollid;
1102 : 2 : break;
1103 : : case T_ScalarArrayOpExpr:
1104 : 1 : coll = ((const ScalarArrayOpExpr *) expr)->inputcollid;
1105 : 1 : break;
1106 : : case T_MinMaxExpr:
1107 : 1 : coll = ((const MinMaxExpr *) expr)->inputcollid;
1108 : 1 : break;
1109 : : default:
1110 : 130 : coll = InvalidOid;
1111 : 130 : break;
1112 : : }
1113 : 200 : return coll;
1114 : 200 : }
1115 : :
1116 : : /*
1117 : : * exprSetCollation -
1118 : : * Assign collation information to an expression tree node.
1119 : : *
1120 : : * Note: since this is only used during parse analysis, we don't need to
1121 : : * worry about subplans, PlaceHolderVars, or ReturningExprs.
1122 : : */
1123 : : void
1124 : 149268 : exprSetCollation(Node *expr, Oid collation)
1125 : : {
1126 [ + - - - : 149268 : switch (nodeTag(expr))
+ + + + +
+ + + + +
+ + + - +
+ + + + +
+ - - + +
+ + + + +
+ + + + -
- - - - ]
1127 : : {
1128 : : case T_Var:
1129 : 0 : ((Var *) expr)->varcollid = collation;
1130 : 0 : break;
1131 : : case T_Const:
1132 : 0 : ((Const *) expr)->constcollid = collation;
1133 : 0 : break;
1134 : : case T_Param:
1135 : 0 : ((Param *) expr)->paramcollid = collation;
1136 : 0 : break;
1137 : : case T_Aggref:
1138 : 5510 : ((Aggref *) expr)->aggcollid = collation;
1139 : 5510 : break;
1140 : : case T_GroupingFunc:
1141 [ + - ]: 59 : Assert(!OidIsValid(collation));
1142 : 59 : break;
1143 : : case T_WindowFunc:
1144 : 619 : ((WindowFunc *) expr)->wincollid = collation;
1145 : 619 : break;
1146 : : case T_MergeSupportFunc:
1147 : 34 : ((MergeSupportFunc *) expr)->msfcollid = collation;
1148 : 34 : break;
1149 : : case T_SubscriptingRef:
1150 : 1604 : ((SubscriptingRef *) expr)->refcollid = collation;
1151 : 1604 : break;
1152 : : case T_FuncExpr:
1153 : 38293 : ((FuncExpr *) expr)->funccollid = collation;
1154 : 38293 : break;
1155 : : case T_NamedArgExpr:
1156 [ + - ]: 315 : Assert(collation == exprCollation((Node *) ((NamedArgExpr *) expr)->arg));
1157 : 315 : break;
1158 : : case T_OpExpr:
1159 : 53980 : ((OpExpr *) expr)->opcollid = collation;
1160 : 53980 : break;
1161 : : case T_DistinctExpr:
1162 : 47 : ((DistinctExpr *) expr)->opcollid = collation;
1163 : 47 : break;
1164 : : case T_NullIfExpr:
1165 : 59 : ((NullIfExpr *) expr)->opcollid = collation;
1166 : 59 : break;
1167 : : case T_ScalarArrayOpExpr:
1168 : : /* ScalarArrayOpExpr's result is boolean ... */
1169 [ + - ]: 3297 : Assert(!OidIsValid(collation)); /* ... so never set a collation */
1170 : 3297 : break;
1171 : : case T_BoolExpr:
1172 : : /* BoolExpr's result is boolean ... */
1173 [ + - ]: 13029 : Assert(!OidIsValid(collation)); /* ... so never set a collation */
1174 : 13029 : break;
1175 : : case T_SubLink:
1176 : : #ifdef USE_ASSERT_CHECKING
1177 : : {
1178 : 5193 : SubLink *sublink = (SubLink *) expr;
1179 : :
1180 [ + + + + ]: 5193 : if (sublink->subLinkType == EXPR_SUBLINK ||
1181 : 2310 : sublink->subLinkType == ARRAY_SUBLINK)
1182 : : {
1183 : : /* get the collation of subselect's first target column */
1184 : 3899 : Query *qtree = (Query *) sublink->subselect;
1185 : 3899 : TargetEntry *tent;
1186 : :
1187 [ + - ]: 3899 : if (!qtree || !IsA(qtree, Query))
1188 [ # # # # ]: 0 : elog(ERROR, "cannot set collation for untransformed sublink");
1189 : 3899 : tent = linitial_node(TargetEntry, qtree->targetList);
1190 [ + - ]: 3899 : Assert(!tent->resjunk);
1191 [ + - ]: 3899 : Assert(collation == exprCollation((Node *) tent->expr));
1192 : 3899 : }
1193 : : else
1194 : : {
1195 : : /* otherwise, result is RECORD or BOOLEAN */
1196 [ + - ]: 1294 : Assert(!OidIsValid(collation));
1197 : : }
1198 : 5193 : }
1199 : : #endif /* USE_ASSERT_CHECKING */
1200 : 5193 : break;
1201 : : case T_FieldSelect:
1202 : 0 : ((FieldSelect *) expr)->resultcollid = collation;
1203 : 0 : break;
1204 : : case T_FieldStore:
1205 : : /* FieldStore's result is composite ... */
1206 [ + - ]: 99 : Assert(!OidIsValid(collation)); /* ... so never set a collation */
1207 : 99 : break;
1208 : : case T_RelabelType:
1209 : 10969 : ((RelabelType *) expr)->resultcollid = collation;
1210 : 10969 : break;
1211 : : case T_CoerceViaIO:
1212 : 3400 : ((CoerceViaIO *) expr)->resultcollid = collation;
1213 : 3400 : break;
1214 : : case T_ArrayCoerceExpr:
1215 : 674 : ((ArrayCoerceExpr *) expr)->resultcollid = collation;
1216 : 674 : break;
1217 : : case T_ConvertRowtypeExpr:
1218 : : /* ConvertRowtypeExpr's result is composite ... */
1219 [ + - ]: 10 : Assert(!OidIsValid(collation)); /* ... so never set a collation */
1220 : 10 : break;
1221 : : case T_CaseExpr:
1222 : 4082 : ((CaseExpr *) expr)->casecollid = collation;
1223 : 4082 : break;
1224 : : case T_ArrayExpr:
1225 : 2130 : ((ArrayExpr *) expr)->array_collid = collation;
1226 : 2130 : break;
1227 : : case T_RowExpr:
1228 : : /* RowExpr's result is composite ... */
1229 [ # # ]: 0 : Assert(!OidIsValid(collation)); /* ... so never set a collation */
1230 : 0 : break;
1231 : : case T_RowCompareExpr:
1232 : : /* RowCompareExpr's result is boolean ... */
1233 [ # # ]: 0 : Assert(!OidIsValid(collation)); /* ... so never set a collation */
1234 : 0 : break;
1235 : : case T_CoalesceExpr:
1236 : 288 : ((CoalesceExpr *) expr)->coalescecollid = collation;
1237 : 288 : break;
1238 : : case T_MinMaxExpr:
1239 : 52 : ((MinMaxExpr *) expr)->minmaxcollid = collation;
1240 : 52 : break;
1241 : : case T_SQLValueFunction:
1242 [ + + + - ]: 201 : Assert((((SQLValueFunction *) expr)->type == NAMEOID) ?
1243 : : (collation == C_COLLATION_OID) :
1244 : : (collation == InvalidOid));
1245 : 201 : break;
1246 : : case T_XmlExpr:
1247 [ + + + - ]: 127 : Assert((((XmlExpr *) expr)->op == IS_XMLSERIALIZE) ?
1248 : : (collation == DEFAULT_COLLATION_OID) :
1249 : : (collation == InvalidOid));
1250 : 127 : break;
1251 : : case T_JsonValueExpr:
1252 : 220 : exprSetCollation((Node *) ((JsonValueExpr *) expr)->formatted_expr,
1253 : 110 : collation);
1254 : 110 : break;
1255 : : case T_JsonConstructorExpr:
1256 : : {
1257 : 216 : JsonConstructorExpr *ctor = (JsonConstructorExpr *) expr;
1258 : :
1259 [ + + ]: 216 : if (ctor->coercion)
1260 : 59 : exprSetCollation((Node *) ctor->coercion, collation);
1261 : : else
1262 [ + - ]: 157 : Assert(!OidIsValid(collation)); /* result is always a
1263 : : * json[b] type */
1264 : 216 : }
1265 : 216 : break;
1266 : : case T_JsonIsPredicate:
1267 [ + - ]: 52 : Assert(!OidIsValid(collation)); /* result is always boolean */
1268 : 52 : break;
1269 : : case T_JsonExpr:
1270 : : {
1271 : 416 : JsonExpr *jexpr = (JsonExpr *) expr;
1272 : :
1273 : 416 : jexpr->collation = collation;
1274 : 416 : }
1275 : 416 : break;
1276 : : case T_JsonBehavior:
1277 [ + + + - ]: 783 : Assert(((JsonBehavior *) expr)->expr == NULL ||
1278 : : exprCollation(((JsonBehavior *) expr)->expr) == collation);
1279 : 783 : break;
1280 : : case T_NullTest:
1281 : : /* NullTest's result is boolean ... */
1282 [ + - ]: 2250 : Assert(!OidIsValid(collation)); /* ... so never set a collation */
1283 : 2250 : break;
1284 : : case T_BooleanTest:
1285 : : /* BooleanTest's result is boolean ... */
1286 [ + - ]: 73 : Assert(!OidIsValid(collation)); /* ... so never set a collation */
1287 : 73 : break;
1288 : : case T_CoerceToDomain:
1289 : 1297 : ((CoerceToDomain *) expr)->resultcollid = collation;
1290 : 1297 : break;
1291 : : case T_CoerceToDomainValue:
1292 : 0 : ((CoerceToDomainValue *) expr)->collation = collation;
1293 : 0 : break;
1294 : : case T_SetToDefault:
1295 : 0 : ((SetToDefault *) expr)->collation = collation;
1296 : 0 : break;
1297 : : case T_CurrentOfExpr:
1298 : : /* CurrentOfExpr's result is boolean ... */
1299 [ # # ]: 0 : Assert(!OidIsValid(collation)); /* ... so never set a collation */
1300 : 0 : break;
1301 : : case T_NextValueExpr:
1302 : : /* NextValueExpr's result is an integer type ... */
1303 [ # # ]: 0 : Assert(!OidIsValid(collation)); /* ... so never set a collation */
1304 : 0 : break;
1305 : : default:
1306 [ # # # # ]: 0 : elog(ERROR, "unrecognized node type: %d", (int) nodeTag(expr));
1307 : 0 : break;
1308 : : }
1309 : 149268 : }
1310 : :
1311 : : /*
1312 : : * exprSetInputCollation -
1313 : : * Assign input-collation information to an expression tree node.
1314 : : *
1315 : : * This is a no-op for node types that don't store their input collation.
1316 : : * Note we omit RowCompareExpr, which needs special treatment since it
1317 : : * contains multiple input collation OIDs.
1318 : : */
1319 : : void
1320 : 147804 : exprSetInputCollation(Node *expr, Oid inputcollation)
1321 : : {
1322 [ + + + + : 147804 : switch (nodeTag(expr))
+ + + +
+ ]
1323 : : {
1324 : : case T_Aggref:
1325 : 5510 : ((Aggref *) expr)->inputcollid = inputcollation;
1326 : 5510 : break;
1327 : : case T_WindowFunc:
1328 : 619 : ((WindowFunc *) expr)->inputcollid = inputcollation;
1329 : 619 : break;
1330 : : case T_FuncExpr:
1331 : 38263 : ((FuncExpr *) expr)->inputcollid = inputcollation;
1332 : 38263 : break;
1333 : : case T_OpExpr:
1334 : 53980 : ((OpExpr *) expr)->inputcollid = inputcollation;
1335 : 53980 : break;
1336 : : case T_DistinctExpr:
1337 : 47 : ((DistinctExpr *) expr)->inputcollid = inputcollation;
1338 : 47 : break;
1339 : : case T_NullIfExpr:
1340 : 59 : ((NullIfExpr *) expr)->inputcollid = inputcollation;
1341 : 59 : break;
1342 : : case T_ScalarArrayOpExpr:
1343 : 3297 : ((ScalarArrayOpExpr *) expr)->inputcollid = inputcollation;
1344 : 3297 : break;
1345 : : case T_MinMaxExpr:
1346 : 52 : ((MinMaxExpr *) expr)->inputcollid = inputcollation;
1347 : 52 : break;
1348 : : default:
1349 : 45977 : break;
1350 : : }
1351 : 147804 : }
1352 : :
1353 : :
1354 : : /*
1355 : : * exprLocation -
1356 : : * returns the parse location of an expression tree, for error reports
1357 : : *
1358 : : * -1 is returned if the location can't be determined.
1359 : : *
1360 : : * For expressions larger than a single token, the intent here is to
1361 : : * return the location of the expression's leftmost token, not necessarily
1362 : : * the topmost Node's location field. For example, an OpExpr's location
1363 : : * field will point at the operator name, but if it is not a prefix operator
1364 : : * then we should return the location of the left-hand operand instead.
1365 : : * The reason is that we want to reference the entire expression not just
1366 : : * that operator, and pointing to its start seems to be the most natural way.
1367 : : *
1368 : : * The location is not perfect --- for example, since the grammar doesn't
1369 : : * explicitly represent parentheses in the parsetree, given something that
1370 : : * had been written "(a + b) * c" we are going to point at "a" not "(".
1371 : : * But it should be plenty good enough for error reporting purposes.
1372 : : *
1373 : : * You might think that this code is overly general, for instance why check
1374 : : * the operands of a FuncExpr node, when the function name can be expected
1375 : : * to be to the left of them? There are a couple of reasons. The grammar
1376 : : * sometimes builds expressions that aren't quite what the user wrote;
1377 : : * for instance x IS NOT BETWEEN ... becomes a NOT-expression whose keyword
1378 : : * pointer is to the right of its leftmost argument. Also, nodes that were
1379 : : * inserted implicitly by parse analysis (such as FuncExprs for implicit
1380 : : * coercions) will have location -1, and so we can have odd combinations of
1381 : : * known and unknown locations in a tree.
1382 : : */
1383 : : int
1384 : 385089 : exprLocation(const Node *expr)
1385 : : {
1386 : 385089 : int loc;
1387 : :
1388 [ + + ]: 385089 : if (expr == NULL)
1389 : 464 : return -1;
1390 [ + - - - : 384625 : switch (nodeTag(expr))
- - + + +
- + + + +
+ + + + +
+ - + + +
- + + + +
+ + - + +
- + + + +
- - + - +
+ + - + +
+ - - + +
+ - + + -
+ - + + +
- - - + -
- - - + -
+ - - - -
- - - - -
- + + ]
1391 : : {
1392 : : case T_RangeVar:
1393 : 5 : loc = ((const RangeVar *) expr)->location;
1394 : 5 : break;
1395 : : case T_TableFunc:
1396 : 0 : loc = ((const TableFunc *) expr)->location;
1397 : 0 : break;
1398 : : case T_Var:
1399 : 196691 : loc = ((const Var *) expr)->location;
1400 : 196691 : break;
1401 : : case T_Const:
1402 : 121201 : loc = ((const Const *) expr)->location;
1403 : 121201 : break;
1404 : : case T_Param:
1405 : 8640 : loc = ((const Param *) expr)->location;
1406 : 8640 : break;
1407 : : case T_Aggref:
1408 : : /* function name should always be the first thing */
1409 : 1211 : loc = ((const Aggref *) expr)->location;
1410 : 1211 : break;
1411 : : case T_GroupingFunc:
1412 : 11 : loc = ((const GroupingFunc *) expr)->location;
1413 : 11 : break;
1414 : : case T_WindowFunc:
1415 : : /* function name should always be the first thing */
1416 : 9 : loc = ((const WindowFunc *) expr)->location;
1417 : 9 : break;
1418 : : case T_MergeSupportFunc:
1419 : 34 : loc = ((const MergeSupportFunc *) expr)->location;
1420 : 34 : break;
1421 : : case T_SubscriptingRef:
1422 : : /* just use container argument's location */
1423 : 35 : loc = exprLocation((Node *) ((const SubscriptingRef *) expr)->refexpr);
1424 : 35 : break;
1425 : : case T_FuncExpr:
1426 : : {
1427 : 7756 : const FuncExpr *fexpr = (const FuncExpr *) expr;
1428 : :
1429 : : /* consider both function name and leftmost arg */
1430 : 15512 : loc = leftmostLoc(fexpr->location,
1431 : 7756 : exprLocation((Node *) fexpr->args));
1432 : 7756 : }
1433 : 7756 : break;
1434 : : case T_NamedArgExpr:
1435 : : {
1436 : 1 : const NamedArgExpr *na = (const NamedArgExpr *) expr;
1437 : :
1438 : : /* consider both argument name and value */
1439 : 2 : loc = leftmostLoc(na->location,
1440 : 1 : exprLocation((Node *) na->arg));
1441 : 1 : }
1442 : 1 : break;
1443 : : case T_OpExpr:
1444 : : case T_DistinctExpr: /* struct-equivalent to OpExpr */
1445 : : case T_NullIfExpr: /* struct-equivalent to OpExpr */
1446 : : {
1447 : 1366 : const OpExpr *opexpr = (const OpExpr *) expr;
1448 : :
1449 : : /* consider both operator name and leftmost arg */
1450 : 2732 : loc = leftmostLoc(opexpr->location,
1451 : 1366 : exprLocation((Node *) opexpr->args));
1452 : 1366 : }
1453 : 1366 : break;
1454 : : case T_ScalarArrayOpExpr:
1455 : : {
1456 : 0 : const ScalarArrayOpExpr *saopexpr = (const ScalarArrayOpExpr *) expr;
1457 : :
1458 : : /* consider both operator name and leftmost arg */
1459 : 0 : loc = leftmostLoc(saopexpr->location,
1460 : 0 : exprLocation((Node *) saopexpr->args));
1461 : 0 : }
1462 : 0 : break;
1463 : : case T_BoolExpr:
1464 : : {
1465 : 9 : const BoolExpr *bexpr = (const BoolExpr *) expr;
1466 : :
1467 : : /*
1468 : : * Same as above, to handle either NOT or AND/OR. We can't
1469 : : * special-case NOT because of the way that it's used for
1470 : : * things like IS NOT BETWEEN.
1471 : : */
1472 : 18 : loc = leftmostLoc(bexpr->location,
1473 : 9 : exprLocation((Node *) bexpr->args));
1474 : 9 : }
1475 : 9 : break;
1476 : : case T_SubLink:
1477 : : {
1478 : 46 : const SubLink *sublink = (const SubLink *) expr;
1479 : :
1480 : : /* check the testexpr, if any, and the operator/keyword */
1481 : 92 : loc = leftmostLoc(exprLocation(sublink->testexpr),
1482 : 46 : sublink->location);
1483 : 46 : }
1484 : 46 : break;
1485 : : case T_FieldSelect:
1486 : : /* just use argument's location */
1487 : 215 : loc = exprLocation((Node *) ((const FieldSelect *) expr)->arg);
1488 : 215 : break;
1489 : : case T_FieldStore:
1490 : : /* just use argument's location */
1491 : 0 : loc = exprLocation((Node *) ((const FieldStore *) expr)->arg);
1492 : 0 : break;
1493 : : case T_RelabelType:
1494 : : {
1495 : 2145 : const RelabelType *rexpr = (const RelabelType *) expr;
1496 : :
1497 : : /* Much as above */
1498 : 4290 : loc = leftmostLoc(rexpr->location,
1499 : 2145 : exprLocation((Node *) rexpr->arg));
1500 : 2145 : }
1501 : 2145 : break;
1502 : : case T_CoerceViaIO:
1503 : : {
1504 : 2969 : const CoerceViaIO *cexpr = (const CoerceViaIO *) expr;
1505 : :
1506 : : /* Much as above */
1507 : 5938 : loc = leftmostLoc(cexpr->location,
1508 : 2969 : exprLocation((Node *) cexpr->arg));
1509 : 2969 : }
1510 : 2969 : break;
1511 : : case T_ArrayCoerceExpr:
1512 : : {
1513 : 1 : const ArrayCoerceExpr *cexpr = (const ArrayCoerceExpr *) expr;
1514 : :
1515 : : /* Much as above */
1516 : 2 : loc = leftmostLoc(cexpr->location,
1517 : 1 : exprLocation((Node *) cexpr->arg));
1518 : 1 : }
1519 : 1 : break;
1520 : : case T_ConvertRowtypeExpr:
1521 : : {
1522 : 2 : const ConvertRowtypeExpr *cexpr = (const ConvertRowtypeExpr *) expr;
1523 : :
1524 : : /* Much as above */
1525 : 4 : loc = leftmostLoc(cexpr->location,
1526 : 2 : exprLocation((Node *) cexpr->arg));
1527 : 2 : }
1528 : 2 : break;
1529 : : case T_CollateExpr:
1530 : : /* just use argument's location */
1531 : 8 : loc = exprLocation((Node *) ((const CollateExpr *) expr)->arg);
1532 : 8 : break;
1533 : : case T_CaseExpr:
1534 : : /* CASE keyword should always be the first thing */
1535 : 611 : loc = ((const CaseExpr *) expr)->location;
1536 : 611 : break;
1537 : : case T_CaseWhen:
1538 : : /* WHEN keyword should always be the first thing */
1539 : 0 : loc = ((const CaseWhen *) expr)->location;
1540 : 0 : break;
1541 : : case T_ArrayExpr:
1542 : : /* the location points at ARRAY or [, which must be leftmost */
1543 : 42 : loc = ((const ArrayExpr *) expr)->location;
1544 : 42 : break;
1545 : : case T_RowExpr:
1546 : : /* the location points at ROW or (, which must be leftmost */
1547 : 46 : loc = ((const RowExpr *) expr)->location;
1548 : 46 : break;
1549 : : case T_RowCompareExpr:
1550 : : /* just use leftmost argument's location */
1551 : 0 : loc = exprLocation((Node *) ((const RowCompareExpr *) expr)->largs);
1552 : 0 : break;
1553 : : case T_CoalesceExpr:
1554 : : /* COALESCE keyword should always be the first thing */
1555 : 26 : loc = ((const CoalesceExpr *) expr)->location;
1556 : 26 : break;
1557 : : case T_MinMaxExpr:
1558 : : /* GREATEST/LEAST keyword should always be the first thing */
1559 : 4 : loc = ((const MinMaxExpr *) expr)->location;
1560 : 4 : break;
1561 : : case T_SQLValueFunction:
1562 : : /* function keyword should always be the first thing */
1563 : 138 : loc = ((const SQLValueFunction *) expr)->location;
1564 : 138 : break;
1565 : : case T_XmlExpr:
1566 : : {
1567 : 35 : const XmlExpr *xexpr = (const XmlExpr *) expr;
1568 : :
1569 : : /* consider both function name and leftmost arg */
1570 : 70 : loc = leftmostLoc(xexpr->location,
1571 : 35 : exprLocation((Node *) xexpr->args));
1572 : 35 : }
1573 : 35 : break;
1574 : : case T_JsonFormat:
1575 : 0 : loc = ((const JsonFormat *) expr)->location;
1576 : 0 : break;
1577 : : case T_JsonValueExpr:
1578 : 0 : loc = exprLocation((Node *) ((const JsonValueExpr *) expr)->raw_expr);
1579 : 0 : break;
1580 : : case T_JsonConstructorExpr:
1581 : 28 : loc = ((const JsonConstructorExpr *) expr)->location;
1582 : 28 : break;
1583 : : case T_JsonIsPredicate:
1584 : 0 : loc = ((const JsonIsPredicate *) expr)->location;
1585 : 0 : break;
1586 : : case T_JsonExpr:
1587 : : {
1588 : 85 : const JsonExpr *jsexpr = (const JsonExpr *) expr;
1589 : :
1590 : : /* consider both function name and leftmost arg */
1591 : 170 : loc = leftmostLoc(jsexpr->location,
1592 : 85 : exprLocation(jsexpr->formatted_expr));
1593 : 85 : }
1594 : 85 : break;
1595 : : case T_JsonBehavior:
1596 : 35 : loc = exprLocation(((JsonBehavior *) expr)->expr);
1597 : 35 : break;
1598 : : case T_NullTest:
1599 : : {
1600 : 2 : const NullTest *nexpr = (const NullTest *) expr;
1601 : :
1602 : : /* Much as above */
1603 : 4 : loc = leftmostLoc(nexpr->location,
1604 : 2 : exprLocation((Node *) nexpr->arg));
1605 : 2 : }
1606 : 2 : break;
1607 : : case T_BooleanTest:
1608 : : {
1609 : 0 : const BooleanTest *bexpr = (const BooleanTest *) expr;
1610 : :
1611 : : /* Much as above */
1612 : 0 : loc = leftmostLoc(bexpr->location,
1613 : 0 : exprLocation((Node *) bexpr->arg));
1614 : 0 : }
1615 : 0 : break;
1616 : : case T_CoerceToDomain:
1617 : : {
1618 : 901 : const CoerceToDomain *cexpr = (const CoerceToDomain *) expr;
1619 : :
1620 : : /* Much as above */
1621 : 1802 : loc = leftmostLoc(cexpr->location,
1622 : 901 : exprLocation((Node *) cexpr->arg));
1623 : 901 : }
1624 : 901 : break;
1625 : : case T_CoerceToDomainValue:
1626 : 111 : loc = ((const CoerceToDomainValue *) expr)->location;
1627 : 111 : break;
1628 : : case T_SetToDefault:
1629 : 4146 : loc = ((const SetToDefault *) expr)->location;
1630 : 4146 : break;
1631 : : case T_ReturningExpr:
1632 : 0 : loc = exprLocation((Node *) ((const ReturningExpr *) expr)->retexpr);
1633 : 0 : break;
1634 : : case T_TargetEntry:
1635 : : /* just use argument's location */
1636 : 0 : loc = exprLocation((Node *) ((const TargetEntry *) expr)->expr);
1637 : 0 : break;
1638 : : case T_IntoClause:
1639 : : /* use the contained RangeVar's location --- close enough */
1640 : 3 : loc = exprLocation((Node *) ((const IntoClause *) expr)->rel);
1641 : 3 : break;
1642 : : case T_List:
1643 : : {
1644 : : /* report location of first list member that has a location */
1645 : 9598 : ListCell *lc;
1646 : :
1647 : 9598 : loc = -1; /* just to suppress compiler warning */
1648 [ + - + + : 19229 : foreach(lc, (const List *) expr)
+ + ]
1649 : : {
1650 : 9631 : loc = exprLocation((Node *) lfirst(lc));
1651 [ + + ]: 9631 : if (loc >= 0)
1652 : 9572 : break;
1653 : 59 : }
1654 : 9598 : }
1655 : 9598 : break;
1656 : : case T_A_Expr:
1657 : : {
1658 : 817 : const A_Expr *aexpr = (const A_Expr *) expr;
1659 : :
1660 : : /* use leftmost of operator or left operand (if any) */
1661 : : /* we assume right operand can't be to left of operator */
1662 : 1634 : loc = leftmostLoc(aexpr->location,
1663 : 817 : exprLocation(aexpr->lexpr));
1664 : 817 : }
1665 : 817 : break;
1666 : : case T_ColumnRef:
1667 : 9483 : loc = ((const ColumnRef *) expr)->location;
1668 : 9483 : break;
1669 : : case T_ParamRef:
1670 : 0 : loc = ((const ParamRef *) expr)->location;
1671 : 0 : break;
1672 : : case T_A_Const:
1673 : 9813 : loc = ((const A_Const *) expr)->location;
1674 : 9813 : break;
1675 : : case T_FuncCall:
1676 : : {
1677 : 608 : const FuncCall *fc = (const FuncCall *) expr;
1678 : :
1679 : : /* consider both function name and leftmost arg */
1680 : : /* (we assume any ORDER BY nodes must be to right of name) */
1681 : 1216 : loc = leftmostLoc(fc->location,
1682 : 608 : exprLocation((Node *) fc->args));
1683 : 608 : }
1684 : 608 : break;
1685 : : case T_A_ArrayExpr:
1686 : : /* the location points at ARRAY or [, which must be leftmost */
1687 : 0 : loc = ((const A_ArrayExpr *) expr)->location;
1688 : 0 : break;
1689 : : case T_ResTarget:
1690 : : /* we need not examine the contained expression (if any) */
1691 : 3 : loc = ((const ResTarget *) expr)->location;
1692 : 3 : break;
1693 : : case T_MultiAssignRef:
1694 : 0 : loc = exprLocation(((const MultiAssignRef *) expr)->source);
1695 : 0 : break;
1696 : : case T_TypeCast:
1697 : : {
1698 : 1272 : const TypeCast *tc = (const TypeCast *) expr;
1699 : :
1700 : : /*
1701 : : * This could represent CAST(), ::, or TypeName 'literal', so
1702 : : * any of the components might be leftmost.
1703 : : */
1704 : 1272 : loc = exprLocation(tc->arg);
1705 : 1272 : loc = leftmostLoc(loc, tc->typeName->location);
1706 : 1272 : loc = leftmostLoc(loc, tc->location);
1707 : 1272 : }
1708 : 1272 : break;
1709 : : case T_CollateClause:
1710 : : /* just use argument's location */
1711 : 123 : loc = exprLocation(((const CollateClause *) expr)->arg);
1712 : 123 : break;
1713 : : case T_SortBy:
1714 : : /* just use argument's location (ignore operator, if any) */
1715 : 2 : loc = exprLocation(((const SortBy *) expr)->node);
1716 : 2 : break;
1717 : : case T_WindowDef:
1718 : 0 : loc = ((const WindowDef *) expr)->location;
1719 : 0 : break;
1720 : : case T_RangeTableSample:
1721 : 0 : loc = ((const RangeTableSample *) expr)->location;
1722 : 0 : break;
1723 : : case T_TypeName:
1724 : 0 : loc = ((const TypeName *) expr)->location;
1725 : 0 : break;
1726 : : case T_ColumnDef:
1727 : 3 : loc = ((const ColumnDef *) expr)->location;
1728 : 3 : break;
1729 : : case T_IndexElem:
1730 : 0 : loc = ((const IndexElem *) expr)->location;
1731 : 0 : break;
1732 : : case T_Constraint:
1733 : 0 : loc = ((const Constraint *) expr)->location;
1734 : 0 : break;
1735 : : case T_FunctionParameter:
1736 : 0 : loc = ((const FunctionParameter *) expr)->location;
1737 : 0 : break;
1738 : : case T_XmlSerialize:
1739 : : /* XMLSERIALIZE keyword should always be the first thing */
1740 : 0 : loc = ((const XmlSerialize *) expr)->location;
1741 : 0 : break;
1742 : : case T_GroupingSet:
1743 : 6 : loc = ((const GroupingSet *) expr)->location;
1744 : 6 : break;
1745 : : case T_WithClause:
1746 : 0 : loc = ((const WithClause *) expr)->location;
1747 : 0 : break;
1748 : : case T_InferClause:
1749 : 0 : loc = ((const InferClause *) expr)->location;
1750 : 0 : break;
1751 : : case T_OnConflictClause:
1752 : 1 : loc = ((const OnConflictClause *) expr)->location;
1753 : 1 : break;
1754 : : case T_CTESearchClause:
1755 : 0 : loc = ((const CTESearchClause *) expr)->location;
1756 : 0 : break;
1757 : : case T_CTECycleClause:
1758 : 0 : loc = ((const CTECycleClause *) expr)->location;
1759 : 0 : break;
1760 : : case T_CommonTableExpr:
1761 : 0 : loc = ((const CommonTableExpr *) expr)->location;
1762 : 0 : break;
1763 : : case T_JsonKeyValue:
1764 : : /* just use the key's location */
1765 : 0 : loc = exprLocation((Node *) ((const JsonKeyValue *) expr)->key);
1766 : 0 : break;
1767 : : case T_JsonObjectConstructor:
1768 : 0 : loc = ((const JsonObjectConstructor *) expr)->location;
1769 : 0 : break;
1770 : : case T_JsonArrayConstructor:
1771 : 0 : loc = ((const JsonArrayConstructor *) expr)->location;
1772 : 0 : break;
1773 : : case T_JsonArrayQueryConstructor:
1774 : 0 : loc = ((const JsonArrayQueryConstructor *) expr)->location;
1775 : 0 : break;
1776 : : case T_JsonAggConstructor:
1777 : 0 : loc = ((const JsonAggConstructor *) expr)->location;
1778 : 0 : break;
1779 : : case T_JsonObjectAgg:
1780 : 0 : loc = exprLocation((Node *) ((const JsonObjectAgg *) expr)->constructor);
1781 : 0 : break;
1782 : : case T_JsonArrayAgg:
1783 : 0 : loc = exprLocation((Node *) ((const JsonArrayAgg *) expr)->constructor);
1784 : 0 : break;
1785 : : case T_PlaceHolderVar:
1786 : : /* just use argument's location */
1787 : 0 : loc = exprLocation((Node *) ((const PlaceHolderVar *) expr)->phexpr);
1788 : 0 : break;
1789 : : case T_InferenceElem:
1790 : : /* just use nested expr's location */
1791 : 0 : loc = exprLocation((Node *) ((const InferenceElem *) expr)->expr);
1792 : 0 : break;
1793 : : case T_PartitionElem:
1794 : 0 : loc = ((const PartitionElem *) expr)->location;
1795 : 0 : break;
1796 : : case T_PartitionSpec:
1797 : 0 : loc = ((const PartitionSpec *) expr)->location;
1798 : 0 : break;
1799 : : case T_PartitionBoundSpec:
1800 : 9 : loc = ((const PartitionBoundSpec *) expr)->location;
1801 : 9 : break;
1802 : : case T_PartitionRangeDatum:
1803 : 8 : loc = ((const PartitionRangeDatum *) expr)->location;
1804 : 8 : break;
1805 : : default:
1806 : : /* for any other node type it's just unknown... */
1807 : 4311 : loc = -1;
1808 : 4311 : break;
1809 : : }
1810 : 384625 : return loc;
1811 : 385089 : }
1812 : :
1813 : : /*
1814 : : * leftmostLoc - support for exprLocation
1815 : : *
1816 : : * Take the minimum of two parse location values, but ignore unknowns
1817 : : */
1818 : : static int
1819 : 19287 : leftmostLoc(int loc1, int loc2)
1820 : : {
1821 [ + + ]: 19287 : if (loc1 < 0)
1822 : 672 : return loc2;
1823 [ + + ]: 18615 : else if (loc2 < 0)
1824 : 463 : return loc1;
1825 : : else
1826 [ + + ]: 18152 : return Min(loc1, loc2);
1827 : 19287 : }
1828 : :
1829 : :
1830 : : /*
1831 : : * fix_opfuncids
1832 : : * Calculate opfuncid field from opno for each OpExpr node in given tree.
1833 : : * The given tree can be anything expression_tree_walker handles.
1834 : : *
1835 : : * The argument is modified in-place. (This is OK since we'd want the
1836 : : * same change for any node, even if it gets visited more than once due to
1837 : : * shared structure.)
1838 : : */
1839 : : void
1840 : 31482 : fix_opfuncids(Node *node)
1841 : : {
1842 : : /* This tree walk requires no special setup, so away we go... */
1843 : 31482 : fix_opfuncids_walker(node, NULL);
1844 : 31482 : }
1845 : :
1846 : : static bool
1847 : 101654 : fix_opfuncids_walker(Node *node, void *context)
1848 : : {
1849 [ + + ]: 101654 : if (node == NULL)
1850 : 4359 : return false;
1851 [ + + ]: 97295 : if (IsA(node, OpExpr))
1852 : 9492 : set_opfuncid((OpExpr *) node);
1853 [ + + ]: 87803 : else if (IsA(node, DistinctExpr))
1854 : 1 : set_opfuncid((OpExpr *) node); /* rely on struct equivalence */
1855 [ + + ]: 87802 : else if (IsA(node, NullIfExpr))
1856 : 97 : set_opfuncid((OpExpr *) node); /* rely on struct equivalence */
1857 [ + + ]: 87705 : else if (IsA(node, ScalarArrayOpExpr))
1858 : 309 : set_sa_opfuncid((ScalarArrayOpExpr *) node);
1859 : 97295 : return expression_tree_walker(node, fix_opfuncids_walker, context);
1860 : 101654 : }
1861 : :
1862 : : /*
1863 : : * set_opfuncid
1864 : : * Set the opfuncid (procedure OID) in an OpExpr node,
1865 : : * if it hasn't been set already.
1866 : : *
1867 : : * Because of struct equivalence, this can also be used for
1868 : : * DistinctExpr and NullIfExpr nodes.
1869 : : */
1870 : : void
1871 : 395726 : set_opfuncid(OpExpr *opexpr)
1872 : : {
1873 [ + + ]: 395726 : if (opexpr->opfuncid == InvalidOid)
1874 : 24920 : opexpr->opfuncid = get_opcode(opexpr->opno);
1875 : 395726 : }
1876 : :
1877 : : /*
1878 : : * set_sa_opfuncid
1879 : : * As above, for ScalarArrayOpExpr nodes.
1880 : : */
1881 : : void
1882 : 23318 : set_sa_opfuncid(ScalarArrayOpExpr *opexpr)
1883 : : {
1884 [ + + ]: 23318 : if (opexpr->opfuncid == InvalidOid)
1885 : 73 : opexpr->opfuncid = get_opcode(opexpr->opno);
1886 : 23318 : }
1887 : :
1888 : :
1889 : : /*
1890 : : * check_functions_in_node -
1891 : : * apply checker() to each function OID contained in given expression node
1892 : : *
1893 : : * Returns true if the checker() function does; for nodes representing more
1894 : : * than one function call, returns true if the checker() function does so
1895 : : * for any of those functions. Returns false if node does not invoke any
1896 : : * SQL-visible function. Caller must not pass node == NULL.
1897 : : *
1898 : : * This function examines only the given node; it does not recurse into any
1899 : : * sub-expressions. Callers typically prefer to keep control of the recursion
1900 : : * for themselves, in case additional checks should be made, or because they
1901 : : * have special rules about which parts of the tree need to be visited.
1902 : : *
1903 : : * Note: we ignore MinMaxExpr, SQLValueFunction, XmlExpr, CoerceToDomain,
1904 : : * and NextValueExpr nodes, because they do not contain SQL function OIDs.
1905 : : * However, they can invoke SQL-visible functions, so callers should take
1906 : : * thought about how to treat them.
1907 : : */
1908 : : bool
1909 : 2161875 : check_functions_in_node(Node *node, check_function_callback checker,
1910 : : void *context)
1911 : : {
1912 [ + + + + : 2161875 : switch (nodeTag(node))
+ + + + ]
1913 : : {
1914 : : case T_Aggref:
1915 : : {
1916 : 13429 : Aggref *expr = (Aggref *) node;
1917 : :
1918 [ + + ]: 13429 : if (checker(expr->aggfnoid, context))
1919 : 167 : return true;
1920 [ + + ]: 13429 : }
1921 : 13262 : break;
1922 : : case T_WindowFunc:
1923 : : {
1924 : 1299 : WindowFunc *expr = (WindowFunc *) node;
1925 : :
1926 [ + + ]: 1299 : if (checker(expr->winfnoid, context))
1927 : 25 : return true;
1928 [ + + ]: 1299 : }
1929 : 1274 : break;
1930 : : case T_FuncExpr:
1931 : : {
1932 : 70531 : FuncExpr *expr = (FuncExpr *) node;
1933 : :
1934 [ + + ]: 70531 : if (checker(expr->funcid, context))
1935 : 5298 : return true;
1936 [ + + ]: 70531 : }
1937 : 65233 : break;
1938 : : case T_OpExpr:
1939 : : case T_DistinctExpr: /* struct-equivalent to OpExpr */
1940 : : case T_NullIfExpr: /* struct-equivalent to OpExpr */
1941 : : {
1942 : 144016 : OpExpr *expr = (OpExpr *) node;
1943 : :
1944 : : /* Set opfuncid if it wasn't set already */
1945 : 144016 : set_opfuncid(expr);
1946 [ + + ]: 144016 : if (checker(expr->opfuncid, context))
1947 : 270 : return true;
1948 [ + + ]: 144016 : }
1949 : 143746 : break;
1950 : : case T_ScalarArrayOpExpr:
1951 : : {
1952 : 7575 : ScalarArrayOpExpr *expr = (ScalarArrayOpExpr *) node;
1953 : :
1954 : 7575 : set_sa_opfuncid(expr);
1955 [ + + ]: 7575 : if (checker(expr->opfuncid, context))
1956 : 15 : return true;
1957 [ + + ]: 7575 : }
1958 : 7560 : break;
1959 : : case T_CoerceViaIO:
1960 : : {
1961 : 5346 : CoerceViaIO *expr = (CoerceViaIO *) node;
1962 : 5346 : Oid iofunc;
1963 : 5346 : Oid typioparam;
1964 : 5346 : bool typisvarlena;
1965 : :
1966 : : /* check the result type's input function */
1967 : 5346 : getTypeInputInfo(expr->resulttype,
1968 : : &iofunc, &typioparam);
1969 [ + + ]: 5346 : if (checker(iofunc, context))
1970 : 4 : return true;
1971 : : /* check the input type's output function */
1972 : 5342 : getTypeOutputInfo(exprType((Node *) expr->arg),
1973 : : &iofunc, &typisvarlena);
1974 [ + + ]: 5342 : if (checker(iofunc, context))
1975 : 26 : return true;
1976 [ + + ]: 5346 : }
1977 : 5316 : break;
1978 : : case T_RowCompareExpr:
1979 : : {
1980 : 55 : RowCompareExpr *rcexpr = (RowCompareExpr *) node;
1981 : 55 : ListCell *opid;
1982 : :
1983 [ + - + + : 182 : foreach(opid, rcexpr->opnos)
+ + - + ]
1984 : : {
1985 : 127 : Oid opfuncid = get_opcode(lfirst_oid(opid));
1986 : :
1987 [ - + ]: 127 : if (checker(opfuncid, context))
1988 : 0 : return true;
1989 [ - + ]: 127 : }
1990 [ - + ]: 55 : }
1991 : 55 : break;
1992 : : default:
1993 : 1919624 : break;
1994 : : }
1995 : 2156070 : return false;
1996 : 2161875 : }
1997 : :
1998 : :
1999 : : /*
2000 : : * Standard expression-tree walking support
2001 : : *
2002 : : * We used to have near-duplicate code in many different routines that
2003 : : * understood how to recurse through an expression node tree. That was
2004 : : * a pain to maintain, and we frequently had bugs due to some particular
2005 : : * routine neglecting to support a particular node type. In most cases,
2006 : : * these routines only actually care about certain node types, and don't
2007 : : * care about other types except insofar as they have to recurse through
2008 : : * non-primitive node types. Therefore, we now provide generic tree-walking
2009 : : * logic to consolidate the redundant "boilerplate" code. There are
2010 : : * two versions: expression_tree_walker() and expression_tree_mutator().
2011 : : */
2012 : :
2013 : : /*
2014 : : * expression_tree_walker() is designed to support routines that traverse
2015 : : * a tree in a read-only fashion (although it will also work for routines
2016 : : * that modify nodes in-place but never add/delete/replace nodes).
2017 : : * A walker routine should look like this:
2018 : : *
2019 : : * bool my_walker (Node *node, my_struct *context)
2020 : : * {
2021 : : * if (node == NULL)
2022 : : * return false;
2023 : : * // check for nodes that special work is required for, eg:
2024 : : * if (IsA(node, Var))
2025 : : * {
2026 : : * ... do special actions for Var nodes
2027 : : * }
2028 : : * else if (IsA(node, ...))
2029 : : * {
2030 : : * ... do special actions for other node types
2031 : : * }
2032 : : * // for any node type not specially processed, do:
2033 : : * return expression_tree_walker(node, my_walker, context);
2034 : : * }
2035 : : *
2036 : : * The "context" argument points to a struct that holds whatever context
2037 : : * information the walker routine needs --- it can be used to return data
2038 : : * gathered by the walker, too. This argument is not touched by
2039 : : * expression_tree_walker, but it is passed down to recursive sub-invocations
2040 : : * of my_walker. The tree walk is started from a setup routine that
2041 : : * fills in the appropriate context struct, calls my_walker with the top-level
2042 : : * node of the tree, and then examines the results.
2043 : : *
2044 : : * The walker routine should return "false" to continue the tree walk, or
2045 : : * "true" to abort the walk and immediately return "true" to the top-level
2046 : : * caller. This can be used to short-circuit the traversal if the walker
2047 : : * has found what it came for. "false" is returned to the top-level caller
2048 : : * iff no invocation of the walker returned "true".
2049 : : *
2050 : : * The node types handled by expression_tree_walker include all those
2051 : : * normally found in target lists and qualifier clauses during the planning
2052 : : * stage. In particular, it handles List nodes since a cnf-ified qual clause
2053 : : * will have List structure at the top level, and it handles TargetEntry nodes
2054 : : * so that a scan of a target list can be handled without additional code.
2055 : : * Also, RangeTblRef, FromExpr, JoinExpr, and SetOperationStmt nodes are
2056 : : * handled, so that query jointrees and setOperation trees can be processed
2057 : : * without additional code.
2058 : : *
2059 : : * expression_tree_walker will handle SubLink nodes by recursing normally
2060 : : * into the "testexpr" subtree (which is an expression belonging to the outer
2061 : : * plan). It will also call the walker on the sub-Query node; however, when
2062 : : * expression_tree_walker itself is called on a Query node, it does nothing
2063 : : * and returns "false". The net effect is that unless the walker does
2064 : : * something special at a Query node, sub-selects will not be visited during
2065 : : * an expression tree walk. This is exactly the behavior wanted in many cases
2066 : : * --- and for those walkers that do want to recurse into sub-selects, special
2067 : : * behavior is typically needed anyway at the entry to a sub-select (such as
2068 : : * incrementing a depth counter). A walker that wants to examine sub-selects
2069 : : * should include code along the lines of:
2070 : : *
2071 : : * if (IsA(node, Query))
2072 : : * {
2073 : : * adjust context for subquery;
2074 : : * result = query_tree_walker((Query *) node, my_walker, context,
2075 : : * 0); // adjust flags as needed
2076 : : * restore context if needed;
2077 : : * return result;
2078 : : * }
2079 : : *
2080 : : * query_tree_walker is a convenience routine (see below) that calls the
2081 : : * walker on all the expression subtrees of the given Query node.
2082 : : *
2083 : : * expression_tree_walker will handle SubPlan nodes by recursing normally
2084 : : * into the "testexpr" and the "args" list (which are expressions belonging to
2085 : : * the outer plan). It will not touch the completed subplan, however. Since
2086 : : * there is no link to the original Query, it is not possible to recurse into
2087 : : * subselects of an already-planned expression tree. This is OK for current
2088 : : * uses, but may need to be revisited in future.
2089 : : */
2090 : :
2091 : : bool
2092 : 13520339 : expression_tree_walker_impl(Node *node,
2093 : : tree_walker_callback walker,
2094 : : void *context)
2095 : : {
2096 : 13520339 : ListCell *temp;
2097 : :
2098 : : /*
2099 : : * The walker has already visited the current node, and so we need only
2100 : : * recurse into any sub-nodes it has.
2101 : : *
2102 : : * We assume that the walker is not interested in List nodes per se, so
2103 : : * when we expect a List we just recurse directly to self without
2104 : : * bothering to call the walker.
2105 : : */
2106 : : #define WALK(n) walker((Node *) (n), context)
2107 : :
2108 : : #define LIST_WALK(l) expression_tree_walker_impl((Node *) (l), walker, context)
2109 : :
2110 [ + + ]: 13520339 : if (node == NULL)
2111 : 218174 : return false;
2112 : :
2113 : : /* Guard against stack overflow due to overly complex expressions */
2114 : 13302165 : check_stack_depth();
2115 : :
2116 [ + + + + : 13302165 : switch (nodeTag(node))
+ + + + +
+ + + + +
+ + + + +
+ + + + +
+ + + + +
+ + + + +
+ + + + +
+ + + - -
- - - - -
+ + + + +
+ + + + -
+ + + + -
+ + + - ]
2117 : : {
2118 : : case T_Var:
2119 : : case T_Const:
2120 : : case T_Param:
2121 : : case T_CaseTestExpr:
2122 : : case T_SQLValueFunction:
2123 : : case T_CoerceToDomainValue:
2124 : : case T_SetToDefault:
2125 : : case T_CurrentOfExpr:
2126 : : case T_NextValueExpr:
2127 : : case T_RangeTblRef:
2128 : : case T_SortGroupClause:
2129 : : case T_CTESearchClause:
2130 : : case T_MergeSupportFunc:
2131 : : /* primitive node types with no expression subnodes */
2132 : 5105618 : break;
2133 : : case T_WithCheckOption:
2134 : 1426 : return WALK(((WithCheckOption *) node)->qual);
2135 : : case T_Aggref:
2136 : : {
2137 : 42133 : Aggref *expr = (Aggref *) node;
2138 : :
2139 : : /* recurse directly on Lists */
2140 [ - + ]: 42133 : if (LIST_WALK(expr->aggdirectargs))
2141 : 0 : return true;
2142 [ + + ]: 42133 : if (LIST_WALK(expr->args))
2143 : 3025 : return true;
2144 [ - + ]: 39108 : if (LIST_WALK(expr->aggorder))
2145 : 0 : return true;
2146 [ - + ]: 39108 : if (LIST_WALK(expr->aggdistinct))
2147 : 0 : return true;
2148 [ + + ]: 39108 : if (WALK(expr->aggfilter))
2149 : 5 : return true;
2150 [ + + ]: 42133 : }
2151 : 39103 : break;
2152 : : case T_GroupingFunc:
2153 : : {
2154 : 627 : GroupingFunc *grouping = (GroupingFunc *) node;
2155 : :
2156 [ + + ]: 627 : if (LIST_WALK(grouping->args))
2157 : 41 : return true;
2158 [ + + ]: 627 : }
2159 : 586 : break;
2160 : : case T_WindowFunc:
2161 : : {
2162 : 3179 : WindowFunc *expr = (WindowFunc *) node;
2163 : :
2164 : : /* recurse directly on List */
2165 [ + + ]: 3179 : if (LIST_WALK(expr->args))
2166 : 112 : return true;
2167 [ + + ]: 3067 : if (WALK(expr->aggfilter))
2168 : 2 : return true;
2169 [ - + ]: 3065 : if (WALK(expr->runCondition))
2170 : 0 : return true;
2171 [ + + ]: 3179 : }
2172 : 3065 : break;
2173 : : case T_WindowFuncRunCondition:
2174 : : {
2175 : 92 : WindowFuncRunCondition *expr = (WindowFuncRunCondition *) node;
2176 : :
2177 [ - + ]: 92 : if (WALK(expr->arg))
2178 : 0 : return true;
2179 [ - + ]: 92 : }
2180 : 92 : break;
2181 : : case T_SubscriptingRef:
2182 : : {
2183 : 33337 : SubscriptingRef *sbsref = (SubscriptingRef *) node;
2184 : :
2185 : : /* recurse directly for upper/lower container index lists */
2186 [ + + ]: 33337 : if (LIST_WALK(sbsref->refupperindexpr))
2187 : 1782 : return true;
2188 [ - + ]: 31555 : if (LIST_WALK(sbsref->reflowerindexpr))
2189 : 0 : return true;
2190 : : /* walker must see the refexpr and refassgnexpr, however */
2191 [ + + ]: 31555 : if (WALK(sbsref->refexpr))
2192 : 1600 : return true;
2193 : :
2194 [ + + ]: 29955 : if (WALK(sbsref->refassgnexpr))
2195 : 25 : return true;
2196 [ + + ]: 33337 : }
2197 : 29930 : break;
2198 : : case T_FuncExpr:
2199 : : {
2200 : 345503 : FuncExpr *expr = (FuncExpr *) node;
2201 : :
2202 [ + + ]: 345503 : if (LIST_WALK(expr->args))
2203 : 8188 : return true;
2204 [ + + ]: 345503 : }
2205 : 337315 : break;
2206 : : case T_NamedArgExpr:
2207 : 486 : return WALK(((NamedArgExpr *) node)->arg);
2208 : : case T_OpExpr:
2209 : : case T_DistinctExpr: /* struct-equivalent to OpExpr */
2210 : : case T_NullIfExpr: /* struct-equivalent to OpExpr */
2211 : : {
2212 : 1199833 : OpExpr *expr = (OpExpr *) node;
2213 : :
2214 [ + + ]: 1199833 : if (LIST_WALK(expr->args))
2215 : 9196 : return true;
2216 [ + + ]: 1199833 : }
2217 : 1190637 : break;
2218 : : case T_ScalarArrayOpExpr:
2219 : : {
2220 : 49411 : ScalarArrayOpExpr *expr = (ScalarArrayOpExpr *) node;
2221 : :
2222 [ + + ]: 49411 : if (LIST_WALK(expr->args))
2223 : 5204 : return true;
2224 [ + + ]: 49411 : }
2225 : 44207 : break;
2226 : : case T_BoolExpr:
2227 : : {
2228 : 97605 : BoolExpr *expr = (BoolExpr *) node;
2229 : :
2230 [ + + ]: 97605 : if (LIST_WALK(expr->args))
2231 : 925 : return true;
2232 [ + + ]: 97605 : }
2233 : 96680 : break;
2234 : : case T_SubLink:
2235 : : {
2236 : 25844 : SubLink *sublink = (SubLink *) node;
2237 : :
2238 [ + + ]: 25844 : if (WALK(sublink->testexpr))
2239 : 11 : return true;
2240 : :
2241 : : /*
2242 : : * Also invoke the walker on the sublink's Query node, so it
2243 : : * can recurse into the sub-query if it wants to.
2244 : : */
2245 : 25833 : return WALK(sublink->subselect);
2246 : 25844 : }
2247 : : break;
2248 : : case T_SubPlan:
2249 : : {
2250 : 13052 : SubPlan *subplan = (SubPlan *) node;
2251 : :
2252 : : /* recurse into the testexpr, but not into the Plan */
2253 [ + + ]: 13052 : if (WALK(subplan->testexpr))
2254 : 12 : return true;
2255 : : /* also examine args list */
2256 [ + + ]: 13040 : if (LIST_WALK(subplan->args))
2257 : 70 : return true;
2258 [ + + ]: 13052 : }
2259 : 12970 : break;
2260 : : case T_AlternativeSubPlan:
2261 : 718 : return LIST_WALK(((AlternativeSubPlan *) node)->subplans);
2262 : : case T_FieldSelect:
2263 : 6601 : return WALK(((FieldSelect *) node)->arg);
2264 : : case T_FieldStore:
2265 : : {
2266 : 501 : FieldStore *fstore = (FieldStore *) node;
2267 : :
2268 [ - + ]: 501 : if (WALK(fstore->arg))
2269 : 0 : return true;
2270 [ + + ]: 501 : if (WALK(fstore->newvals))
2271 : 2 : return true;
2272 [ + + ]: 501 : }
2273 : 499 : break;
2274 : : case T_RelabelType:
2275 : 117973 : return WALK(((RelabelType *) node)->arg);
2276 : : case T_CoerceViaIO:
2277 : 30557 : return WALK(((CoerceViaIO *) node)->arg);
2278 : : case T_ArrayCoerceExpr:
2279 : : {
2280 : 6278 : ArrayCoerceExpr *acoerce = (ArrayCoerceExpr *) node;
2281 : :
2282 [ + + ]: 6278 : if (WALK(acoerce->arg))
2283 : 569 : return true;
2284 [ + + ]: 5709 : if (WALK(acoerce->elemexpr))
2285 : 4 : return true;
2286 [ + + ]: 6278 : }
2287 : 5705 : break;
2288 : : case T_ConvertRowtypeExpr:
2289 : 435 : return WALK(((ConvertRowtypeExpr *) node)->arg);
2290 : : case T_CollateExpr:
2291 : 5362 : return WALK(((CollateExpr *) node)->arg);
2292 : : case T_CaseExpr:
2293 : : {
2294 : 44289 : CaseExpr *caseexpr = (CaseExpr *) node;
2295 : :
2296 [ + + ]: 44289 : if (WALK(caseexpr->arg))
2297 : 12 : return true;
2298 : : /* we assume walker doesn't care about CaseWhens, either */
2299 [ + - + + : 123384 : foreach(temp, caseexpr->args)
+ + + + ]
2300 : : {
2301 : 79107 : CaseWhen *when = lfirst_node(CaseWhen, temp);
2302 : :
2303 [ + + ]: 79107 : if (WALK(when->expr))
2304 : 210 : return true;
2305 [ + + ]: 78897 : if (WALK(when->result))
2306 : 810 : return true;
2307 [ + + ]: 79107 : }
2308 [ + + ]: 43257 : if (WALK(caseexpr->defresult))
2309 : 1120 : return true;
2310 [ + + ]: 44289 : }
2311 : 42137 : break;
2312 : : case T_ArrayExpr:
2313 : 18794 : return WALK(((ArrayExpr *) node)->elements);
2314 : : case T_RowExpr:
2315 : : /* Assume colnames isn't interesting */
2316 : 4525 : return WALK(((RowExpr *) node)->args);
2317 : : case T_RowCompareExpr:
2318 : : {
2319 : 439 : RowCompareExpr *rcexpr = (RowCompareExpr *) node;
2320 : :
2321 [ - + ]: 439 : if (WALK(rcexpr->largs))
2322 : 0 : return true;
2323 [ - + ]: 439 : if (WALK(rcexpr->rargs))
2324 : 0 : return true;
2325 [ - + ]: 439 : }
2326 : 439 : break;
2327 : : case T_CoalesceExpr:
2328 : 6381 : return WALK(((CoalesceExpr *) node)->args);
2329 : : case T_MinMaxExpr:
2330 : 696 : return WALK(((MinMaxExpr *) node)->args);
2331 : : case T_XmlExpr:
2332 : : {
2333 : 842 : XmlExpr *xexpr = (XmlExpr *) node;
2334 : :
2335 [ + + ]: 842 : if (WALK(xexpr->named_args))
2336 : 2 : return true;
2337 : : /* we assume walker doesn't care about arg_names */
2338 [ + + ]: 840 : if (WALK(xexpr->args))
2339 : 4 : return true;
2340 [ + + ]: 842 : }
2341 : 836 : break;
2342 : : case T_JsonValueExpr:
2343 : : {
2344 : 641 : JsonValueExpr *jve = (JsonValueExpr *) node;
2345 : :
2346 [ + + ]: 641 : if (WALK(jve->raw_expr))
2347 : 8 : return true;
2348 [ - + ]: 633 : if (WALK(jve->formatted_expr))
2349 : 0 : return true;
2350 [ + + ]: 641 : }
2351 : 633 : break;
2352 : : case T_JsonConstructorExpr:
2353 : : {
2354 : 1542 : JsonConstructorExpr *ctor = (JsonConstructorExpr *) node;
2355 : :
2356 [ + + ]: 1542 : if (WALK(ctor->args))
2357 : 12 : return true;
2358 [ + + ]: 1530 : if (WALK(ctor->func))
2359 : 16 : return true;
2360 [ + + ]: 1514 : if (WALK(ctor->coercion))
2361 : 2 : return true;
2362 [ + + ]: 1542 : }
2363 : 1512 : break;
2364 : : case T_JsonIsPredicate:
2365 : 372 : return WALK(((JsonIsPredicate *) node)->expr);
2366 : : case T_JsonExpr:
2367 : : {
2368 : 3010 : JsonExpr *jexpr = (JsonExpr *) node;
2369 : :
2370 [ + + ]: 3010 : if (WALK(jexpr->formatted_expr))
2371 : 14 : return true;
2372 [ + + ]: 2996 : if (WALK(jexpr->path_spec))
2373 : 1 : return true;
2374 [ + + ]: 2995 : if (WALK(jexpr->passing_values))
2375 : 1 : return true;
2376 : : /* we assume walker doesn't care about passing_names */
2377 [ + + ]: 2994 : if (WALK(jexpr->on_empty))
2378 : 6 : return true;
2379 [ + + ]: 2988 : if (WALK(jexpr->on_error))
2380 : 5 : return true;
2381 [ + + ]: 3010 : }
2382 : 2983 : break;
2383 : : case T_JsonBehavior:
2384 : : {
2385 : 5279 : JsonBehavior *behavior = (JsonBehavior *) node;
2386 : :
2387 [ + + ]: 5279 : if (WALK(behavior->expr))
2388 : 11 : return true;
2389 [ + + ]: 5279 : }
2390 : 5268 : break;
2391 : : case T_NullTest:
2392 : 27643 : return WALK(((NullTest *) node)->arg);
2393 : : case T_BooleanTest:
2394 : 1511 : return WALK(((BooleanTest *) node)->arg);
2395 : : case T_CoerceToDomain:
2396 : 22913 : return WALK(((CoerceToDomain *) node)->arg);
2397 : : case T_TargetEntry:
2398 : 2369339 : return WALK(((TargetEntry *) node)->expr);
2399 : : case T_Query:
2400 : : /* Do nothing with a sub-Query, per discussion above */
2401 : : break;
2402 : : case T_WindowClause:
2403 : : {
2404 : 15 : WindowClause *wc = (WindowClause *) node;
2405 : :
2406 [ - + ]: 15 : if (WALK(wc->partitionClause))
2407 : 0 : return true;
2408 [ - + ]: 15 : if (WALK(wc->orderClause))
2409 : 0 : return true;
2410 [ - + ]: 15 : if (WALK(wc->startOffset))
2411 : 0 : return true;
2412 [ - + ]: 15 : if (WALK(wc->endOffset))
2413 : 0 : return true;
2414 [ - + ]: 15 : }
2415 : 15 : break;
2416 : : case T_CTECycleClause:
2417 : : {
2418 : 14 : CTECycleClause *cc = (CTECycleClause *) node;
2419 : :
2420 [ - + ]: 14 : if (WALK(cc->cycle_mark_value))
2421 : 0 : return true;
2422 [ - + ]: 14 : if (WALK(cc->cycle_mark_default))
2423 : 0 : return true;
2424 [ - + ]: 14 : }
2425 : 14 : break;
2426 : : case T_CommonTableExpr:
2427 : : {
2428 : 627 : CommonTableExpr *cte = (CommonTableExpr *) node;
2429 : :
2430 : : /*
2431 : : * Invoke the walker on the CTE's Query node, so it can
2432 : : * recurse into the sub-query if it wants to.
2433 : : */
2434 [ + + ]: 627 : if (WALK(cte->ctequery))
2435 : 15 : return true;
2436 : :
2437 [ - + ]: 612 : if (WALK(cte->search_clause))
2438 : 0 : return true;
2439 [ - + ]: 612 : if (WALK(cte->cycle_clause))
2440 : 0 : return true;
2441 [ + + ]: 627 : }
2442 : 612 : break;
2443 : : case T_JsonKeyValue:
2444 : : {
2445 : 0 : JsonKeyValue *kv = (JsonKeyValue *) node;
2446 : :
2447 [ # # ]: 0 : if (WALK(kv->key))
2448 : 0 : return true;
2449 [ # # ]: 0 : if (WALK(kv->value))
2450 : 0 : return true;
2451 [ # # ]: 0 : }
2452 : 0 : break;
2453 : : case T_JsonObjectConstructor:
2454 : : {
2455 : 0 : JsonObjectConstructor *ctor = (JsonObjectConstructor *) node;
2456 : :
2457 [ # # ]: 0 : if (LIST_WALK(ctor->exprs))
2458 : 0 : return true;
2459 [ # # ]: 0 : }
2460 : 0 : break;
2461 : : case T_JsonArrayConstructor:
2462 : : {
2463 : 0 : JsonArrayConstructor *ctor = (JsonArrayConstructor *) node;
2464 : :
2465 [ # # ]: 0 : if (LIST_WALK(ctor->exprs))
2466 : 0 : return true;
2467 [ # # ]: 0 : }
2468 : 0 : break;
2469 : : case T_JsonArrayQueryConstructor:
2470 : : {
2471 : 0 : JsonArrayQueryConstructor *ctor = (JsonArrayQueryConstructor *) node;
2472 : :
2473 [ # # ]: 0 : if (WALK(ctor->query))
2474 : 0 : return true;
2475 [ # # ]: 0 : }
2476 : 0 : break;
2477 : : case T_JsonAggConstructor:
2478 : : {
2479 : 0 : JsonAggConstructor *ctor = (JsonAggConstructor *) node;
2480 : :
2481 [ # # ]: 0 : if (WALK(ctor->agg_filter))
2482 : 0 : return true;
2483 [ # # ]: 0 : if (WALK(ctor->agg_order))
2484 : 0 : return true;
2485 [ # # ]: 0 : if (WALK(ctor->over))
2486 : 0 : return true;
2487 [ # # ]: 0 : }
2488 : 0 : break;
2489 : : case T_JsonObjectAgg:
2490 : : {
2491 : 0 : JsonObjectAgg *ctor = (JsonObjectAgg *) node;
2492 : :
2493 [ # # ]: 0 : if (WALK(ctor->constructor))
2494 : 0 : return true;
2495 [ # # ]: 0 : if (WALK(ctor->arg))
2496 : 0 : return true;
2497 [ # # ]: 0 : }
2498 : 0 : break;
2499 : : case T_JsonArrayAgg:
2500 : : {
2501 : 0 : JsonArrayAgg *ctor = (JsonArrayAgg *) node;
2502 : :
2503 [ # # ]: 0 : if (WALK(ctor->constructor))
2504 : 0 : return true;
2505 [ # # ]: 0 : if (WALK(ctor->arg))
2506 : 0 : return true;
2507 [ # # ]: 0 : }
2508 : 0 : break;
2509 : :
2510 : : case T_PartitionBoundSpec:
2511 : : {
2512 : 338 : PartitionBoundSpec *pbs = (PartitionBoundSpec *) node;
2513 : :
2514 [ - + ]: 338 : if (WALK(pbs->listdatums))
2515 : 0 : return true;
2516 [ - + ]: 338 : if (WALK(pbs->lowerdatums))
2517 : 0 : return true;
2518 [ - + ]: 338 : if (WALK(pbs->upperdatums))
2519 : 0 : return true;
2520 [ - + ]: 338 : }
2521 : 338 : break;
2522 : : case T_PartitionRangeDatum:
2523 : : {
2524 : 476 : PartitionRangeDatum *prd = (PartitionRangeDatum *) node;
2525 : :
2526 [ - + ]: 476 : if (WALK(prd->value))
2527 : 0 : return true;
2528 [ - + ]: 476 : }
2529 : 476 : break;
2530 : : case T_List:
2531 [ + - + + : 10899562 : foreach(temp, (List *) node)
+ + + + ]
2532 : : {
2533 [ + + ]: 7397768 : if (WALK(lfirst(temp)))
2534 : 84458 : return true;
2535 : 7313310 : }
2536 : 3417336 : break;
2537 : : case T_FromExpr:
2538 : : {
2539 : 140965 : FromExpr *from = (FromExpr *) node;
2540 : :
2541 [ + + ]: 140965 : if (LIST_WALK(from->fromlist))
2542 : 5896 : return true;
2543 [ + + ]: 135069 : if (WALK(from->quals))
2544 : 319 : return true;
2545 [ + + ]: 140965 : }
2546 : 134750 : break;
2547 : : case T_OnConflictExpr:
2548 : : {
2549 : 453 : OnConflictExpr *onconflict = (OnConflictExpr *) node;
2550 : :
2551 [ - + ]: 453 : if (WALK(onconflict->arbiterElems))
2552 : 0 : return true;
2553 [ - + ]: 453 : if (WALK(onconflict->arbiterWhere))
2554 : 0 : return true;
2555 [ - + ]: 453 : if (WALK(onconflict->onConflictSet))
2556 : 0 : return true;
2557 [ - + ]: 453 : if (WALK(onconflict->onConflictWhere))
2558 : 0 : return true;
2559 [ - + ]: 453 : if (WALK(onconflict->exclRelTlist))
2560 : 0 : return true;
2561 [ - + ]: 453 : }
2562 : 453 : break;
2563 : : case T_MergeAction:
2564 : : {
2565 : 1058 : MergeAction *action = (MergeAction *) node;
2566 : :
2567 [ + + ]: 1058 : if (WALK(action->qual))
2568 : 23 : return true;
2569 [ + + ]: 1035 : if (WALK(action->targetList))
2570 : 55 : return true;
2571 [ + + ]: 1058 : }
2572 : 980 : break;
2573 : : case T_PartitionPruneStepOp:
2574 : : {
2575 : 128 : PartitionPruneStepOp *opstep = (PartitionPruneStepOp *) node;
2576 : :
2577 [ - + ]: 128 : if (WALK(opstep->exprs))
2578 : 0 : return true;
2579 [ - + ]: 128 : }
2580 : 128 : break;
2581 : : case T_PartitionPruneStepCombine:
2582 : : /* no expression subnodes */
2583 : : break;
2584 : : case T_JoinExpr:
2585 : : {
2586 : 25749 : JoinExpr *join = (JoinExpr *) node;
2587 : :
2588 [ + + ]: 25749 : if (WALK(join->larg))
2589 : 1677 : return true;
2590 [ + + ]: 24072 : if (WALK(join->rarg))
2591 : 2185 : return true;
2592 [ + + ]: 21887 : if (WALK(join->quals))
2593 : 10 : return true;
2594 : :
2595 : : /*
2596 : : * alias clause, using list are deemed uninteresting.
2597 : : */
2598 [ + + ]: 25749 : }
2599 : 21877 : break;
2600 : : case T_SetOperationStmt:
2601 : : {
2602 : 2966 : SetOperationStmt *setop = (SetOperationStmt *) node;
2603 : :
2604 [ - + ]: 2966 : if (WALK(setop->larg))
2605 : 0 : return true;
2606 [ - + ]: 2966 : if (WALK(setop->rarg))
2607 : 0 : return true;
2608 : :
2609 : : /* groupClauses are deemed uninteresting */
2610 [ - + ]: 2966 : }
2611 : 2966 : break;
2612 : : case T_IndexClause:
2613 : : {
2614 : 0 : IndexClause *iclause = (IndexClause *) node;
2615 : :
2616 [ # # ]: 0 : if (WALK(iclause->rinfo))
2617 : 0 : return true;
2618 [ # # ]: 0 : if (LIST_WALK(iclause->indexquals))
2619 : 0 : return true;
2620 [ # # ]: 0 : }
2621 : 0 : break;
2622 : : case T_PlaceHolderVar:
2623 : 4779 : return WALK(((PlaceHolderVar *) node)->phexpr);
2624 : : case T_InferenceElem:
2625 : 474 : return WALK(((InferenceElem *) node)->expr);
2626 : : case T_ReturningExpr:
2627 : 579 : return WALK(((ReturningExpr *) node)->retexpr);
2628 : : case T_AppendRelInfo:
2629 : : {
2630 : 88 : AppendRelInfo *appinfo = (AppendRelInfo *) node;
2631 : :
2632 [ - + ]: 88 : if (LIST_WALK(appinfo->translated_vars))
2633 : 0 : return true;
2634 [ - + ]: 88 : }
2635 : 88 : break;
2636 : : case T_PlaceHolderInfo:
2637 : 0 : return WALK(((PlaceHolderInfo *) node)->ph_var);
2638 : : case T_RangeTblFunction:
2639 : 13435 : return WALK(((RangeTblFunction *) node)->funcexpr);
2640 : : case T_TableSampleClause:
2641 : : {
2642 : 114 : TableSampleClause *tsc = (TableSampleClause *) node;
2643 : :
2644 [ - + ]: 114 : if (LIST_WALK(tsc->args))
2645 : 0 : return true;
2646 [ - + ]: 114 : if (WALK(tsc->repeatable))
2647 : 0 : return true;
2648 [ - + ]: 114 : }
2649 : 114 : break;
2650 : : case T_TableFunc:
2651 : : {
2652 : 530 : TableFunc *tf = (TableFunc *) node;
2653 : :
2654 [ - + ]: 530 : if (WALK(tf->ns_uris))
2655 : 0 : return true;
2656 [ + + ]: 530 : if (WALK(tf->docexpr))
2657 : 15 : return true;
2658 [ - + ]: 515 : if (WALK(tf->rowexpr))
2659 : 0 : return true;
2660 [ - + ]: 515 : if (WALK(tf->colexprs))
2661 : 0 : return true;
2662 [ - + ]: 515 : if (WALK(tf->coldefexprs))
2663 : 0 : return true;
2664 [ - + ]: 515 : if (WALK(tf->colvalexprs))
2665 : 0 : return true;
2666 [ - + ]: 515 : if (WALK(tf->passingvalexprs))
2667 : 0 : return true;
2668 [ + + ]: 530 : }
2669 : 515 : break;
2670 : : default:
2671 [ # # # # ]: 0 : elog(ERROR, "unrecognized node type: %d",
2672 : : (int) nodeTag(node));
2673 : 0 : break;
2674 : : }
2675 : 10513673 : return false;
2676 : :
2677 : : /* The WALK() macro can be re-used below, but LIST_WALK() not so much */
2678 : : #undef LIST_WALK
2679 : 13520339 : }
2680 : :
2681 : : /*
2682 : : * query_tree_walker --- initiate a walk of a Query's expressions
2683 : : *
2684 : : * This routine exists just to reduce the number of places that need to know
2685 : : * where all the expression subtrees of a Query are. Note it can be used
2686 : : * for starting a walk at top level of a Query regardless of whether the
2687 : : * walker intends to descend into subqueries. It is also useful for
2688 : : * descending into subqueries within a walker.
2689 : : *
2690 : : * Some callers want to suppress visitation of certain items in the sub-Query,
2691 : : * typically because they need to process them specially, or don't actually
2692 : : * want to recurse into subqueries. This is supported by the flags argument,
2693 : : * which is the bitwise OR of flag values to add or suppress visitation of
2694 : : * indicated items. (More flag bits may be added as needed.)
2695 : : */
2696 : : bool
2697 : 175729 : query_tree_walker_impl(Query *query,
2698 : : tree_walker_callback walker,
2699 : : void *context,
2700 : : int flags)
2701 : : {
2702 [ + - ]: 175729 : Assert(query != NULL && IsA(query, Query));
2703 : :
2704 : : /*
2705 : : * We don't walk any utilityStmt here. However, we can't easily assert
2706 : : * that it is absent, since there are at least two code paths by which
2707 : : * action statements from CREATE RULE end up here, and NOTIFY is allowed
2708 : : * in a rule action.
2709 : : */
2710 : :
2711 [ + + ]: 175729 : if (WALK(query->targetList))
2712 : 35040 : return true;
2713 [ - + ]: 140689 : if (WALK(query->withCheckOptions))
2714 : 0 : return true;
2715 [ - + ]: 140689 : if (WALK(query->onConflict))
2716 : 0 : return true;
2717 [ + + ]: 140689 : if (WALK(query->mergeActionList))
2718 : 78 : return true;
2719 [ + + ]: 140611 : if (WALK(query->mergeJoinCondition))
2720 : 55 : return true;
2721 [ + + ]: 140556 : if (WALK(query->returningList))
2722 : 12 : return true;
2723 [ + + ]: 140544 : if (WALK(query->jointree))
2724 : 6140 : return true;
2725 [ - + ]: 134404 : if (WALK(query->setOperations))
2726 : 0 : return true;
2727 [ - + ]: 134404 : if (WALK(query->havingQual))
2728 : 0 : return true;
2729 [ + + ]: 134404 : if (WALK(query->limitOffset))
2730 : 1 : return true;
2731 [ - + ]: 134403 : if (WALK(query->limitCount))
2732 : 0 : return true;
2733 : :
2734 : : /*
2735 : : * Most callers aren't interested in SortGroupClause nodes since those
2736 : : * don't contain actual expressions. However they do contain OIDs which
2737 : : * may be needed by dependency walkers etc.
2738 : : */
2739 [ + + ]: 134403 : if ((flags & QTW_EXAMINE_SORTGROUP))
2740 : : {
2741 [ - + ]: 2042 : if (WALK(query->groupClause))
2742 : 0 : return true;
2743 [ - + ]: 2042 : if (WALK(query->windowClause))
2744 : 0 : return true;
2745 [ - + ]: 2042 : if (WALK(query->sortClause))
2746 : 0 : return true;
2747 [ - + ]: 2042 : if (WALK(query->distinctClause))
2748 : 0 : return true;
2749 : 2042 : }
2750 : : else
2751 : : {
2752 : : /*
2753 : : * But we need to walk the expressions under WindowClause nodes even
2754 : : * if we're not interested in SortGroupClause nodes.
2755 : : */
2756 : 132361 : ListCell *lc;
2757 : :
2758 [ + + + + : 133364 : foreach(lc, query->windowClause)
+ + + + ]
2759 : : {
2760 : 1003 : WindowClause *wc = lfirst_node(WindowClause, lc);
2761 : :
2762 [ + + ]: 1003 : if (WALK(wc->startOffset))
2763 : 1 : return true;
2764 [ - + ]: 1002 : if (WALK(wc->endOffset))
2765 : 0 : return true;
2766 [ + + ]: 1003 : }
2767 [ - + + ]: 132361 : }
2768 : :
2769 : : /*
2770 : : * groupingSets and rowMarks are not walked:
2771 : : *
2772 : : * groupingSets contain only ressortgrouprefs (integers) which are
2773 : : * meaningless without the corresponding groupClause or tlist.
2774 : : * Accordingly, any walker that needs to care about them needs to handle
2775 : : * them itself in its Query processing.
2776 : : *
2777 : : * rowMarks is not walked because it contains only rangetable indexes (and
2778 : : * flags etc.) and therefore should be handled at Query level similarly.
2779 : : */
2780 : :
2781 [ + + ]: 134402 : if (!(flags & QTW_IGNORE_CTE_SUBQUERIES))
2782 : : {
2783 [ + + ]: 73967 : if (WALK(query->cteList))
2784 : 14 : return true;
2785 : 73953 : }
2786 [ + + ]: 134388 : if (!(flags & QTW_IGNORE_RANGE_TABLE))
2787 : : {
2788 [ + + ]: 78848 : if (range_table_walker(query->rtable, walker, context, flags))
2789 : 1243 : return true;
2790 : 77605 : }
2791 : 133145 : return false;
2792 : 175729 : }
2793 : :
2794 : : /*
2795 : : * range_table_walker is just the part of query_tree_walker that scans
2796 : : * a query's rangetable. This is split out since it can be useful on
2797 : : * its own.
2798 : : */
2799 : : bool
2800 : 79482 : range_table_walker_impl(List *rtable,
2801 : : tree_walker_callback walker,
2802 : : void *context,
2803 : : int flags)
2804 : : {
2805 : 79482 : ListCell *rt;
2806 : :
2807 [ + + + + : 187093 : foreach(rt, rtable)
+ + + + ]
2808 : : {
2809 : 107611 : RangeTblEntry *rte = lfirst_node(RangeTblEntry, rt);
2810 : :
2811 [ + + ]: 107611 : if (range_table_entry_walker(rte, walker, context, flags))
2812 : 1243 : return true;
2813 [ + + ]: 107611 : }
2814 : 78239 : return false;
2815 : 79482 : }
2816 : :
2817 : : /*
2818 : : * Some callers even want to scan the expressions in individual RTEs.
2819 : : */
2820 : : bool
2821 : 107615 : range_table_entry_walker_impl(RangeTblEntry *rte,
2822 : : tree_walker_callback walker,
2823 : : void *context,
2824 : : int flags)
2825 : : {
2826 : : /*
2827 : : * Walkers might need to examine the RTE node itself either before or
2828 : : * after visiting its contents (or, conceivably, both). Note that if you
2829 : : * specify neither flag, the walker won't be called on the RTE at all.
2830 : : */
2831 [ + + ]: 107615 : if (flags & QTW_EXAMINE_RTES_BEFORE)
2832 [ + + ]: 10905 : if (WALK(rte))
2833 : 2 : return true;
2834 : :
2835 [ + - + + : 107613 : switch (rte->rtekind)
+ + + +
+ ]
2836 : : {
2837 : : case RTE_RELATION:
2838 [ - + ]: 67089 : if (WALK(rte->tablesample))
2839 : 0 : return true;
2840 : 67089 : break;
2841 : : case RTE_SUBQUERY:
2842 [ + + ]: 12242 : if (!(flags & QTW_IGNORE_RT_SUBQUERIES))
2843 [ + + ]: 12055 : if (WALK(rte->subquery))
2844 : 141 : return true;
2845 : 12101 : break;
2846 : : case RTE_JOIN:
2847 [ + + ]: 13829 : if (!(flags & QTW_IGNORE_JOINALIASES))
2848 [ - + ]: 13425 : if (WALK(rte->joinaliasvars))
2849 : 0 : return true;
2850 : 13829 : break;
2851 : : case RTE_FUNCTION:
2852 [ + + ]: 5803 : if (WALK(rte->functions))
2853 : 1088 : return true;
2854 : 4715 : break;
2855 : : case RTE_TABLEFUNC:
2856 [ - + ]: 162 : if (WALK(rte->tablefunc))
2857 : 0 : return true;
2858 : 162 : break;
2859 : : case RTE_VALUES:
2860 [ + + ]: 4038 : if (WALK(rte->values_lists))
2861 : 9 : return true;
2862 : 4029 : break;
2863 : : case RTE_CTE:
2864 : : case RTE_NAMEDTUPLESTORE:
2865 : : case RTE_RESULT:
2866 : : /* nothing to do */
2867 : 3472 : break;
2868 : : case RTE_GROUP:
2869 [ - + ]: 978 : if (!(flags & QTW_IGNORE_GROUPEXPRS))
2870 [ + + ]: 978 : if (WALK(rte->groupexprs))
2871 : 4 : return true;
2872 : 974 : break;
2873 : : }
2874 : :
2875 [ + + ]: 106371 : if (WALK(rte->securityQuals))
2876 : 3 : return true;
2877 : :
2878 [ + + ]: 106368 : if (flags & QTW_EXAMINE_RTES_AFTER)
2879 [ - + ]: 541 : if (WALK(rte))
2880 : 0 : return true;
2881 : :
2882 : 106368 : return false;
2883 : 107615 : }
2884 : :
2885 : :
2886 : : /*
2887 : : * expression_tree_mutator() is designed to support routines that make a
2888 : : * modified copy of an expression tree, with some nodes being added,
2889 : : * removed, or replaced by new subtrees. The original tree is (normally)
2890 : : * not changed. Each recursion level is responsible for returning a copy of
2891 : : * (or appropriately modified substitute for) the subtree it is handed.
2892 : : * A mutator routine should look like this:
2893 : : *
2894 : : * Node * my_mutator (Node *node, my_struct *context)
2895 : : * {
2896 : : * if (node == NULL)
2897 : : * return NULL;
2898 : : * // check for nodes that special work is required for, eg:
2899 : : * if (IsA(node, Var))
2900 : : * {
2901 : : * ... create and return modified copy of Var node
2902 : : * }
2903 : : * else if (IsA(node, ...))
2904 : : * {
2905 : : * ... do special transformations of other node types
2906 : : * }
2907 : : * // for any node type not specially processed, do:
2908 : : * return expression_tree_mutator(node, my_mutator, context);
2909 : : * }
2910 : : *
2911 : : * The "context" argument points to a struct that holds whatever context
2912 : : * information the mutator routine needs --- it can be used to return extra
2913 : : * data gathered by the mutator, too. This argument is not touched by
2914 : : * expression_tree_mutator, but it is passed down to recursive sub-invocations
2915 : : * of my_mutator. The tree walk is started from a setup routine that
2916 : : * fills in the appropriate context struct, calls my_mutator with the
2917 : : * top-level node of the tree, and does any required post-processing.
2918 : : *
2919 : : * Each level of recursion must return an appropriately modified Node.
2920 : : * If expression_tree_mutator() is called, it will make an exact copy
2921 : : * of the given Node, but invoke my_mutator() to copy the sub-node(s)
2922 : : * of that Node. In this way, my_mutator() has full control over the
2923 : : * copying process but need not directly deal with expression trees
2924 : : * that it has no interest in.
2925 : : *
2926 : : * Just as for expression_tree_walker, the node types handled by
2927 : : * expression_tree_mutator include all those normally found in target lists
2928 : : * and qualifier clauses during the planning stage.
2929 : : *
2930 : : * expression_tree_mutator will handle SubLink nodes by recursing normally
2931 : : * into the "testexpr" subtree (which is an expression belonging to the outer
2932 : : * plan). It will also call the mutator on the sub-Query node; however, when
2933 : : * expression_tree_mutator itself is called on a Query node, it does nothing
2934 : : * and returns the unmodified Query node. The net effect is that unless the
2935 : : * mutator does something special at a Query node, sub-selects will not be
2936 : : * visited or modified; the original sub-select will be linked to by the new
2937 : : * SubLink node. Mutators that want to descend into sub-selects will usually
2938 : : * do so by recognizing Query nodes and calling query_tree_mutator (below).
2939 : : *
2940 : : * expression_tree_mutator will handle a SubPlan node by recursing into the
2941 : : * "testexpr" and the "args" list (which belong to the outer plan), but it
2942 : : * will simply copy the link to the inner plan, since that's typically what
2943 : : * expression tree mutators want. A mutator that wants to modify the subplan
2944 : : * can force appropriate behavior by recognizing SubPlan expression nodes
2945 : : * and doing the right thing.
2946 : : */
2947 : :
2948 : : Node *
2949 : 1976587 : expression_tree_mutator_impl(Node *node,
2950 : : tree_mutator_callback mutator,
2951 : : void *context)
2952 : : {
2953 : : /*
2954 : : * The mutator has already decided not to modify the current node, but we
2955 : : * must call the mutator for any sub-nodes.
2956 : : */
2957 : :
2958 : : #define FLATCOPY(newnode, node, nodetype) \
2959 : : ( (newnode) = palloc_object(nodetype), \
2960 : : memcpy((newnode), (node), sizeof(nodetype)) )
2961 : :
2962 : : #define MUTATE(newfield, oldfield, fieldtype) \
2963 : : ( (newfield) = (fieldtype) mutator((Node *) (oldfield), context) )
2964 : :
2965 [ + + ]: 1976587 : if (node == NULL)
2966 : 1442 : return NULL;
2967 : :
2968 : : /* Guard against stack overflow due to overly complex expressions */
2969 : 1975145 : check_stack_depth();
2970 : :
2971 [ + + + + : 1975145 : switch (nodeTag(node))
+ + - + +
- + + + +
+ + + + +
+ + + + +
+ + + + +
+ + + + +
+ + + + +
+ + + + +
+ - - + -
+ + + + +
+ + + - +
+ - + + +
+ + + - ]
2972 : : {
2973 : : /*
2974 : : * Primitive node types with no expression subnodes. Var and
2975 : : * Const are frequent enough to deserve special cases, the others
2976 : : * we just use copyObject for.
2977 : : */
2978 : : case T_Var:
2979 : : {
2980 : 352754 : Var *var = (Var *) node;
2981 : 352754 : Var *newnode;
2982 : :
2983 : 352754 : FLATCOPY(newnode, var, Var);
2984 : : /* Assume we need not copy the varnullingrels bitmapset */
2985 : 352754 : return (Node *) newnode;
2986 : 352754 : }
2987 : : break;
2988 : : case T_Const:
2989 : : {
2990 : 308338 : Const *oldnode = (Const *) node;
2991 : 308338 : Const *newnode;
2992 : :
2993 : 308338 : FLATCOPY(newnode, oldnode, Const);
2994 : : /* XXX we don't bother with datumCopy; should we? */
2995 : 308338 : return (Node *) newnode;
2996 : 308338 : }
2997 : : break;
2998 : : case T_Param:
2999 : : case T_CaseTestExpr:
3000 : : case T_SQLValueFunction:
3001 : : case T_JsonFormat:
3002 : : case T_CoerceToDomainValue:
3003 : : case T_SetToDefault:
3004 : : case T_CurrentOfExpr:
3005 : : case T_NextValueExpr:
3006 : : case T_RangeTblRef:
3007 : : case T_SortGroupClause:
3008 : : case T_CTESearchClause:
3009 : : case T_MergeSupportFunc:
3010 : 16994 : return copyObject(node);
3011 : : case T_WithCheckOption:
3012 : : {
3013 : 231 : WithCheckOption *wco = (WithCheckOption *) node;
3014 : 231 : WithCheckOption *newnode;
3015 : :
3016 : 231 : FLATCOPY(newnode, wco, WithCheckOption);
3017 : 231 : MUTATE(newnode->qual, wco->qual, Node *);
3018 : 231 : return (Node *) newnode;
3019 : 231 : }
3020 : : case T_Aggref:
3021 : : {
3022 : 23062 : Aggref *aggref = (Aggref *) node;
3023 : 23062 : Aggref *newnode;
3024 : :
3025 : 23062 : FLATCOPY(newnode, aggref, Aggref);
3026 : : /* assume mutation doesn't change types of arguments */
3027 : 23062 : newnode->aggargtypes = list_copy(aggref->aggargtypes);
3028 : 23062 : MUTATE(newnode->aggdirectargs, aggref->aggdirectargs, List *);
3029 : 23062 : MUTATE(newnode->args, aggref->args, List *);
3030 : 23062 : MUTATE(newnode->aggorder, aggref->aggorder, List *);
3031 : 23062 : MUTATE(newnode->aggdistinct, aggref->aggdistinct, List *);
3032 : 23062 : MUTATE(newnode->aggfilter, aggref->aggfilter, Expr *);
3033 : 23062 : return (Node *) newnode;
3034 : 23062 : }
3035 : : break;
3036 : : case T_GroupingFunc:
3037 : : {
3038 : 238 : GroupingFunc *grouping = (GroupingFunc *) node;
3039 : 238 : GroupingFunc *newnode;
3040 : :
3041 : 238 : FLATCOPY(newnode, grouping, GroupingFunc);
3042 : 238 : MUTATE(newnode->args, grouping->args, List *);
3043 : :
3044 : : /*
3045 : : * We assume here that mutating the arguments does not change
3046 : : * the semantics, i.e. that the arguments are not mutated in a
3047 : : * way that makes them semantically different from their
3048 : : * previously matching expressions in the GROUP BY clause.
3049 : : *
3050 : : * If a mutator somehow wanted to do this, it would have to
3051 : : * handle the refs and cols lists itself as appropriate.
3052 : : */
3053 : 238 : newnode->refs = list_copy(grouping->refs);
3054 : 238 : newnode->cols = list_copy(grouping->cols);
3055 : :
3056 : 238 : return (Node *) newnode;
3057 : 238 : }
3058 : : break;
3059 : : case T_WindowFunc:
3060 : : {
3061 : 877 : WindowFunc *wfunc = (WindowFunc *) node;
3062 : 877 : WindowFunc *newnode;
3063 : :
3064 : 877 : FLATCOPY(newnode, wfunc, WindowFunc);
3065 : 877 : MUTATE(newnode->args, wfunc->args, List *);
3066 : 877 : MUTATE(newnode->aggfilter, wfunc->aggfilter, Expr *);
3067 : 877 : return (Node *) newnode;
3068 : 877 : }
3069 : : break;
3070 : : case T_WindowFuncRunCondition:
3071 : : {
3072 : 0 : WindowFuncRunCondition *wfuncrc = (WindowFuncRunCondition *) node;
3073 : 0 : WindowFuncRunCondition *newnode;
3074 : :
3075 : 0 : FLATCOPY(newnode, wfuncrc, WindowFuncRunCondition);
3076 : 0 : MUTATE(newnode->arg, wfuncrc->arg, Expr *);
3077 : 0 : return (Node *) newnode;
3078 : 0 : }
3079 : : break;
3080 : : case T_SubscriptingRef:
3081 : : {
3082 : 7739 : SubscriptingRef *sbsref = (SubscriptingRef *) node;
3083 : 7739 : SubscriptingRef *newnode;
3084 : :
3085 : 7739 : FLATCOPY(newnode, sbsref, SubscriptingRef);
3086 : 7739 : MUTATE(newnode->refupperindexpr, sbsref->refupperindexpr,
3087 : : List *);
3088 : 7739 : MUTATE(newnode->reflowerindexpr, sbsref->reflowerindexpr,
3089 : : List *);
3090 : 7739 : MUTATE(newnode->refexpr, sbsref->refexpr,
3091 : : Expr *);
3092 : 7739 : MUTATE(newnode->refassgnexpr, sbsref->refassgnexpr,
3093 : : Expr *);
3094 : :
3095 : 7739 : return (Node *) newnode;
3096 : 7739 : }
3097 : : break;
3098 : : case T_FuncExpr:
3099 : : {
3100 : 37795 : FuncExpr *expr = (FuncExpr *) node;
3101 : 37795 : FuncExpr *newnode;
3102 : :
3103 : 37795 : FLATCOPY(newnode, expr, FuncExpr);
3104 : 37795 : MUTATE(newnode->args, expr->args, List *);
3105 : 37795 : return (Node *) newnode;
3106 : 37795 : }
3107 : : break;
3108 : : case T_NamedArgExpr:
3109 : : {
3110 : 0 : NamedArgExpr *nexpr = (NamedArgExpr *) node;
3111 : 0 : NamedArgExpr *newnode;
3112 : :
3113 : 0 : FLATCOPY(newnode, nexpr, NamedArgExpr);
3114 : 0 : MUTATE(newnode->arg, nexpr->arg, Expr *);
3115 : 0 : return (Node *) newnode;
3116 : 0 : }
3117 : : break;
3118 : : case T_OpExpr:
3119 : : {
3120 : 129272 : OpExpr *expr = (OpExpr *) node;
3121 : 129272 : OpExpr *newnode;
3122 : :
3123 : 129272 : FLATCOPY(newnode, expr, OpExpr);
3124 : 129272 : MUTATE(newnode->args, expr->args, List *);
3125 : 129272 : return (Node *) newnode;
3126 : 129272 : }
3127 : : break;
3128 : : case T_DistinctExpr:
3129 : : {
3130 : 55 : DistinctExpr *expr = (DistinctExpr *) node;
3131 : 55 : DistinctExpr *newnode;
3132 : :
3133 : 55 : FLATCOPY(newnode, expr, DistinctExpr);
3134 : 55 : MUTATE(newnode->args, expr->args, List *);
3135 : 55 : return (Node *) newnode;
3136 : 55 : }
3137 : : break;
3138 : : case T_NullIfExpr:
3139 : : {
3140 : 140 : NullIfExpr *expr = (NullIfExpr *) node;
3141 : 140 : NullIfExpr *newnode;
3142 : :
3143 : 140 : FLATCOPY(newnode, expr, NullIfExpr);
3144 : 140 : MUTATE(newnode->args, expr->args, List *);
3145 : 140 : return (Node *) newnode;
3146 : 140 : }
3147 : : break;
3148 : : case T_ScalarArrayOpExpr:
3149 : : {
3150 : 9549 : ScalarArrayOpExpr *expr = (ScalarArrayOpExpr *) node;
3151 : 9549 : ScalarArrayOpExpr *newnode;
3152 : :
3153 : 9549 : FLATCOPY(newnode, expr, ScalarArrayOpExpr);
3154 : 9549 : MUTATE(newnode->args, expr->args, List *);
3155 : 9549 : return (Node *) newnode;
3156 : 9549 : }
3157 : : break;
3158 : : case T_BoolExpr:
3159 : : {
3160 : 14261 : BoolExpr *expr = (BoolExpr *) node;
3161 : 14261 : BoolExpr *newnode;
3162 : :
3163 : 14261 : FLATCOPY(newnode, expr, BoolExpr);
3164 : 14261 : MUTATE(newnode->args, expr->args, List *);
3165 : 14261 : return (Node *) newnode;
3166 : 14261 : }
3167 : : break;
3168 : : case T_SubLink:
3169 : : {
3170 : 5406 : SubLink *sublink = (SubLink *) node;
3171 : 5406 : SubLink *newnode;
3172 : :
3173 : 5406 : FLATCOPY(newnode, sublink, SubLink);
3174 : 5406 : MUTATE(newnode->testexpr, sublink->testexpr, Node *);
3175 : :
3176 : : /*
3177 : : * Also invoke the mutator on the sublink's Query node, so it
3178 : : * can recurse into the sub-query if it wants to.
3179 : : */
3180 : 5406 : MUTATE(newnode->subselect, sublink->subselect, Node *);
3181 : 5406 : return (Node *) newnode;
3182 : 5406 : }
3183 : : break;
3184 : : case T_SubPlan:
3185 : : {
3186 : 2205 : SubPlan *subplan = (SubPlan *) node;
3187 : 2205 : SubPlan *newnode;
3188 : :
3189 : 2205 : FLATCOPY(newnode, subplan, SubPlan);
3190 : : /* transform testexpr */
3191 : 2205 : MUTATE(newnode->testexpr, subplan->testexpr, Node *);
3192 : : /* transform args list (params to be passed to subplan) */
3193 : 2205 : MUTATE(newnode->args, subplan->args, List *);
3194 : : /* but not the sub-Plan itself, which is referenced as-is */
3195 : 2205 : return (Node *) newnode;
3196 : 2205 : }
3197 : : break;
3198 : : case T_AlternativeSubPlan:
3199 : : {
3200 : 36 : AlternativeSubPlan *asplan = (AlternativeSubPlan *) node;
3201 : 36 : AlternativeSubPlan *newnode;
3202 : :
3203 : 36 : FLATCOPY(newnode, asplan, AlternativeSubPlan);
3204 : 36 : MUTATE(newnode->subplans, asplan->subplans, List *);
3205 : 36 : return (Node *) newnode;
3206 : 36 : }
3207 : : break;
3208 : : case T_FieldSelect:
3209 : : {
3210 : 756 : FieldSelect *fselect = (FieldSelect *) node;
3211 : 756 : FieldSelect *newnode;
3212 : :
3213 : 756 : FLATCOPY(newnode, fselect, FieldSelect);
3214 : 756 : MUTATE(newnode->arg, fselect->arg, Expr *);
3215 : 756 : return (Node *) newnode;
3216 : 756 : }
3217 : : break;
3218 : : case T_FieldStore:
3219 : : {
3220 : 71 : FieldStore *fstore = (FieldStore *) node;
3221 : 71 : FieldStore *newnode;
3222 : :
3223 : 71 : FLATCOPY(newnode, fstore, FieldStore);
3224 : 71 : MUTATE(newnode->arg, fstore->arg, Expr *);
3225 : 71 : MUTATE(newnode->newvals, fstore->newvals, List *);
3226 : 71 : newnode->fieldnums = list_copy(fstore->fieldnums);
3227 : 71 : return (Node *) newnode;
3228 : 71 : }
3229 : : break;
3230 : : case T_RelabelType:
3231 : : {
3232 : 16286 : RelabelType *relabel = (RelabelType *) node;
3233 : 16286 : RelabelType *newnode;
3234 : :
3235 : 16286 : FLATCOPY(newnode, relabel, RelabelType);
3236 : 16286 : MUTATE(newnode->arg, relabel->arg, Expr *);
3237 : 16286 : return (Node *) newnode;
3238 : 16286 : }
3239 : : break;
3240 : : case T_CoerceViaIO:
3241 : : {
3242 : 4372 : CoerceViaIO *iocoerce = (CoerceViaIO *) node;
3243 : 4372 : CoerceViaIO *newnode;
3244 : :
3245 : 4372 : FLATCOPY(newnode, iocoerce, CoerceViaIO);
3246 : 4372 : MUTATE(newnode->arg, iocoerce->arg, Expr *);
3247 : 4372 : return (Node *) newnode;
3248 : 4372 : }
3249 : : break;
3250 : : case T_ArrayCoerceExpr:
3251 : : {
3252 : 1671 : ArrayCoerceExpr *acoerce = (ArrayCoerceExpr *) node;
3253 : 1671 : ArrayCoerceExpr *newnode;
3254 : :
3255 : 1671 : FLATCOPY(newnode, acoerce, ArrayCoerceExpr);
3256 : 1671 : MUTATE(newnode->arg, acoerce->arg, Expr *);
3257 : 1671 : MUTATE(newnode->elemexpr, acoerce->elemexpr, Expr *);
3258 : 1671 : return (Node *) newnode;
3259 : 1671 : }
3260 : : break;
3261 : : case T_ConvertRowtypeExpr:
3262 : : {
3263 : 51 : ConvertRowtypeExpr *convexpr = (ConvertRowtypeExpr *) node;
3264 : 51 : ConvertRowtypeExpr *newnode;
3265 : :
3266 : 51 : FLATCOPY(newnode, convexpr, ConvertRowtypeExpr);
3267 : 51 : MUTATE(newnode->arg, convexpr->arg, Expr *);
3268 : 51 : return (Node *) newnode;
3269 : 51 : }
3270 : : break;
3271 : : case T_CollateExpr:
3272 : : {
3273 : 1126 : CollateExpr *collate = (CollateExpr *) node;
3274 : 1126 : CollateExpr *newnode;
3275 : :
3276 : 1126 : FLATCOPY(newnode, collate, CollateExpr);
3277 : 1126 : MUTATE(newnode->arg, collate->arg, Expr *);
3278 : 1126 : return (Node *) newnode;
3279 : 1126 : }
3280 : : break;
3281 : : case T_CaseExpr:
3282 : : {
3283 : 8911 : CaseExpr *caseexpr = (CaseExpr *) node;
3284 : 8911 : CaseExpr *newnode;
3285 : :
3286 : 8911 : FLATCOPY(newnode, caseexpr, CaseExpr);
3287 : 8911 : MUTATE(newnode->arg, caseexpr->arg, Expr *);
3288 : 8911 : MUTATE(newnode->args, caseexpr->args, List *);
3289 : 8911 : MUTATE(newnode->defresult, caseexpr->defresult, Expr *);
3290 : 8911 : return (Node *) newnode;
3291 : 8911 : }
3292 : : break;
3293 : : case T_CaseWhen:
3294 : : {
3295 : 16404 : CaseWhen *casewhen = (CaseWhen *) node;
3296 : 16404 : CaseWhen *newnode;
3297 : :
3298 : 16404 : FLATCOPY(newnode, casewhen, CaseWhen);
3299 : 16404 : MUTATE(newnode->expr, casewhen->expr, Expr *);
3300 : 16404 : MUTATE(newnode->result, casewhen->result, Expr *);
3301 : 16404 : return (Node *) newnode;
3302 : 16404 : }
3303 : : break;
3304 : : case T_ArrayExpr:
3305 : : {
3306 : 4196 : ArrayExpr *arrayexpr = (ArrayExpr *) node;
3307 : 4196 : ArrayExpr *newnode;
3308 : :
3309 : 4196 : FLATCOPY(newnode, arrayexpr, ArrayExpr);
3310 : 4196 : MUTATE(newnode->elements, arrayexpr->elements, List *);
3311 : 4196 : return (Node *) newnode;
3312 : 4196 : }
3313 : : break;
3314 : : case T_RowExpr:
3315 : : {
3316 : 1214 : RowExpr *rowexpr = (RowExpr *) node;
3317 : 1214 : RowExpr *newnode;
3318 : :
3319 : 1214 : FLATCOPY(newnode, rowexpr, RowExpr);
3320 : 1214 : MUTATE(newnode->args, rowexpr->args, List *);
3321 : : /* Assume colnames needn't be duplicated */
3322 : 1214 : return (Node *) newnode;
3323 : 1214 : }
3324 : : break;
3325 : : case T_RowCompareExpr:
3326 : : {
3327 : 98 : RowCompareExpr *rcexpr = (RowCompareExpr *) node;
3328 : 98 : RowCompareExpr *newnode;
3329 : :
3330 : 98 : FLATCOPY(newnode, rcexpr, RowCompareExpr);
3331 : 98 : MUTATE(newnode->largs, rcexpr->largs, List *);
3332 : 98 : MUTATE(newnode->rargs, rcexpr->rargs, List *);
3333 : 98 : return (Node *) newnode;
3334 : 98 : }
3335 : : break;
3336 : : case T_CoalesceExpr:
3337 : : {
3338 : 1758 : CoalesceExpr *coalesceexpr = (CoalesceExpr *) node;
3339 : 1758 : CoalesceExpr *newnode;
3340 : :
3341 : 1758 : FLATCOPY(newnode, coalesceexpr, CoalesceExpr);
3342 : 1758 : MUTATE(newnode->args, coalesceexpr->args, List *);
3343 : 1758 : return (Node *) newnode;
3344 : 1758 : }
3345 : : break;
3346 : : case T_MinMaxExpr:
3347 : : {
3348 : 221 : MinMaxExpr *minmaxexpr = (MinMaxExpr *) node;
3349 : 221 : MinMaxExpr *newnode;
3350 : :
3351 : 221 : FLATCOPY(newnode, minmaxexpr, MinMaxExpr);
3352 : 221 : MUTATE(newnode->args, minmaxexpr->args, List *);
3353 : 221 : return (Node *) newnode;
3354 : 221 : }
3355 : : break;
3356 : : case T_XmlExpr:
3357 : : {
3358 : 137 : XmlExpr *xexpr = (XmlExpr *) node;
3359 : 137 : XmlExpr *newnode;
3360 : :
3361 : 137 : FLATCOPY(newnode, xexpr, XmlExpr);
3362 : 137 : MUTATE(newnode->named_args, xexpr->named_args, List *);
3363 : : /* assume mutator does not care about arg_names */
3364 : 137 : MUTATE(newnode->args, xexpr->args, List *);
3365 : 137 : return (Node *) newnode;
3366 : 137 : }
3367 : : break;
3368 : : case T_JsonReturning:
3369 : : {
3370 : 404 : JsonReturning *jr = (JsonReturning *) node;
3371 : 404 : JsonReturning *newnode;
3372 : :
3373 : 404 : FLATCOPY(newnode, jr, JsonReturning);
3374 : 404 : MUTATE(newnode->format, jr->format, JsonFormat *);
3375 : :
3376 : 404 : return (Node *) newnode;
3377 : 404 : }
3378 : : case T_JsonValueExpr:
3379 : : {
3380 : 28 : JsonValueExpr *jve = (JsonValueExpr *) node;
3381 : 28 : JsonValueExpr *newnode;
3382 : :
3383 : 28 : FLATCOPY(newnode, jve, JsonValueExpr);
3384 : 28 : MUTATE(newnode->raw_expr, jve->raw_expr, Expr *);
3385 : 28 : MUTATE(newnode->formatted_expr, jve->formatted_expr, Expr *);
3386 : 28 : MUTATE(newnode->format, jve->format, JsonFormat *);
3387 : :
3388 : 28 : return (Node *) newnode;
3389 : 28 : }
3390 : : case T_JsonConstructorExpr:
3391 : : {
3392 : 404 : JsonConstructorExpr *jce = (JsonConstructorExpr *) node;
3393 : 404 : JsonConstructorExpr *newnode;
3394 : :
3395 : 404 : FLATCOPY(newnode, jce, JsonConstructorExpr);
3396 : 404 : MUTATE(newnode->args, jce->args, List *);
3397 : 404 : MUTATE(newnode->func, jce->func, Expr *);
3398 : 404 : MUTATE(newnode->coercion, jce->coercion, Expr *);
3399 : 404 : MUTATE(newnode->returning, jce->returning, JsonReturning *);
3400 : :
3401 : 404 : return (Node *) newnode;
3402 : 404 : }
3403 : : case T_JsonIsPredicate:
3404 : : {
3405 : 77 : JsonIsPredicate *pred = (JsonIsPredicate *) node;
3406 : 77 : JsonIsPredicate *newnode;
3407 : :
3408 : 77 : FLATCOPY(newnode, pred, JsonIsPredicate);
3409 : 77 : MUTATE(newnode->expr, pred->expr, Node *);
3410 : 77 : MUTATE(newnode->format, pred->format, JsonFormat *);
3411 : :
3412 : 77 : return (Node *) newnode;
3413 : 77 : }
3414 : : case T_JsonExpr:
3415 : : {
3416 : 645 : JsonExpr *jexpr = (JsonExpr *) node;
3417 : 645 : JsonExpr *newnode;
3418 : :
3419 : 645 : FLATCOPY(newnode, jexpr, JsonExpr);
3420 : 645 : MUTATE(newnode->formatted_expr, jexpr->formatted_expr, Node *);
3421 : 645 : MUTATE(newnode->path_spec, jexpr->path_spec, Node *);
3422 : 645 : MUTATE(newnode->passing_values, jexpr->passing_values, List *);
3423 : : /* assume mutator does not care about passing_names */
3424 : 645 : MUTATE(newnode->on_empty, jexpr->on_empty, JsonBehavior *);
3425 : 645 : MUTATE(newnode->on_error, jexpr->on_error, JsonBehavior *);
3426 : 645 : return (Node *) newnode;
3427 : 645 : }
3428 : : break;
3429 : : case T_JsonBehavior:
3430 : : {
3431 : 1132 : JsonBehavior *behavior = (JsonBehavior *) node;
3432 : 1132 : JsonBehavior *newnode;
3433 : :
3434 : 1132 : FLATCOPY(newnode, behavior, JsonBehavior);
3435 : 1132 : MUTATE(newnode->expr, behavior->expr, Node *);
3436 : 1132 : return (Node *) newnode;
3437 : 1132 : }
3438 : : break;
3439 : : case T_NullTest:
3440 : : {
3441 : 5392 : NullTest *ntest = (NullTest *) node;
3442 : 5392 : NullTest *newnode;
3443 : :
3444 : 5392 : FLATCOPY(newnode, ntest, NullTest);
3445 : 5392 : MUTATE(newnode->arg, ntest->arg, Expr *);
3446 : 5392 : return (Node *) newnode;
3447 : 5392 : }
3448 : : break;
3449 : : case T_BooleanTest:
3450 : : {
3451 : 133 : BooleanTest *btest = (BooleanTest *) node;
3452 : 133 : BooleanTest *newnode;
3453 : :
3454 : 133 : FLATCOPY(newnode, btest, BooleanTest);
3455 : 133 : MUTATE(newnode->arg, btest->arg, Expr *);
3456 : 133 : return (Node *) newnode;
3457 : 133 : }
3458 : : break;
3459 : : case T_CoerceToDomain:
3460 : : {
3461 : 2649 : CoerceToDomain *ctest = (CoerceToDomain *) node;
3462 : 2649 : CoerceToDomain *newnode;
3463 : :
3464 : 2649 : FLATCOPY(newnode, ctest, CoerceToDomain);
3465 : 2649 : MUTATE(newnode->arg, ctest->arg, Expr *);
3466 : 2649 : return (Node *) newnode;
3467 : 2649 : }
3468 : : break;
3469 : : case T_ReturningExpr:
3470 : : {
3471 : 313 : ReturningExpr *rexpr = (ReturningExpr *) node;
3472 : 313 : ReturningExpr *newnode;
3473 : :
3474 : 313 : FLATCOPY(newnode, rexpr, ReturningExpr);
3475 : 313 : MUTATE(newnode->retexpr, rexpr->retexpr, Expr *);
3476 : 313 : return (Node *) newnode;
3477 : 313 : }
3478 : : break;
3479 : : case T_TargetEntry:
3480 : : {
3481 : 413561 : TargetEntry *targetentry = (TargetEntry *) node;
3482 : 413561 : TargetEntry *newnode;
3483 : :
3484 : 413561 : FLATCOPY(newnode, targetentry, TargetEntry);
3485 : 413561 : MUTATE(newnode->expr, targetentry->expr, Expr *);
3486 : 413561 : return (Node *) newnode;
3487 : 413561 : }
3488 : : break;
3489 : : case T_Query:
3490 : : /* Do nothing with a sub-Query, per discussion above */
3491 : 3949 : return node;
3492 : : case T_WindowClause:
3493 : : {
3494 : 0 : WindowClause *wc = (WindowClause *) node;
3495 : 0 : WindowClause *newnode;
3496 : :
3497 : 0 : FLATCOPY(newnode, wc, WindowClause);
3498 : 0 : MUTATE(newnode->partitionClause, wc->partitionClause, List *);
3499 : 0 : MUTATE(newnode->orderClause, wc->orderClause, List *);
3500 : 0 : MUTATE(newnode->startOffset, wc->startOffset, Node *);
3501 : 0 : MUTATE(newnode->endOffset, wc->endOffset, Node *);
3502 : 0 : return (Node *) newnode;
3503 : 0 : }
3504 : : break;
3505 : : case T_CTECycleClause:
3506 : : {
3507 : 0 : CTECycleClause *cc = (CTECycleClause *) node;
3508 : 0 : CTECycleClause *newnode;
3509 : :
3510 : 0 : FLATCOPY(newnode, cc, CTECycleClause);
3511 : 0 : MUTATE(newnode->cycle_mark_value, cc->cycle_mark_value, Node *);
3512 : 0 : MUTATE(newnode->cycle_mark_default, cc->cycle_mark_default, Node *);
3513 : 0 : return (Node *) newnode;
3514 : 0 : }
3515 : : break;
3516 : : case T_CommonTableExpr:
3517 : : {
3518 : 19 : CommonTableExpr *cte = (CommonTableExpr *) node;
3519 : 19 : CommonTableExpr *newnode;
3520 : :
3521 : 19 : FLATCOPY(newnode, cte, CommonTableExpr);
3522 : :
3523 : : /*
3524 : : * Also invoke the mutator on the CTE's Query node, so it can
3525 : : * recurse into the sub-query if it wants to.
3526 : : */
3527 : 19 : MUTATE(newnode->ctequery, cte->ctequery, Node *);
3528 : :
3529 : 19 : MUTATE(newnode->search_clause, cte->search_clause, CTESearchClause *);
3530 : 19 : MUTATE(newnode->cycle_clause, cte->cycle_clause, CTECycleClause *);
3531 : :
3532 : 19 : return (Node *) newnode;
3533 : 19 : }
3534 : : break;
3535 : : case T_PartitionBoundSpec:
3536 : : {
3537 : 0 : PartitionBoundSpec *pbs = (PartitionBoundSpec *) node;
3538 : 0 : PartitionBoundSpec *newnode;
3539 : :
3540 : 0 : FLATCOPY(newnode, pbs, PartitionBoundSpec);
3541 : 0 : MUTATE(newnode->listdatums, pbs->listdatums, List *);
3542 : 0 : MUTATE(newnode->lowerdatums, pbs->lowerdatums, List *);
3543 : 0 : MUTATE(newnode->upperdatums, pbs->upperdatums, List *);
3544 : 0 : return (Node *) newnode;
3545 : 0 : }
3546 : : break;
3547 : : case T_PartitionRangeDatum:
3548 : : {
3549 : 0 : PartitionRangeDatum *prd = (PartitionRangeDatum *) node;
3550 : 0 : PartitionRangeDatum *newnode;
3551 : :
3552 : 0 : FLATCOPY(newnode, prd, PartitionRangeDatum);
3553 : 0 : MUTATE(newnode->value, prd->value, Node *);
3554 : 0 : return (Node *) newnode;
3555 : 0 : }
3556 : : break;
3557 : : case T_List:
3558 : : {
3559 : : /*
3560 : : * We assume the mutator isn't interested in the list nodes
3561 : : * per se, so just invoke it on each list element. NOTE: this
3562 : : * would fail badly on a list with integer elements!
3563 : : */
3564 : 562960 : List *resultlist;
3565 : 562960 : ListCell *temp;
3566 : :
3567 : 562960 : resultlist = NIL;
3568 [ + - + + : 1775791 : foreach(temp, (List *) node)
+ + ]
3569 : : {
3570 : 2425662 : resultlist = lappend(resultlist,
3571 : 2425662 : mutator((Node *) lfirst(temp),
3572 : 1212831 : context));
3573 : 1212831 : }
3574 : 562960 : return (Node *) resultlist;
3575 : 562960 : }
3576 : : break;
3577 : : case T_FromExpr:
3578 : : {
3579 : 3293 : FromExpr *from = (FromExpr *) node;
3580 : 3293 : FromExpr *newnode;
3581 : :
3582 : 3293 : FLATCOPY(newnode, from, FromExpr);
3583 : 3293 : MUTATE(newnode->fromlist, from->fromlist, List *);
3584 : 3293 : MUTATE(newnode->quals, from->quals, Node *);
3585 : 3293 : return (Node *) newnode;
3586 : 3293 : }
3587 : : break;
3588 : : case T_OnConflictExpr:
3589 : : {
3590 : 60 : OnConflictExpr *oc = (OnConflictExpr *) node;
3591 : 60 : OnConflictExpr *newnode;
3592 : :
3593 : 60 : FLATCOPY(newnode, oc, OnConflictExpr);
3594 : 60 : MUTATE(newnode->arbiterElems, oc->arbiterElems, List *);
3595 : 60 : MUTATE(newnode->arbiterWhere, oc->arbiterWhere, Node *);
3596 : 60 : MUTATE(newnode->onConflictSet, oc->onConflictSet, List *);
3597 : 60 : MUTATE(newnode->onConflictWhere, oc->onConflictWhere, Node *);
3598 : 60 : MUTATE(newnode->exclRelTlist, oc->exclRelTlist, List *);
3599 : :
3600 : 60 : return (Node *) newnode;
3601 : 60 : }
3602 : : break;
3603 : : case T_MergeAction:
3604 : : {
3605 : 177 : MergeAction *action = (MergeAction *) node;
3606 : 177 : MergeAction *newnode;
3607 : :
3608 : 177 : FLATCOPY(newnode, action, MergeAction);
3609 : 177 : MUTATE(newnode->qual, action->qual, Node *);
3610 : 177 : MUTATE(newnode->targetList, action->targetList, List *);
3611 : :
3612 : 177 : return (Node *) newnode;
3613 : 177 : }
3614 : : break;
3615 : : case T_PartitionPruneStepOp:
3616 : : {
3617 : 26 : PartitionPruneStepOp *opstep = (PartitionPruneStepOp *) node;
3618 : 26 : PartitionPruneStepOp *newnode;
3619 : :
3620 : 26 : FLATCOPY(newnode, opstep, PartitionPruneStepOp);
3621 : 26 : MUTATE(newnode->exprs, opstep->exprs, List *);
3622 : :
3623 : 26 : return (Node *) newnode;
3624 : 26 : }
3625 : : break;
3626 : : case T_PartitionPruneStepCombine:
3627 : : /* no expression sub-nodes */
3628 : 2 : return copyObject(node);
3629 : : case T_JoinExpr:
3630 : : {
3631 : 673 : JoinExpr *join = (JoinExpr *) node;
3632 : 673 : JoinExpr *newnode;
3633 : :
3634 : 673 : FLATCOPY(newnode, join, JoinExpr);
3635 : 673 : MUTATE(newnode->larg, join->larg, Node *);
3636 : 673 : MUTATE(newnode->rarg, join->rarg, Node *);
3637 : 673 : MUTATE(newnode->quals, join->quals, Node *);
3638 : : /* We do not mutate alias or using by default */
3639 : 673 : return (Node *) newnode;
3640 : 673 : }
3641 : : break;
3642 : : case T_SetOperationStmt:
3643 : : {
3644 : 29 : SetOperationStmt *setop = (SetOperationStmt *) node;
3645 : 29 : SetOperationStmt *newnode;
3646 : :
3647 : 29 : FLATCOPY(newnode, setop, SetOperationStmt);
3648 : 29 : MUTATE(newnode->larg, setop->larg, Node *);
3649 : 29 : MUTATE(newnode->rarg, setop->rarg, Node *);
3650 : : /* We do not mutate groupClauses by default */
3651 : 29 : return (Node *) newnode;
3652 : 29 : }
3653 : : break;
3654 : : case T_IndexClause:
3655 : : {
3656 : 99 : IndexClause *iclause = (IndexClause *) node;
3657 : 99 : IndexClause *newnode;
3658 : :
3659 : 99 : FLATCOPY(newnode, iclause, IndexClause);
3660 : 99 : MUTATE(newnode->rinfo, iclause->rinfo, RestrictInfo *);
3661 : 99 : MUTATE(newnode->indexquals, iclause->indexquals, List *);
3662 : 99 : return (Node *) newnode;
3663 : 99 : }
3664 : : break;
3665 : : case T_PlaceHolderVar:
3666 : : {
3667 : 2204 : PlaceHolderVar *phv = (PlaceHolderVar *) node;
3668 : 2204 : PlaceHolderVar *newnode;
3669 : :
3670 : 2204 : FLATCOPY(newnode, phv, PlaceHolderVar);
3671 : 2204 : MUTATE(newnode->phexpr, phv->phexpr, Expr *);
3672 : : /* Assume we need not copy the relids bitmapsets */
3673 : 2204 : return (Node *) newnode;
3674 : 2204 : }
3675 : : break;
3676 : : case T_InferenceElem:
3677 : : {
3678 : 401 : InferenceElem *inferenceelemdexpr = (InferenceElem *) node;
3679 : 401 : InferenceElem *newnode;
3680 : :
3681 : 401 : FLATCOPY(newnode, inferenceelemdexpr, InferenceElem);
3682 : 401 : MUTATE(newnode->expr, newnode->expr, Node *);
3683 : 401 : return (Node *) newnode;
3684 : 401 : }
3685 : : break;
3686 : : case T_AppendRelInfo:
3687 : : {
3688 : 2597 : AppendRelInfo *appinfo = (AppendRelInfo *) node;
3689 : 2597 : AppendRelInfo *newnode;
3690 : :
3691 : 2597 : FLATCOPY(newnode, appinfo, AppendRelInfo);
3692 : 2597 : MUTATE(newnode->translated_vars, appinfo->translated_vars, List *);
3693 : : /* Assume nothing need be done with parent_colnos[] */
3694 : 2597 : return (Node *) newnode;
3695 : 2597 : }
3696 : : break;
3697 : : case T_PlaceHolderInfo:
3698 : : {
3699 : 0 : PlaceHolderInfo *phinfo = (PlaceHolderInfo *) node;
3700 : 0 : PlaceHolderInfo *newnode;
3701 : :
3702 : 0 : FLATCOPY(newnode, phinfo, PlaceHolderInfo);
3703 : 0 : MUTATE(newnode->ph_var, phinfo->ph_var, PlaceHolderVar *);
3704 : : /* Assume we need not copy the relids bitmapsets */
3705 : 0 : return (Node *) newnode;
3706 : 0 : }
3707 : : break;
3708 : : case T_RangeTblFunction:
3709 : : {
3710 : 7457 : RangeTblFunction *rtfunc = (RangeTblFunction *) node;
3711 : 7457 : RangeTblFunction *newnode;
3712 : :
3713 : 7457 : FLATCOPY(newnode, rtfunc, RangeTblFunction);
3714 : 7457 : MUTATE(newnode->funcexpr, rtfunc->funcexpr, Node *);
3715 : : /* Assume we need not copy the coldef info lists */
3716 : 7457 : return (Node *) newnode;
3717 : 7457 : }
3718 : : break;
3719 : : case T_TableSampleClause:
3720 : : {
3721 : 67 : TableSampleClause *tsc = (TableSampleClause *) node;
3722 : 67 : TableSampleClause *newnode;
3723 : :
3724 : 67 : FLATCOPY(newnode, tsc, TableSampleClause);
3725 : 67 : MUTATE(newnode->args, tsc->args, List *);
3726 : 67 : MUTATE(newnode->repeatable, tsc->repeatable, Expr *);
3727 : 67 : return (Node *) newnode;
3728 : 67 : }
3729 : : break;
3730 : : case T_TableFunc:
3731 : : {
3732 : 170 : TableFunc *tf = (TableFunc *) node;
3733 : 170 : TableFunc *newnode;
3734 : :
3735 : 170 : FLATCOPY(newnode, tf, TableFunc);
3736 : 170 : MUTATE(newnode->ns_uris, tf->ns_uris, List *);
3737 : 170 : MUTATE(newnode->docexpr, tf->docexpr, Node *);
3738 : 170 : MUTATE(newnode->rowexpr, tf->rowexpr, Node *);
3739 : 170 : MUTATE(newnode->colexprs, tf->colexprs, List *);
3740 : 170 : MUTATE(newnode->coldefexprs, tf->coldefexprs, List *);
3741 : 170 : MUTATE(newnode->colvalexprs, tf->colvalexprs, List *);
3742 : 170 : MUTATE(newnode->passingvalexprs, tf->passingvalexprs, List *);
3743 : 170 : return (Node *) newnode;
3744 : 170 : }
3745 : : break;
3746 : : default:
3747 [ # # # # ]: 0 : elog(ERROR, "unrecognized node type: %d",
3748 : : (int) nodeTag(node));
3749 : 0 : break;
3750 : : }
3751 : : /* can't get here, but keep compiler happy */
3752 : 0 : return NULL;
3753 : 1976587 : }
3754 : :
3755 : :
3756 : : /*
3757 : : * query_tree_mutator --- initiate modification of a Query's expressions
3758 : : *
3759 : : * This routine exists just to reduce the number of places that need to know
3760 : : * where all the expression subtrees of a Query are. Note it can be used
3761 : : * for starting a walk at top level of a Query regardless of whether the
3762 : : * mutator intends to descend into subqueries. It is also useful for
3763 : : * descending into subqueries within a mutator.
3764 : : *
3765 : : * Some callers want to suppress mutating of certain items in the Query,
3766 : : * typically because they need to process them specially, or don't actually
3767 : : * want to recurse into subqueries. This is supported by the flags argument,
3768 : : * which is the bitwise OR of flag values to suppress mutating of
3769 : : * indicated items. (More flag bits may be added as needed.)
3770 : : *
3771 : : * Normally the top-level Query node itself is copied, but some callers want
3772 : : * it to be modified in-place; they must pass QTW_DONT_COPY_QUERY in flags.
3773 : : * All modified substructure is safely copied in any case.
3774 : : */
3775 : : Query *
3776 : 3250 : query_tree_mutator_impl(Query *query,
3777 : : tree_mutator_callback mutator,
3778 : : void *context,
3779 : : int flags)
3780 : : {
3781 [ + - ]: 3250 : Assert(query != NULL && IsA(query, Query));
3782 : :
3783 [ + - ]: 3250 : if (!(flags & QTW_DONT_COPY_QUERY))
3784 : : {
3785 : 3250 : Query *newquery;
3786 : :
3787 : 3250 : FLATCOPY(newquery, query, Query);
3788 : 3250 : query = newquery;
3789 : 3250 : }
3790 : :
3791 : 3250 : MUTATE(query->targetList, query->targetList, List *);
3792 : 3250 : MUTATE(query->withCheckOptions, query->withCheckOptions, List *);
3793 : 3250 : MUTATE(query->onConflict, query->onConflict, OnConflictExpr *);
3794 : 3250 : MUTATE(query->mergeActionList, query->mergeActionList, List *);
3795 : 3250 : MUTATE(query->mergeJoinCondition, query->mergeJoinCondition, Node *);
3796 : 3250 : MUTATE(query->returningList, query->returningList, List *);
3797 : 3250 : MUTATE(query->jointree, query->jointree, FromExpr *);
3798 : 3250 : MUTATE(query->setOperations, query->setOperations, Node *);
3799 : 3250 : MUTATE(query->havingQual, query->havingQual, Node *);
3800 : 3250 : MUTATE(query->limitOffset, query->limitOffset, Node *);
3801 : 3250 : MUTATE(query->limitCount, query->limitCount, Node *);
3802 : :
3803 : : /*
3804 : : * Most callers aren't interested in SortGroupClause nodes since those
3805 : : * don't contain actual expressions. However they do contain OIDs, which
3806 : : * may be of interest to some mutators.
3807 : : */
3808 : :
3809 [ + - ]: 3250 : if ((flags & QTW_EXAMINE_SORTGROUP))
3810 : : {
3811 : 0 : MUTATE(query->groupClause, query->groupClause, List *);
3812 : 0 : MUTATE(query->windowClause, query->windowClause, List *);
3813 : 0 : MUTATE(query->sortClause, query->sortClause, List *);
3814 : 0 : MUTATE(query->distinctClause, query->distinctClause, List *);
3815 : 0 : }
3816 : : else
3817 : : {
3818 : : /*
3819 : : * But we need to mutate the expressions under WindowClause nodes even
3820 : : * if we're not interested in SortGroupClause nodes.
3821 : : */
3822 : 3250 : List *resultlist;
3823 : 3250 : ListCell *temp;
3824 : :
3825 : 3250 : resultlist = NIL;
3826 [ + + + + : 3258 : foreach(temp, query->windowClause)
+ + ]
3827 : : {
3828 : 8 : WindowClause *wc = lfirst_node(WindowClause, temp);
3829 : 8 : WindowClause *newnode;
3830 : :
3831 : 8 : FLATCOPY(newnode, wc, WindowClause);
3832 : 8 : MUTATE(newnode->startOffset, wc->startOffset, Node *);
3833 : 8 : MUTATE(newnode->endOffset, wc->endOffset, Node *);
3834 : :
3835 : 8 : resultlist = lappend(resultlist, (Node *) newnode);
3836 : 8 : }
3837 : 3250 : query->windowClause = resultlist;
3838 : 3250 : }
3839 : :
3840 : : /*
3841 : : * groupingSets and rowMarks are not mutated:
3842 : : *
3843 : : * groupingSets contain only ressortgroup refs (integers) which are
3844 : : * meaningless without the groupClause or tlist. Accordingly, any mutator
3845 : : * that needs to care about them needs to handle them itself in its Query
3846 : : * processing.
3847 : : *
3848 : : * rowMarks contains only rangetable indexes (and flags etc.) and
3849 : : * therefore should be handled at Query level similarly.
3850 : : */
3851 : :
3852 [ + - ]: 3250 : if (!(flags & QTW_IGNORE_CTE_SUBQUERIES))
3853 : 3250 : MUTATE(query->cteList, query->cteList, List *);
3854 : : else /* else copy CTE list as-is */
3855 : 0 : query->cteList = copyObject(query->cteList);
3856 : 3250 : query->rtable = range_table_mutator(query->rtable,
3857 : : mutator, context, flags);
3858 : 3250 : return query;
3859 : : }
3860 : :
3861 : : /*
3862 : : * range_table_mutator is just the part of query_tree_mutator that processes
3863 : : * a query's rangetable. This is split out since it can be useful on
3864 : : * its own.
3865 : : */
3866 : : List *
3867 : 3250 : range_table_mutator_impl(List *rtable,
3868 : : tree_mutator_callback mutator,
3869 : : void *context,
3870 : : int flags)
3871 : : {
3872 : 3250 : List *newrt = NIL;
3873 : 3250 : ListCell *rt;
3874 : :
3875 [ + + + + : 9984 : foreach(rt, rtable)
+ + ]
3876 : : {
3877 : 6734 : RangeTblEntry *rte = (RangeTblEntry *) lfirst(rt);
3878 : 6734 : RangeTblEntry *newrte;
3879 : :
3880 : 6734 : FLATCOPY(newrte, rte, RangeTblEntry);
3881 [ + + - + : 6734 : switch (rte->rtekind)
+ - + +
+ ]
3882 : : {
3883 : : case RTE_RELATION:
3884 : 4269 : MUTATE(newrte->tablesample, rte->tablesample,
3885 : : TableSampleClause *);
3886 : : /* we don't bother to copy eref, aliases, etc; OK? */
3887 : 4269 : break;
3888 : : case RTE_SUBQUERY:
3889 [ - + ]: 529 : if (!(flags & QTW_IGNORE_RT_SUBQUERIES))
3890 : 529 : MUTATE(newrte->subquery, rte->subquery, Query *);
3891 : : else
3892 : : {
3893 : : /* else, copy RT subqueries as-is */
3894 : 0 : newrte->subquery = copyObject(rte->subquery);
3895 : : }
3896 : 529 : break;
3897 : : case RTE_JOIN:
3898 [ + + ]: 677 : if (!(flags & QTW_IGNORE_JOINALIASES))
3899 : 663 : MUTATE(newrte->joinaliasvars, rte->joinaliasvars, List *);
3900 : : else
3901 : : {
3902 : : /* else, copy join aliases as-is */
3903 : 14 : newrte->joinaliasvars = copyObject(rte->joinaliasvars);
3904 : : }
3905 : 677 : break;
3906 : : case RTE_FUNCTION:
3907 : 925 : MUTATE(newrte->functions, rte->functions, List *);
3908 : 925 : break;
3909 : : case RTE_TABLEFUNC:
3910 : 0 : MUTATE(newrte->tablefunc, rte->tablefunc, TableFunc *);
3911 : 0 : break;
3912 : : case RTE_VALUES:
3913 : 229 : MUTATE(newrte->values_lists, rte->values_lists, List *);
3914 : 229 : break;
3915 : : case RTE_CTE:
3916 : : case RTE_NAMEDTUPLESTORE:
3917 : : case RTE_RESULT:
3918 : : /* nothing to do */
3919 : 64 : break;
3920 : : case RTE_GROUP:
3921 [ - + ]: 41 : if (!(flags & QTW_IGNORE_GROUPEXPRS))
3922 : 41 : MUTATE(newrte->groupexprs, rte->groupexprs, List *);
3923 : : else
3924 : : {
3925 : : /* else, copy grouping exprs as-is */
3926 : 0 : newrte->groupexprs = copyObject(rte->groupexprs);
3927 : : }
3928 : 41 : break;
3929 : : }
3930 : 6734 : MUTATE(newrte->securityQuals, rte->securityQuals, List *);
3931 : 6734 : newrt = lappend(newrt, newrte);
3932 : 6734 : }
3933 : 6500 : return newrt;
3934 : 3250 : }
3935 : :
3936 : : /*
3937 : : * query_or_expression_tree_walker --- hybrid form
3938 : : *
3939 : : * This routine will invoke query_tree_walker if called on a Query node,
3940 : : * else will invoke the walker directly. This is a useful way of starting
3941 : : * the recursion when the walker's normal change of state is not appropriate
3942 : : * for the outermost Query node.
3943 : : */
3944 : : bool
3945 : 405805 : query_or_expression_tree_walker_impl(Node *node,
3946 : : tree_walker_callback walker,
3947 : : void *context,
3948 : : int flags)
3949 : : {
3950 [ + + + + ]: 405805 : if (node && IsA(node, Query))
3951 : 47631 : return query_tree_walker((Query *) node,
3952 : : walker,
3953 : : context,
3954 : : flags);
3955 : : else
3956 : 358174 : return WALK(node);
3957 : 405805 : }
3958 : :
3959 : : /*
3960 : : * query_or_expression_tree_mutator --- hybrid form
3961 : : *
3962 : : * This routine will invoke query_tree_mutator if called on a Query node,
3963 : : * else will invoke the mutator directly. This is a useful way of starting
3964 : : * the recursion when the mutator's normal change of state is not appropriate
3965 : : * for the outermost Query node.
3966 : : */
3967 : : Node *
3968 : 58943 : query_or_expression_tree_mutator_impl(Node *node,
3969 : : tree_mutator_callback mutator,
3970 : : void *context,
3971 : : int flags)
3972 : : {
3973 [ + + + + ]: 58943 : if (node && IsA(node, Query))
3974 : 1274 : return (Node *) query_tree_mutator((Query *) node,
3975 : : mutator,
3976 : : context,
3977 : : flags);
3978 : : else
3979 : 57669 : return mutator(node, context);
3980 : 58943 : }
3981 : :
3982 : :
3983 : : /*
3984 : : * raw_expression_tree_walker --- walk raw parse trees
3985 : : *
3986 : : * This has exactly the same API as expression_tree_walker, but instead of
3987 : : * walking post-analysis parse trees, it knows how to walk the node types
3988 : : * found in raw grammar output. (There is not currently any need for a
3989 : : * combined walker, so we keep them separate in the name of efficiency.)
3990 : : * Unlike expression_tree_walker, there is no special rule about query
3991 : : * boundaries: we descend to everything that's possibly interesting.
3992 : : *
3993 : : * Currently, the node type coverage here extends only to DML statements
3994 : : * (SELECT/INSERT/UPDATE/DELETE/MERGE) and nodes that can appear in them,
3995 : : * because this is used mainly during analysis of CTEs, and only DML
3996 : : * statements can appear in CTEs.
3997 : : */
3998 : : bool
3999 : 4527 : raw_expression_tree_walker_impl(Node *node,
4000 : : tree_walker_callback walker,
4001 : : void *context)
4002 : : {
4003 : 4527 : ListCell *temp;
4004 : :
4005 : : /*
4006 : : * The walker has already visited the current node, and so we need only
4007 : : * recurse into any sub-nodes it has.
4008 : : */
4009 [ + - ]: 4527 : if (node == NULL)
4010 : 0 : return false;
4011 : :
4012 : : /* Guard against stack overflow due to overly complex expressions */
4013 : 4527 : check_stack_depth();
4014 : :
4015 [ + + - - : 4527 : switch (nodeTag(node))
+ - + - -
- - - - -
- - - - -
- - - - -
- + - + +
+ + + + +
+ - + + +
- - - + +
- + + + +
+ - - - -
+ - - - +
- - - - +
- - - - -
- - - - ]
4016 : : {
4017 : : case T_JsonFormat:
4018 : : case T_SetToDefault:
4019 : : case T_CurrentOfExpr:
4020 : : case T_SQLValueFunction:
4021 : : case T_Integer:
4022 : : case T_Float:
4023 : : case T_Boolean:
4024 : : case T_String:
4025 : : case T_BitString:
4026 : : case T_ParamRef:
4027 : : case T_A_Const:
4028 : : case T_A_Star:
4029 : : case T_MergeSupportFunc:
4030 : : case T_ReturningOption:
4031 : : /* primitive node types with no subnodes */
4032 : 507 : break;
4033 : : case T_Alias:
4034 : : /* we assume the colnames list isn't interesting */
4035 : : break;
4036 : : case T_RangeVar:
4037 : 0 : return WALK(((RangeVar *) node)->alias);
4038 : : case T_GroupingFunc:
4039 : 0 : return WALK(((GroupingFunc *) node)->args);
4040 : : case T_SubLink:
4041 : : {
4042 : 5 : SubLink *sublink = (SubLink *) node;
4043 : :
4044 [ - + ]: 5 : if (WALK(sublink->testexpr))
4045 : 0 : return true;
4046 : : /* we assume the operName is not interesting */
4047 [ - + ]: 5 : if (WALK(sublink->subselect))
4048 : 0 : return true;
4049 [ - + ]: 5 : }
4050 : 5 : break;
4051 : : case T_CaseExpr:
4052 : : {
4053 : 0 : CaseExpr *caseexpr = (CaseExpr *) node;
4054 : :
4055 [ # # ]: 0 : if (WALK(caseexpr->arg))
4056 : 0 : return true;
4057 : : /* we assume walker doesn't care about CaseWhens, either */
4058 [ # # # # : 0 : foreach(temp, caseexpr->args)
# # # # ]
4059 : : {
4060 : 0 : CaseWhen *when = lfirst_node(CaseWhen, temp);
4061 : :
4062 [ # # ]: 0 : if (WALK(when->expr))
4063 : 0 : return true;
4064 [ # # ]: 0 : if (WALK(when->result))
4065 : 0 : return true;
4066 [ # # ]: 0 : }
4067 [ # # ]: 0 : if (WALK(caseexpr->defresult))
4068 : 0 : return true;
4069 [ # # ]: 0 : }
4070 : 0 : break;
4071 : : case T_RowExpr:
4072 : : /* Assume colnames isn't interesting */
4073 : 18 : return WALK(((RowExpr *) node)->args);
4074 : : case T_CoalesceExpr:
4075 : 0 : return WALK(((CoalesceExpr *) node)->args);
4076 : : case T_MinMaxExpr:
4077 : 0 : return WALK(((MinMaxExpr *) node)->args);
4078 : : case T_XmlExpr:
4079 : : {
4080 : 0 : XmlExpr *xexpr = (XmlExpr *) node;
4081 : :
4082 [ # # ]: 0 : if (WALK(xexpr->named_args))
4083 : 0 : return true;
4084 : : /* we assume walker doesn't care about arg_names */
4085 [ # # ]: 0 : if (WALK(xexpr->args))
4086 : 0 : return true;
4087 [ # # ]: 0 : }
4088 : 0 : break;
4089 : : case T_JsonReturning:
4090 : 0 : return WALK(((JsonReturning *) node)->format);
4091 : : case T_JsonValueExpr:
4092 : : {
4093 : 0 : JsonValueExpr *jve = (JsonValueExpr *) node;
4094 : :
4095 [ # # ]: 0 : if (WALK(jve->raw_expr))
4096 : 0 : return true;
4097 [ # # ]: 0 : if (WALK(jve->formatted_expr))
4098 : 0 : return true;
4099 [ # # ]: 0 : if (WALK(jve->format))
4100 : 0 : return true;
4101 [ # # ]: 0 : }
4102 : 0 : break;
4103 : : case T_JsonParseExpr:
4104 : : {
4105 : 0 : JsonParseExpr *jpe = (JsonParseExpr *) node;
4106 : :
4107 [ # # ]: 0 : if (WALK(jpe->expr))
4108 : 0 : return true;
4109 [ # # ]: 0 : if (WALK(jpe->output))
4110 : 0 : return true;
4111 [ # # ]: 0 : }
4112 : 0 : break;
4113 : : case T_JsonScalarExpr:
4114 : : {
4115 : 0 : JsonScalarExpr *jse = (JsonScalarExpr *) node;
4116 : :
4117 [ # # ]: 0 : if (WALK(jse->expr))
4118 : 0 : return true;
4119 [ # # ]: 0 : if (WALK(jse->output))
4120 : 0 : return true;
4121 [ # # ]: 0 : }
4122 : 0 : break;
4123 : : case T_JsonSerializeExpr:
4124 : : {
4125 : 0 : JsonSerializeExpr *jse = (JsonSerializeExpr *) node;
4126 : :
4127 [ # # ]: 0 : if (WALK(jse->expr))
4128 : 0 : return true;
4129 [ # # ]: 0 : if (WALK(jse->output))
4130 : 0 : return true;
4131 [ # # ]: 0 : }
4132 : 0 : break;
4133 : : case T_JsonConstructorExpr:
4134 : : {
4135 : 0 : JsonConstructorExpr *ctor = (JsonConstructorExpr *) node;
4136 : :
4137 [ # # ]: 0 : if (WALK(ctor->args))
4138 : 0 : return true;
4139 [ # # ]: 0 : if (WALK(ctor->func))
4140 : 0 : return true;
4141 [ # # ]: 0 : if (WALK(ctor->coercion))
4142 : 0 : return true;
4143 [ # # ]: 0 : if (WALK(ctor->returning))
4144 : 0 : return true;
4145 [ # # ]: 0 : }
4146 : 0 : break;
4147 : : case T_JsonIsPredicate:
4148 : 0 : return WALK(((JsonIsPredicate *) node)->expr);
4149 : : case T_JsonArgument:
4150 : 0 : return WALK(((JsonArgument *) node)->val);
4151 : : case T_JsonFuncExpr:
4152 : : {
4153 : 0 : JsonFuncExpr *jfe = (JsonFuncExpr *) node;
4154 : :
4155 [ # # ]: 0 : if (WALK(jfe->context_item))
4156 : 0 : return true;
4157 [ # # ]: 0 : if (WALK(jfe->pathspec))
4158 : 0 : return true;
4159 [ # # ]: 0 : if (WALK(jfe->passing))
4160 : 0 : return true;
4161 [ # # ]: 0 : if (WALK(jfe->output))
4162 : 0 : return true;
4163 [ # # ]: 0 : if (WALK(jfe->on_empty))
4164 : 0 : return true;
4165 [ # # ]: 0 : if (WALK(jfe->on_error))
4166 : 0 : return true;
4167 [ # # ]: 0 : }
4168 : 0 : break;
4169 : : case T_JsonBehavior:
4170 : : {
4171 : 0 : JsonBehavior *jb = (JsonBehavior *) node;
4172 : :
4173 [ # # ]: 0 : if (WALK(jb->expr))
4174 : 0 : return true;
4175 [ # # ]: 0 : }
4176 : 0 : break;
4177 : : case T_JsonTable:
4178 : : {
4179 : 0 : JsonTable *jt = (JsonTable *) node;
4180 : :
4181 [ # # ]: 0 : if (WALK(jt->context_item))
4182 : 0 : return true;
4183 [ # # ]: 0 : if (WALK(jt->pathspec))
4184 : 0 : return true;
4185 [ # # ]: 0 : if (WALK(jt->passing))
4186 : 0 : return true;
4187 [ # # ]: 0 : if (WALK(jt->columns))
4188 : 0 : return true;
4189 [ # # ]: 0 : if (WALK(jt->on_error))
4190 : 0 : return true;
4191 [ # # ]: 0 : }
4192 : 0 : break;
4193 : : case T_JsonTableColumn:
4194 : : {
4195 : 0 : JsonTableColumn *jtc = (JsonTableColumn *) node;
4196 : :
4197 [ # # ]: 0 : if (WALK(jtc->typeName))
4198 : 0 : return true;
4199 [ # # ]: 0 : if (WALK(jtc->on_empty))
4200 : 0 : return true;
4201 [ # # ]: 0 : if (WALK(jtc->on_error))
4202 : 0 : return true;
4203 [ # # ]: 0 : if (WALK(jtc->columns))
4204 : 0 : return true;
4205 [ # # ]: 0 : }
4206 : 0 : break;
4207 : : case T_JsonTablePathSpec:
4208 : 0 : return WALK(((JsonTablePathSpec *) node)->string);
4209 : : case T_NullTest:
4210 : 0 : return WALK(((NullTest *) node)->arg);
4211 : : case T_BooleanTest:
4212 : 0 : return WALK(((BooleanTest *) node)->arg);
4213 : : case T_JoinExpr:
4214 : : {
4215 : 14 : JoinExpr *join = (JoinExpr *) node;
4216 : :
4217 [ - + ]: 14 : if (WALK(join->larg))
4218 : 0 : return true;
4219 [ - + ]: 14 : if (WALK(join->rarg))
4220 : 0 : return true;
4221 [ - + ]: 14 : if (WALK(join->quals))
4222 : 0 : return true;
4223 [ - + ]: 14 : if (WALK(join->alias))
4224 : 0 : return true;
4225 : : /* using list is deemed uninteresting */
4226 [ - + ]: 14 : }
4227 : 14 : break;
4228 : : case T_IntoClause:
4229 : : {
4230 : 0 : IntoClause *into = (IntoClause *) node;
4231 : :
4232 [ # # ]: 0 : if (WALK(into->rel))
4233 : 0 : return true;
4234 : : /* colNames, options are deemed uninteresting */
4235 : : /* viewQuery should be null in raw parsetree, but check it */
4236 [ # # ]: 0 : if (WALK(into->viewQuery))
4237 : 0 : return true;
4238 [ # # ]: 0 : }
4239 : 0 : break;
4240 : : case T_List:
4241 [ + - + + : 2671 : foreach(temp, (List *) node)
+ + - + ]
4242 : : {
4243 [ - + ]: 1472 : if (WALK((Node *) lfirst(temp)))
4244 : 0 : return true;
4245 : 1472 : }
4246 : 1199 : break;
4247 : : case T_InsertStmt:
4248 : : {
4249 : 7 : InsertStmt *stmt = (InsertStmt *) node;
4250 : :
4251 [ - + ]: 7 : if (WALK(stmt->relation))
4252 : 0 : return true;
4253 [ - + ]: 7 : if (WALK(stmt->cols))
4254 : 0 : return true;
4255 [ - + ]: 7 : if (WALK(stmt->selectStmt))
4256 : 0 : return true;
4257 [ - + ]: 7 : if (WALK(stmt->onConflictClause))
4258 : 0 : return true;
4259 [ - + ]: 7 : if (WALK(stmt->returningClause))
4260 : 0 : return true;
4261 [ - + ]: 7 : if (WALK(stmt->withClause))
4262 : 0 : return true;
4263 [ - + ]: 7 : }
4264 : 7 : break;
4265 : : case T_DeleteStmt:
4266 : : {
4267 : 1 : DeleteStmt *stmt = (DeleteStmt *) node;
4268 : :
4269 [ - + ]: 1 : if (WALK(stmt->relation))
4270 : 0 : return true;
4271 [ - + ]: 1 : if (WALK(stmt->usingClause))
4272 : 0 : return true;
4273 [ - + ]: 1 : if (WALK(stmt->whereClause))
4274 : 0 : return true;
4275 [ - + ]: 1 : if (WALK(stmt->returningClause))
4276 : 0 : return true;
4277 [ - + ]: 1 : if (WALK(stmt->withClause))
4278 : 0 : return true;
4279 [ - + ]: 1 : }
4280 : 1 : break;
4281 : : case T_UpdateStmt:
4282 : : {
4283 : 1 : UpdateStmt *stmt = (UpdateStmt *) node;
4284 : :
4285 [ - + ]: 1 : if (WALK(stmt->relation))
4286 : 0 : return true;
4287 [ - + ]: 1 : if (WALK(stmt->targetList))
4288 : 0 : return true;
4289 [ - + ]: 1 : if (WALK(stmt->whereClause))
4290 : 0 : return true;
4291 [ - + ]: 1 : if (WALK(stmt->fromClause))
4292 : 0 : return true;
4293 [ - + ]: 1 : if (WALK(stmt->returningClause))
4294 : 0 : return true;
4295 [ - + ]: 1 : if (WALK(stmt->withClause))
4296 : 0 : return true;
4297 [ - + ]: 1 : }
4298 : 1 : break;
4299 : : case T_MergeStmt:
4300 : : {
4301 : 1 : MergeStmt *stmt = (MergeStmt *) node;
4302 : :
4303 [ - + ]: 1 : if (WALK(stmt->relation))
4304 : 0 : return true;
4305 [ - + ]: 1 : if (WALK(stmt->sourceRelation))
4306 : 0 : return true;
4307 [ - + ]: 1 : if (WALK(stmt->joinCondition))
4308 : 0 : return true;
4309 [ - + ]: 1 : if (WALK(stmt->mergeWhenClauses))
4310 : 0 : return true;
4311 [ - + ]: 1 : if (WALK(stmt->returningClause))
4312 : 0 : return true;
4313 [ - + ]: 1 : if (WALK(stmt->withClause))
4314 : 0 : return true;
4315 [ - + ]: 1 : }
4316 : 1 : break;
4317 : : case T_MergeWhenClause:
4318 : : {
4319 : 1 : MergeWhenClause *mergeWhenClause = (MergeWhenClause *) node;
4320 : :
4321 [ - + ]: 1 : if (WALK(mergeWhenClause->condition))
4322 : 0 : return true;
4323 [ - + ]: 1 : if (WALK(mergeWhenClause->targetList))
4324 : 0 : return true;
4325 [ - + ]: 1 : if (WALK(mergeWhenClause->values))
4326 : 0 : return true;
4327 [ - + ]: 1 : }
4328 : 1 : break;
4329 : : case T_ReturningClause:
4330 : : {
4331 : 9 : ReturningClause *returning = (ReturningClause *) node;
4332 : :
4333 [ - + ]: 9 : if (WALK(returning->options))
4334 : 0 : return true;
4335 [ - + ]: 9 : if (WALK(returning->exprs))
4336 : 0 : return true;
4337 [ - + ]: 9 : }
4338 : 9 : break;
4339 : : case T_SelectStmt:
4340 : : {
4341 : 741 : SelectStmt *stmt = (SelectStmt *) node;
4342 : :
4343 [ - + ]: 741 : if (WALK(stmt->distinctClause))
4344 : 0 : return true;
4345 [ - + ]: 741 : if (WALK(stmt->intoClause))
4346 : 0 : return true;
4347 [ - + ]: 741 : if (WALK(stmt->targetList))
4348 : 0 : return true;
4349 [ - + ]: 741 : if (WALK(stmt->fromClause))
4350 : 0 : return true;
4351 [ - + ]: 741 : if (WALK(stmt->whereClause))
4352 : 0 : return true;
4353 [ - + ]: 741 : if (WALK(stmt->groupClause))
4354 : 0 : return true;
4355 [ - + ]: 741 : if (WALK(stmt->havingClause))
4356 : 0 : return true;
4357 [ - + ]: 741 : if (WALK(stmt->windowClause))
4358 : 0 : return true;
4359 [ - + ]: 741 : if (WALK(stmt->valuesLists))
4360 : 0 : return true;
4361 [ - + ]: 741 : if (WALK(stmt->sortClause))
4362 : 0 : return true;
4363 [ - + ]: 741 : if (WALK(stmt->limitOffset))
4364 : 0 : return true;
4365 [ - + ]: 741 : if (WALK(stmt->limitCount))
4366 : 0 : return true;
4367 [ - + ]: 741 : if (WALK(stmt->lockingClause))
4368 : 0 : return true;
4369 [ - + ]: 741 : if (WALK(stmt->withClause))
4370 : 0 : return true;
4371 [ - + ]: 741 : if (WALK(stmt->larg))
4372 : 0 : return true;
4373 [ - + ]: 741 : if (WALK(stmt->rarg))
4374 : 0 : return true;
4375 [ - + ]: 741 : }
4376 : 741 : break;
4377 : : case T_PLAssignStmt:
4378 : : {
4379 : 0 : PLAssignStmt *stmt = (PLAssignStmt *) node;
4380 : :
4381 [ # # ]: 0 : if (WALK(stmt->indirection))
4382 : 0 : return true;
4383 [ # # ]: 0 : if (WALK(stmt->val))
4384 : 0 : return true;
4385 [ # # ]: 0 : }
4386 : 0 : break;
4387 : : case T_A_Expr:
4388 : : {
4389 : 375 : A_Expr *expr = (A_Expr *) node;
4390 : :
4391 [ - + ]: 375 : if (WALK(expr->lexpr))
4392 : 0 : return true;
4393 [ - + ]: 375 : if (WALK(expr->rexpr))
4394 : 0 : return true;
4395 : : /* operator name is deemed uninteresting */
4396 [ - + ]: 375 : }
4397 : 375 : break;
4398 : : case T_BoolExpr:
4399 : : {
4400 : 14 : BoolExpr *expr = (BoolExpr *) node;
4401 : :
4402 [ - + ]: 14 : if (WALK(expr->args))
4403 : 0 : return true;
4404 [ - + ]: 14 : }
4405 : 14 : break;
4406 : : case T_ColumnRef:
4407 : : /* we assume the fields contain nothing interesting */
4408 : : break;
4409 : : case T_FuncCall:
4410 : : {
4411 : 22 : FuncCall *fcall = (FuncCall *) node;
4412 : :
4413 [ - + ]: 22 : if (WALK(fcall->args))
4414 : 0 : return true;
4415 [ - + ]: 22 : if (WALK(fcall->agg_order))
4416 : 0 : return true;
4417 [ - + ]: 22 : if (WALK(fcall->agg_filter))
4418 : 0 : return true;
4419 [ - + ]: 22 : if (WALK(fcall->over))
4420 : 0 : return true;
4421 : : /* function name is deemed uninteresting */
4422 [ - + ]: 22 : }
4423 : 22 : break;
4424 : : case T_NamedArgExpr:
4425 : 0 : return WALK(((NamedArgExpr *) node)->arg);
4426 : : case T_A_Indices:
4427 : : {
4428 : 0 : A_Indices *indices = (A_Indices *) node;
4429 : :
4430 [ # # ]: 0 : if (WALK(indices->lidx))
4431 : 0 : return true;
4432 [ # # ]: 0 : if (WALK(indices->uidx))
4433 : 0 : return true;
4434 [ # # ]: 0 : }
4435 : 0 : break;
4436 : : case T_A_Indirection:
4437 : : {
4438 : 0 : A_Indirection *indir = (A_Indirection *) node;
4439 : :
4440 [ # # ]: 0 : if (WALK(indir->arg))
4441 : 0 : return true;
4442 [ # # ]: 0 : if (WALK(indir->indirection))
4443 : 0 : return true;
4444 [ # # ]: 0 : }
4445 : 0 : break;
4446 : : case T_A_ArrayExpr:
4447 : 12 : return WALK(((A_ArrayExpr *) node)->elements);
4448 : : case T_ResTarget:
4449 : : {
4450 : 630 : ResTarget *rt = (ResTarget *) node;
4451 : :
4452 [ - + ]: 630 : if (WALK(rt->indirection))
4453 : 0 : return true;
4454 [ - + ]: 630 : if (WALK(rt->val))
4455 : 0 : return true;
4456 [ - + ]: 630 : }
4457 : 630 : break;
4458 : : case T_MultiAssignRef:
4459 : 0 : return WALK(((MultiAssignRef *) node)->source);
4460 : : case T_TypeCast:
4461 : : {
4462 : 28 : TypeCast *tc = (TypeCast *) node;
4463 : :
4464 [ - + ]: 28 : if (WALK(tc->arg))
4465 : 0 : return true;
4466 [ - + ]: 28 : if (WALK(tc->typeName))
4467 : 0 : return true;
4468 [ - + ]: 28 : }
4469 : 28 : break;
4470 : : case T_CollateClause:
4471 : 8 : return WALK(((CollateClause *) node)->arg);
4472 : : case T_SortBy:
4473 : 2 : return WALK(((SortBy *) node)->node);
4474 : : case T_WindowDef:
4475 : : {
4476 : 4 : WindowDef *wd = (WindowDef *) node;
4477 : :
4478 [ - + ]: 4 : if (WALK(wd->partitionClause))
4479 : 0 : return true;
4480 [ - + ]: 4 : if (WALK(wd->orderClause))
4481 : 0 : return true;
4482 [ - + ]: 4 : if (WALK(wd->startOffset))
4483 : 0 : return true;
4484 [ - + ]: 4 : if (WALK(wd->endOffset))
4485 : 0 : return true;
4486 [ - + ]: 4 : }
4487 : 4 : break;
4488 : : case T_RangeSubselect:
4489 : : {
4490 : 12 : RangeSubselect *rs = (RangeSubselect *) node;
4491 : :
4492 [ - + ]: 12 : if (WALK(rs->subquery))
4493 : 0 : return true;
4494 [ - + ]: 12 : if (WALK(rs->alias))
4495 : 0 : return true;
4496 [ - + ]: 12 : }
4497 : 12 : break;
4498 : : case T_RangeFunction:
4499 : : {
4500 : 0 : RangeFunction *rf = (RangeFunction *) node;
4501 : :
4502 [ # # ]: 0 : if (WALK(rf->functions))
4503 : 0 : return true;
4504 [ # # ]: 0 : if (WALK(rf->alias))
4505 : 0 : return true;
4506 [ # # ]: 0 : if (WALK(rf->coldeflist))
4507 : 0 : return true;
4508 [ # # ]: 0 : }
4509 : 0 : break;
4510 : : case T_RangeTableSample:
4511 : : {
4512 : 0 : RangeTableSample *rts = (RangeTableSample *) node;
4513 : :
4514 [ # # ]: 0 : if (WALK(rts->relation))
4515 : 0 : return true;
4516 : : /* method name is deemed uninteresting */
4517 [ # # ]: 0 : if (WALK(rts->args))
4518 : 0 : return true;
4519 [ # # ]: 0 : if (WALK(rts->repeatable))
4520 : 0 : return true;
4521 [ # # ]: 0 : }
4522 : 0 : break;
4523 : : case T_RangeTableFunc:
4524 : : {
4525 : 0 : RangeTableFunc *rtf = (RangeTableFunc *) node;
4526 : :
4527 [ # # ]: 0 : if (WALK(rtf->docexpr))
4528 : 0 : return true;
4529 [ # # ]: 0 : if (WALK(rtf->rowexpr))
4530 : 0 : return true;
4531 [ # # ]: 0 : if (WALK(rtf->namespaces))
4532 : 0 : return true;
4533 [ # # ]: 0 : if (WALK(rtf->columns))
4534 : 0 : return true;
4535 [ # # ]: 0 : if (WALK(rtf->alias))
4536 : 0 : return true;
4537 [ # # ]: 0 : }
4538 : 0 : break;
4539 : : case T_RangeTableFuncCol:
4540 : : {
4541 : 0 : RangeTableFuncCol *rtfc = (RangeTableFuncCol *) node;
4542 : :
4543 [ # # ]: 0 : if (WALK(rtfc->colexpr))
4544 : 0 : return true;
4545 [ # # ]: 0 : if (WALK(rtfc->coldefexpr))
4546 : 0 : return true;
4547 [ # # ]: 0 : }
4548 : 0 : break;
4549 : : case T_TypeName:
4550 : : {
4551 : 28 : TypeName *tn = (TypeName *) node;
4552 : :
4553 [ - + ]: 28 : if (WALK(tn->typmods))
4554 : 0 : return true;
4555 [ - + ]: 28 : if (WALK(tn->arrayBounds))
4556 : 0 : return true;
4557 : : /* type name itself is deemed uninteresting */
4558 [ - + ]: 28 : }
4559 : 28 : break;
4560 : : case T_ColumnDef:
4561 : : {
4562 : 0 : ColumnDef *coldef = (ColumnDef *) node;
4563 : :
4564 [ # # ]: 0 : if (WALK(coldef->typeName))
4565 : 0 : return true;
4566 [ # # ]: 0 : if (WALK(coldef->raw_default))
4567 : 0 : return true;
4568 [ # # ]: 0 : if (WALK(coldef->collClause))
4569 : 0 : return true;
4570 : : /* for now, constraints are ignored */
4571 [ # # ]: 0 : }
4572 : 0 : break;
4573 : : case T_IndexElem:
4574 : : {
4575 : 0 : IndexElem *indelem = (IndexElem *) node;
4576 : :
4577 [ # # ]: 0 : if (WALK(indelem->expr))
4578 : 0 : return true;
4579 : : /* collation and opclass names are deemed uninteresting */
4580 [ # # ]: 0 : }
4581 : 0 : break;
4582 : : case T_GroupingSet:
4583 : 0 : return WALK(((GroupingSet *) node)->content);
4584 : : case T_LockingClause:
4585 : 1 : return WALK(((LockingClause *) node)->lockedRels);
4586 : : case T_XmlSerialize:
4587 : : {
4588 : 0 : XmlSerialize *xs = (XmlSerialize *) node;
4589 : :
4590 [ # # ]: 0 : if (WALK(xs->expr))
4591 : 0 : return true;
4592 [ # # ]: 0 : if (WALK(xs->typeName))
4593 : 0 : return true;
4594 [ # # ]: 0 : }
4595 : 0 : break;
4596 : : case T_WithClause:
4597 : 0 : return WALK(((WithClause *) node)->ctes);
4598 : : case T_InferClause:
4599 : : {
4600 : 0 : InferClause *stmt = (InferClause *) node;
4601 : :
4602 [ # # ]: 0 : if (WALK(stmt->indexElems))
4603 : 0 : return true;
4604 [ # # ]: 0 : if (WALK(stmt->whereClause))
4605 : 0 : return true;
4606 [ # # ]: 0 : }
4607 : 0 : break;
4608 : : case T_OnConflictClause:
4609 : : {
4610 : 0 : OnConflictClause *stmt = (OnConflictClause *) node;
4611 : :
4612 [ # # ]: 0 : if (WALK(stmt->infer))
4613 : 0 : return true;
4614 [ # # ]: 0 : if (WALK(stmt->targetList))
4615 : 0 : return true;
4616 [ # # ]: 0 : if (WALK(stmt->whereClause))
4617 : 0 : return true;
4618 [ # # ]: 0 : }
4619 : 0 : break;
4620 : : case T_CommonTableExpr:
4621 : : /* search_clause and cycle_clause are not interesting here */
4622 : 3 : return WALK(((CommonTableExpr *) node)->ctequery);
4623 : : case T_JsonOutput:
4624 : : {
4625 : 0 : JsonOutput *out = (JsonOutput *) node;
4626 : :
4627 [ # # ]: 0 : if (WALK(out->typeName))
4628 : 0 : return true;
4629 [ # # ]: 0 : if (WALK(out->returning))
4630 : 0 : return true;
4631 [ # # ]: 0 : }
4632 : 0 : break;
4633 : : case T_JsonKeyValue:
4634 : : {
4635 : 0 : JsonKeyValue *jkv = (JsonKeyValue *) node;
4636 : :
4637 [ # # ]: 0 : if (WALK(jkv->key))
4638 : 0 : return true;
4639 [ # # ]: 0 : if (WALK(jkv->value))
4640 : 0 : return true;
4641 [ # # ]: 0 : }
4642 : 0 : break;
4643 : : case T_JsonObjectConstructor:
4644 : : {
4645 : 0 : JsonObjectConstructor *joc = (JsonObjectConstructor *) node;
4646 : :
4647 [ # # ]: 0 : if (WALK(joc->output))
4648 : 0 : return true;
4649 [ # # ]: 0 : if (WALK(joc->exprs))
4650 : 0 : return true;
4651 [ # # ]: 0 : }
4652 : 0 : break;
4653 : : case T_JsonArrayConstructor:
4654 : : {
4655 : 0 : JsonArrayConstructor *jac = (JsonArrayConstructor *) node;
4656 : :
4657 [ # # ]: 0 : if (WALK(jac->output))
4658 : 0 : return true;
4659 [ # # ]: 0 : if (WALK(jac->exprs))
4660 : 0 : return true;
4661 [ # # ]: 0 : }
4662 : 0 : break;
4663 : : case T_JsonAggConstructor:
4664 : : {
4665 : 0 : JsonAggConstructor *ctor = (JsonAggConstructor *) node;
4666 : :
4667 [ # # ]: 0 : if (WALK(ctor->output))
4668 : 0 : return true;
4669 [ # # ]: 0 : if (WALK(ctor->agg_order))
4670 : 0 : return true;
4671 [ # # ]: 0 : if (WALK(ctor->agg_filter))
4672 : 0 : return true;
4673 [ # # ]: 0 : if (WALK(ctor->over))
4674 : 0 : return true;
4675 [ # # ]: 0 : }
4676 : 0 : break;
4677 : : case T_JsonObjectAgg:
4678 : : {
4679 : 0 : JsonObjectAgg *joa = (JsonObjectAgg *) node;
4680 : :
4681 [ # # ]: 0 : if (WALK(joa->constructor))
4682 : 0 : return true;
4683 [ # # ]: 0 : if (WALK(joa->arg))
4684 : 0 : return true;
4685 [ # # ]: 0 : }
4686 : 0 : break;
4687 : : case T_JsonArrayAgg:
4688 : : {
4689 : 0 : JsonArrayAgg *jaa = (JsonArrayAgg *) node;
4690 : :
4691 [ # # ]: 0 : if (WALK(jaa->constructor))
4692 : 0 : return true;
4693 [ # # ]: 0 : if (WALK(jaa->arg))
4694 : 0 : return true;
4695 [ # # ]: 0 : }
4696 : 0 : break;
4697 : : case T_JsonArrayQueryConstructor:
4698 : : {
4699 : 0 : JsonArrayQueryConstructor *jaqc = (JsonArrayQueryConstructor *) node;
4700 : :
4701 [ # # ]: 0 : if (WALK(jaqc->output))
4702 : 0 : return true;
4703 [ # # ]: 0 : if (WALK(jaqc->query))
4704 : 0 : return true;
4705 [ # # ]: 0 : }
4706 : 0 : break;
4707 : : default:
4708 [ # # # # ]: 0 : elog(ERROR, "unrecognized node type: %d",
4709 : : (int) nodeTag(node));
4710 : 0 : break;
4711 : : }
4712 : 4483 : return false;
4713 : 4527 : }
4714 : :
4715 : : /*
4716 : : * planstate_tree_walker --- walk plan state trees
4717 : : *
4718 : : * The walker has already visited the current node, and so we need only
4719 : : * recurse into any sub-nodes it has.
4720 : : */
4721 : : bool
4722 : 932217 : planstate_tree_walker_impl(PlanState *planstate,
4723 : : planstate_tree_walker_callback walker,
4724 : : void *context)
4725 : : {
4726 : 932217 : Plan *plan = planstate->plan;
4727 : 932217 : ListCell *lc;
4728 : :
4729 : : /* We don't need implicit coercions to Node here */
4730 : : #define PSWALK(n) walker(n, context)
4731 : :
4732 : : /* Guard against stack overflow due to overly complex plan trees */
4733 : 932217 : check_stack_depth();
4734 : :
4735 : : /* initPlan-s */
4736 [ - + ]: 932217 : if (planstate_walk_subplans(planstate->initPlan, walker, context))
4737 : 0 : return true;
4738 : :
4739 : : /* lefttree */
4740 [ + + ]: 932217 : if (outerPlanState(planstate))
4741 : : {
4742 [ - + ]: 453709 : if (PSWALK(outerPlanState(planstate)))
4743 : 0 : return true;
4744 : 453709 : }
4745 : :
4746 : : /* righttree */
4747 [ + + ]: 932217 : if (innerPlanState(planstate))
4748 : : {
4749 [ - + ]: 13972 : if (PSWALK(innerPlanState(planstate)))
4750 : 0 : return true;
4751 : 13972 : }
4752 : :
4753 : : /* special child plans */
4754 [ + + + + : 932217 : switch (nodeTag(plan))
+ + - ]
4755 : : {
4756 : : case T_Append:
4757 [ - + - + ]: 5578 : if (planstate_walk_members(((AppendState *) planstate)->appendplans,
4758 : 2789 : ((AppendState *) planstate)->as_nplans,
4759 : 2789 : walker, context))
4760 : 0 : return true;
4761 : 2789 : break;
4762 : : case T_MergeAppend:
4763 [ - + - + ]: 204 : if (planstate_walk_members(((MergeAppendState *) planstate)->mergeplans,
4764 : 102 : ((MergeAppendState *) planstate)->ms_nplans,
4765 : 102 : walker, context))
4766 : 0 : return true;
4767 : 102 : break;
4768 : : case T_BitmapAnd:
4769 [ - + - + ]: 34 : if (planstate_walk_members(((BitmapAndState *) planstate)->bitmapplans,
4770 : 17 : ((BitmapAndState *) planstate)->nplans,
4771 : 17 : walker, context))
4772 : 0 : return true;
4773 : 17 : break;
4774 : : case T_BitmapOr:
4775 [ - + - + ]: 68 : if (planstate_walk_members(((BitmapOrState *) planstate)->bitmapplans,
4776 : 34 : ((BitmapOrState *) planstate)->nplans,
4777 : 34 : walker, context))
4778 : 0 : return true;
4779 : 34 : break;
4780 : : case T_SubqueryScan:
4781 [ - + ]: 1519 : if (PSWALK(((SubqueryScanState *) planstate)->subplan))
4782 : 0 : return true;
4783 : 1519 : break;
4784 : : case T_CustomScan:
4785 [ # # # # : 0 : foreach(lc, ((CustomScanState *) planstate)->custom_ps)
# # # # ]
4786 : : {
4787 [ # # ]: 0 : if (PSWALK(lfirst(lc)))
4788 : 0 : return true;
4789 : 0 : }
4790 : 0 : break;
4791 : : default:
4792 : 927756 : break;
4793 : : }
4794 : :
4795 : : /* subPlan-s */
4796 [ - + ]: 932217 : if (planstate_walk_subplans(planstate->subPlan, walker, context))
4797 : 0 : return true;
4798 : :
4799 : 932217 : return false;
4800 : 932217 : }
4801 : :
4802 : : /*
4803 : : * Walk a list of SubPlans (or initPlans, which also use SubPlan nodes).
4804 : : */
4805 : : static bool
4806 : 1864434 : planstate_walk_subplans(List *plans,
4807 : : planstate_tree_walker_callback walker,
4808 : : void *context)
4809 : : {
4810 : 1864434 : ListCell *lc;
4811 : :
4812 [ + + + + : 1868838 : foreach(lc, plans)
+ + - + ]
4813 : : {
4814 : 4404 : SubPlanState *sps = lfirst_node(SubPlanState, lc);
4815 : :
4816 [ - + ]: 4404 : if (PSWALK(sps->planstate))
4817 : 0 : return true;
4818 [ - + ]: 4404 : }
4819 : :
4820 : 1864434 : return false;
4821 : 1864434 : }
4822 : :
4823 : : /*
4824 : : * Walk the constituent plans of a ModifyTable, Append, MergeAppend,
4825 : : * BitmapAnd, or BitmapOr node.
4826 : : */
4827 : : static bool
4828 : 2942 : planstate_walk_members(PlanState **planstates, int nplans,
4829 : : planstate_tree_walker_callback walker,
4830 : : void *context)
4831 : : {
4832 [ + + - - : 11452 : for (int j = 0; j < nplans; j++)
+ ]
4833 : : {
4834 [ - + ]: 8510 : if (PSWALK(planstates[j]))
4835 : 0 : return true;
4836 : 8510 : }
4837 : :
4838 : 2942 : return false;
4839 : 2942 : }
|
