Branch data Line data Source code
1 : : /*-------------------------------------------------------------------------
2 : : *
3 : : * rewriteSearchCycle.c
4 : : * Support for rewriting SEARCH and CYCLE clauses.
5 : : *
6 : : * Portions Copyright (c) 1996-2026, PostgreSQL Global Development Group
7 : : * Portions Copyright (c) 1994, Regents of the University of California
8 : : *
9 : : * IDENTIFICATION
10 : : * src/backend/rewrite/rewriteSearchCycle.c
11 : : *
12 : : *-------------------------------------------------------------------------
13 : : */
14 : : #include "postgres.h"
15 : :
16 : : #include "catalog/pg_operator_d.h"
17 : : #include "catalog/pg_type_d.h"
18 : : #include "nodes/makefuncs.h"
19 : : #include "nodes/parsenodes.h"
20 : : #include "nodes/pg_list.h"
21 : : #include "nodes/primnodes.h"
22 : : #include "parser/analyze.h"
23 : : #include "parser/parsetree.h"
24 : : #include "rewrite/rewriteManip.h"
25 : : #include "rewrite/rewriteSearchCycle.h"
26 : : #include "utils/fmgroids.h"
27 : :
28 : :
29 : : /*----------
30 : : * Rewrite a CTE with SEARCH or CYCLE clause
31 : : *
32 : : * Consider a CTE like
33 : : *
34 : : * WITH RECURSIVE ctename (col1, col2, col3) AS (
35 : : * query1
36 : : * UNION [ALL]
37 : : * SELECT trosl FROM ctename
38 : : * )
39 : : *
40 : : * With a search clause
41 : : *
42 : : * SEARCH BREADTH FIRST BY col1, col2 SET sqc
43 : : *
44 : : * the CTE is rewritten to
45 : : *
46 : : * WITH RECURSIVE ctename (col1, col2, col3, sqc) AS (
47 : : * SELECT col1, col2, col3, -- original WITH column list
48 : : * ROW(0, col1, col2) -- initial row of search columns
49 : : * FROM (query1) "*TLOCRN*" (col1, col2, col3)
50 : : * UNION [ALL]
51 : : * SELECT col1, col2, col3, -- same as above
52 : : * ROW(sqc.depth + 1, col1, col2) -- count depth
53 : : * FROM (SELECT trosl, ctename.sqc FROM ctename) "*TROCRN*" (col1, col2, col3, sqc)
54 : : * )
55 : : *
56 : : * (This isn't quite legal SQL: sqc.depth is meant to refer to the first
57 : : * column of sqc, which has a row type, but the field names are not defined
58 : : * here. Representing this properly in SQL would be more complicated (and the
59 : : * SQL standard actually does it in that more complicated way), but the
60 : : * internal representation allows us to construct it this way.)
61 : : *
62 : : * With a search clause
63 : : *
64 : : * SEARCH DEPTH FIRST BY col1, col2 SET sqc
65 : : *
66 : : * the CTE is rewritten to
67 : : *
68 : : * WITH RECURSIVE ctename (col1, col2, col3, sqc) AS (
69 : : * SELECT col1, col2, col3, -- original WITH column list
70 : : * ARRAY[ROW(col1, col2)] -- initial row of search columns
71 : : * FROM (query1) "*TLOCRN*" (col1, col2, col3)
72 : : * UNION [ALL]
73 : : * SELECT col1, col2, col3, -- same as above
74 : : * sqc || ARRAY[ROW(col1, col2)] -- record rows seen
75 : : * FROM (SELECT trosl, ctename.sqc FROM ctename) "*TROCRN*" (col1, col2, col3, sqc)
76 : : * )
77 : : *
78 : : * With a cycle clause
79 : : *
80 : : * CYCLE col1, col2 SET cmc TO 'Y' DEFAULT 'N' USING cpa
81 : : *
82 : : * (cmc = cycle mark column, cpa = cycle path) the CTE is rewritten to
83 : : *
84 : : * WITH RECURSIVE ctename (col1, col2, col3, cmc, cpa) AS (
85 : : * SELECT col1, col2, col3, -- original WITH column list
86 : : * 'N', -- cycle mark default
87 : : * ARRAY[ROW(col1, col2)] -- initial row of cycle columns
88 : : * FROM (query1) "*TLOCRN*" (col1, col2, col3)
89 : : * UNION [ALL]
90 : : * SELECT col1, col2, col3, -- same as above
91 : : * CASE WHEN ROW(col1, col2) = ANY (ARRAY[cpa]) THEN 'Y' ELSE 'N' END, -- compute cycle mark column
92 : : * cpa || ARRAY[ROW(col1, col2)] -- record rows seen
93 : : * FROM (SELECT trosl, ctename.cmc, ctename.cpa FROM ctename) "*TROCRN*" (col1, col2, col3, cmc, cpa)
94 : : * WHERE cmc <> 'Y'
95 : : * )
96 : : *
97 : : * The expression to compute the cycle mark column in the right-hand query is
98 : : * written as
99 : : *
100 : : * CASE WHEN ROW(col1, col2) IN (SELECT p.* FROM TABLE(cpa) p) THEN cmv ELSE cmd END
101 : : *
102 : : * in the SQL standard, but in PostgreSQL we can use the scalar-array operator
103 : : * expression shown above.
104 : : *
105 : : * Also, in some of the cases where operators are shown above we actually
106 : : * directly produce the underlying function call.
107 : : *
108 : : * If both a search clause and a cycle clause is specified, then the search
109 : : * clause column is added before the cycle clause columns.
110 : : */
111 : :
112 : : /*
113 : : * Make a RowExpr from the specified column names, which have to be among the
114 : : * output columns of the CTE.
115 : : */
116 : : static RowExpr *
117 : 104 : make_path_rowexpr(const CommonTableExpr *cte, const List *col_list)
118 : : {
119 : : RowExpr *rowexpr;
120 : : ListCell *lc;
121 : :
122 : 104 : rowexpr = makeNode(RowExpr);
123 : 104 : rowexpr->row_typeid = RECORDOID;
124 : 104 : rowexpr->row_format = COERCE_IMPLICIT_CAST;
125 : 104 : rowexpr->location = -1;
126 : :
127 [ + - + + : 276 : foreach(lc, col_list)
+ + ]
128 : : {
129 : 172 : char *colname = strVal(lfirst(lc));
130 : :
131 [ + - ]: 240 : for (int i = 0; i < list_length(cte->ctecolnames); i++)
132 : : {
133 : 240 : char *colname2 = strVal(list_nth(cte->ctecolnames, i));
134 : :
135 [ + + ]: 240 : if (strcmp(colname, colname2) == 0)
136 : : {
137 : : Var *var;
138 : :
139 : 172 : var = makeVar(1, i + 1,
140 : 172 : list_nth_oid(cte->ctecoltypes, i),
141 : 172 : list_nth_int(cte->ctecoltypmods, i),
142 : 172 : list_nth_oid(cte->ctecolcollations, i),
143 : : 0);
144 : 172 : rowexpr->args = lappend(rowexpr->args, var);
145 : 172 : rowexpr->colnames = lappend(rowexpr->colnames, makeString(colname));
146 : 172 : break;
147 : : }
148 : : }
149 : : }
150 : :
151 : 104 : return rowexpr;
152 : : }
153 : :
154 : : /*
155 : : * Wrap a RowExpr in an ArrayExpr, for the initial search depth first or cycle
156 : : * row.
157 : : */
158 : : static Expr *
159 : 80 : make_path_initial_array(RowExpr *rowexpr)
160 : : {
161 : : ArrayExpr *arr;
162 : :
163 : 80 : arr = makeNode(ArrayExpr);
164 : 80 : arr->array_typeid = RECORDARRAYOID;
165 : 80 : arr->element_typeid = RECORDOID;
166 : 80 : arr->location = -1;
167 : 80 : arr->elements = list_make1(rowexpr);
168 : :
169 : 80 : return (Expr *) arr;
170 : : }
171 : :
172 : : /*
173 : : * Make an array catenation expression like
174 : : *
175 : : * cpa || ARRAY[ROW(cols)]
176 : : *
177 : : * where the varattno of cpa is provided as path_varattno.
178 : : */
179 : : static Expr *
180 : 76 : make_path_cat_expr(RowExpr *rowexpr, AttrNumber path_varattno)
181 : : {
182 : : ArrayExpr *arr;
183 : : FuncExpr *fexpr;
184 : :
185 : 76 : arr = makeNode(ArrayExpr);
186 : 76 : arr->array_typeid = RECORDARRAYOID;
187 : 76 : arr->element_typeid = RECORDOID;
188 : 76 : arr->location = -1;
189 : 76 : arr->elements = list_make1(rowexpr);
190 : :
191 : 152 : fexpr = makeFuncExpr(F_ARRAY_CAT, RECORDARRAYOID,
192 : 76 : list_make2(makeVar(1, path_varattno, RECORDARRAYOID, -1, 0, 0),
193 : : arr),
194 : : InvalidOid, InvalidOid, COERCE_EXPLICIT_CALL);
195 : :
196 : 76 : return (Expr *) fexpr;
197 : : }
198 : :
199 : : /*
200 : : * The real work happens here.
201 : : */
202 : : CommonTableExpr *
203 : 96 : rewriteSearchAndCycle(CommonTableExpr *cte)
204 : : {
205 : : Query *ctequery;
206 : : SetOperationStmt *sos;
207 : : int rti1,
208 : : rti2;
209 : : RangeTblEntry *rte1,
210 : : *rte2,
211 : : *newrte;
212 : : Query *newq1,
213 : : *newq2;
214 : : Query *newsubquery;
215 : : RangeTblRef *rtr;
216 : 96 : Oid search_seq_type = InvalidOid;
217 : 96 : AttrNumber sqc_attno = InvalidAttrNumber;
218 : 96 : AttrNumber cmc_attno = InvalidAttrNumber;
219 : 96 : AttrNumber cpa_attno = InvalidAttrNumber;
220 : : TargetEntry *tle;
221 : 96 : RowExpr *cycle_col_rowexpr = NULL;
222 : 96 : RowExpr *search_col_rowexpr = NULL;
223 : : List *ewcl;
224 : 96 : int cte_rtindex = -1;
225 : :
226 : : Assert(cte->search_clause || cte->cycle_clause);
227 : :
228 : 96 : cte = copyObject(cte);
229 : :
230 : 96 : ctequery = castNode(Query, cte->ctequery);
231 : :
232 : : /*
233 : : * The top level of the CTE's query should be a UNION. Find the two
234 : : * subqueries.
235 : : */
236 : : Assert(ctequery->setOperations);
237 : 96 : sos = castNode(SetOperationStmt, ctequery->setOperations);
238 : : Assert(sos->op == SETOP_UNION);
239 : :
240 : 96 : rti1 = castNode(RangeTblRef, sos->larg)->rtindex;
241 : 96 : rti2 = castNode(RangeTblRef, sos->rarg)->rtindex;
242 : :
243 : 96 : rte1 = rt_fetch(rti1, ctequery->rtable);
244 : 96 : rte2 = rt_fetch(rti2, ctequery->rtable);
245 : :
246 : : Assert(rte1->rtekind == RTE_SUBQUERY);
247 : : Assert(rte2->rtekind == RTE_SUBQUERY);
248 : :
249 : : /*
250 : : * We'll need this a few times later.
251 : : */
252 [ + + ]: 96 : if (cte->search_clause)
253 : : {
254 [ + + ]: 56 : if (cte->search_clause->search_breadth_first)
255 : 24 : search_seq_type = RECORDOID;
256 : : else
257 : 32 : search_seq_type = RECORDARRAYOID;
258 : : }
259 : :
260 : : /*
261 : : * Attribute numbers of the added columns in the CTE's column list
262 : : */
263 [ + + ]: 96 : if (cte->search_clause)
264 : 56 : sqc_attno = list_length(cte->ctecolnames) + 1;
265 [ + + ]: 96 : if (cte->cycle_clause)
266 : : {
267 : 48 : cmc_attno = list_length(cte->ctecolnames) + 1;
268 : 48 : cpa_attno = list_length(cte->ctecolnames) + 2;
269 [ + + ]: 48 : if (cte->search_clause)
270 : : {
271 : 8 : cmc_attno++;
272 : 8 : cpa_attno++;
273 : : }
274 : : }
275 : :
276 : : /*
277 : : * Make new left subquery
278 : : */
279 : 96 : newq1 = makeNode(Query);
280 : 96 : newq1->commandType = CMD_SELECT;
281 : 96 : newq1->canSetTag = true;
282 : :
283 : 96 : newrte = makeNode(RangeTblEntry);
284 : 96 : newrte->rtekind = RTE_SUBQUERY;
285 : 96 : newrte->alias = NULL;
286 : 96 : newrte->eref = makeAlias("*TLOCRN*", cte->ctecolnames);
287 : 96 : newsubquery = copyObject(rte1->subquery);
288 : 96 : IncrementVarSublevelsUp((Node *) newsubquery, 1, 1);
289 : 96 : newrte->subquery = newsubquery;
290 : 96 : newrte->inFromCl = true;
291 : 96 : newq1->rtable = list_make1(newrte);
292 : :
293 : 96 : rtr = makeNode(RangeTblRef);
294 : 96 : rtr->rtindex = 1;
295 : 96 : newq1->jointree = makeFromExpr(list_make1(rtr), NULL);
296 : :
297 : : /*
298 : : * Make target list
299 : : */
300 [ + + ]: 312 : for (int i = 0; i < list_length(cte->ctecolnames); i++)
301 : : {
302 : : Var *var;
303 : :
304 : 216 : var = makeVar(1, i + 1,
305 : 216 : list_nth_oid(cte->ctecoltypes, i),
306 : 216 : list_nth_int(cte->ctecoltypmods, i),
307 : 216 : list_nth_oid(cte->ctecolcollations, i),
308 : : 0);
309 : 216 : tle = makeTargetEntry((Expr *) var, i + 1, strVal(list_nth(cte->ctecolnames, i)), false);
310 : 216 : tle->resorigtbl = list_nth_node(TargetEntry, rte1->subquery->targetList, i)->resorigtbl;
311 : 216 : tle->resorigcol = list_nth_node(TargetEntry, rte1->subquery->targetList, i)->resorigcol;
312 : 216 : newq1->targetList = lappend(newq1->targetList, tle);
313 : : }
314 : :
315 [ + + ]: 96 : if (cte->search_clause)
316 : : {
317 : : Expr *texpr;
318 : :
319 : 56 : search_col_rowexpr = make_path_rowexpr(cte, cte->search_clause->search_col_list);
320 [ + + ]: 56 : if (cte->search_clause->search_breadth_first)
321 : : {
322 : 24 : search_col_rowexpr->args = lcons(makeConst(INT8OID, -1, InvalidOid, sizeof(int64),
323 : : Int64GetDatum(0), false, true),
324 : : search_col_rowexpr->args);
325 : 24 : search_col_rowexpr->colnames = lcons(makeString("*DEPTH*"), search_col_rowexpr->colnames);
326 : 24 : texpr = (Expr *) search_col_rowexpr;
327 : : }
328 : : else
329 : 32 : texpr = make_path_initial_array(search_col_rowexpr);
330 : 112 : tle = makeTargetEntry(texpr,
331 : 56 : list_length(newq1->targetList) + 1,
332 : 56 : cte->search_clause->search_seq_column,
333 : : false);
334 : 56 : newq1->targetList = lappend(newq1->targetList, tle);
335 : : }
336 [ + + ]: 96 : if (cte->cycle_clause)
337 : : {
338 : 96 : tle = makeTargetEntry((Expr *) cte->cycle_clause->cycle_mark_default,
339 : 48 : list_length(newq1->targetList) + 1,
340 : 48 : cte->cycle_clause->cycle_mark_column,
341 : : false);
342 : 48 : newq1->targetList = lappend(newq1->targetList, tle);
343 : 48 : cycle_col_rowexpr = make_path_rowexpr(cte, cte->cycle_clause->cycle_col_list);
344 : 48 : tle = makeTargetEntry(make_path_initial_array(cycle_col_rowexpr),
345 : 48 : list_length(newq1->targetList) + 1,
346 : 48 : cte->cycle_clause->cycle_path_column,
347 : : false);
348 : 48 : newq1->targetList = lappend(newq1->targetList, tle);
349 : : }
350 : :
351 : 96 : rte1->subquery = newq1;
352 : :
353 [ + + ]: 96 : if (cte->search_clause)
354 : : {
355 : 56 : rte1->eref->colnames = lappend(rte1->eref->colnames, makeString(cte->search_clause->search_seq_column));
356 : : }
357 [ + + ]: 96 : if (cte->cycle_clause)
358 : : {
359 : 48 : rte1->eref->colnames = lappend(rte1->eref->colnames, makeString(cte->cycle_clause->cycle_mark_column));
360 : 48 : rte1->eref->colnames = lappend(rte1->eref->colnames, makeString(cte->cycle_clause->cycle_path_column));
361 : : }
362 : :
363 : : /*
364 : : * Make new right subquery
365 : : */
366 : 96 : newq2 = makeNode(Query);
367 : 96 : newq2->commandType = CMD_SELECT;
368 : 96 : newq2->canSetTag = true;
369 : :
370 : 96 : newrte = makeNode(RangeTblEntry);
371 : 96 : newrte->rtekind = RTE_SUBQUERY;
372 : 96 : ewcl = copyObject(cte->ctecolnames);
373 [ + + ]: 96 : if (cte->search_clause)
374 : : {
375 : 56 : ewcl = lappend(ewcl, makeString(cte->search_clause->search_seq_column));
376 : : }
377 [ + + ]: 96 : if (cte->cycle_clause)
378 : : {
379 : 48 : ewcl = lappend(ewcl, makeString(cte->cycle_clause->cycle_mark_column));
380 : 48 : ewcl = lappend(ewcl, makeString(cte->cycle_clause->cycle_path_column));
381 : : }
382 : 96 : newrte->alias = NULL;
383 : 96 : newrte->eref = makeAlias("*TROCRN*", ewcl);
384 : :
385 : : /*
386 : : * Find the reference to the recursive CTE in the right UNION subquery's
387 : : * range table. We expect it to be two levels up from the UNION subquery
388 : : * (and must check that to avoid being fooled by sub-WITHs with the same
389 : : * CTE name). There will not be more than one such reference, because the
390 : : * parser would have rejected that (see checkWellFormedRecursion() in
391 : : * parse_cte.c). However, the parser doesn't insist that the reference
392 : : * appear in the UNION subquery's topmost range table, so we might fail to
393 : : * find it at all. That's an unimplemented case for the moment.
394 : : */
395 [ + + ]: 160 : for (int rti = 1; rti <= list_length(rte2->subquery->rtable); rti++)
396 : : {
397 : 156 : RangeTblEntry *e = rt_fetch(rti, rte2->subquery->rtable);
398 : :
399 [ + + ]: 156 : if (e->rtekind == RTE_CTE &&
400 [ + + ]: 100 : strcmp(cte->ctename, e->ctename) == 0 &&
401 [ + + ]: 96 : e->ctelevelsup == 2)
402 : : {
403 : 92 : cte_rtindex = rti;
404 : 92 : break;
405 : : }
406 : : }
407 [ + + ]: 96 : if (cte_rtindex <= 0)
408 [ + - ]: 4 : ereport(ERROR,
409 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
410 : : errmsg("with a SEARCH or CYCLE clause, the recursive reference to WITH query \"%s\" must be at the top level of its right-hand SELECT",
411 : : cte->ctename)));
412 : :
413 : 92 : newsubquery = copyObject(rte2->subquery);
414 : 92 : IncrementVarSublevelsUp((Node *) newsubquery, 1, 1);
415 : :
416 : : /*
417 : : * Add extra columns to target list of subquery of right subquery
418 : : */
419 [ + + ]: 92 : if (cte->search_clause)
420 : : {
421 : : Var *var;
422 : :
423 : : /* ctename.sqc */
424 : 52 : var = makeVar(cte_rtindex, sqc_attno,
425 : : search_seq_type, -1, InvalidOid, 0);
426 : 104 : tle = makeTargetEntry((Expr *) var,
427 : 52 : list_length(newsubquery->targetList) + 1,
428 : 52 : cte->search_clause->search_seq_column,
429 : : false);
430 : 52 : newsubquery->targetList = lappend(newsubquery->targetList, tle);
431 : : }
432 [ + + ]: 92 : if (cte->cycle_clause)
433 : : {
434 : : Var *var;
435 : :
436 : : /* ctename.cmc */
437 : 48 : var = makeVar(cte_rtindex, cmc_attno,
438 : 48 : cte->cycle_clause->cycle_mark_type,
439 : 48 : cte->cycle_clause->cycle_mark_typmod,
440 : 48 : cte->cycle_clause->cycle_mark_collation, 0);
441 : 96 : tle = makeTargetEntry((Expr *) var,
442 : 48 : list_length(newsubquery->targetList) + 1,
443 : 48 : cte->cycle_clause->cycle_mark_column,
444 : : false);
445 : 48 : newsubquery->targetList = lappend(newsubquery->targetList, tle);
446 : :
447 : : /* ctename.cpa */
448 : 48 : var = makeVar(cte_rtindex, cpa_attno,
449 : : RECORDARRAYOID, -1, InvalidOid, 0);
450 : 96 : tle = makeTargetEntry((Expr *) var,
451 : 48 : list_length(newsubquery->targetList) + 1,
452 : 48 : cte->cycle_clause->cycle_path_column,
453 : : false);
454 : 48 : newsubquery->targetList = lappend(newsubquery->targetList, tle);
455 : : }
456 : :
457 : 92 : newrte->subquery = newsubquery;
458 : 92 : newrte->inFromCl = true;
459 : 92 : newq2->rtable = list_make1(newrte);
460 : :
461 : 92 : rtr = makeNode(RangeTblRef);
462 : 92 : rtr->rtindex = 1;
463 : :
464 [ + + ]: 92 : if (cte->cycle_clause)
465 : : {
466 : : Expr *expr;
467 : :
468 : : /*
469 : : * Add cmc <> cmv condition
470 : : */
471 : 48 : expr = make_opclause(cte->cycle_clause->cycle_mark_neop, BOOLOID, false,
472 : 48 : (Expr *) makeVar(1, cmc_attno,
473 : 48 : cte->cycle_clause->cycle_mark_type,
474 : 48 : cte->cycle_clause->cycle_mark_typmod,
475 : 48 : cte->cycle_clause->cycle_mark_collation, 0),
476 : 48 : (Expr *) cte->cycle_clause->cycle_mark_value,
477 : : InvalidOid,
478 : 48 : cte->cycle_clause->cycle_mark_collation);
479 : :
480 : 48 : newq2->jointree = makeFromExpr(list_make1(rtr), (Node *) expr);
481 : : }
482 : : else
483 : 44 : newq2->jointree = makeFromExpr(list_make1(rtr), NULL);
484 : :
485 : : /*
486 : : * Make target list
487 : : */
488 [ + + ]: 304 : for (int i = 0; i < list_length(cte->ctecolnames); i++)
489 : : {
490 : : Var *var;
491 : :
492 : 212 : var = makeVar(1, i + 1,
493 : 212 : list_nth_oid(cte->ctecoltypes, i),
494 : 212 : list_nth_int(cte->ctecoltypmods, i),
495 : 212 : list_nth_oid(cte->ctecolcollations, i),
496 : : 0);
497 : 212 : tle = makeTargetEntry((Expr *) var, i + 1, strVal(list_nth(cte->ctecolnames, i)), false);
498 : 212 : tle->resorigtbl = list_nth_node(TargetEntry, rte2->subquery->targetList, i)->resorigtbl;
499 : 212 : tle->resorigcol = list_nth_node(TargetEntry, rte2->subquery->targetList, i)->resorigcol;
500 : 212 : newq2->targetList = lappend(newq2->targetList, tle);
501 : : }
502 : :
503 [ + + ]: 92 : if (cte->search_clause)
504 : : {
505 : : Expr *texpr;
506 : :
507 [ + + ]: 52 : if (cte->search_clause->search_breadth_first)
508 : : {
509 : : FieldSelect *fs;
510 : : FuncExpr *fexpr;
511 : :
512 : : /*
513 : : * ROW(sqc.depth + 1, cols)
514 : : */
515 : :
516 : 24 : search_col_rowexpr = copyObject(search_col_rowexpr);
517 : :
518 : 24 : fs = makeNode(FieldSelect);
519 : 24 : fs->arg = (Expr *) makeVar(1, sqc_attno, RECORDOID, -1, 0, 0);
520 : 24 : fs->fieldnum = 1;
521 : 24 : fs->resulttype = INT8OID;
522 : 24 : fs->resulttypmod = -1;
523 : :
524 : 24 : fexpr = makeFuncExpr(F_INT8INC, INT8OID, list_make1(fs), InvalidOid, InvalidOid, COERCE_EXPLICIT_CALL);
525 : :
526 : 24 : linitial(search_col_rowexpr->args) = fexpr;
527 : :
528 : 24 : texpr = (Expr *) search_col_rowexpr;
529 : : }
530 : : else
531 : : {
532 : : /*
533 : : * sqc || ARRAY[ROW(cols)]
534 : : */
535 : 28 : texpr = make_path_cat_expr(search_col_rowexpr, sqc_attno);
536 : : }
537 : 104 : tle = makeTargetEntry(texpr,
538 : 52 : list_length(newq2->targetList) + 1,
539 : 52 : cte->search_clause->search_seq_column,
540 : : false);
541 : 52 : newq2->targetList = lappend(newq2->targetList, tle);
542 : : }
543 : :
544 [ + + ]: 92 : if (cte->cycle_clause)
545 : : {
546 : : ScalarArrayOpExpr *saoe;
547 : : CaseExpr *caseexpr;
548 : : CaseWhen *casewhen;
549 : :
550 : : /*
551 : : * CASE WHEN ROW(cols) = ANY (ARRAY[cpa]) THEN cmv ELSE cmd END
552 : : */
553 : :
554 : 48 : saoe = makeNode(ScalarArrayOpExpr);
555 : 48 : saoe->location = -1;
556 : 48 : saoe->opno = RECORD_EQ_OP;
557 : 48 : saoe->useOr = true;
558 : 48 : saoe->args = list_make2(cycle_col_rowexpr,
559 : : makeVar(1, cpa_attno, RECORDARRAYOID, -1, 0, 0));
560 : :
561 : 48 : caseexpr = makeNode(CaseExpr);
562 : 48 : caseexpr->location = -1;
563 : 48 : caseexpr->casetype = cte->cycle_clause->cycle_mark_type;
564 : 48 : caseexpr->casecollid = cte->cycle_clause->cycle_mark_collation;
565 : 48 : casewhen = makeNode(CaseWhen);
566 : 48 : casewhen->location = -1;
567 : 48 : casewhen->expr = (Expr *) saoe;
568 : 48 : casewhen->result = (Expr *) cte->cycle_clause->cycle_mark_value;
569 : 48 : caseexpr->args = list_make1(casewhen);
570 : 48 : caseexpr->defresult = (Expr *) cte->cycle_clause->cycle_mark_default;
571 : :
572 : 96 : tle = makeTargetEntry((Expr *) caseexpr,
573 : 48 : list_length(newq2->targetList) + 1,
574 : 48 : cte->cycle_clause->cycle_mark_column,
575 : : false);
576 : 48 : newq2->targetList = lappend(newq2->targetList, tle);
577 : :
578 : : /*
579 : : * cpa || ARRAY[ROW(cols)]
580 : : */
581 : 48 : tle = makeTargetEntry(make_path_cat_expr(cycle_col_rowexpr, cpa_attno),
582 : 48 : list_length(newq2->targetList) + 1,
583 : 48 : cte->cycle_clause->cycle_path_column,
584 : : false);
585 : 48 : newq2->targetList = lappend(newq2->targetList, tle);
586 : : }
587 : :
588 : 92 : rte2->subquery = newq2;
589 : :
590 [ + + ]: 92 : if (cte->search_clause)
591 : : {
592 : 52 : rte2->eref->colnames = lappend(rte2->eref->colnames, makeString(cte->search_clause->search_seq_column));
593 : : }
594 [ + + ]: 92 : if (cte->cycle_clause)
595 : : {
596 : 48 : rte2->eref->colnames = lappend(rte2->eref->colnames, makeString(cte->cycle_clause->cycle_mark_column));
597 : 48 : rte2->eref->colnames = lappend(rte2->eref->colnames, makeString(cte->cycle_clause->cycle_path_column));
598 : : }
599 : :
600 : : /*
601 : : * Add the additional columns to the SetOperationStmt
602 : : */
603 [ + + ]: 92 : if (cte->search_clause)
604 : : {
605 : 52 : sos->colTypes = lappend_oid(sos->colTypes, search_seq_type);
606 : 52 : sos->colTypmods = lappend_int(sos->colTypmods, -1);
607 : 52 : sos->colCollations = lappend_oid(sos->colCollations, InvalidOid);
608 [ + + ]: 52 : if (!sos->all)
609 : 8 : sos->groupClauses = lappend(sos->groupClauses,
610 : 8 : makeSortGroupClauseForSetOp(search_seq_type, true));
611 : : }
612 [ + + ]: 92 : if (cte->cycle_clause)
613 : : {
614 : 48 : sos->colTypes = lappend_oid(sos->colTypes, cte->cycle_clause->cycle_mark_type);
615 : 48 : sos->colTypmods = lappend_int(sos->colTypmods, cte->cycle_clause->cycle_mark_typmod);
616 : 48 : sos->colCollations = lappend_oid(sos->colCollations, cte->cycle_clause->cycle_mark_collation);
617 [ + + ]: 48 : if (!sos->all)
618 : 4 : sos->groupClauses = lappend(sos->groupClauses,
619 : 4 : makeSortGroupClauseForSetOp(cte->cycle_clause->cycle_mark_type, true));
620 : :
621 : 48 : sos->colTypes = lappend_oid(sos->colTypes, RECORDARRAYOID);
622 : 48 : sos->colTypmods = lappend_int(sos->colTypmods, -1);
623 : 48 : sos->colCollations = lappend_oid(sos->colCollations, InvalidOid);
624 [ + + ]: 48 : if (!sos->all)
625 : 4 : sos->groupClauses = lappend(sos->groupClauses,
626 : 4 : makeSortGroupClauseForSetOp(RECORDARRAYOID, true));
627 : : }
628 : :
629 : : /*
630 : : * Add the additional columns to the CTE query's target list
631 : : */
632 [ + + ]: 92 : if (cte->search_clause)
633 : : {
634 : 52 : ctequery->targetList = lappend(ctequery->targetList,
635 : 52 : makeTargetEntry((Expr *) makeVar(1, sqc_attno,
636 : : search_seq_type, -1, InvalidOid, 0),
637 : 52 : list_length(ctequery->targetList) + 1,
638 : 52 : cte->search_clause->search_seq_column,
639 : : false));
640 : : }
641 [ + + ]: 92 : if (cte->cycle_clause)
642 : : {
643 : 48 : ctequery->targetList = lappend(ctequery->targetList,
644 : 48 : makeTargetEntry((Expr *) makeVar(1, cmc_attno,
645 : 48 : cte->cycle_clause->cycle_mark_type,
646 : 48 : cte->cycle_clause->cycle_mark_typmod,
647 : 48 : cte->cycle_clause->cycle_mark_collation, 0),
648 : 48 : list_length(ctequery->targetList) + 1,
649 : 48 : cte->cycle_clause->cycle_mark_column,
650 : : false));
651 : 48 : ctequery->targetList = lappend(ctequery->targetList,
652 : 48 : makeTargetEntry((Expr *) makeVar(1, cpa_attno,
653 : : RECORDARRAYOID, -1, InvalidOid, 0),
654 : 48 : list_length(ctequery->targetList) + 1,
655 : 48 : cte->cycle_clause->cycle_path_column,
656 : : false));
657 : : }
658 : :
659 : : /*
660 : : * Add the additional columns to the CTE's output columns
661 : : */
662 : 92 : cte->ctecolnames = ewcl;
663 [ + + ]: 92 : if (cte->search_clause)
664 : : {
665 : 52 : cte->ctecoltypes = lappend_oid(cte->ctecoltypes, search_seq_type);
666 : 52 : cte->ctecoltypmods = lappend_int(cte->ctecoltypmods, -1);
667 : 52 : cte->ctecolcollations = lappend_oid(cte->ctecolcollations, InvalidOid);
668 : : }
669 [ + + ]: 92 : if (cte->cycle_clause)
670 : : {
671 : 48 : cte->ctecoltypes = lappend_oid(cte->ctecoltypes, cte->cycle_clause->cycle_mark_type);
672 : 48 : cte->ctecoltypmods = lappend_int(cte->ctecoltypmods, cte->cycle_clause->cycle_mark_typmod);
673 : 48 : cte->ctecolcollations = lappend_oid(cte->ctecolcollations, cte->cycle_clause->cycle_mark_collation);
674 : :
675 : 48 : cte->ctecoltypes = lappend_oid(cte->ctecoltypes, RECORDARRAYOID);
676 : 48 : cte->ctecoltypmods = lappend_int(cte->ctecoltypmods, -1);
677 : 48 : cte->ctecolcollations = lappend_oid(cte->ctecolcollations, InvalidOid);
678 : : }
679 : :
680 : 92 : return cte;
681 : : }
|
