mirror of https://github.com/postgres/postgres
ExecReScan for nodeAgg, nodeHash, nodeHashjoin, nodeNestloop and nodeResult. Fixed ExecReScan for nodeMaterial. Get rid of #ifdef INDEXSCAN_PATCH. Get rid of ExecMarkPos and ExecRestrPos in nodeNestloop.pull/50/head
Vadim B. Mikheev
28 years ago
parent
13637df458
commit
1a105cefbd
@ -0,0 +1,280 @@ |
||||
/*-------------------------------------------------------------------------
|
||||
* |
||||
* nodeSubplan.c-- |
||||
* routines to support subselects |
||||
* |
||||
*------------------------------------------------------------------------- |
||||
*/ |
||||
/*
|
||||
* INTERFACE ROUTINES |
||||
* ExecSubPlan - process a subselect |
||||
* ExecInitSubPlan - initialize a subselect |
||||
* ExecEndSubPlan - shut down a subselect |
||||
*/ |
||||
#include "postgres.h" |
||||
|
||||
#include "access/heapam.h" |
||||
#include "tcop/pquery.h" |
||||
#include "executor/executor.h" |
||||
#include "executor/execdebug.h" |
||||
#include "executor/nodeSubplan.h" |
||||
|
||||
/* ----------------------------------------------------------------
|
||||
* ExecSubPlan(node) |
||||
* |
||||
* ---------------------------------------------------------------- |
||||
*/ |
||||
Datum |
||||
ExecSubPlan(SubPlan *node, List *pvar, ExprContext *econtext) |
||||
{ |
||||
Plan *plan = node->plan; |
||||
SubLink *sublink = node->sublink; |
||||
TupleTableSlot *slot; |
||||
List *lst; |
||||
bool result = false; |
||||
bool found = false; |
||||
|
||||
if ( node->setParam != NULL ) |
||||
elog (ERROR, "ExecSubPlan: can't set parent params from subquery"); |
||||
|
||||
/*
|
||||
* Set Params of this plan from parent plan correlation Vars |
||||
*/ |
||||
if ( node->parParam != NULL ) |
||||
{ |
||||
foreach (lst, node->parParam) |
||||
{ |
||||
ParamExecData *prm = &(econtext->ecxt_param_exec_vals[lfirsti(lst)]); |
||||
|
||||
prm->value = ExecEvalExpr ((Node*) lfirst(pvar),
|
||||
econtext,
|
||||
&(prm->isnull), NULL); |
||||
pvar = lnext (pvar); |
||||
} |
||||
plan->chgParam = nconc (plan->chgParam, listCopy(node->parParam)); |
||||
} |
||||
|
||||
ExecReScan (plan, (ExprContext*) NULL, plan); |
||||
|
||||
for (slot = ExecProcNode (plan, plan);
|
||||
!TupIsNull(slot);
|
||||
slot = ExecProcNode (plan, plan)) |
||||
{ |
||||
HeapTuple tup = slot->val; |
||||
TupleDesc tdesc = slot->ttc_tupleDescriptor; |
||||
int i = 1; |
||||
|
||||
if ( sublink->subLinkType == EXPR_SUBLINK && found ) |
||||
{ |
||||
elog (ERROR, "ExecSubPlan: more than one tuple returned by expression subselect"); |
||||
return ((Datum) false); |
||||
} |
||||
|
||||
if ( sublink->subLinkType == EXISTS_SUBLINK ) |
||||
return ((Datum) true); |
||||
|
||||
found = true; |
||||
|
||||
foreach (lst, sublink->oper) |
||||
{ |
||||
Expr *expr = (Expr*) lfirst(lst); |
||||
Const *con = lsecond(expr->args); |
||||
bool isnull; |
||||
|
||||
con->constvalue = heap_getattr (tup, i, tdesc, &(con->constisnull)); |
||||
result = (bool) ExecEvalExpr ((Node*) expr, econtext, &isnull, (bool*) NULL); |
||||
if ( isnull ) |
||||
result = false; |
||||
if ( (!result && !(sublink->useor)) || (result && sublink->useor) ) |
||||
break; |
||||
i++; |
||||
} |
||||
|
||||
if ( (!result && sublink->subLinkType == ALL_SUBLINK) || |
||||
(result && sublink->subLinkType == ANY_SUBLINK) ) |
||||
break; |
||||
} |
||||
|
||||
return ((Datum) result); |
||||
} |
||||
|
||||
/* ----------------------------------------------------------------
|
||||
* ExecInitSubPlan |
||||
* |
||||
* ---------------------------------------------------------------- |
||||
*/ |
||||
bool |
||||
ExecInitSubPlan(SubPlan *node, EState *estate, Plan *parent) |
||||
{ |
||||
EState *sp_estate = CreateExecutorState (); |
||||
|
||||
sp_estate->es_range_table = node->rtable; |
||||
sp_estate->es_param_list_info = estate->es_param_list_info; |
||||
sp_estate->es_param_exec_vals = estate->es_param_exec_vals; |
||||
sp_estate->es_tupleTable =
|
||||
ExecCreateTupleTable (ExecCountSlotsNode(node->plan) + 10); |
||||
pfree (sp_estate->es_refcount); |
||||
sp_estate->es_refcount = estate->es_refcount; |
||||
|
||||
if ( !ExecInitNode (node->plan, sp_estate, NULL) ) |
||||
return (false); |
||||
|
||||
node->shutdown = true; |
||||
|
||||
/*
|
||||
* If this plan is un-correlated or undirect correlated one and
|
||||
* want to set params for parent plan then prepare parameters. |
||||
*/ |
||||
if ( node->setParam != NULL ) |
||||
{ |
||||
List *lst; |
||||
|
||||
foreach (lst, node->setParam) |
||||
{ |
||||
ParamExecData *prm = &(estate->es_param_exec_vals[lfirsti(lst)]); |
||||
|
||||
prm->execPlan = node; |
||||
} |
||||
/*
|
||||
* Note that in the case of un-correlated subqueries we don't care |
||||
* about setting parent->chgParam here: indices take care about it, |
||||
* for others - it doesn't matter... |
||||
*/ |
||||
} |
||||
|
||||
return (true); |
||||
} |
||||
|
||||
/* ----------------------------------------------------------------
|
||||
* ExecSetParamPlan |
||||
* |
||||
* Executes plan of node and sets parameters. |
||||
* ---------------------------------------------------------------- |
||||
*/ |
||||
void |
||||
ExecSetParamPlan (SubPlan *node) |
||||
{ |
||||
Plan *plan = node->plan; |
||||
SubLink *sublink = node->sublink; |
||||
TupleTableSlot *slot; |
||||
List *lst; |
||||
bool found = false; |
||||
|
||||
if ( sublink->subLinkType == ANY_SUBLINK || |
||||
sublink->subLinkType == ALL_SUBLINK ) |
||||
elog (ERROR, "ExecSetParamPlan: ANY/ALL subselect unsupported"); |
||||
|
||||
if ( plan->chgParam != NULL ) |
||||
ExecReScan (plan, (ExprContext*) NULL, plan); |
||||
|
||||
for (slot = ExecProcNode (plan, plan);
|
||||
!TupIsNull(slot);
|
||||
slot = ExecProcNode (plan, plan)) |
||||
{ |
||||
HeapTuple tup = slot->val; |
||||
TupleDesc tdesc = slot->ttc_tupleDescriptor; |
||||
int i = 1; |
||||
|
||||
if ( sublink->subLinkType == EXPR_SUBLINK && found ) |
||||
{ |
||||
elog (ERROR, "ExecSetParamPlan: more than one tuple returned by expression subselect"); |
||||
return; |
||||
} |
||||
|
||||
found = true; |
||||
|
||||
if ( sublink->subLinkType == EXISTS_SUBLINK ) |
||||
{ |
||||
ParamExecData *prm = &(plan->state->es_param_exec_vals[lfirsti(node->setParam)]); |
||||
|
||||
prm->execPlan = NULL; |
||||
prm->value = (Datum) true; |
||||
prm->isnull = false; |
||||
break; |
||||
} |
||||
|
||||
foreach (lst, node->setParam) |
||||
{ |
||||
ParamExecData *prm = &(plan->state->es_param_exec_vals[lfirsti(lst)]); |
||||
|
||||
prm->execPlan = NULL; |
||||
prm->value = heap_getattr (tup, i, tdesc, &(prm->isnull)); |
||||
i++; |
||||
} |
||||
} |
||||
|
||||
if ( !found ) |
||||
{ |
||||
if ( sublink->subLinkType == EXISTS_SUBLINK ) |
||||
{ |
||||
ParamExecData *prm = &(plan->state->es_param_exec_vals[lfirsti(node->setParam)]); |
||||
|
||||
prm->execPlan = NULL; |
||||
prm->value = (Datum) false; |
||||
prm->isnull = false; |
||||
} |
||||
else |
||||
{ |
||||
foreach (lst, node->setParam) |
||||
{ |
||||
ParamExecData *prm = &(plan->state->es_param_exec_vals[lfirsti(lst)]); |
||||
|
||||
prm->execPlan = NULL; |
||||
prm->value = (Datum) NULL; |
||||
prm->isnull = true; |
||||
} |
||||
} |
||||
} |
||||
|
||||
if ( plan->extParam == NULL ) /* un-correlated ... */ |
||||
{ |
||||
ExecEndNode (plan, plan); |
||||
node->shutdown = false; |
||||
} |
||||
} |
||||
|
||||
/* ----------------------------------------------------------------
|
||||
* ExecEndSubPlan |
||||
* ---------------------------------------------------------------- |
||||
*/ |
||||
void |
||||
ExecEndSubPlan(SubPlan *node) |
||||
{ |
||||
|
||||
if ( node->shutdown ) |
||||
{ |
||||
ExecEndNode (node->plan, node->plan); |
||||
node->shutdown = false; |
||||
} |
||||
|
||||
} |
||||
|
||||
void
|
||||
ExecReScanSetParamPlan (SubPlan *node, Plan *parent) |
||||
{ |
||||
Plan *plan = node->plan; |
||||
List *lst; |
||||
|
||||
if ( node->parParam != NULL ) |
||||
elog (ERROR, "ExecReScanSetParamPlan: direct correlated subquery unsupported, yet"); |
||||
if ( node->setParam == NULL ) |
||||
elog (ERROR, "ExecReScanSetParamPlan: setParam list is NULL"); |
||||
if ( plan->extParam == NULL ) |
||||
elog (ERROR, "ExecReScanSetParamPlan: extParam list of plan is NULL"); |
||||
|
||||
/*
|
||||
* Don't actual re-scan: ExecSetParamPlan does re-scan if
|
||||
* node->plan->chgParam is not NULL... |
||||
ExecReScan (plan, NULL, plan); |
||||
*/ |
||||
|
||||
foreach (lst, node->setParam) |
||||
{ |
||||
ParamExecData *prm = &(plan->state->es_param_exec_vals[lfirsti(lst)]); |
||||
|
||||
prm->execPlan = node; |
||||
} |
||||
|
||||
parent->chgParam = nconc (parent->chgParam, listCopy(node->setParam)); |
||||
|
||||
} |
||||
Loading…
Reference in new issue