rewriteSearchCycle.c

Go to the documentation of this file.

/*-------------------------------------------------------------------------
 *
 * rewriteSearchCycle.c
 *      Support for rewriting SEARCH and CYCLE clauses.
 *
 * Portions Copyright (c) 1996-2026, PostgreSQL Global Development Group
 * Portions Copyright (c) 1994, Regents of the University of California
 *
 * IDENTIFICATION
 *    src/backend/rewrite/rewriteSearchCycle.c
 *
 *-------------------------------------------------------------------------
 */
#include "postgres.h"
 
#include "catalog/pg_operator_d.h"
#include "catalog/pg_type_d.h"
#include "nodes/makefuncs.h"
#include "nodes/parsenodes.h"
#include "nodes/pg_list.h"
#include "nodes/primnodes.h"
#include "parser/analyze.h"
#include "parser/parsetree.h"
#include "rewrite/rewriteManip.h"
#include "rewrite/rewriteSearchCycle.h"
#include "utils/fmgroids.h"
 
 
/*----------
 * Rewrite a CTE with SEARCH or CYCLE clause
 *
 * Consider a CTE like
 *
 * WITH RECURSIVE ctename (col1, col2, col3) AS (
 *     query1
 *   UNION [ALL]
 *     SELECT trosl FROM ctename
 * )
 *
 * With a search clause
 *
 * SEARCH BREADTH FIRST BY col1, col2 SET sqc
 *
 * the CTE is rewritten to
 *
 * WITH RECURSIVE ctename (col1, col2, col3, sqc) AS (
 *     SELECT col1, col2, col3,               -- original WITH column list
 *            ROW(0, col1, col2)              -- initial row of search columns
 *       FROM (query1) "*TLOCRN*" (col1, col2, col3)
 *   UNION [ALL]
 *     SELECT col1, col2, col3,               -- same as above
 *            ROW(sqc.depth + 1, col1, col2)  -- count depth
 *       FROM (SELECT trosl, ctename.sqc FROM ctename) "*TROCRN*" (col1, col2, col3, sqc)
 * )
 *
 * (This isn't quite legal SQL: sqc.depth is meant to refer to the first
 * column of sqc, which has a row type, but the field names are not defined
 * here.  Representing this properly in SQL would be more complicated (and the
 * SQL standard actually does it in that more complicated way), but the
 * internal representation allows us to construct it this way.)
 *
 * With a search clause
 *
 * SEARCH DEPTH FIRST BY col1, col2 SET sqc
 *
 * the CTE is rewritten to
 *
 * WITH RECURSIVE ctename (col1, col2, col3, sqc) AS (
 *     SELECT col1, col2, col3,               -- original WITH column list
 *            ARRAY[ROW(col1, col2)]          -- initial row of search columns
 *       FROM (query1) "*TLOCRN*" (col1, col2, col3)
 *   UNION [ALL]
 *     SELECT col1, col2, col3,               -- same as above
 *            sqc || ARRAY[ROW(col1, col2)]   -- record rows seen
 *       FROM (SELECT trosl, ctename.sqc FROM ctename) "*TROCRN*" (col1, col2, col3, sqc)
 * )
 *
 * With a cycle clause
 *
 * CYCLE col1, col2 SET cmc TO 'Y' DEFAULT 'N' USING cpa
 *
 * (cmc = cycle mark column, cpa = cycle path) the CTE is rewritten to
 *
 * WITH RECURSIVE ctename (col1, col2, col3, cmc, cpa) AS (
 *     SELECT col1, col2, col3,               -- original WITH column list
 *            'N',                            -- cycle mark default
 *            ARRAY[ROW(col1, col2)]          -- initial row of cycle columns
 *       FROM (query1) "*TLOCRN*" (col1, col2, col3)
 *   UNION [ALL]
 *     SELECT col1, col2, col3,               -- same as above
 *            CASE WHEN ROW(col1, col2) = ANY (ARRAY[cpa]) THEN 'Y' ELSE 'N' END,  -- compute cycle mark column
 *            cpa || ARRAY[ROW(col1, col2)]   -- record rows seen
 *       FROM (SELECT trosl, ctename.cmc, ctename.cpa FROM ctename) "*TROCRN*" (col1, col2, col3, cmc, cpa)
 *       WHERE cmc <> 'Y'
 * )
 *
 * The expression to compute the cycle mark column in the right-hand query is
 * written as
 *
 * CASE WHEN ROW(col1, col2) IN (SELECT p.* FROM TABLE(cpa) p) THEN cmv ELSE cmd END
 *
 * in the SQL standard, but in PostgreSQL we can use the scalar-array operator
 * expression shown above.
 *
 * Also, in some of the cases where operators are shown above we actually
 * directly produce the underlying function call.
 *
 * If both a search clause and a cycle clause is specified, then the search
 * clause column is added before the cycle clause columns.
 */
 
