PostgreSQL Source Code  git master
nodeSubplan.c File Reference
#include "postgres.h"
#include <math.h>
#include "access/htup_details.h"
#include "executor/executor.h"
#include "executor/nodeSubplan.h"
#include "miscadmin.h"
#include "nodes/makefuncs.h"
#include "nodes/nodeFuncs.h"
#include "optimizer/optimizer.h"
#include "utils/array.h"
#include "utils/lsyscache.h"
#include "utils/memutils.h"
Include dependency graph for nodeSubplan.c:

Go to the source code of this file.

Functions

static Datum ExecHashSubPlan (SubPlanState *node, ExprContext *econtext, bool *isNull)
 
static Datum ExecScanSubPlan (SubPlanState *node, ExprContext *econtext, bool *isNull)
 
static void buildSubPlanHash (SubPlanState *node, ExprContext *econtext)
 
static bool findPartialMatch (TupleHashTable hashtable, TupleTableSlot *slot, FmgrInfo *eqfunctions)
 
static bool slotAllNulls (TupleTableSlot *slot)
 
static bool slotNoNulls (TupleTableSlot *slot)
 
Datum ExecSubPlan (SubPlanState *node, ExprContext *econtext, bool *isNull)
 
static bool execTuplesUnequal (TupleTableSlot *slot1, TupleTableSlot *slot2, int numCols, AttrNumber *matchColIdx, FmgrInfo *eqfunctions, const Oid *collations, MemoryContext evalContext)
 
SubPlanStateExecInitSubPlan (SubPlan *subplan, PlanState *parent)
 
void ExecSetParamPlan (SubPlanState *node, ExprContext *econtext)
 
void ExecSetParamPlanMulti (const Bitmapset *params, ExprContext *econtext)
 
void ExecReScanSetParamPlan (SubPlanState *node, PlanState *parent)
 

Function Documentation

◆ buildSubPlanHash()

static void buildSubPlanHash ( SubPlanState node,
ExprContext econtext 
)
static

Definition at line 504 of file nodeSubplan.c.

505 {
506  SubPlan *subplan = node->subplan;
507  PlanState *planstate = node->planstate;
508  int ncols = node->numCols;
509  ExprContext *innerecontext = node->innerecontext;
510  MemoryContext oldcontext;
511  long nbuckets;
512  TupleTableSlot *slot;
513 
514  Assert(subplan->subLinkType == ANY_SUBLINK);
515 
516  /*
517  * If we already had any hash tables, reset 'em; otherwise create empty
518  * hash table(s).
519  *
520  * If we need to distinguish accurately between FALSE and UNKNOWN (i.e.,
521  * NULL) results of the IN operation, then we have to store subplan output
522  * rows that are partly or wholly NULL. We store such rows in a separate
523  * hash table that we expect will be much smaller than the main table. (We
524  * can use hashing to eliminate partly-null rows that are not distinct. We
525  * keep them separate to minimize the cost of the inevitable full-table
526  * searches; see findPartialMatch.)
527  *
528  * If it's not necessary to distinguish FALSE and UNKNOWN, then we don't
529  * need to store subplan output rows that contain NULL.
530  */
532  node->havehashrows = false;
533  node->havenullrows = false;
534 
535  nbuckets = clamp_cardinality_to_long(planstate->plan->plan_rows);
536  if (nbuckets < 1)
537  nbuckets = 1;
538 
539  if (node->hashtable)
541  else
543  node->descRight,
544  ncols,
545  node->keyColIdx,
546  node->tab_eq_funcoids,
547  node->tab_hash_funcs,
548  node->tab_collations,
549  nbuckets,
550  0,
551  node->planstate->state->es_query_cxt,
552  node->hashtablecxt,
553  node->hashtempcxt,
554  false);
555 
556  if (!subplan->unknownEqFalse)
557  {
558  if (ncols == 1)
559  nbuckets = 1; /* there can only be one entry */
560  else
561  {
562  nbuckets /= 16;
563  if (nbuckets < 1)
564  nbuckets = 1;
565  }
566 
567  if (node->hashnulls)
569  else
571  node->descRight,
572  ncols,
573  node->keyColIdx,
574  node->tab_eq_funcoids,
575  node->tab_hash_funcs,
576  node->tab_collations,
577  nbuckets,
578  0,
579  node->planstate->state->es_query_cxt,
580  node->hashtablecxt,
581  node->hashtempcxt,
582  false);
583  }
584  else
585  node->hashnulls = NULL;
586 
587  /*
588  * We are probably in a short-lived expression-evaluation context. Switch
589  * to the per-query context for manipulating the child plan.
590  */
591  oldcontext = MemoryContextSwitchTo(econtext->ecxt_per_query_memory);
592 
593  /*
594  * Reset subplan to start.
595  */
596  ExecReScan(planstate);
597 
598  /*
599  * Scan the subplan and load the hash table(s). Note that when there are
600  * duplicate rows coming out of the sub-select, only one copy is stored.
601  */
602  for (slot = ExecProcNode(planstate);
603  !TupIsNull(slot);
604  slot = ExecProcNode(planstate))
605  {
606  int col = 1;
607  ListCell *plst;
608  bool isnew;
609 
610  /*
611  * Load up the Params representing the raw sub-select outputs, then
612  * form the projection tuple to store in the hashtable.
613  */
614  foreach(plst, subplan->paramIds)
615  {
616  int paramid = lfirst_int(plst);
617  ParamExecData *prmdata;
618 
619  prmdata = &(innerecontext->ecxt_param_exec_vals[paramid]);
620  Assert(prmdata->execPlan == NULL);
621  prmdata->value = slot_getattr(slot, col,
622  &(prmdata->isnull));
623  col++;
624  }
625  slot = ExecProject(node->projRight);
626 
627  /*
628  * If result contains any nulls, store separately or not at all.
629  */
630  if (slotNoNulls(slot))
631  {
632  (void) LookupTupleHashEntry(node->hashtable, slot, &isnew, NULL);
633  node->havehashrows = true;
634  }
635  else if (node->hashnulls)
636  {
637  (void) LookupTupleHashEntry(node->hashnulls, slot, &isnew, NULL);
638  node->havenullrows = true;
639  }
640 
641  /*
642  * Reset innerecontext after each inner tuple to free any memory used
643  * during ExecProject.
644  */
645  ResetExprContext(innerecontext);
646  }
647 
648  /*
649  * Since the projected tuples are in the sub-query's context and not the
650  * main context, we'd better clear the tuple slot before there's any
651  * chance of a reset of the sub-query's context. Else we will have the
652  * potential for a double free attempt. (XXX possibly no longer needed,
653  * but can't hurt.)
654  */
656 
657  MemoryContextSwitchTo(oldcontext);
658 }
long clamp_cardinality_to_long(Cardinality x)
Definition: costsize.c:254
void ExecReScan(PlanState *node)
Definition: execAmi.c:76
TupleHashEntry LookupTupleHashEntry(TupleHashTable hashtable, TupleTableSlot *slot, bool *isnew, uint32 *hash)
Definition: execGrouping.c:305
TupleHashTable BuildTupleHashTableExt(PlanState *parent, TupleDesc inputDesc, int numCols, AttrNumber *keyColIdx, const Oid *eqfuncoids, FmgrInfo *hashfunctions, Oid *collations, long nbuckets, Size additionalsize, MemoryContext metacxt, MemoryContext tablecxt, MemoryContext tempcxt, bool use_variable_hash_iv)
Definition: execGrouping.c:153
void ResetTupleHashTable(TupleHashTable hashtable)
Definition: execGrouping.c:284
static TupleTableSlot * ExecProject(ProjectionInfo *projInfo)
Definition: executor.h:376
#define ResetExprContext(econtext)
Definition: executor.h:544
static TupleTableSlot * ExecProcNode(PlanState *node)
Definition: executor.h:269
Assert(fmt[strlen(fmt) - 1] !='\n')
void MemoryContextReset(MemoryContext context)
Definition: mcxt.c:326
static bool slotNoNulls(TupleTableSlot *slot)
Definition: nodeSubplan.c:799
static MemoryContext MemoryContextSwitchTo(MemoryContext context)
Definition: palloc.h:124
#define lfirst_int(lc)
Definition: pg_list.h:173
@ ANY_SUBLINK
Definition: primnodes.h:934
MemoryContext es_query_cxt
Definition: execnodes.h:665
ParamExecData * ecxt_param_exec_vals
Definition: execnodes.h:266
MemoryContext ecxt_per_query_memory
Definition: execnodes.h:262
TupleTableSlot * resultslot
Definition: execnodes.h:97
bool isnull
Definition: params.h:150
Datum value
Definition: params.h:149
void * execPlan
Definition: params.h:148
Plan * plan
Definition: execnodes.h:1043
EState * state
Definition: execnodes.h:1045
Cardinality plan_rows
Definition: plannodes.h:134
ExprState pi_state
Definition: execnodes.h:362
TupleHashTable hashtable
Definition: execnodes.h:969
MemoryContext hashtablecxt
Definition: execnodes.h:973
Oid * tab_eq_funcoids
Definition: execnodes.h:979
ExprContext * innerecontext
Definition: execnodes.h:975
FmgrInfo * tab_hash_funcs
Definition: execnodes.h:982
MemoryContext hashtempcxt
Definition: execnodes.h:974
AttrNumber * keyColIdx
Definition: execnodes.h:978
struct PlanState * planstate
Definition: execnodes.h:959
TupleDesc descRight
Definition: execnodes.h:966
SubPlan * subplan
Definition: execnodes.h:958
ProjectionInfo * projRight
Definition: execnodes.h:968
bool havenullrows
Definition: execnodes.h:972
struct PlanState * parent
Definition: execnodes.h:960
Oid * tab_collations
Definition: execnodes.h:981
TupleHashTable hashnulls
Definition: execnodes.h:970
bool havehashrows
Definition: execnodes.h:971
List * paramIds
Definition: primnodes.h:1003
bool unknownEqFalse
Definition: primnodes.h:1016
SubLinkType subLinkType
Definition: primnodes.h:1000
static TupleTableSlot * ExecClearTuple(TupleTableSlot *slot)
Definition: tuptable.h:433
static Datum slot_getattr(TupleTableSlot *slot, int attnum, bool *isnull)
Definition: tuptable.h:389
#define TupIsNull(slot)
Definition: tuptable.h:300

