]> granicus.if.org Git - postgresql/blobdiff - src/backend/executor/execScan.c
Fix inconsistencies in the code
[postgresql] / src / backend / executor / execScan.c
index 85ddda64710ff6096b75b412a499fc0f336e9ed2..c0e4a5376c33741022134281200d86e9d0e64039 100644 (file)
 /*-------------------------------------------------------------------------
  *
- * execScan.c--
- *    This code provides support for generalized relation scans. ExecScan
- *    is passed a node and a pointer to a function to "do the right thing"
- *    and return a tuple from the relation. ExecScan then does the tedious
- *    stuff - checking the qualification and projecting the tuple
- *    appropriately.
+ * execScan.c
+ *       This code provides support for generalized relation scans. ExecScan
+ *       is passed a node and a pointer to a function to "do the right thing"
+ *       and return a tuple from the relation. ExecScan then does the tedious
+ *       stuff - checking the qualification and projecting the tuple
+ *       appropriately.
  *
- * Copyright (c) 1994, Regents of the University of California
+ * Portions Copyright (c) 1996-2019, PostgreSQL Global Development Group
+ * Portions Copyright (c) 1994, Regents of the University of California
  *
  *
  * IDENTIFICATION
- *    $Header: /cvsroot/pgsql/src/backend/executor/execScan.c,v 1.2 1996/10/31 10:11:34 scrappy Exp $
+ *       src/backend/executor/execScan.c
  *
  *-------------------------------------------------------------------------
  */
-
-#include <sys/file.h>
 #include "postgres.h"
 
 #include "executor/executor.h"
+#include "miscadmin.h"
+#include "utils/memutils.h"
+
+
+
+/*
+ * ExecScanFetch -- check interrupts & fetch next potential tuple
+ *
+ * This routine is concerned with substituting a test tuple if we are
+ * inside an EvalPlanQual recheck.  If we aren't, just execute
+ * the access method's next-tuple routine.
+ */
+static inline TupleTableSlot *
+ExecScanFetch(ScanState *node,
+                         ExecScanAccessMtd accessMtd,
+                         ExecScanRecheckMtd recheckMtd)
+{
+       EState     *estate = node->ps.state;
+
+       CHECK_FOR_INTERRUPTS();
+
+       if (estate->es_epqTupleSlot != NULL)
+       {
+               /*
+                * We are inside an EvalPlanQual recheck.  Return the test tuple if
+                * one is available, after rechecking any access-method-specific
+                * conditions.
+                */
+               Index           scanrelid = ((Scan *) node->ps.plan)->scanrelid;
+
+               if (scanrelid == 0)
+               {
+                       TupleTableSlot *slot = node->ss_ScanTupleSlot;
+
+                       /*
+                        * This is a ForeignScan or CustomScan which has pushed down a
+                        * join to the remote side.  The recheck method is responsible not
+                        * only for rechecking the scan/join quals but also for storing
+                        * the correct tuple in the slot.
+                        */
+                       if (!(*recheckMtd) (node, slot))
+                               ExecClearTuple(slot);   /* would not be returned by scan */
+                       return slot;
+               }
+               else if (estate->es_epqTupleSlot[scanrelid - 1] != NULL)
+               {
+                       TupleTableSlot *slot = node->ss_ScanTupleSlot;
+
+                       /* Return empty slot if we already returned a tuple */
+                       if (estate->es_epqScanDone[scanrelid - 1])
+                               return ExecClearTuple(slot);
+                       /* Else mark to remember that we shouldn't return more */
+                       estate->es_epqScanDone[scanrelid - 1] = true;
+
+                       slot = estate->es_epqTupleSlot[scanrelid - 1];
+
+                       /* Return empty slot if we haven't got a test tuple */
+                       if (TupIsNull(slot))
+                               return NULL;
+
+                       /* Check if it meets the access-method conditions */
+                       if (!(*recheckMtd) (node, slot))
+                               return ExecClearTuple(slot);    /* would not be returned by
+                                                                                                * scan */
+
+                       return slot;
+               }
+       }
+
+       /*
+        * Run the node-type-specific access method function to get the next tuple
+        */
+       return (*accessMtd) (node);
+}
 
 /* ----------------------------------------------------------------
- *     ExecScan
- *   
- *     Scans the relation using the 'access method' indicated and 
- *     returns the next qualifying tuple in the direction specified
- *     in the global variable ExecDirection.
- *     The access method returns the next tuple and execScan() is 
- *     responisble for checking the tuple returned against the qual-clause.
- *     
- *     Conditions:
- *       -- the "cursor" maintained by the AMI is positioned at the tuple
- *          returned previously.
- *   
- *     Initial States:
- *       -- the relation indicated is opened for scanning so that the 
- *          "cursor" is positioned before the first qualifying tuple.
+ *             ExecScan
+ *
+ *             Scans the relation using the 'access method' indicated and
+ *             returns the next qualifying tuple.
+ *             The access method returns the next tuple and ExecScan() is
+ *             responsible for checking the tuple returned against the qual-clause.
  *
- *     May need to put startmmgr  and endmmgr in here.
+ *             A 'recheck method' must also be provided that can check an
+ *             arbitrary tuple of the relation against any qual conditions
+ *             that are implemented internal to the access method.
+ *
+ *             Conditions:
+ *               -- the "cursor" maintained by the AMI is positioned at the tuple
+ *                      returned previously.
+ *
+ *             Initial States:
+ *               -- the relation indicated is opened for scanning so that the
+ *                      "cursor" is positioned before the first qualifying tuple.
  * ----------------------------------------------------------------
  */
 TupleTableSlot *