/*
 * Make a RowExpr from the specified column names, which have to be among the
 * output columns of the CTE.
 */
static RowExpr *
make_path_rowexpr(const CommonTableExpr *cte, const List *col_list)
{
    RowExpr    *rowexpr;
    ListCell   *lc;
 
    rowexpr = makeNode(RowExpr);
    rowexpr->row_typeid = RECORDOID;
    rowexpr->row_format = COERCE_IMPLICIT_CAST;
    rowexpr->location = -1;
 
    foreach(lc, col_list)
    {
        char       *colname = strVal(lfirst(lc));
 
        for (int i = 0; i < list_length(cte->ctecolnames); i++)
        {
            char       *colname2 = strVal(list_nth(cte->ctecolnames, i));
 
            if (strcmp(colname, colname2) == 0)
            {
                Var        *var;
 
                var = makeVar(1, i + 1,
                              list_nth_oid(cte->ctecoltypes, i),
                              list_nth_int(cte->ctecoltypmods, i),
                              list_nth_oid(cte->ctecolcollations, i),
                              0);
                rowexpr->args = lappend(rowexpr->args, var);
                rowexpr->colnames = lappend(rowexpr->colnames, makeString(colname));
                break;
            }
        }
    }
 
    return rowexpr;
}
 
/*
 * Wrap a RowExpr in an ArrayExpr, for the initial search depth first or cycle
 * row.
 */
static Expr *
make_path_initial_array(RowExpr *rowexpr)
{
    ArrayExpr  *arr;
 
    arr = makeNode(ArrayExpr);
    arr->array_typeid = RECORDARRAYOID;
    arr->element_typeid = RECORDOID;
    arr->location = -1;
    arr->elements = list_make1(rowexpr);
 
    return (Expr *) arr;
}
 
/*
 * Make an array catenation expression like
 *
 * cpa || ARRAY[ROW(cols)]
 *
 * where the varattno of cpa is provided as path_varattno.
 */
static Expr *
make_path_cat_expr(RowExpr *rowexpr, AttrNumber path_varattno)
{
    ArrayExpr  *arr;
    FuncExpr   *fexpr;
 
    arr = makeNode(ArrayExpr);
    arr->array_typeid = RECORDARRAYOID;
    arr->element_typeid = RECORDOID;
    arr->location = -1;
    arr->elements = list_make1(rowexpr);
 
    fexpr = makeFuncExpr(F_ARRAY_CAT, RECORDARRAYOID,
                         list_make2(makeVar(1, path_varattno, RECORDARRAYOID, -1, 0, 0),
                                    arr),
                         InvalidOid, InvalidOid, COERCE_EXPLICIT_CALL);
 
    return (Expr *) fexpr;
}
 
/*
 * The real work happens here.
 */