References ANY_SUBLINK, Assert(), BuildTupleHashTableExt(), clamp_cardinality_to_long(), SubPlanState::descRight, ExprContext::ecxt_param_exec_vals, ExprContext::ecxt_per_query_memory, EState::es_query_cxt, ExecClearTuple(), ParamExecData::execPlan, ExecProcNode(), ExecProject(), ExecReScan(), SubPlanState::hashnulls, SubPlanState::hashtable, SubPlanState::hashtablecxt, SubPlanState::hashtempcxt, SubPlanState::havehashrows, SubPlanState::havenullrows, SubPlanState::innerecontext, ParamExecData::isnull, SubPlanState::keyColIdx, lfirst_int, LookupTupleHashEntry(), MemoryContextReset(), MemoryContextSwitchTo(), SubPlanState::numCols, SubPlan::paramIds, SubPlanState::parent, ProjectionInfo::pi_state, PlanState::plan, Plan::plan_rows, SubPlanState::planstate, SubPlanState::projRight, ResetExprContext, ResetTupleHashTable(), ExprState::resultslot, slot_getattr(), slotNoNulls(), PlanState::state, SubPlan::subLinkType, SubPlanState::subplan, SubPlanState::tab_collations, SubPlanState::tab_eq_funcoids, SubPlanState::tab_hash_funcs, TupIsNull, SubPlan::unknownEqFalse, and ParamExecData::value.

Referenced by ExecHashSubPlan().

◆ ExecHashSubPlan()

static Datum ExecHashSubPlan ( SubPlanState node,
ExprContext econtext,
bool isNull 
)
static

Definition at line 101 of file nodeSubplan.c.

104 {
105  SubPlan *subplan = node->subplan;
106  PlanState *planstate = node->planstate;
107  TupleTableSlot *slot;
108 
109  /* Shouldn't have any direct correlation Vars */
110  if (subplan->parParam != NIL || node->args != NIL)
111  elog(ERROR, "hashed subplan with direct correlation not supported");
112 
113  /*
114  * If first time through or we need to rescan the subplan, build the hash
115  * table.
116  */
117  if (node->hashtable == NULL || planstate->chgParam != NULL)
118  buildSubPlanHash(node, econtext);
119 
120  /*
121  * The result for an empty subplan is always FALSE; no need to evaluate
122  * lefthand side.
123  */
124  *isNull = false;
125  if (!node->havehashrows && !node->havenullrows)
126  return BoolGetDatum(false);
127 
128  /*
129  * Evaluate lefthand expressions and form a projection tuple. First we
130  * have to set the econtext to use (hack alert!).
131  */
132  node->projLeft->pi_exprContext = econtext;
133  slot = ExecProject(node->projLeft);
134 
135  /*
136  * Note: because we are typically called in a per-tuple context, we have
137  * to explicitly clear the projected tuple before returning. Otherwise,
138  * we'll have a double-free situation: the per-tuple context will probably
139  * be reset before we're called again, and then the tuple slot will think
140  * it still needs to free the tuple.
141  */
142 
143  /*
144  * If the LHS is all non-null, probe for an exact match in the main hash
145  * table. If we find one, the result is TRUE. Otherwise, scan the
146  * partly-null table to see if there are any rows that aren't provably
147  * unequal to the LHS; if so, the result is UNKNOWN. (We skip that part
148  * if we don't care about UNKNOWN.) Otherwise, the result is FALSE.
149  *
150  * Note: the reason we can avoid a full scan of the main hash table is
151  * that the combining operators are assumed never to yield NULL when both
152  * inputs are non-null. If they were to do so, we might need to produce
153  * UNKNOWN instead of FALSE because of an UNKNOWN result in comparing the
154  * LHS to some main-table entry --- which is a comparison we will not even
155  * make, unless there's a chance match of hash keys.
156  */
157  if (slotNoNulls(slot))
158  {
159  if (node->havehashrows &&
161  slot,
162  node->cur_eq_comp,
163  node->lhs_hash_funcs) != NULL)
164  {
165  ExecClearTuple(slot);
166  return BoolGetDatum(true);
167  }
168  if (node->havenullrows &&
169  findPartialMatch(node->hashnulls, slot, node->cur_eq_funcs))
170  {
171  ExecClearTuple(slot);
172  *isNull = true;
173  return BoolGetDatum(false);
174  }
175  ExecClearTuple(slot);
176  return BoolGetDatum(false);
177  }
178 
179  /*
180  * When the LHS is partly or wholly NULL, we can never return TRUE. If we
181  * don't care about UNKNOWN, just return FALSE. Otherwise, if the LHS is
182  * wholly NULL, immediately return UNKNOWN. (Since the combining
183  * operators are strict, the result could only be FALSE if the sub-select
184  * were empty, but we already handled that case.) Otherwise, we must scan
185  * both the main and partly-null tables to see if there are any rows that
186  * aren't provably unequal to the LHS; if so, the result is UNKNOWN.
187  * Otherwise, the result is FALSE.
188  */
189  if (node->hashnulls == NULL)
190  {
191  ExecClearTuple(slot);
192  return BoolGetDatum(false);
193  }
194  if (slotAllNulls(slot))
195  {
196  ExecClearTuple(slot);
197  *isNull = true;
198  return BoolGetDatum(false);
199  }
200  /* Scan partly-null table first, since more likely to get a match */
201  if (node->havenullrows &&
202  findPartialMatch(node->hashnulls, slot, node->cur_eq_funcs))
203  {
204  ExecClearTuple(slot);
205  *isNull = true;
206  return BoolGetDatum(false);
207  }
208  if (node->havehashrows &&
209  findPartialMatch(node->hashtable, slot, node->cur_eq_funcs))
210  {
211  ExecClearTuple(slot);
212  *isNull = true;
213  return BoolGetDatum(false);
214  }
215  ExecClearTuple(slot);
216  return BoolGetDatum(false);
217 }
#define ERROR
Definition: elog.h:39
#define elog(elevel,...)
Definition: elog.h:224
TupleHashEntry FindTupleHashEntry(TupleHashTable hashtable, TupleTableSlot *slot, ExprState *eqcomp, FmgrInfo *hashfunctions)
Definition: execGrouping.c:392
static void buildSubPlanHash(SubPlanState *node, ExprContext *econtext)
Definition: nodeSubplan.c:504
static bool slotAllNulls(TupleTableSlot *slot)
Definition: nodeSubplan.c:779
static bool findPartialMatch(TupleHashTable hashtable, TupleTableSlot *slot, FmgrInfo *eqfunctions)
Definition: nodeSubplan.c:744
#define NIL
Definition: pg_list.h:68
static Datum BoolGetDatum(bool X)
Definition: postgres.h:102
Bitmapset * chgParam
Definition: execnodes.h:1075
ExprContext * pi_exprContext
Definition: execnodes.h:364
ExprState * cur_eq_comp
Definition: execnodes.h:986
FmgrInfo * cur_eq_funcs
Definition: execnodes.h:985
List * args
Definition: execnodes.h:962
FmgrInfo * lhs_hash_funcs
Definition: execnodes.h:984
ProjectionInfo * projLeft
Definition: execnodes.h:967
List * parParam
Definition: primnodes.h:1025