-ExecScan(Scan *node, 
-        TupleTableSlot* (*accessMtd)())        /* function returning a tuple */
+ExecScan(ScanState *node,
+                ExecScanAccessMtd accessMtd,   /* function returning a tuple */
+                ExecScanRecheckMtd recheckMtd)
 {
-    CommonScanState    *scanstate;
-    EState             *estate;
-    List               *qual;
-    bool               isDone;
-
-    TupleTableSlot     *slot;
-    TupleTableSlot     *resultSlot;
-    HeapTuple          newTuple;
-
-    ExprContext                *econtext;
-    ProjectionInfo     *projInfo;
-
-
-    /* ----------------
-     * initialize misc variables
-     * ----------------
-     */
-    newTuple =         NULL;
-    slot =     NULL;
-
-    estate =   node->plan.state;
-    scanstate =        node->scanstate;
-
-    /* ----------------
-     * get the expression context
-     * ----------------
-     */
-    econtext =         scanstate->cstate.cs_ExprContext;
-
-    /* ----------------
-     * initialize fields in ExprContext which don't change
-     * in the course of the scan..
-     * ----------------
-     */
-    qual = node->plan.qual;
-    econtext->ecxt_relation = scanstate->css_currentRelation;
-    econtext->ecxt_relid = node->scanrelid;
-
-    if (scanstate->cstate.cs_TupFromTlist) {
-       projInfo = scanstate->cstate.cs_ProjInfo;
-       resultSlot = ExecProject(projInfo, &isDone);
-       if (!isDone)
-         return resultSlot;
-    }
-    /* 
-     * get a tuple from the access method 
-     * loop until we obtain a tuple which passes the qualification.
-     */
-    for(;;) {
-       slot = (TupleTableSlot *) (*accessMtd)(node);
-
-       /* ----------------
-        *  if the slot returned by the accessMtd contains
-        *  NULL, then it means there is nothing more to scan
-        *  so we just return the empty slot.
-        * ----------------
+       ExprContext *econtext;
+       ExprState  *qual;
+       ProjectionInfo *projInfo;
+
+       /*
+        * Fetch data from node
         */
-       if (TupIsNull(slot)) return slot;
-       
-       /* ----------------
-        *   place the current tuple into the expr context
-        * ----------------
+       qual = node->ps.qual;
+       projInfo = node->ps.ps_ProjInfo;
+       econtext = node->ps.ps_ExprContext;
+
+       /* interrupt checks are in ExecScanFetch */
+
+       /*
+        * If we have neither a qual to check nor a projection to do, just skip
+        * all the overhead and return the raw scan tuple.
         */
-       econtext->ecxt_scantuple = slot;
-       
-       /* ----------------
-        *  check that the current tuple satisfies the qual-clause
-        *  if our qualification succeeds then we
-        *  leave the loop.
-        * ----------------
+       if (!qual && !projInfo)
+       {
+               ResetExprContext(econtext);
+               return ExecScanFetch(node, accessMtd, recheckMtd);
+       }
+
+       /*
+        * Reset per-tuple memory context to free any expression evaluation
+        * storage allocated in the previous tuple cycle.
         */
+       ResetExprContext(econtext);
 
-       /* add a check for non-nil qual here to avoid a
-          function call to ExecQual() when the qual is nil */
-       if (!qual || ExecQual(qual, econtext) == true)
-           break;
-    }
+       /*
+        * get a tuple from the access method.  Loop until we obtain a tuple that
+        * passes the qualification.
+        */
+       for (;;)
+       {
+               TupleTableSlot *slot;
+
+               slot = ExecScanFetch(node, accessMtd, recheckMtd);
 
-    /* ----------------
-     * form a projection tuple, store it in the result tuple
-     *  slot and return it.
-     * ----------------
-     */
-    projInfo = scanstate->cstate.cs_ProjInfo;
+               /*
+                * if the slot returned by the accessMtd contains NULL, then it means
+                * there is nothing more to scan so we just return an empty slot,
+                * being careful to use the projection result slot so it has correct
+                * tupleDesc.
+                */
+               if (TupIsNull(slot))
+               {
+                       if (projInfo)
+                               return ExecClearTuple(projInfo->pi_state.resultslot);
+                       else
+                               return slot;
+               }
 
-    resultSlot = ExecProject(projInfo, &isDone);
-    scanstate->cstate.cs_TupFromTlist = !isDone;
+               /*
+                * place the current tuple into the expr context
+                */
+               econtext->ecxt_scantuple = slot;
 
-    return resultSlot;
+               /*
+                * check that the current tuple satisfies the qual-clause
+                *
+                * check for non-null qual here to avoid a function call to ExecQual()
+                * when the qual is null ... saves only a few cycles, but they add up
+                * ...
+                */
+               if (qual == NULL || ExecQual(qual, econtext))
+               {
+                       /*
+                        * Found a satisfactory scan tuple.
+                        */
+                       if (projInfo)
+                       {
+                               /*
+                                * Form a projection tuple, store it in the result tuple slot
+                                * and return it.
+                                */
+                               return ExecProject(projInfo);
+                       }
+                       else
+                       {
+                               /*
+                                * Here, we aren't projecting, so just return scan tuple.
+                                */
+                               return slot;
+                       }
+               }
+               else
+                       InstrCountFiltered1(node, 1);
+
+               /*
+                * Tuple fails qual, so free per-tuple memory and try again.
+                */
+               ResetExprContext(econtext);
+       }
 }
 
