nodeProjectSet.c File Reference

#include "postgres.h"
#include "executor/executor.h"
#include "executor/nodeProjectSet.h"
#include "miscadmin.h"
#include "nodes/nodeFuncs.h"

Include dependency graph for nodeProjectSet.c:

Go to the source code of this file.

Functions
static TupleTableSlot *	ExecProjectSRF (ProjectSetState *node, bool continuing)
static TupleTableSlot *	ExecProjectSet (PlanState *pstate)
ProjectSetState *	ExecInitProjectSet (ProjectSet *node, EState *estate, int eflags)
void	ExecEndProjectSet (ProjectSetState *node)
void	ExecReScanProjectSet (ProjectSetState *node)

Function Documentation

ExecEndProjectSet()

void ExecEndProjectSet

(

ProjectSetState *

node

)

Definition at line 328 of file nodeProjectSet.c.

{
    /*
     * shut down subplans
     */
    ExecEndNode(outerPlanState(node));
}

References ExecEndNode(), and outerPlanState.

Referenced by ExecEndNode().

ExecInitProjectSet()

ProjectSetState * ExecInitProjectSet	(	ProjectSet *	node,
		EState *	estate,
		int	eflags
	)

Definition at line 227 of file nodeProjectSet.c.

{
    ProjectSetState *state;
    ListCell   *lc;
    int         off;
 
    /* check for unsupported flags */
    Assert(!(eflags & (EXEC_FLAG_MARK | EXEC_FLAG_BACKWARD)));
 
    /*
     * create state structure
     */
    state = makeNode(ProjectSetState);
    state->ps.plan = (Plan *) node;
    state->ps.state = estate;
    state->ps.ExecProcNode = ExecProjectSet;
 
    state->pending_srf_tuples = false;
 
    /*
     * Miscellaneous initialization
     *
     * create expression context for node
     */
    ExecAssignExprContext(estate, &state->ps);
 
    /*
     * initialize child nodes
     */
    outerPlanState(state) = ExecInitNode(outerPlan(node), estate, eflags);
 
    /*
     * we don't use inner plan
     */
    Assert(innerPlan(node) == NULL);
 
    /*
     * tuple table and result type initialization
     */
    ExecInitResultTupleSlotTL(&state->ps, &TTSOpsVirtual);
 
    /* Create workspace for per-tlist-entry expr state & SRF-is-done state */
    state->nelems = list_length(node->plan.targetlist);
    state->elems = (Node **)
        palloc(sizeof(Node *) * state->nelems);
    state->elemdone = (ExprDoneCond *)
        palloc(sizeof(ExprDoneCond) * state->nelems);
 
    /*
     * Build expressions to evaluate targetlist.  We can't use
     * ExecBuildProjectionInfo here, since that doesn't deal with SRFs.
     * Instead compile each expression separately, using
     * ExecInitFunctionResultSet where applicable.
     */
    off = 0;
    foreach(lc, node->plan.targetlist)
    {
        TargetEntry *te = (TargetEntry *) lfirst(lc);
        Expr       *expr = te->expr;
 
        if ((IsA(expr, FuncExpr) && ((FuncExpr *) expr)->funcretset) ||
            (IsA(expr, OpExpr) && ((OpExpr *) expr)->opretset))
        {
            state->elems[off] = (Node *)
                ExecInitFunctionResultSet(expr, state->ps.ps_ExprContext,
                                          &state->ps);
        }
        else
        {
            Assert(!expression_returns_set((Node *) expr));
            state->elems[off] = (Node *) ExecInitExpr(expr, &state->ps);
        }
 
        off++;
    }
 
    /* We don't support any qual on ProjectSet nodes */
    Assert(node->plan.qual == NIL);
 
    /*
     * Create a memory context that ExecMakeFunctionResultSet can use to
     * evaluate function arguments in.  We can't use the per-tuple context for
     * this because it gets reset too often; but we don't want to leak
     * evaluation results into the query-lifespan context either.  We use one
     * context for the arguments of all tSRFs, as they have roughly equivalent
     * lifetimes.
     */
    state->argcontext = AllocSetContextCreate(CurrentMemoryContext,
                                              "tSRF function arguments",
                                              ALLOCSET_DEFAULT_SIZES);
 
    return state;
}