References SubPlanState::args, BoolGetDatum(), buildSubPlanHash(), PlanState::chgParam, SubPlanState::cur_eq_comp, SubPlanState::cur_eq_funcs, elog, ERROR, ExecClearTuple(), ExecProject(), findPartialMatch(), FindTupleHashEntry(), SubPlanState::hashnulls, SubPlanState::hashtable, SubPlanState::havehashrows, SubPlanState::havenullrows, SubPlanState::lhs_hash_funcs, NIL, SubPlan::parParam, ProjectionInfo::pi_exprContext, SubPlanState::planstate, SubPlanState::projLeft, slotAllNulls(), slotNoNulls(), and SubPlanState::subplan.

Referenced by ExecSubPlan().

◆ ExecInitSubPlan()

SubPlanState* ExecInitSubPlan ( SubPlan subplan,
PlanState parent 
)

Definition at line 823 of file nodeSubplan.c.

824 {
826  EState *estate = parent->state;
827 
828  sstate->subplan = subplan;
829 
830  /* Link the SubPlanState to already-initialized subplan */
831  sstate->planstate = (PlanState *) list_nth(estate->es_subplanstates,
832  subplan->plan_id - 1);
833 
834  /*
835  * This check can fail if the planner mistakenly puts a parallel-unsafe
836  * subplan into a parallelized subquery; see ExecSerializePlan.
837  */
838  if (sstate->planstate == NULL)
839  elog(ERROR, "subplan \"%s\" was not initialized",
840  subplan->plan_name);
841 
842  /* Link to parent's state, too */
843  sstate->parent = parent;
844 
845  /* Initialize subexpressions */
846  sstate->testexpr = ExecInitExpr((Expr *) subplan->testexpr, parent);
847  sstate->args = ExecInitExprList(subplan->args, parent);
848 
849  /*
850  * initialize my state
851  */
852  sstate->curTuple = NULL;
853  sstate->curArray = PointerGetDatum(NULL);
854  sstate->projLeft = NULL;
855  sstate->projRight = NULL;
856  sstate->hashtable = NULL;
857  sstate->hashnulls = NULL;
858  sstate->hashtablecxt = NULL;
859  sstate->hashtempcxt = NULL;
860  sstate->innerecontext = NULL;
861  sstate->keyColIdx = NULL;
862  sstate->tab_eq_funcoids = NULL;
863  sstate->tab_hash_funcs = NULL;
864  sstate->tab_eq_funcs = NULL;
865  sstate->tab_collations = NULL;
866  sstate->lhs_hash_funcs = NULL;
867  sstate->cur_eq_funcs = NULL;
868 
869  /*
870  * If this is an initplan, it has output parameters that the parent plan
871  * will use, so mark those parameters as needing evaluation. We don't
872  * actually run the subplan until we first need one of its outputs.
873  *
874  * A CTE subplan's output parameter is never to be evaluated in the normal
875  * way, so skip this in that case.
876  *
877  * Note that we don't set parent->chgParam here: the parent plan hasn't
878  * been run yet, so no need to force it to re-run.
879  */
880  if (subplan->setParam != NIL && subplan->parParam == NIL &&
881  subplan->subLinkType != CTE_SUBLINK)
882  {
883  ListCell *lst;
884 
885  foreach(lst, subplan->setParam)
886  {
887  int paramid = lfirst_int(lst);
888  ParamExecData *prm = &(estate->es_param_exec_vals[paramid]);
889 
890  prm->execPlan = sstate;
891  }
892  }
893 
894  /*
895  * If we are going to hash the subquery output, initialize relevant stuff.
896  * (We don't create the hashtable until needed, though.)
897  */
898  if (subplan->useHashTable)
899  {
900  int ncols,
901  i;
902  TupleDesc tupDescLeft;
903  TupleDesc tupDescRight;
904  Oid *cross_eq_funcoids;
905  TupleTableSlot *slot;
906  List *oplist,
907  *lefttlist,
908  *righttlist;
909  ListCell *l;
910 
911  /* We need a memory context to hold the hash table(s) */
912  sstate->hashtablecxt =
914  "Subplan HashTable Context",
916  /* and a small one for the hash tables to use as temp storage */
917  sstate->hashtempcxt =
919  "Subplan HashTable Temp Context",
921  /* and a short-lived exprcontext for function evaluation */
922  sstate->innerecontext = CreateExprContext(estate);
923 
924  /*
925  * We use ExecProject to evaluate the lefthand and righthand
926  * expression lists and form tuples. (You might think that we could
927  * use the sub-select's output tuples directly, but that is not the
928  * case if we had to insert any run-time coercions of the sub-select's
929  * output datatypes; anyway this avoids storing any resjunk columns
930  * that might be in the sub-select's output.) Run through the
931  * combining expressions to build tlists for the lefthand and
932  * righthand sides.
933  *
934  * We also extract the combining operators themselves to initialize
935  * the equality and hashing functions for the hash tables.
936  */
937  if (IsA(subplan->testexpr, OpExpr))
938  {
939  /* single combining operator */
940  oplist = list_make1(subplan->testexpr);
941  }
942  else if (is_andclause(subplan->testexpr))
943  {
944  /* multiple combining operators */
945  oplist = castNode(BoolExpr, subplan->testexpr)->args;
946  }
947  else
948  {
949  /* shouldn't see anything else in a hashable subplan */
950  elog(ERROR, "unrecognized testexpr type: %d",
951  (int) nodeTag(subplan->testexpr));
952  oplist = NIL; /* keep compiler quiet */
953  }
954  ncols = list_length(oplist);
955 
956  lefttlist = righttlist = NIL;
957  sstate->numCols = ncols;
958  sstate->keyColIdx = (AttrNumber *) palloc(ncols * sizeof(AttrNumber));
959  sstate->tab_eq_funcoids = (Oid *) palloc(ncols * sizeof(Oid));
960  sstate->tab_collations = (Oid *) palloc(ncols * sizeof(Oid));
961  sstate->tab_hash_funcs = (FmgrInfo *) palloc(ncols * sizeof(FmgrInfo));
962  sstate->tab_eq_funcs = (FmgrInfo *) palloc(ncols * sizeof(FmgrInfo));
963  sstate->lhs_hash_funcs = (FmgrInfo *) palloc(ncols * sizeof(FmgrInfo));
964  sstate->cur_eq_funcs = (FmgrInfo *) palloc(ncols * sizeof(FmgrInfo));
965  /* we'll need the cross-type equality fns below, but not in sstate */
966  cross_eq_funcoids = (Oid *) palloc(ncols * sizeof(Oid));
967 
968  i = 1;
969  foreach(l, oplist)
970  {
971  OpExpr *opexpr = lfirst_node(OpExpr, l);
972  Expr *expr;
973  TargetEntry *tle;
974  Oid rhs_eq_oper;
975  Oid left_hashfn;
976  Oid right_hashfn;
977 
978  Assert(list_length(opexpr->args) == 2);
979 
980  /* Process lefthand argument */
981  expr = (Expr *) linitial(opexpr->args);
982  tle = makeTargetEntry(expr,
983  i,
984  NULL,
985  false);
986  lefttlist = lappend(lefttlist, tle);
987 
988  /* Process righthand argument */
989  expr = (Expr *) lsecond(opexpr->args);
990  tle = makeTargetEntry(expr,
991  i,
992  NULL,
993  false);
994  righttlist = lappend(righttlist, tle);
995 
996  /* Lookup the equality function (potentially cross-type) */
997  cross_eq_funcoids[i - 1] = opexpr->opfuncid;
998  fmgr_info(opexpr->opfuncid, &sstate->cur_eq_funcs[i - 1]);
999  fmgr_info_set_expr((Node *) opexpr, &sstate->cur_eq_funcs[i - 1]);
1000 
1001  /* Look up the equality function for the RHS type */
1002  if (!get_compatible_hash_operators(opexpr->opno,
1003  NULL, &rhs_eq_oper))
1004  elog(ERROR, "could not find compatible hash operator for operator %u",
1005  opexpr->opno);
1006  sstate->tab_eq_funcoids[i - 1] = get_opcode(rhs_eq_oper);
1007  fmgr_info(sstate->tab_eq_funcoids[i - 1],
1008  &sstate->tab_eq_funcs[i - 1]);
1009 
1010  /* Lookup the associated hash functions */
1011  if (!get_op_hash_functions(opexpr->opno,
1012  &left_hashfn, &right_hashfn))
1013  elog(ERROR, "could not find hash function for hash operator %u",
1014  opexpr->opno);
1015  fmgr_info(left_hashfn, &sstate->lhs_hash_funcs[i - 1]);
1016  fmgr_info(right_hashfn, &sstate->tab_hash_funcs[i - 1]);
1017 
1018  /* Set collation */
1019  sstate->tab_collations[i - 1] = opexpr->inputcollid;
1020 
1021  /* keyColIdx is just column numbers 1..n */
1022  sstate->keyColIdx[i - 1] = i;
1023 
1024  i++;
1025  }
1026 
1027  /*
1028  * Construct tupdescs, slots and projection nodes for left and right
1029  * sides. The lefthand expressions will be evaluated in the parent
1030  * plan node's exprcontext, which we don't have access to here.
1031  * Fortunately we can just pass NULL for now and fill it in later
1032  * (hack alert!). The righthand expressions will be evaluated in our
1033  * own innerecontext.
1034  */
1035  tupDescLeft = ExecTypeFromTL(lefttlist);
1036  slot = ExecInitExtraTupleSlot(estate, tupDescLeft, &TTSOpsVirtual);
1037  sstate->projLeft = ExecBuildProjectionInfo(lefttlist,
1038  NULL,
1039  slot,
1040  parent,
1041  NULL);
1042 
1043  sstate->descRight = tupDescRight = ExecTypeFromTL(righttlist);
1044  slot = ExecInitExtraTupleSlot(estate, tupDescRight, &TTSOpsVirtual);
1045  sstate->projRight = ExecBuildProjectionInfo(righttlist,
1046  sstate->innerecontext,
1047  slot,
1048  sstate->planstate,
1049  NULL);
1050 
1051  /*
1052  * Create comparator for lookups of rows in the table (potentially
1053  * cross-type comparisons).
1054  */
1055  sstate->cur_eq_comp = ExecBuildGroupingEqual(tupDescLeft, tupDescRight,
1057  ncols,
1058  sstate->keyColIdx,
1059  cross_eq_funcoids,
1060  sstate->tab_collations,
1061  parent);
1062  }
1063 
1064  return sstate;
1065 }
int16 AttrNumber
Definition: attnum.h:21
List * ExecInitExprList(List *nodes, PlanState *parent)
Definition: execExpr.c:319
ExprState * ExecInitExpr(Expr *node, PlanState *parent)
Definition: execExpr.c:127
ProjectionInfo * ExecBuildProjectionInfo(List *targetList, ExprContext *econtext, TupleTableSlot *slot, PlanState *parent, TupleDesc inputDesc)
Definition: execExpr.c:354
ExprState * ExecBuildGroupingEqual(TupleDesc ldesc, TupleDesc rdesc, const TupleTableSlotOps *lops, const TupleTableSlotOps *rops, int numCols, const AttrNumber *keyColIdx, const Oid *eqfunctions, const Oid *collations, PlanState *parent)
Definition: execExpr.c:3909
const TupleTableSlotOps TTSOpsVirtual
Definition: execTuples.c:83
TupleTableSlot * ExecInitExtraTupleSlot(EState *estate, TupleDesc tupledesc, const TupleTableSlotOps *tts_ops)
Definition: execTuples.c:1830
const TupleTableSlotOps TTSOpsMinimalTuple
Definition: execTuples.c:85
TupleDesc ExecTypeFromTL(List *targetList)
Definition: execTuples.c:1937
ExprContext * CreateExprContext(EState *estate)
Definition: execUtils.c:304
void fmgr_info(Oid functionId, FmgrInfo *finfo)
Definition: fmgr.c:127
#define fmgr_info_set_expr(expr, finfo)
Definition: fmgr.h:135
int i
Definition: isn.c:73
List * lappend(List *list, void *datum)
Definition: list.c:339
bool get_compatible_hash_operators(Oid opno, Oid *lhs_opno, Oid *rhs_opno)
Definition: lsyscache.c:410
RegProcedure get_opcode(Oid opno)
Definition: lsyscache.c:1263
bool get_op_hash_functions(Oid opno, RegProcedure *lhs_procno, RegProcedure *rhs_procno)
Definition: lsyscache.c:510
TargetEntry * makeTargetEntry(Expr *expr, AttrNumber resno, char *resname, bool resjunk)
Definition: makefuncs.c:240
MemoryContext CurrentMemoryContext
Definition: mcxt.c:131
void * palloc(Size size)
Definition: mcxt.c:1197
#define AllocSetContextCreate
Definition: memutils.h:128
#define ALLOCSET_DEFAULT_SIZES
Definition: memutils.h:152
#define ALLOCSET_SMALL_SIZES
Definition: memutils.h:162
static bool is_andclause(const void *clause)
Definition: nodeFuncs.h:105
#define IsA(nodeptr, _type_)
Definition: nodes.h:158
#define nodeTag(nodeptr)
Definition: nodes.h:133
#define makeNode(_type_)
Definition: nodes.h:155
#define castNode(_type_, nodeptr)
Definition: nodes.h:176
#define lfirst_node(type, lc)
Definition: pg_list.h:176
static int list_length(const List *l)
Definition: pg_list.h:152
#define list_make1(x1)
Definition: pg_list.h:212
#define linitial(l)
Definition: pg_list.h:178
#define lsecond(l)
Definition: pg_list.h:183
static void * list_nth(const List *list, int n)
Definition: pg_list.h:299
static Datum PointerGetDatum(const void *X)
Definition: postgres.h:322
unsigned int Oid
Definition: postgres_ext.h:31
@ CTE_SUBLINK
Definition: primnodes.h:939
ParamExecData * es_param_exec_vals
Definition: execnodes.h:660
List * es_subplanstates
Definition: execnodes.h:680
Definition: fmgr.h:57
Definition: pg_list.h:54
Definition: nodes.h:129
Oid opno
Definition: primnodes.h:753
List * args
Definition: primnodes.h:771
FmgrInfo * tab_eq_funcs
Definition: execnodes.h:983
HeapTuple curTuple
Definition: execnodes.h:963
ExprState * testexpr
Definition: execnodes.h:961
Datum curArray
Definition: execnodes.h:964
int plan_id
Definition: primnodes.h:1005
char * plan_name
Definition: primnodes.h:1007
List * args
Definition: primnodes.h:1026
bool useHashTable
Definition: primnodes.h:1014
Node * testexpr
Definition: primnodes.h:1002
List * setParam
Definition: primnodes.h:1023