+/*
+ * ExecAssignScanProjectionInfo
+ *             Set up projection info for a scan node, if necessary.
+ *
+ * We can avoid a projection step if the requested tlist exactly matches
+ * the underlying tuple type.  If so, we just set ps_ProjInfo to NULL.
+ * Note that this case occurs not only for simple "SELECT * FROM ...", but
+ * also in most cases where there are joins or other processing nodes above
+ * the scan node, because the planner will preferentially generate a matching
+ * tlist.
+ *
+ * The scan slot's descriptor must have been set already.
+ */
+void
+ExecAssignScanProjectionInfo(ScanState *node)
+{
+       Scan       *scan = (Scan *) node->ps.plan;
+       TupleDesc       tupdesc = node->ss_ScanTupleSlot->tts_tupleDescriptor;
+
+       ExecConditionalAssignProjectionInfo(&node->ps, tupdesc, scan->scanrelid);
+}
+
+/*
+ * ExecAssignScanProjectionInfoWithVarno
+ *             As above, but caller can specify varno expected in Vars in the tlist.
+ */
+void
+ExecAssignScanProjectionInfoWithVarno(ScanState *node, Index varno)
+{
+       TupleDesc       tupdesc = node->ss_ScanTupleSlot->tts_tupleDescriptor;
+
+       ExecConditionalAssignProjectionInfo(&node->ps, tupdesc, varno);
+}
+
+/*
+ * ExecScanReScan
+ *
+ * This must be called within the ReScan function of any plan node type
+ * that uses ExecScan().
+ */
+void
+ExecScanReScan(ScanState *node)
+{
+       EState     *estate = node->ps.state;
+
+       /*
+        * We must clear the scan tuple so that observers (e.g., execCurrent.c)
+        * can tell that this plan node is not positioned on a tuple.
+        */
+       ExecClearTuple(node->ss_ScanTupleSlot);
+
+       /* Rescan EvalPlanQual tuple if we're inside an EvalPlanQual recheck */
+       if (estate->es_epqScanDone != NULL)
+       {
+               Index           scanrelid = ((Scan *) node->ps.plan)->scanrelid;
+
+               if (scanrelid > 0)
+                       estate->es_epqScanDone[scanrelid - 1] = false;
+               else
+               {
+                       Bitmapset  *relids;
+                       int                     rtindex = -1;
+
+                       /*
+                        * If an FDW or custom scan provider has replaced the join with a
+                        * scan, there are multiple RTIs; reset the epqScanDone flag for
+                        * all of them.
+                        */
+                       if (IsA(node->ps.plan, ForeignScan))
+                               relids = ((ForeignScan *) node->ps.plan)->fs_relids;
+                       else if (IsA(node->ps.plan, CustomScan))
+                               relids = ((CustomScan *) node->ps.plan)->custom_relids;
+                       else
+                               elog(ERROR, "unexpected scan node: %d",
+                                        (int) nodeTag(node->ps.plan));
+
+                       while ((rtindex = bms_next_member(relids, rtindex)) >= 0)
+                       {
+                               Assert(rtindex > 0);
+                               estate->es_epqScanDone[rtindex - 1] = false;
+                       }
+               }
+       }
+}