References ALLOCSET_DEFAULT_SIZES, AllocSetContextCreate, Assert, CurrentMemoryContext, EXEC_FLAG_BACKWARD, EXEC_FLAG_MARK, ExecAssignExprContext(), ExecInitExpr(), ExecInitFunctionResultSet(), ExecInitNode(), ExecInitResultTupleSlotTL(), ExecProjectSet(), TargetEntry::expr, expression_returns_set(), innerPlan, IsA, lfirst, list_length(), makeNode, NIL, outerPlan, outerPlanState, palloc(), ProjectSet::plan, Plan::qual, Plan::targetlist, and TTSOpsVirtual.

Referenced by ExecInitNode().

ExecProjectSet()

static TupleTableSlot * ExecProjectSet ( PlanState *  pstate )

static

Definition at line 42 of file nodeProjectSet.c.

{
    ProjectSetState *node = castNode(ProjectSetState, pstate);
    TupleTableSlot *outerTupleSlot;
    TupleTableSlot *resultSlot;
    PlanState  *outerPlan;
    ExprContext *econtext;
 
    CHECK_FOR_INTERRUPTS();
 
    econtext = node->ps.ps_ExprContext;
 
    /*
     * Reset per-tuple context to free expression-evaluation storage allocated
     * for a potentially previously returned tuple. Note that the SRF argument
     * context has a different lifetime and is reset below.
     */
    ResetExprContext(econtext);
 
    /*
     * Check to see if we're still projecting out tuples from a previous scan
     * tuple (because there is a function-returning-set in the projection
     * expressions).  If so, try to project another one.
     */
    if (node->pending_srf_tuples)
    {
        resultSlot = ExecProjectSRF(node, true);
 
        if (resultSlot != NULL)
            return resultSlot;
    }
 
    /*
     * Get another input tuple and project SRFs from it.
     */
    for (;;)
    {
        /*
         * Reset argument context to free any expression evaluation storage
         * allocated in the previous tuple cycle.  Note this can't happen
         * until we're done projecting out tuples from a scan tuple, as
         * ValuePerCall functions are allowed to reference the arguments for
         * each returned tuple.  However, if we loop around after finding that
         * no rows are produced from a scan tuple, we should reset, to avoid
         * leaking memory when many successive scan tuples produce no rows.
         */
        MemoryContextReset(node->argcontext);
 
        /*
         * Retrieve tuples from the outer plan until there are no more.
         */
        outerPlan = outerPlanState(node);
        outerTupleSlot = ExecProcNode(outerPlan);
 
        if (TupIsNull(outerTupleSlot))
            return NULL;
 
        /*
         * Prepare to compute projection expressions, which will expect to
         * access the input tuples as varno OUTER.
         */
        econtext->ecxt_outertuple = outerTupleSlot;
 
        /* Evaluate the expressions */
        resultSlot = ExecProjectSRF(node, false);
 
        /*
         * Return the tuple unless the projection produced no rows (due to an
         * empty set), in which case we must loop back to see if there are
         * more outerPlan tuples.
         */
        if (resultSlot)
            return resultSlot;
 
        /*
         * When we do loop back, we'd better reset the econtext again, just in
         * case the SRF leaked some memory there.
         */
        ResetExprContext(econtext);
    }
 
    return NULL;
}

References ProjectSetState::argcontext, castNode, CHECK_FOR_INTERRUPTS, ExprContext::ecxt_outertuple, ExecProcNode(), ExecProjectSRF(), MemoryContextReset(), outerPlan, outerPlanState, ProjectSetState::pending_srf_tuples, ProjectSetState::ps, PlanState::ps_ExprContext, ResetExprContext, and TupIsNull.

Referenced by ExecInitProjectSet().