References ALLOCSET_DEFAULT_SIZES, ALLOCSET_SMALL_SIZES, AllocSetContextCreate, SubPlanState::args, OpExpr::args, SubPlan::args, Assert(), castNode, CreateExprContext(), CTE_SUBLINK, SubPlanState::cur_eq_comp, SubPlanState::cur_eq_funcs, SubPlanState::curArray, CurrentMemoryContext, SubPlanState::curTuple, SubPlanState::descRight, elog, ERROR, EState::es_param_exec_vals, EState::es_subplanstates, ExecBuildGroupingEqual(), ExecBuildProjectionInfo(), ExecInitExpr(), ExecInitExprList(), ExecInitExtraTupleSlot(), ParamExecData::execPlan, ExecTypeFromTL(), fmgr_info(), fmgr_info_set_expr, get_compatible_hash_operators(), get_op_hash_functions(), get_opcode(), SubPlanState::hashnulls, SubPlanState::hashtable, SubPlanState::hashtablecxt, SubPlanState::hashtempcxt, i, SubPlanState::innerecontext, is_andclause(), IsA, SubPlanState::keyColIdx, lappend(), lfirst_int, lfirst_node, SubPlanState::lhs_hash_funcs, linitial, list_length(), list_make1, list_nth(), lsecond, makeNode, makeTargetEntry(), NIL, nodeTag, SubPlanState::numCols, OpExpr::opno, palloc(), SubPlanState::parent, SubPlan::parParam, SubPlan::plan_id, SubPlan::plan_name, SubPlanState::planstate, PointerGetDatum(), SubPlanState::projLeft, SubPlanState::projRight, SubPlan::setParam, PlanState::state, SubPlan::subLinkType, SubPlanState::subplan, SubPlanState::tab_collations, SubPlanState::tab_eq_funcoids, SubPlanState::tab_eq_funcs, SubPlanState::tab_hash_funcs, SubPlanState::testexpr, SubPlan::testexpr, TTSOpsMinimalTuple, TTSOpsVirtual, and SubPlan::useHashTable.