CommonTableExpr *
rewriteSearchAndCycle(CommonTableExpr *cte)
{
    Query      *ctequery;
    SetOperationStmt *sos;
    int         rti1,
                rti2;
    RangeTblEntry *rte1,
               *rte2,
               *newrte;
    Query      *newq1,
               *newq2;
    Query      *newsubquery;
    RangeTblRef *rtr;
    Oid         search_seq_type = InvalidOid;
    AttrNumber  sqc_attno = InvalidAttrNumber;
    AttrNumber  cmc_attno = InvalidAttrNumber;
    AttrNumber  cpa_attno = InvalidAttrNumber;
    TargetEntry *tle;
    RowExpr    *cycle_col_rowexpr = NULL;
    RowExpr    *search_col_rowexpr = NULL;
    List       *ewcl;
    int         cte_rtindex = -1;
 
    Assert(cte->search_clause || cte->cycle_clause);
 
    cte = copyObject(cte);
 
    ctequery = castNode(Query, cte->ctequery);
 
    /*
     * The top level of the CTE's query should be a UNION.  Find the two
     * subqueries.
     */
    Assert(ctequery->setOperations);
    sos = castNode(SetOperationStmt, ctequery->setOperations);
    Assert(sos->op == SETOP_UNION);
 
    rti1 = castNode(RangeTblRef, sos->larg)->rtindex;
    rti2 = castNode(RangeTblRef, sos->rarg)->rtindex;
 
    rte1 = rt_fetch(rti1, ctequery->rtable);
    rte2 = rt_fetch(rti2, ctequery->rtable);
 
    Assert(rte1->rtekind == RTE_SUBQUERY);
    Assert(rte2->rtekind == RTE_SUBQUERY);
 
    /*
     * We'll need this a few times later.
     */
    if (cte->search_clause)
    {
        if (cte->search_clause->search_breadth_first)
            search_seq_type = RECORDOID;
        else
            search_seq_type = RECORDARRAYOID;
    }
 
    /*
     * Attribute numbers of the added columns in the CTE's column list
     */
    if (cte->search_clause)
        sqc_attno = list_length(cte->ctecolnames) + 1;
    if (cte->cycle_clause)
    {
        cmc_attno = list_length(cte->ctecolnames) + 1;
        cpa_attno = list_length(cte->ctecolnames) + 2;
        if (cte->search_clause)
        {
            cmc_attno++;
            cpa_attno++;
        }
    }
 
    /*
     * Make new left subquery
     */
    newq1 = makeNode(Query);
    newq1->commandType = CMD_SELECT;
    newq1->canSetTag = true;
 
    newrte = makeNode(RangeTblEntry);
    newrte->rtekind = RTE_SUBQUERY;
    newrte->alias = NULL;
    newrte->eref = makeAlias("*TLOCRN*", cte->ctecolnames);
    newsubquery = copyObject(rte1->subquery);
    IncrementVarSublevelsUp((Node *) newsubquery, 1, 1);
    newrte->subquery = newsubquery;
    newrte->inFromCl = true;
    newq1->rtable = list_make1(newrte);
 
    rtr = makeNode(RangeTblRef);
    rtr->rtindex = 1;
    newq1->jointree = makeFromExpr(list_make1(rtr), NULL);
 
    /*
     * Make target list
     */
    for (int i = 0; i < list_length(cte->ctecolnames); i++)
    {
        Var        *var;
 
        var = makeVar(1, i + 1,
                      list_nth_oid(cte->ctecoltypes, i),
                      list_nth_int(cte->ctecoltypmods, i),
                      list_nth_oid(cte->ctecolcollations, i),
                      0);
        tle = makeTargetEntry((Expr *) var, i + 1, strVal(list_nth(cte->ctecolnames, i)), false);
        tle->resorigtbl = list_nth_node(TargetEntry, rte1->subquery->targetList, i)->resorigtbl;
        tle->resorigcol = list_nth_node(TargetEntry, rte1->subquery->targetList, i)->resorigcol;
        newq1->targetList = lappend(newq1->targetList, tle);
    }
 
    if (cte->search_clause)
    {
        Expr       *texpr;
 
        search_col_rowexpr = make_path_rowexpr(cte, cte->search_clause->search_col_list);
        if (cte->search_clause->search_breadth_first)
        {
            search_col_rowexpr->args = lcons(makeConst(INT8OID, -1, InvalidOid, sizeof(int64),
                                                       Int64GetDatum(0), false, true),
                                             search_col_rowexpr->args);
            search_col_rowexpr->colnames = lcons(makeString("*DEPTH*"), search_col_rowexpr->colnames);
            texpr = (Expr *) search_col_rowexpr;
        }
        else
            texpr = make_path_initial_array(search_col_rowexpr);
        tle = makeTargetEntry(texpr,
                              list_length(newq1->targetList) + 1,
                              cte->search_clause->search_seq_column,
                              false);
        newq1->targetList = lappend(newq1->targetList, tle);
    }
    if (cte->cycle_clause)
    {
        tle = makeTargetEntry((Expr *) cte->cycle_clause->cycle_mark_default,
                              list_length(newq1->targetList) + 1,
                              cte->cycle_clause->cycle_mark_column,
                              false);
        newq1->targetList = lappend(newq1->targetList, tle);
        cycle_col_rowexpr = make_path_rowexpr(cte, cte->cycle_clause->cycle_col_list);
        tle = makeTargetEntry(make_path_initial_array(cycle_col_rowexpr),
                              list_length(newq1->targetList) + 1,
                              cte->cycle_clause->cycle_path_column,
                              false);
        newq1->targetList = lappend(newq1->targetList, tle);
    }
 
    rte1->subquery = newq1;
 
    if (cte->search_clause)
    {
        rte1->eref->colnames = lappend(rte1->eref->colnames, makeString(cte->search_clause->search_seq_column));
    }
    if (cte->cycle_clause)
    {
        rte1->eref->colnames = lappend(rte1->eref->colnames, makeString(cte->cycle_clause->cycle_mark_column));
        rte1->eref->colnames = lappend(rte1->eref->colnames, makeString(cte->cycle_clause->cycle_path_column));
    }
 
    /*
     * Make new right subquery
     */
    newq2 = makeNode(Query);
    newq2->commandType = CMD_SELECT;
    newq2->canSetTag = true;
 
    newrte = makeNode(RangeTblEntry);
    newrte->rtekind = RTE_SUBQUERY;
    ewcl = copyObject(cte->ctecolnames);
    if (cte->search_clause)
    {
        ewcl = lappend(ewcl, makeString(cte->search_clause->search_seq_column));
    }
    if (cte->cycle_clause)
    {
        ewcl = lappend(ewcl, makeString(cte->cycle_clause->cycle_mark_column));
        ewcl = lappend(ewcl, makeString(cte->cycle_clause->cycle_path_column));
    }
    newrte->alias = NULL;
    newrte->eref = makeAlias("*TROCRN*", ewcl);
 
    /*
     * Find the reference to the recursive CTE in the right UNION subquery's
     * range table.  We expect it to be two levels up from the UNION subquery
     * (and must check that to avoid being fooled by sub-WITHs with the same
     * CTE name).  There will not be more than one such reference, because the
     * parser would have rejected that (see checkWellFormedRecursion() in
     * parse_cte.c).  However, the parser doesn't insist that the reference
     * appear in the UNION subquery's topmost range table, so we might fail to
     * find it at all.  That's an unimplemented case for the moment.
     */
    for (int rti = 1; rti <= list_length(rte2->subquery->rtable); rti++)
    {
        RangeTblEntry *e = rt_fetch(rti, rte2->subquery->rtable);
 
        if (e->rtekind == RTE_CTE &&
            strcmp(cte->ctename, e->ctename) == 0 &&
            e->ctelevelsup == 2)
        {
            cte_rtindex = rti;
            break;
        }
    }
    if (cte_rtindex <= 0)
        ereport(ERROR,
                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                 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",
                        cte->ctename)));
 
    newsubquery = copyObject(rte2->subquery);
    IncrementVarSublevelsUp((Node *) newsubquery, 1, 1);
 
    /*
     * Add extra columns to target list of subquery of right subquery
     */
    if (cte->search_clause)
    {
        Var        *var;
 
        /* ctename.sqc */
        var = makeVar(cte_rtindex, sqc_attno,
                      search_seq_type, -1, InvalidOid, 0);
        tle = makeTargetEntry((Expr *) var,
                              list_length(newsubquery->targetList) + 1,
                              cte->search_clause->search_seq_column,
                              false);
        newsubquery->targetList = lappend(newsubquery->targetList, tle);
    }
    if (cte->cycle_clause)
    {
        Var        *var;
 
        /* ctename.cmc */
        var = makeVar(cte_rtindex, cmc_attno,
                      cte->cycle_clause->cycle_mark_type,
                      cte->cycle_clause->cycle_mark_typmod,
                      cte->cycle_clause->cycle_mark_collation, 0);
        tle = makeTargetEntry((Expr *) var,
                              list_length(newsubquery->targetList) + 1,
                              cte->cycle_clause->cycle_mark_column,
                              false);
        newsubquery->targetList = lappend(newsubquery->targetList, tle);
 
        /* ctename.cpa */
        var = makeVar(cte_rtindex, cpa_attno,
                      RECORDARRAYOID, -1, InvalidOid, 0);
        tle = makeTargetEntry((Expr *) var,
                              list_length(newsubquery->targetList) + 1,
                              cte->cycle_clause->cycle_path_column,
                              false);
        newsubquery->targetList = lappend(newsubquery->targetList, tle);
    }
 
    newrte->subquery = newsubquery;
    newrte->inFromCl = true;
    newq2->rtable = list_make1(newrte);
 
    rtr = makeNode(RangeTblRef);
    rtr->rtindex = 1;
 
    if (cte->cycle_clause)
    {
        Expr       *expr;
 
        /*
         * Add cmc <> cmv condition
         */
        expr = make_opclause(cte->cycle_clause->cycle_mark_neop, BOOLOID, false,
                             (Expr *) makeVar(1, cmc_attno,
                                              cte->cycle_clause->cycle_mark_type,
                                              cte->cycle_clause->cycle_mark_typmod,
                                              cte->cycle_clause->cycle_mark_collation, 0),
                             (Expr *) cte->cycle_clause->cycle_mark_value,
                             InvalidOid,
                             cte->cycle_clause->cycle_mark_collation);
 
        newq2->jointree = makeFromExpr(list_make1(rtr), (Node *) expr);
    }
    else
        newq2->jointree = makeFromExpr(list_make1(rtr), NULL);
 
    /*
     * Make target list
     */
    for (int i = 0; i < list_length(cte->ctecolnames); i++)
    {
        Var        *var;
 
        var = makeVar(1, i + 1,
                      list_nth_oid(cte->ctecoltypes, i),
                      list_nth_int(cte->ctecoltypmods, i),
                      list_nth_oid(cte->ctecolcollations, i),
                      0);
        tle = makeTargetEntry((Expr *) var, i + 1, strVal(list_nth(cte->ctecolnames, i)), false);
        tle->resorigtbl = list_nth_node(TargetEntry, rte2->subquery->targetList, i)->resorigtbl;
        tle->resorigcol = list_nth_node(TargetEntry, rte2->subquery->targetList, i)->resorigcol;
        newq2->targetList = lappend(newq2->targetList, tle);
    }
 
    if (cte->search_clause)
    {
        Expr       *texpr;
 
        if (cte->search_clause->search_breadth_first)
        {
            FieldSelect *fs;
            FuncExpr   *fexpr;
 
            /*
             * ROW(sqc.depth + 1, cols)
             */
 
            search_col_rowexpr = copyObject(search_col_rowexpr);
 
            fs = makeNode(FieldSelect);
            fs->arg = (Expr *) makeVar(1, sqc_attno, RECORDOID, -1, 0, 0);
            fs->fieldnum = 1;
            fs->resulttype = INT8OID;
            fs->resulttypmod = -1;
 
            fexpr = makeFuncExpr(F_INT8INC, INT8OID, list_make1(fs), InvalidOid, InvalidOid, COERCE_EXPLICIT_CALL);
 
            linitial(search_col_rowexpr->args) = fexpr;
 
            texpr = (Expr *) search_col_rowexpr;
        }
        else
        {
            /*
             * sqc || ARRAY[ROW(cols)]
             */
            texpr = make_path_cat_expr(search_col_rowexpr, sqc_attno);
        }
        tle = makeTargetEntry(texpr,
                              list_length(newq2->targetList) + 1,
                              cte->search_clause->search_seq_column,
                              false);
        newq2->targetList = lappend(newq2->targetList, tle);
    }
 
    if (cte->cycle_clause)
    {
        ScalarArrayOpExpr *saoe;
        CaseExpr   *caseexpr;
        CaseWhen   *casewhen;
 
        /*
         * CASE WHEN ROW(cols) = ANY (ARRAY[cpa]) THEN cmv ELSE cmd END
         */
 
        saoe = makeNode(ScalarArrayOpExpr);
        saoe->location = -1;
        saoe->opno = RECORD_EQ_OP;
        saoe->useOr = true;
        saoe->args = list_make2(cycle_col_rowexpr,
                                makeVar(1, cpa_attno, RECORDARRAYOID, -1, 0, 0));
 
        caseexpr = makeNode(CaseExpr);
        caseexpr->location = -1;
        caseexpr->casetype = cte->cycle_clause->cycle_mark_type;
        caseexpr->casecollid = cte->cycle_clause->cycle_mark_collation;
        casewhen = makeNode(CaseWhen);
        casewhen->location = -1;
        casewhen->expr = (Expr *) saoe;
        casewhen->result = (Expr *) cte->cycle_clause->cycle_mark_value;
        caseexpr->args = list_make1(casewhen);
        caseexpr->defresult = (Expr *) cte->cycle_clause->cycle_mark_default;
 
        tle = makeTargetEntry((Expr *) caseexpr,
                              list_length(newq2->targetList) + 1,
                              cte->cycle_clause->cycle_mark_column,
                              false);
        newq2->targetList = lappend(newq2->targetList, tle);
 
        /*
         * cpa || ARRAY[ROW(cols)]
         */
        tle = makeTargetEntry(make_path_cat_expr(cycle_col_rowexpr, cpa_attno),
                              list_length(newq2->targetList) + 1,
                              cte->cycle_clause->cycle_path_column,
                              false);
        newq2->targetList = lappend(newq2->targetList, tle);
    }
 
    rte2->subquery = newq2;
 
    if (cte->search_clause)
    {
        rte2->eref->colnames = lappend(rte2->eref->colnames, makeString(cte->search_clause->search_seq_column));
    }
    if (cte->cycle_clause)
    {
        rte2->eref->colnames = lappend(rte2->eref->colnames, makeString(cte->cycle_clause->cycle_mark_column));
        rte2->eref->colnames = lappend(rte2->eref->colnames, makeString(cte->cycle_clause->cycle_path_column));
    }
 
    /*
     * Add the additional columns to the SetOperationStmt
     */
    if (cte->search_clause)
    {
        sos->colTypes = lappend_oid(sos->colTypes, search_seq_type);
        sos->colTypmods = lappend_int(sos->colTypmods, -1);
        sos->colCollations = lappend_oid(sos->colCollations, InvalidOid);
        if (!sos->all)
            sos->groupClauses = lappend(sos->groupClauses,
                                        makeSortGroupClauseForSetOp(search_seq_type, true));
    }
    if (cte->cycle_clause)
    {
        sos->colTypes = lappend_oid(sos->colTypes, cte->cycle_clause->cycle_mark_type);
        sos->colTypmods = lappend_int(sos->colTypmods, cte->cycle_clause->cycle_mark_typmod);
        sos->colCollations = lappend_oid(sos->colCollations, cte->cycle_clause->cycle_mark_collation);
        if (!sos->all)
            sos->groupClauses = lappend(sos->groupClauses,
                                        makeSortGroupClauseForSetOp(cte->cycle_clause->cycle_mark_type, true));
 
        sos->colTypes = lappend_oid(sos->colTypes, RECORDARRAYOID);
        sos->colTypmods = lappend_int(sos->colTypmods, -1);
        sos->colCollations = lappend_oid(sos->colCollations, InvalidOid);
        if (!sos->all)
            sos->groupClauses = lappend(sos->groupClauses,
                                        makeSortGroupClauseForSetOp(RECORDARRAYOID, true));
    }
 
    /*
     * Add the additional columns to the CTE query's target list
     */
    if (cte->search_clause)
    {
        ctequery->targetList = lappend(ctequery->targetList,
                                       makeTargetEntry((Expr *) makeVar(1, sqc_attno,
                                                                        search_seq_type, -1, InvalidOid, 0),
                                                       list_length(ctequery->targetList) + 1,
                                                       cte->search_clause->search_seq_column,
                                                       false));
    }
    if (cte->cycle_clause)
    {
        ctequery->targetList = lappend(ctequery->targetList,
                                       makeTargetEntry((Expr *) makeVar(1, cmc_attno,
                                                                        cte->cycle_clause->cycle_mark_type,
                                                                        cte->cycle_clause->cycle_mark_typmod,
                                                                        cte->cycle_clause->cycle_mark_collation, 0),
                                                       list_length(ctequery->targetList) + 1,
                                                       cte->cycle_clause->cycle_mark_column,
                                                       false));
        ctequery->targetList = lappend(ctequery->targetList,
                                       makeTargetEntry((Expr *) makeVar(1, cpa_attno,
                                                                        RECORDARRAYOID, -1, InvalidOid, 0),
                                                       list_length(ctequery->targetList) + 1,
                                                       cte->cycle_clause->cycle_path_column,
                                                       false));
    }
 
    /*
     * Add the additional columns to the CTE's output columns
     */
    cte->ctecolnames = ewcl;
    if (cte->search_clause)
    {
        cte->ctecoltypes = lappend_oid(cte->ctecoltypes, search_seq_type);
        cte->ctecoltypmods = lappend_int(cte->ctecoltypmods, -1);
        cte->ctecolcollations = lappend_oid(cte->ctecolcollations, InvalidOid);
    }
    if (cte->cycle_clause)
    {
        cte->ctecoltypes = lappend_oid(cte->ctecoltypes, cte->cycle_clause->cycle_mark_type);
        cte->ctecoltypmods = lappend_int(cte->ctecoltypmods, cte->cycle_clause->cycle_mark_typmod);
        cte->ctecolcollations = lappend_oid(cte->ctecolcollations, cte->cycle_clause->cycle_mark_collation);
 
        cte->ctecoltypes = lappend_oid(cte->ctecoltypes, RECORDARRAYOID);
        cte->ctecoltypmods = lappend_int(cte->ctecoltypmods, -1);
        cte->ctecolcollations = lappend_oid(cte->ctecolcollations, InvalidOid);
    }
 
    return cte;
}