ExecProjectSRF()

static TupleTableSlot * ExecProjectSRF ( ProjectSetState *  node, bool  continuing  )

static

Definition at line 139 of file nodeProjectSet.c.

{
    TupleTableSlot *resultSlot = node->ps.ps_ResultTupleSlot;
    ExprContext *econtext = node->ps.ps_ExprContext;
    MemoryContext oldcontext;
    bool        hassrf PG_USED_FOR_ASSERTS_ONLY;
    bool        hasresult;
    int         argno;
 
    ExecClearTuple(resultSlot);
 
    /* Call SRFs, as well as plain expressions, in per-tuple context */
    oldcontext = MemoryContextSwitchTo(econtext->ecxt_per_tuple_memory);
 
    /*
     * Assume no further tuples are produced unless an ExprMultipleResult is
     * encountered from a set returning function.
     */
    node->pending_srf_tuples = false;
 
    hassrf = hasresult = false;
    for (argno = 0; argno < node->nelems; argno++)
    {
        Node       *elem = node->elems[argno];
        ExprDoneCond *isdone = &node->elemdone[argno];
        Datum      *result = &resultSlot->tts_values[argno];
        bool       *isnull = &resultSlot->tts_isnull[argno];
 
        if (continuing && *isdone == ExprEndResult)
        {
            /*
             * If we're continuing to project output rows from a source tuple,
             * return NULLs once the SRF has been exhausted.
             */
            *result = (Datum) 0;
            *isnull = true;
            hassrf = true;
        }
        else if (IsA(elem, SetExprState))
        {
            /*
             * Evaluate SRF - possibly continuing previously started output.
             */
            *result = ExecMakeFunctionResultSet((SetExprState *) elem,
                                                econtext, node->argcontext,
                                                isnull, isdone);
 
            if (*isdone != ExprEndResult)
                hasresult = true;
            if (*isdone == ExprMultipleResult)
                node->pending_srf_tuples = true;
            hassrf = true;
        }
        else
        {
            /* Non-SRF tlist expression, just evaluate normally. */
            *result = ExecEvalExpr((ExprState *) elem, econtext, isnull);
            *isdone = ExprSingleResult;
        }
    }
 
    MemoryContextSwitchTo(oldcontext);
 
    /* ProjectSet should not be used if there's no SRFs */
    Assert(hassrf);
 
    /*
     * If all the SRFs returned ExprEndResult, we consider that as no row
     * being produced.
     */
    if (hasresult)
    {
        ExecStoreVirtualTuple(resultSlot);
        return resultSlot;
    }
 
    return NULL;
}

References ProjectSetState::argcontext, Assert, ExprContext::ecxt_per_tuple_memory, ProjectSetState::elemdone, ProjectSetState::elems, ExecClearTuple(), ExecEvalExpr(), ExecMakeFunctionResultSet(), ExecStoreVirtualTuple(), ExprEndResult, ExprMultipleResult, ExprSingleResult, IsA, MemoryContextSwitchTo(), ProjectSetState::nelems, ProjectSetState::pending_srf_tuples, PG_USED_FOR_ASSERTS_ONLY, ProjectSetState::ps, PlanState::ps_ExprContext, PlanState::ps_ResultTupleSlot, TupleTableSlot::tts_isnull, and TupleTableSlot::tts_values.

Referenced by ExecProjectSet().

ExecReScanProjectSet()

void ExecReScanProjectSet

(

ProjectSetState *

node

)

Definition at line 337 of file nodeProjectSet.c.

{
    PlanState  *outerPlan = outerPlanState(node);
 
    /* Forget any incompletely-evaluated SRFs */
    node->pending_srf_tuples = false;
 
    /*
     * If chgParam of subnode is not null then plan will be re-scanned by
     * first ExecProcNode.
     */
    if (outerPlan->chgParam == NULL)
        ExecReScan(outerPlan);
}

References ExecReScan(), outerPlan, outerPlanState, and ProjectSetState::pending_srf_tuples.

Referenced by ExecReScan().