Referenced by ExecInitExprRec(), ExecInitNode(), and ExecPushExprSetupSteps().

◆ ExecReScanSetParamPlan()

void ExecReScanSetParamPlan ( SubPlanState node,
PlanState parent 
)

Definition at line 1291 of file nodeSubplan.c.

1292 {
1293  PlanState *planstate = node->planstate;
1294  SubPlan *subplan = node->subplan;
1295  EState *estate = parent->state;
1296  ListCell *l;
1297 
1298  /* sanity checks */
1299  if (subplan->parParam != NIL)
1300  elog(ERROR, "direct correlated subquery unsupported as initplan");
1301  if (subplan->setParam == NIL)
1302  elog(ERROR, "setParam list of initplan is empty");
1303  if (bms_is_empty(planstate->plan->extParam))
1304  elog(ERROR, "extParam set of initplan is empty");
1305 
1306  /*
1307  * Don't actually re-scan: it'll happen inside ExecSetParamPlan if needed.
1308  */
1309 
1310  /*
1311  * Mark this subplan's output parameters as needing recalculation.
1312  *
1313  * CTE subplans are never executed via parameter recalculation; instead
1314  * they get run when called by nodeCtescan.c. So don't mark the output
1315  * parameter of a CTE subplan as dirty, but do set the chgParam bit for it
1316  * so that dependent plan nodes will get told to rescan.
1317  */
1318  foreach(l, subplan->setParam)
1319  {
1320  int paramid = lfirst_int(l);
1321  ParamExecData *prm = &(estate->es_param_exec_vals[paramid]);
1322 
1323  if (subplan->subLinkType != CTE_SUBLINK)
1324  prm->execPlan = node;
1325 
1326  parent->chgParam = bms_add_member(parent->chgParam, paramid);
1327  }
1328 }
Bitmapset * bms_add_member(Bitmapset *a, int x)
Definition: bitmapset.c:828
#define bms_is_empty(a)
Definition: bitmapset.h:105
Bitmapset * extParam
Definition: plannodes.h:170

References bms_add_member(), bms_is_empty, PlanState::chgParam, CTE_SUBLINK, elog, ERROR, EState::es_param_exec_vals, ParamExecData::execPlan, Plan::extParam, lfirst_int, NIL, SubPlan::parParam, PlanState::plan, SubPlanState::planstate, SubPlan::setParam, PlanState::state, SubPlan::subLinkType, and SubPlanState::subplan.

Referenced by ExecReScan().

◆ ExecScanSubPlan()

static Datum ExecScanSubPlan ( SubPlanState node,
ExprContext econtext,
bool isNull 
)
static

Definition at line 223 of file nodeSubplan.c.

