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