226 {
227  SubPlan *subplan = node->subplan;
228  PlanState *planstate = node->planstate;
229  SubLinkType subLinkType = subplan->subLinkType;
230  MemoryContext oldcontext;
231  TupleTableSlot *slot;
232  Datum result;
233  bool found = false; /* true if got at least one subplan tuple */
234  ListCell *pvar;
235  ListCell *l;
236  ArrayBuildStateAny *astate = NULL;
237 
238  /* Initialize ArrayBuildStateAny in caller's context, if needed */
239  if (subLinkType == ARRAY_SUBLINK)
240  astate = initArrayResultAny(subplan->firstColType,
241  CurrentMemoryContext, true);
242 
243  /*
244  * We are probably in a short-lived expression-evaluation context. Switch
245  * to the per-query context for manipulating the child plan's chgParam,
246  * calling ExecProcNode on it, etc.
247  */
248  oldcontext = MemoryContextSwitchTo(econtext->ecxt_per_query_memory);
249 
250  /*
251  * Set Params of this plan from parent plan correlation values. (Any
252  * calculation we have to do is done in the parent econtext, since the
253  * Param values don't need to have per-query lifetime.)
254  */
255  Assert(list_length(subplan->parParam) == list_length(node->args));
256 
257  forboth(l, subplan->parParam, pvar, node->args)
258  {
259  int paramid = lfirst_int(l);
260  ParamExecData *prm = &(econtext->ecxt_param_exec_vals[paramid]);
261 
263  econtext,
264  &(prm->isnull));
265  planstate->chgParam = bms_add_member(planstate->chgParam, paramid);
266  }
267 
268  /*
269  * Now that we've set up its parameters, we can reset the subplan.
270  */
271  ExecReScan(planstate);
272 
273  /*
274  * For all sublink types except EXPR_SUBLINK and ARRAY_SUBLINK, the result
275  * is boolean as are the results of the combining operators. We combine
276  * results across tuples (if the subplan produces more than one) using OR
277  * semantics for ANY_SUBLINK or AND semantics for ALL_SUBLINK.
278  * (ROWCOMPARE_SUBLINK doesn't allow multiple tuples from the subplan.)
279  * NULL results from the combining operators are handled according to the
280  * usual SQL semantics for OR and AND. The result for no input tuples is
281  * FALSE for ANY_SUBLINK, TRUE for ALL_SUBLINK, NULL for
282  * ROWCOMPARE_SUBLINK.
283  *
284  * For EXPR_SUBLINK we require the subplan to produce no more than one
285  * tuple, else an error is raised. If zero tuples are produced, we return
286  * NULL. Assuming we get a tuple, we just use its first column (there can
287  * be only one non-junk column in this case).
288  *
289  * For MULTIEXPR_SUBLINK, we push the per-column subplan outputs out to
290  * the setParams and then return a dummy false value. There must not be
291  * multiple tuples returned from the subplan; if zero tuples are produced,
292  * set the setParams to NULL.
293  *
294  * For ARRAY_SUBLINK we allow the subplan to produce any number of tuples,
295  * and form an array of the first column's values. Note in particular
296  * that we produce a zero-element array if no tuples are produced (this is
297  * a change from pre-8.3 behavior of returning NULL).
298  */
299  result = BoolGetDatum(subLinkType == ALL_SUBLINK);
300  *isNull = false;
301 
302  for (slot = ExecProcNode(planstate);
303  !TupIsNull(slot);
304  slot = ExecProcNode(planstate))
305  {
306  TupleDesc tdesc = slot->tts_tupleDescriptor;
307  Datum rowresult;
308  bool rownull;
309  int col;
310  ListCell *plst;
311 
312  if (subLinkType == EXISTS_SUBLINK)
313  {
314  found = true;
315  result = BoolGetDatum(true);
316  break;
317  }
318 
319  if (subLinkType == EXPR_SUBLINK)
320  {
321  /* cannot allow multiple input tuples for EXPR sublink */
322  if (found)
323  ereport(ERROR,
324  (errcode(ERRCODE_CARDINALITY_VIOLATION),
325  errmsg("more than one row returned by a subquery used as an expression")));
326  found = true;
327 
328  /*
329  * We need to copy the subplan's tuple in case the result is of
330  * pass-by-ref type --- our return value will point into this
331  * copied tuple! Can't use the subplan's instance of the tuple
332  * since it won't still be valid after next ExecProcNode() call.
333  * node->curTuple keeps track of the copied tuple for eventual
334  * freeing.
335  */
336  if (node->curTuple)
337  heap_freetuple(node->curTuple);
338  node->curTuple = ExecCopySlotHeapTuple(slot);
339 
340  result = heap_getattr(node->curTuple, 1, tdesc, isNull);
341  /* keep scanning subplan to make sure there's only one tuple */
342  continue;
343  }
344 
345  if (subLinkType == MULTIEXPR_SUBLINK)
346  {
347  /* cannot allow multiple input tuples for MULTIEXPR sublink */
348  if (found)
349  ereport(ERROR,
350  (errcode(ERRCODE_CARDINALITY_VIOLATION),
351  errmsg("more than one row returned by a subquery used as an expression")));
352  found = true;
353 
354  /*
355  * We need to copy the subplan's tuple in case any result is of
356  * pass-by-ref type --- our output values will point into this
357  * copied tuple! Can't use the subplan's instance of the tuple
358  * since it won't still be valid after next ExecProcNode() call.
359  * node->curTuple keeps track of the copied tuple for eventual
360  * freeing.
361  */
362  if (node->curTuple)
363  heap_freetuple(node->curTuple);
364  node->curTuple = ExecCopySlotHeapTuple(slot);
365 
366  /*
367  * Now set all the setParam params from the columns of the tuple
368  */
369  col = 1;
370  foreach(plst, subplan->setParam)
371  {
372  int paramid = lfirst_int(plst);
373  ParamExecData *prmdata;
374 
375  prmdata = &(econtext->ecxt_param_exec_vals[paramid]);
376  Assert(prmdata->execPlan == NULL);
377  prmdata->value = heap_getattr(node->curTuple, col, tdesc,
378  &(prmdata->isnull));
379  col++;
380  }
381 
382  /* keep scanning subplan to make sure there's only one tuple */
383  continue;
384  }
385 
386  if (subLinkType == ARRAY_SUBLINK)
387  {
388  Datum dvalue;
389  bool disnull;
390 
391  found = true;
392  /* stash away current value */
393  Assert(subplan->firstColType == TupleDescAttr(tdesc, 0)->atttypid);
394  dvalue = slot_getattr(slot, 1, &disnull);
395  astate = accumArrayResultAny(astate, dvalue, disnull,
396  subplan->firstColType, oldcontext);
397  /* keep scanning subplan to collect all values */
398  continue;
399  }
400 
401  /* cannot allow multiple input tuples for ROWCOMPARE sublink either */
402  if (subLinkType == ROWCOMPARE_SUBLINK && found)
403  ereport(ERROR,
404  (errcode(ERRCODE_CARDINALITY_VIOLATION),
405  errmsg("more than one row returned by a subquery used as an expression")));
406 
407  found = true;
408 
409  /*
410  * For ALL, ANY, and ROWCOMPARE sublinks, load up the Params
411  * representing the columns of the sub-select, and then evaluate the
412  * combining expression.
413  */
414  col = 1;
415  foreach(plst, subplan->paramIds)
416  {
417  int paramid = lfirst_int(plst);
418  ParamExecData *prmdata;
419 
420  prmdata = &(econtext->ecxt_param_exec_vals[paramid]);
421  Assert(prmdata->execPlan == NULL);
422  prmdata->value = slot_getattr(slot, col, &(prmdata->isnull));
423  col++;
424  }
425 
426  rowresult = ExecEvalExprSwitchContext(node->testexpr, econtext,
427  &rownull);
428 
429  if (subLinkType == ANY_SUBLINK)
430  {
431  /* combine across rows per OR semantics */
432  if (rownull)
433  *isNull = true;
434  else if (DatumGetBool(rowresult))
435  {
436  result = BoolGetDatum(true);
437  *isNull = false;
438  break; /* needn't look at any more rows */
439  }
440  }
441  else if (subLinkType == ALL_SUBLINK)
442  {
443  /* combine across rows per AND semantics */
444  if (rownull)
445  *isNull = true;
446  else if (!DatumGetBool(rowresult))
447  {
448  result = BoolGetDatum(false);
449  *isNull = false;
450  break; /* needn't look at any more rows */
451  }
452  }
453  else
454  {
455  /* must be ROWCOMPARE_SUBLINK */
456  result = rowresult;
457  *isNull = rownull;
458  }
459  }
460 
461  MemoryContextSwitchTo(oldcontext);
462 
463  if (subLinkType == ARRAY_SUBLINK)
464  {
465  /* We return the result in the caller's context */
466  result = makeArrayResultAny(astate, oldcontext, true);
467  }
468  else if (!found)
469  {
470  /*
471  * deal with empty subplan result. result/isNull were previously
472  * initialized correctly for all sublink types except EXPR and
473  * ROWCOMPARE; for those, return NULL.
474  */
475  if (subLinkType == EXPR_SUBLINK ||
476  subLinkType == ROWCOMPARE_SUBLINK)
477  {
478  result = (Datum) 0;
479  *isNull = true;
480  }
481  else if (subLinkType == MULTIEXPR_SUBLINK)
482  {
483  /* We don't care about function result, but set the setParams */
484  foreach(l, subplan->setParam)
485  {
486  int paramid = lfirst_int(l);
487  ParamExecData *prmdata;
488 
489  prmdata = &(econtext->ecxt_param_exec_vals[paramid]);
490  Assert(prmdata->execPlan == NULL);
491  prmdata->value = (Datum) 0;
492  prmdata->isnull = true;
493  }
494  }
495  }
496 
497  return result;
498 }
ArrayBuildStateAny * initArrayResultAny(Oid input_type, MemoryContext rcontext, bool subcontext)
Definition: arrayfuncs.c:5763
ArrayBuildStateAny * accumArrayResultAny(ArrayBuildStateAny *astate, Datum dvalue, bool disnull, Oid input_type, MemoryContext rcontext)
Definition: arrayfuncs.c:5808
Datum makeArrayResultAny(ArrayBuildStateAny *astate, MemoryContext rcontext, bool release)
Definition: arrayfuncs.c:5836
int errcode(int sqlerrcode)
Definition: elog.c:859
int errmsg(const char *fmt,...)
Definition: elog.c:1072
#define ereport(elevel,...)
Definition: elog.h:149
static Datum ExecEvalExprSwitchContext(ExprState *state, ExprContext *econtext, bool *isNull)
Definition: executor.h:348
void heap_freetuple(HeapTuple htup)
Definition: heaptuple.c:1434
static Datum heap_getattr(HeapTuple tup, int attnum, TupleDesc tupleDesc, bool *isnull)
Definition: htup_details.h:792
#define lfirst(lc)
Definition: pg_list.h:172
#define forboth(cell1, list1, cell2, list2)
Definition: pg_list.h:518
static bool DatumGetBool(Datum X)
Definition: postgres.h:90
uintptr_t Datum
Definition: postgres.h:64
SubLinkType
Definition: primnodes.h:931
@ ARRAY_SUBLINK
Definition: primnodes.h:938
@ MULTIEXPR_SUBLINK
Definition: primnodes.h:937
@ EXPR_SUBLINK
Definition: primnodes.h:936
@ ROWCOMPARE_SUBLINK
Definition: primnodes.h:935
@ ALL_SUBLINK
Definition: primnodes.h:933
@ EXISTS_SUBLINK
Definition: primnodes.h:932
Oid firstColType
Definition: primnodes.h:1009
TupleDesc tts_tupleDescriptor
Definition: tuptable.h:123
#define TupleDescAttr(tupdesc, i)
Definition: tupdesc.h:92
static HeapTuple ExecCopySlotHeapTuple(TupleTableSlot *slot)
Definition: tuptable.h:460

References accumArrayResultAny(), ALL_SUBLINK, ANY_SUBLINK, SubPlanState::args, ARRAY_SUBLINK, Assert(), bms_add_member(), BoolGetDatum(), PlanState::chgParam, CurrentMemoryContext, SubPlanState::curTuple, DatumGetBool(), ExprContext::ecxt_param_exec_vals, ExprContext::ecxt_per_query_memory, ereport, errcode(), errmsg(), ERROR, ExecCopySlotHeapTuple(), ExecEvalExprSwitchContext(), ParamExecData::execPlan, ExecProcNode(), ExecReScan(), EXISTS_SUBLINK, EXPR_SUBLINK, SubPlan::firstColType, forboth, heap_freetuple(), heap_getattr(), initArrayResultAny(), ParamExecData::isnull, lfirst, lfirst_int, list_length(), makeArrayResultAny(), MemoryContextSwitchTo(), MULTIEXPR_SUBLINK, SubPlan::paramIds, SubPlan::parParam, SubPlanState::planstate, ROWCOMPARE_SUBLINK, SubPlan::setParam, slot_getattr(), SubPlan::subLinkType, SubPlanState::subplan, SubPlanState::testexpr, TupleTableSlot::tts_tupleDescriptor, TupIsNull, TupleDescAttr, and ParamExecData::value.

Referenced by ExecSubPlan().

◆ ExecSetParamPlan()

void ExecSetParamPlan ( SubPlanState node,
ExprContext econtext 
)

Definition at line 1092 of file nodeSubplan.c.

1093 {
1094  SubPlan *subplan = node->subplan;
1095  PlanState *planstate = node->planstate;
1096  SubLinkType subLinkType = subplan->subLinkType;
1097  EState *estate = planstate->state;
1098  ScanDirection dir = estate->es_direction;
1099  MemoryContext oldcontext;
1100  TupleTableSlot *slot;
1101  ListCell *l;
1102  bool found = false;
1103  ArrayBuildStateAny *astate = NULL;
1104 
1105  if (subLinkType == ANY_SUBLINK ||
1106  subLinkType == ALL_SUBLINK)
1107  elog(ERROR, "ANY/ALL subselect unsupported as initplan");
1108  if (subLinkType == CTE_SUBLINK)
1109  elog(ERROR, "CTE subplans should not be executed via ExecSetParamPlan");
1110  if (subplan->parParam || node->args)
1111  elog(ERROR, "correlated subplans should not be executed via ExecSetParamPlan");
1112 
1113  /*
1114  * Enforce forward scan direction regardless of caller. It's hard but not
1115  * impossible to get here in backward scan, so make it work anyway.
1116  */
1118 
1119  /* Initialize ArrayBuildStateAny in caller's context, if needed */
1120  if (subLinkType == ARRAY_SUBLINK)
1121  astate = initArrayResultAny(subplan->firstColType,
1122  CurrentMemoryContext, true);
1123 
1124  /*
1125  * Must switch to per-query memory context.
1126  */
1127  oldcontext = MemoryContextSwitchTo(econtext->ecxt_per_query_memory);
1128 
1129  /*
1130  * Run the plan. (If it needs to be rescanned, the first ExecProcNode
1131  * call will take care of that.)
1132  */
1133  for (slot = ExecProcNode(planstate);
1134  !TupIsNull(slot);
1135  slot = ExecProcNode(planstate))
1136  {
1137  TupleDesc tdesc = slot->tts_tupleDescriptor;
1138  int i = 1;
1139 
1140  if (subLinkType == EXISTS_SUBLINK)
1141  {
1142  /* There can be only one setParam... */
1143  int paramid = linitial_int(subplan->setParam);
1144  ParamExecData *prm = &(econtext->ecxt_param_exec_vals[paramid]);
1145 
1146  prm->execPlan = NULL;
1147  prm->value = BoolGetDatum(true);
1148  prm->isnull = false;
1149  found = true;
1150  break;
1151  }
1152 
1153  if (subLinkType == ARRAY_SUBLINK)
1154  {
1155  Datum dvalue;
1156  bool disnull;
1157 
1158  found = true;
1159  /* stash away current value */
1160  Assert(subplan->firstColType == TupleDescAttr(tdesc, 0)->atttypid);
1161  dvalue = slot_getattr(slot, 1, &disnull);
1162  astate = accumArrayResultAny(astate, dvalue, disnull,
1163  subplan->firstColType, oldcontext);
1164  /* keep scanning subplan to collect all values */
1165  continue;
1166  }
1167 
1168  if (found &&
1169  (subLinkType == EXPR_SUBLINK ||
1170  subLinkType == MULTIEXPR_SUBLINK ||
1171  subLinkType == ROWCOMPARE_SUBLINK))
1172  ereport(ERROR,
1173  (errcode(ERRCODE_CARDINALITY_VIOLATION),
1174  errmsg("more than one row returned by a subquery used as an expression")));
1175 
1176  found = true;
1177 
1178  /*
1179  * We need to copy the subplan's tuple into our own context, in case
1180  * any of the params are pass-by-ref type --- the pointers stored in
1181  * the param structs will point at this copied tuple! node->curTuple
1182  * keeps track of the copied tuple for eventual freeing.
1183  */
1184  if (node->curTuple)
1185  heap_freetuple(node->curTuple);
1186  node->curTuple = ExecCopySlotHeapTuple(slot);
1187 
1188  /*
1189  * Now set all the setParam params from the columns of the tuple
1190  */
1191  foreach(l, subplan->setParam)
1192  {
1193  int paramid = lfirst_int(l);
1194  ParamExecData *prm = &(econtext->ecxt_param_exec_vals[paramid]);
1195 
1196  prm->execPlan = NULL;
1197  prm->value = heap_getattr(node->curTuple, i, tdesc,
1198  &(prm->isnull));
1199  i++;
1200  }
1201  }
1202 
1203  if (subLinkType == ARRAY_SUBLINK)
1204  {
1205  /* There can be only one setParam... */
1206  int paramid = linitial_int(subplan->setParam);
1207  ParamExecData *prm = &(econtext->ecxt_param_exec_vals[paramid]);
1208 
1209  /*
1210  * We build the result array in query context so it won't disappear;
1211  * to avoid leaking memory across repeated calls, we have to remember
1212  * the latest value, much as for curTuple above.
1213  */
1214  if (node->curArray != PointerGetDatum(NULL))
1215  pfree(DatumGetPointer(node->curArray));
1216  node->curArray = makeArrayResultAny(astate,
1217  econtext->ecxt_per_query_memory,
1218  true);
1219  prm->execPlan = NULL;
1220  prm->value = node->curArray;
1221  prm->isnull = false;
1222  }
1223  else if (!found)
1224  {
1225  if (subLinkType == EXISTS_SUBLINK)
1226  {
1227  /* There can be only one setParam... */
1228  int paramid = linitial_int(subplan->setParam);
1229  ParamExecData *prm = &(econtext->ecxt_param_exec_vals[paramid]);
1230 
1231  prm->execPlan = NULL;
1232  prm->value = BoolGetDatum(false);
1233  prm->isnull = false;
1234  }
1235  else
1236  {
1237  /* For other sublink types, set all the output params to NULL */
1238  foreach(l, subplan->setParam)
1239  {
1240  int paramid = lfirst_int(l);
1241  ParamExecData *prm = &(econtext->ecxt_param_exec_vals[paramid]);
1242 
1243  prm->execPlan = NULL;
1244  prm->value = (Datum) 0;
1245  prm->isnull = true;
1246  }
1247  }
1248  }
1249 
1250  MemoryContextSwitchTo(oldcontext);
1251 
1252  /* restore scan direction */
1253  estate->es_direction = dir;
1254 }
void pfree(void *pointer)
Definition: mcxt.c:1401
#define linitial_int(l)
Definition: pg_list.h:179
static Pointer DatumGetPointer(Datum X)
Definition: postgres.h:312
ScanDirection
Definition: sdir.h:25
@ ForwardScanDirection
Definition: sdir.h:28
ScanDirection es_direction
Definition: execnodes.h:621

References accumArrayResultAny(), ALL_SUBLINK, ANY_SUBLINK, SubPlanState::args, ARRAY_SUBLINK, Assert(), BoolGetDatum(), CTE_SUBLINK, SubPlanState::curArray, CurrentMemoryContext, SubPlanState::curTuple, DatumGetPointer(), ExprContext::ecxt_param_exec_vals, ExprContext::ecxt_per_query_memory, elog, ereport, errcode(), errmsg(), ERROR, EState::es_direction, ExecCopySlotHeapTuple(), ParamExecData::execPlan, ExecProcNode(), EXISTS_SUBLINK, EXPR_SUBLINK, SubPlan::firstColType, ForwardScanDirection, heap_freetuple(), heap_getattr(), i, initArrayResultAny(), ParamExecData::isnull, lfirst_int, linitial_int, makeArrayResultAny(), MemoryContextSwitchTo(), MULTIEXPR_SUBLINK, SubPlan::parParam, pfree(), SubPlanState::planstate, PointerGetDatum(), ROWCOMPARE_SUBLINK, SubPlan::setParam, slot_getattr(), PlanState::state, SubPlan::subLinkType, SubPlanState::subplan, TupleTableSlot::tts_tupleDescriptor, TupIsNull, TupleDescAttr, and ParamExecData::value.

Referenced by ExecEvalParamExec(), and ExecSetParamPlanMulti().

◆ ExecSetParamPlanMulti()

void ExecSetParamPlanMulti ( const Bitmapset params,
ExprContext econtext 
)

Definition at line 1268 of file nodeSubplan.c.

1269 {
1270  int paramid;
1271 
1272  paramid = -1;
1273  while ((paramid = bms_next_member(params, paramid)) >= 0)
1274  {
1275  ParamExecData *prm = &(econtext->ecxt_param_exec_vals[paramid]);
1276 
1277  if (prm->execPlan != NULL)
1278  {
1279  /* Parameter not evaluated yet, so go do it */
1280  ExecSetParamPlan(prm->execPlan, econtext);
1281  /* ExecSetParamPlan should have processed this param... */
1282  Assert(prm->execPlan == NULL);
1283  }
1284  }
1285 }
int bms_next_member(const Bitmapset *a, int prevbit)
Definition: bitmapset.c:1319
void ExecSetParamPlan(SubPlanState *node, ExprContext *econtext)
Definition: nodeSubplan.c:1092

References Assert(), bms_next_member(), ExprContext::ecxt_param_exec_vals, ParamExecData::execPlan, and ExecSetParamPlan().

Referenced by EvalPlanQualBegin(), EvalPlanQualStart(), ExecInitParallelPlan(), and ExecParallelReinitialize().

◆ ExecSubPlan()

Datum ExecSubPlan ( SubPlanState node,
ExprContext econtext,
bool isNull 
)

Definition at line 62 of file nodeSubplan.c.

65 {
66  SubPlan *subplan = node->subplan;
67  EState *estate = node->planstate->state;
68  ScanDirection dir = estate->es_direction;
69  Datum retval;
70 
72 
73  /* Set non-null as default */
74  *isNull = false;
75 
76  /* Sanity checks */
77  if (subplan->subLinkType == CTE_SUBLINK)
78  elog(ERROR, "CTE subplans should not be executed via ExecSubPlan");
79  if (subplan->setParam != NIL && subplan->subLinkType != MULTIEXPR_SUBLINK)
80  elog(ERROR, "cannot set parent params from subquery");
81 
82  /* Force forward-scan mode for evaluation */
84 
85  /* Select appropriate evaluation strategy */
86  if (subplan->useHashTable)
87  retval = ExecHashSubPlan(node, econtext, isNull);
88  else
89  retval = ExecScanSubPlan(node, econtext, isNull);
90 
91  /* restore scan direction */
92  estate->es_direction = dir;
93 
94  return retval;
95 }
#define CHECK_FOR_INTERRUPTS()
Definition: miscadmin.h:122
static Datum ExecHashSubPlan(SubPlanState *node, ExprContext *econtext, bool *isNull)
Definition: nodeSubplan.c:101
static Datum ExecScanSubPlan(SubPlanState *node, ExprContext *econtext, bool *isNull)
Definition: nodeSubplan.c:223

References CHECK_FOR_INTERRUPTS, CTE_SUBLINK, elog, ERROR, EState::es_direction, ExecHashSubPlan(), ExecScanSubPlan(), ForwardScanDirection, MULTIEXPR_SUBLINK, NIL, SubPlanState::planstate, SubPlan::setParam, PlanState::state, SubPlan::subLinkType, SubPlanState::subplan, and SubPlan::useHashTable.

Referenced by ExecEvalSubPlan().

◆ execTuplesUnequal()

static bool execTuplesUnequal ( TupleTableSlot slot1,
TupleTableSlot slot2,
int  numCols,
AttrNumber matchColIdx,
FmgrInfo eqfunctions,
const Oid collations,
MemoryContext  evalContext 
)
static

Definition at line 675 of file nodeSubplan.c.

682 {
683  MemoryContext oldContext;
684  bool result;
685  int i;
686 
687  /* Reset and switch into the temp context. */
688  MemoryContextReset(evalContext);
689  oldContext = MemoryContextSwitchTo(evalContext);
690 
691  /*
692  * We cannot report a match without checking all the fields, but we can
693  * report a non-match as soon as we find unequal fields. So, start
694  * comparing at the last field (least significant sort key). That's the
695  * most likely to be different if we are dealing with sorted input.
696  */
697  result = false;
698 
699  for (i = numCols; --i >= 0;)
700  {
701  AttrNumber att = matchColIdx[i];
702  Datum attr1,
703  attr2;
704  bool isNull1,
705  isNull2;
706 
707  attr1 = slot_getattr(slot1, att, &isNull1);
708 
709  if (isNull1)
710  continue; /* can't prove anything here */
711 
712  attr2 = slot_getattr(slot2, att, &isNull2);
713 
714  if (isNull2)
715  continue; /* can't prove anything here */
716 
717  /* Apply the type-specific equality function */
718  if (!DatumGetBool(FunctionCall2Coll(&eqfunctions[i],
719  collations[i],
720  attr1, attr2)))
721  {
722  result = true; /* they are unequal */
723  break;
724  }
725  }
726 
727  MemoryContextSwitchTo(oldContext);
728 
729  return result;
730 }
Datum FunctionCall2Coll(FmgrInfo *flinfo, Oid collation, Datum arg1, Datum arg2)
Definition: fmgr.c:1149

References DatumGetBool(), FunctionCall2Coll(), i, MemoryContextReset(), MemoryContextSwitchTo(), and slot_getattr().

Referenced by findPartialMatch().

◆ findPartialMatch()

static bool findPartialMatch ( TupleHashTable  hashtable,
TupleTableSlot slot,
FmgrInfo eqfunctions 
)
static

Definition at line 744 of file nodeSubplan.c.

746 {
747  int numCols = hashtable->numCols;
748  AttrNumber *keyColIdx = hashtable->keyColIdx;
749  TupleHashIterator hashiter;
750  TupleHashEntry entry;
751 
752  InitTupleHashIterator(hashtable, &hashiter);
753  while ((entry = ScanTupleHashTable(hashtable, &hashiter)) != NULL)
754  {
756 
757  ExecStoreMinimalTuple(entry->firstTuple, hashtable->tableslot, false);
758  if (!execTuplesUnequal(slot, hashtable->tableslot,
759  numCols, keyColIdx,
760  eqfunctions,
761  hashtable->tab_collations,
762  hashtable->tempcxt))
763  {
764  TermTupleHashIterator(&hashiter);
765  return true;
766  }
767  }
768  /* No TermTupleHashIterator call needed here */
769  return false;
770 }
TupleTableSlot * ExecStoreMinimalTuple(MinimalTuple mtup, TupleTableSlot *slot, bool shouldFree)
Definition: execTuples.c:1445
#define ScanTupleHashTable(htable, iter)
Definition: execnodes.h:846
#define TermTupleHashIterator(iter)
Definition: execnodes.h:842
#define InitTupleHashIterator(htable, iter)
Definition: execnodes.h:840
tuplehash_iterator TupleHashIterator
Definition: execnodes.h:833
static bool execTuplesUnequal(TupleTableSlot *slot1, TupleTableSlot *slot2, int numCols, AttrNumber *matchColIdx, FmgrInfo *eqfunctions, const Oid *collations, MemoryContext evalContext)
Definition: nodeSubplan.c:675
MinimalTuple firstTuple
Definition: execnodes.h:799
AttrNumber * keyColIdx
Definition: execnodes.h:817
MemoryContext tempcxt
Definition: execnodes.h:822
TupleTableSlot * tableslot
Definition: execnodes.h:824

References CHECK_FOR_INTERRUPTS, ExecStoreMinimalTuple(), execTuplesUnequal(), TupleHashEntryData::firstTuple, InitTupleHashIterator, TupleHashTableData::keyColIdx, TupleHashTableData::numCols, ScanTupleHashTable, TupleHashTableData::tab_collations, TupleHashTableData::tableslot, TupleHashTableData::tempcxt, and TermTupleHashIterator.

Referenced by ExecHashSubPlan().

◆ slotAllNulls()

static bool slotAllNulls ( TupleTableSlot slot)
static

Definition at line 779 of file nodeSubplan.c.

780 {
781  int ncols = slot->tts_tupleDescriptor->natts;
782  int i;
783 
784  for (i = 1; i <= ncols; i++)
785  {
786  if (!slot_attisnull(slot, i))
787  return false;
788  }
789  return true;
790 }
static bool slot_attisnull(TupleTableSlot *slot, int attnum)
Definition: tuptable.h:375

References i, TupleDescData::natts, slot_attisnull(), and TupleTableSlot::tts_tupleDescriptor.

Referenced by ExecHashSubPlan().

◆ slotNoNulls()

static bool slotNoNulls ( TupleTableSlot slot)
static

Definition at line 799 of file nodeSubplan.c.

800 {
801  int ncols = slot->tts_tupleDescriptor->natts;
802  int i;
803 
804  for (i = 1; i <= ncols; i++)
805  {
806  if (slot_attisnull(slot, i))
807  return false;
808  }
809  return true;
810 }

References i, TupleDescData::natts, slot_attisnull(), and TupleTableSlot::tts_tupleDescriptor.

Referenced by buildSubPlanHash(), and ExecHashSubPlan().