LCOV - differential code coverage report
Current view: top level - src/backend/executor - nodeSetOp.c (source / functions) Coverage Total Hit UNC LBC UIC UBC GIC GNC CBC EUB ECB
Current: Differential Code Coverage HEAD vs 15 Lines: 94.7 % 206 195 1 2 2 6 8 3 184 5 9
Current Date: 2023-04-08 17:13:01 Functions: 100.0 % 12 12 1 11
Baseline: 15 Line coverage date bins:
Baseline Date: 2023-04-08 15:09:40 (240..) days: 94.7 % 206 195 1 2 2 6 8 3 184 4 9
Legend: Lines: hit not hit Function coverage date bins:
(240..) days: 100.0 % 12 12 1 11

 Age         Owner                  TLA  Line data    Source code
                                  1                 : /*-------------------------------------------------------------------------
                                  2                 :  *
                                  3                 :  * nodeSetOp.c
                                  4                 :  *    Routines to handle INTERSECT and EXCEPT selection
                                  5                 :  *
                                  6                 :  * The input of a SetOp node consists of tuples from two relations,
                                  7                 :  * which have been combined into one dataset, with a junk attribute added
                                  8                 :  * that shows which relation each tuple came from.  In SETOP_SORTED mode,
                                  9                 :  * the input has furthermore been sorted according to all the grouping
                                 10                 :  * columns (ie, all the non-junk attributes).  The SetOp node scans each
                                 11                 :  * group of identical tuples to determine how many came from each input
                                 12                 :  * relation.  Then it is a simple matter to emit the output demanded by the
                                 13                 :  * SQL spec for INTERSECT, INTERSECT ALL, EXCEPT, or EXCEPT ALL.
                                 14                 :  *
                                 15                 :  * In SETOP_HASHED mode, the input is delivered in no particular order,
                                 16                 :  * except that we know all the tuples from one input relation will come before
                                 17                 :  * all the tuples of the other.  The planner guarantees that the first input
                                 18                 :  * relation is the left-hand one for EXCEPT, and tries to make the smaller
                                 19                 :  * input relation come first for INTERSECT.  We build a hash table in memory
                                 20                 :  * with one entry for each group of identical tuples, and count the number of
                                 21                 :  * tuples in the group from each relation.  After seeing all the input, we
                                 22                 :  * scan the hashtable and generate the correct output using those counts.
                                 23                 :  * We can avoid making hashtable entries for any tuples appearing only in the
                                 24                 :  * second input relation, since they cannot result in any output.
                                 25                 :  *
                                 26                 :  * This node type is not used for UNION or UNION ALL, since those can be
                                 27                 :  * implemented more cheaply (there's no need for the junk attribute to
                                 28                 :  * identify the source relation).
                                 29                 :  *
                                 30                 :  * Note that SetOp does no qual checking nor projection.  The delivered
                                 31                 :  * output tuples are just copies of the first-to-arrive tuple in each
                                 32                 :  * input group.
                                 33                 :  *
                                 34                 :  *
                                 35                 :  * Portions Copyright (c) 1996-2023, PostgreSQL Global Development Group
                                 36                 :  * Portions Copyright (c) 1994, Regents of the University of California
                                 37                 :  *
                                 38                 :  *
                                 39                 :  * IDENTIFICATION
                                 40                 :  *    src/backend/executor/nodeSetOp.c
                                 41                 :  *
                                 42                 :  *-------------------------------------------------------------------------
                                 43                 :  */
                                 44                 : 
                                 45                 : #include "postgres.h"
                                 46                 : 
                                 47                 : #include "access/htup_details.h"
                                 48                 : #include "executor/executor.h"
                                 49                 : #include "executor/nodeSetOp.h"
                                 50                 : #include "miscadmin.h"
                                 51                 : #include "utils/memutils.h"
                                 52                 : 
                                 53                 : 
                                 54                 : /*
                                 55                 :  * SetOpStatePerGroupData - per-group working state
                                 56                 :  *
                                 57                 :  * These values are working state that is initialized at the start of
                                 58                 :  * an input tuple group and updated for each input tuple.
                                 59                 :  *
                                 60                 :  * In SETOP_SORTED mode, we need only one of these structs, and it's kept in
                                 61                 :  * the plan state node.  In SETOP_HASHED mode, the hash table contains one
                                 62                 :  * of these for each tuple group.
                                 63                 :  */
                                 64                 : typedef struct SetOpStatePerGroupData
                                 65                 : {
                                 66                 :     long        numLeft;        /* number of left-input dups in group */
                                 67                 :     long        numRight;       /* number of right-input dups in group */
                                 68                 : }           SetOpStatePerGroupData;
                                 69                 : 
                                 70                 : 
                                 71                 : static TupleTableSlot *setop_retrieve_direct(SetOpState *setopstate);
                                 72                 : static void setop_fill_hash_table(SetOpState *setopstate);
                                 73                 : static TupleTableSlot *setop_retrieve_hash_table(SetOpState *setopstate);
                                 74                 : 
                                 75                 : 
                                 76                 : /*
                                 77                 :  * Initialize state for a new group of input values.
                                 78                 :  */
                                 79                 : static inline void
 5358 tgl                        80 CBC      235468 : initialize_counts(SetOpStatePerGroup pergroup)
                                 81                 : {
                                 82          235468 :     pergroup->numLeft = pergroup->numRight = 0;
                                 83          235468 : }
                                 84                 : 
                                 85                 : /*
                                 86                 :  * Advance the appropriate counter for one input tuple.
                                 87                 :  */
                                 88                 : static inline void
                                 89          485769 : advance_counts(SetOpStatePerGroup pergroup, int flag)
                                 90                 : {
                                 91          485769 :     if (flag)
                                 92          250110 :         pergroup->numRight++;
                                 93                 :     else
                                 94          235659 :         pergroup->numLeft++;
                                 95          485769 : }
                                 96                 : 
                                 97                 : /*
                                 98                 :  * Fetch the "flag" column from an input tuple.
                                 99                 :  * This is an integer column with value 0 for left side, 1 for right side.
                                100                 :  */
                                101                 : static int
                                102          501192 : fetch_tuple_flag(SetOpState *setopstate, TupleTableSlot *inputslot)
                                103                 : {
                                104          501192 :     SetOp      *node = (SetOp *) setopstate->ps.plan;
                                105                 :     int         flag;
                                106                 :     bool        isNull;
                                107                 : 
                                108          501192 :     flag = DatumGetInt32(slot_getattr(inputslot,
                                109          501192 :                                       node->flagColIdx,
                                110                 :                                       &isNull));
                                111          501192 :     Assert(!isNull);
                                112          501192 :     Assert(flag == 0 || flag == 1);
                                113          501192 :     return flag;
                                114                 : }
                                115                 : 
                                116                 : /*
                                117                 :  * Initialize the hash table to empty.
                                118                 :  */
                                119                 : static void
                                120             234 : build_hash_table(SetOpState *setopstate)
                                121                 : {
                                122             234 :     SetOp      *node = (SetOp *) setopstate->ps.plan;
 1879 andres                    123             234 :     ExprContext *econtext = setopstate->ps.ps_ExprContext;
                                124             234 :     TupleDesc   desc = ExecGetResultType(outerPlanState(setopstate));
                                125                 : 
 5358 tgl                       126             234 :     Assert(node->strategy == SETOP_HASHED);
                                127             234 :     Assert(node->numGroups > 0);
                                128                 : 
 1520 andres                    129             468 :     setopstate->hashtable = BuildTupleHashTableExt(&setopstate->ps,
                                130                 :                                                    desc,
                                131                 :                                                    node->numCols,
                                132                 :                                                    node->dupColIdx,
                                133             234 :                                                    setopstate->eqfuncoids,
                                134                 :                                                    setopstate->hashfunctions,
                                135                 :                                                    node->dupCollations,
                                136                 :                                                    node->numGroups,
                                137                 :                                                    0,
                                138             234 :                                                    setopstate->ps.state->es_query_cxt,
                                139                 :                                                    setopstate->tableContext,
                                140                 :                                                    econtext->ecxt_per_tuple_memory,
                                141                 :                                                    false);
 5358 tgl                       142             234 : }
                                143                 : 
                                144                 : /*
                                145                 :  * We've completed processing a tuple group.  Decide how many copies (if any)
                                146                 :  * of its representative row to emit, and store the count into numOutput.
                                147                 :  * This logic is straight from the SQL92 specification.
                                148                 :  */
                                149                 : static void
                                150          235468 : set_output_count(SetOpState *setopstate, SetOpStatePerGroup pergroup)
                                151                 : {
                                152          235468 :     SetOp      *plannode = (SetOp *) setopstate->ps.plan;
                                153                 : 
                                154          235468 :     switch (plannode->cmd)
                                155                 :     {
                                156           45168 :         case SETOPCMD_INTERSECT:
                                157           45168 :             if (pergroup->numLeft > 0 && pergroup->numRight > 0)
                                158           30102 :                 setopstate->numOutput = 1;
                                159                 :             else
                                160           15066 :                 setopstate->numOutput = 0;
                                161           45168 :             break;
                                162              18 :         case SETOPCMD_INTERSECT_ALL:
                                163              18 :             setopstate->numOutput =
                                164              18 :                 (pergroup->numLeft < pergroup->numRight) ?
                                165              18 :                 pergroup->numLeft : pergroup->numRight;
                                166              18 :             break;
                                167          184240 :         case SETOPCMD_EXCEPT:
                                168          184240 :             if (pergroup->numLeft > 0 && pergroup->numRight == 0)
                                169             420 :                 setopstate->numOutput = 1;
                                170                 :             else
                                171          183820 :                 setopstate->numOutput = 0;
                                172          184240 :             break;
                                173            6042 :         case SETOPCMD_EXCEPT_ALL:
                                174            6042 :             setopstate->numOutput =
                                175            6042 :                 (pergroup->numLeft < pergroup->numRight) ?
                                176            6042 :                 0 : (pergroup->numLeft - pergroup->numRight);
                                177            6042 :             break;
 5358 tgl                       178 UBC           0 :         default:
                                179               0 :             elog(ERROR, "unrecognized set op: %d", (int) plannode->cmd);
                                180                 :             break;
                                181                 :     }
 5358 tgl                       182 CBC      235468 : }
                                183                 : 
                                184                 : 
                                185                 : /* ----------------------------------------------------------------
                                186                 :  *      ExecSetOp
                                187                 :  * ----------------------------------------------------------------
                                188                 :  */
                                189                 : static TupleTableSlot *         /* return: a tuple or NULL */
 2092 andres                    190           31146 : ExecSetOp(PlanState *pstate)
                                191                 : {
                                192           31146 :     SetOpState *node = castNode(SetOpState, pstate);
 7430 tgl                       193           31146 :     SetOp      *plannode = (SetOp *) node->ps.plan;
 5358                           194           31146 :     TupleTableSlot *resultTupleSlot = node->ps.ps_ResultTupleSlot;
                                195                 : 
 2084 andres                    196           31146 :     CHECK_FOR_INTERRUPTS();
                                197                 : 
                                198                 :     /*
                                199                 :      * If the previously-returned tuple needs to be returned more than once,
                                200                 :      * keep returning it.
                                201                 :      */
 7430 tgl                       202           31146 :     if (node->numOutput > 0)
                                203                 :     {
                                204              24 :         node->numOutput--;
 8221                           205              24 :         return resultTupleSlot;
                                206                 :     }
                                207                 : 
                                208                 :     /* Otherwise, we're done if we are out of groups */
 5358                           209           31122 :     if (node->setop_done)
                                210               9 :         return NULL;
                                211                 : 
                                212                 :     /* Fetch the next tuple group according to the correct strategy */
                                213           31113 :     if (plannode->strategy == SETOP_HASHED)
                                214                 :     {
                                215           16044 :         if (!node->table_filled)
                                216             513 :             setop_fill_hash_table(node);
                                217           16044 :         return setop_retrieve_hash_table(node);
                                218                 :     }
                                219                 :     else
                                220           15069 :         return setop_retrieve_direct(node);
                                221                 : }
                                222                 : 
                                223                 : /*
                                224                 :  * ExecSetOp for non-hashed case
                                225                 :  */
                                226                 : static TupleTableSlot *
                                227           15069 : setop_retrieve_direct(SetOpState *setopstate)
                                228                 : {
                                229                 :     PlanState  *outerPlan;
                                230                 :     SetOpStatePerGroup pergroup;
                                231                 :     TupleTableSlot *outerslot;
                                232                 :     TupleTableSlot *resultTupleSlot;
 1879 andres                    233           15069 :     ExprContext *econtext = setopstate->ps.ps_ExprContext;
                                234                 : 
                                235                 :     /*
                                236                 :      * get state info from node
                                237                 :      */
 5358 tgl                       238           15069 :     outerPlan = outerPlanState(setopstate);
 2368 andres                    239           15069 :     pergroup = (SetOpStatePerGroup) setopstate->pergroup;
 5358 tgl                       240           15069 :     resultTupleSlot = setopstate->ps.ps_ResultTupleSlot;
                                241                 : 
                                242                 :     /*
                                243                 :      * We loop retrieving groups until we find one we should return
                                244                 :      */
                                245           60111 :     while (!setopstate->setop_done)
                                246                 :     {
                                247                 :         /*
                                248                 :          * If we don't already have the first tuple of the new group, fetch it
                                249                 :          * from the outer plan.
                                250                 :          */
                                251           60078 :         if (setopstate->grp_firstTuple == NULL)
                                252                 :         {
                                253              42 :             outerslot = ExecProcNode(outerPlan);
                                254              42 :             if (!TupIsNull(outerslot))
                                255                 :             {
                                256                 :                 /* Make a copy of the first input tuple */
 1605 andres                    257              42 :                 setopstate->grp_firstTuple = ExecCopySlotHeapTuple(outerslot);
                                258                 :             }
                                259                 :             else
                                260                 :             {
                                261                 :                 /* outer plan produced no tuples at all */
 5358 tgl                       262 UBC           0 :                 setopstate->setop_done = true;
                                263               0 :                 return NULL;
                                264                 :             }
                                265                 :         }
                                266                 : 
                                267                 :         /*
                                268                 :          * Store the copied first input tuple in the tuple table slot reserved
                                269                 :          * for it.  The tuple will be deleted when it is cleared from the
                                270                 :          * slot.
                                271                 :          */
 1657 andres                    272 CBC       60078 :         ExecStoreHeapTuple(setopstate->grp_firstTuple,
                                273                 :                            resultTupleSlot,
                                274                 :                            true);
 2118 tgl                       275           60078 :         setopstate->grp_firstTuple = NULL;   /* don't keep two pointers */
                                276                 : 
                                277                 :         /* Initialize working state for a new input tuple group */
 5358                           278           60078 :         initialize_counts(pergroup);
                                279                 : 
                                280                 :         /* Count the first input tuple */
                                281           60078 :         advance_counts(pergroup,
                                282                 :                        fetch_tuple_flag(setopstate, resultTupleSlot));
                                283                 : 
                                284                 :         /*
                                285                 :          * Scan the outer plan until we exhaust it or cross a group boundary.
                                286                 :          */
                                287                 :         for (;;)
                                288                 :         {
                                289          120177 :             outerslot = ExecProcNode(outerPlan);
                                290          120177 :             if (TupIsNull(outerslot))
                                291                 :             {
                                292                 :                 /* no more outer-plan tuples available */
                                293              42 :                 setopstate->setop_done = true;
                                294              42 :                 break;
                                295                 :             }
                                296                 : 
                                297                 :             /*
                                298                 :              * Check whether we've crossed a group boundary.
                                299                 :              */
 1879 andres                    300          120135 :             econtext->ecxt_outertuple = resultTupleSlot;
                                301          120135 :             econtext->ecxt_innertuple = outerslot;
                                302                 : 
                                303          120135 :             if (!ExecQualAndReset(setopstate->eqfunction, econtext))
                                304                 :             {
                                305                 :                 /*
                                306                 :                  * Save the first input tuple of the next group.
                                307                 :                  */
 1605                           308           60036 :                 setopstate->grp_firstTuple = ExecCopySlotHeapTuple(outerslot);
 8221 tgl                       309           60036 :                 break;
                                310                 :             }
                                311                 : 
                                312                 :             /* Still in same group, so count this tuple */
 5358                           313           60099 :             advance_counts(pergroup,
                                314                 :                            fetch_tuple_flag(setopstate, outerslot));
                                315                 :         }
                                316                 : 
                                317                 :         /*
                                318                 :          * Done scanning input tuple group.  See if we should emit any copies
                                319                 :          * of result tuple, and if so return the first copy.
                                320                 :          */
                                321           60078 :         set_output_count(setopstate, pergroup);
                                322                 : 
                                323           60078 :         if (setopstate->numOutput > 0)
                                324                 :         {
                                325           15036 :             setopstate->numOutput--;
                                326           15036 :             return resultTupleSlot;
                                327                 :         }
                                328                 :     }
                                329                 : 
                                330                 :     /* No more groups */
                                331              33 :     ExecClearTuple(resultTupleSlot);
                                332              33 :     return NULL;
                                333                 : }
                                334                 : 
                                335                 : /*
                                336                 :  * ExecSetOp for hashed case: phase 1, read input and build hash table
                                337                 :  */
                                338                 : static void
                                339             513 : setop_fill_hash_table(SetOpState *setopstate)
                                340                 : {
                                341             513 :     SetOp      *node = (SetOp *) setopstate->ps.plan;
                                342                 :     PlanState  *outerPlan;
                                343                 :     int         firstFlag;
                                344                 :     bool        in_first_rel PG_USED_FOR_ASSERTS_ONLY;
 1879 andres                    345             513 :     ExprContext *econtext = setopstate->ps.ps_ExprContext;
                                346                 : 
                                347                 :     /*
                                348                 :      * get state info from node
                                349                 :      */
 5358 tgl                       350             513 :     outerPlan = outerPlanState(setopstate);
                                351             513 :     firstFlag = node->firstFlag;
                                352                 :     /* verify planner didn't mess up */
                                353             513 :     Assert(firstFlag == 0 ||
                                354                 :            (firstFlag == 1 &&
                                355                 :             (node->cmd == SETOPCMD_INTERSECT ||
                                356                 :              node->cmd == SETOPCMD_INTERSECT_ALL)));
                                357                 : 
                                358                 :     /*
                                359                 :      * Process each outer-plan tuple, and then fetch the next one, until we
                                360                 :      * exhaust the outer plan.
                                361                 :      */
                                362             513 :     in_first_rel = true;
                                363                 :     for (;;)
                                364          381015 :     {
                                365                 :         TupleTableSlot *outerslot;
                                366                 :         int         flag;
                                367                 :         TupleHashEntryData *entry;
                                368                 :         bool        isnew;
                                369                 : 
                                370          381528 :         outerslot = ExecProcNode(outerPlan);
                                371          381528 :         if (TupIsNull(outerslot))
                                372                 :             break;
                                373                 : 
                                374                 :         /* Identify whether it's left or right input */
                                375          381015 :         flag = fetch_tuple_flag(setopstate, outerslot);
                                376                 : 
                                377          381015 :         if (flag == firstFlag)
                                378                 :         {
                                379                 :             /* (still) in first input relation */
                                380          190557 :             Assert(in_first_rel);
                                381                 : 
                                382                 :             /* Find or build hashtable entry for this tuple's group */
 2368 andres                    383          190557 :             entry = LookupTupleHashEntry(setopstate->hashtable, outerslot,
                                384                 :                                          &isnew, NULL);
                                385                 : 
                                386                 :             /* If new tuple group, initialize counts */
 5358 tgl                       387          190557 :             if (isnew)
                                388                 :             {
 2368 andres                    389          175390 :                 entry->additional = (SetOpStatePerGroup)
                                390          175390 :                     MemoryContextAlloc(setopstate->hashtable->tablecxt,
                                391                 :                                        sizeof(SetOpStatePerGroupData));
                                392          175390 :                 initialize_counts((SetOpStatePerGroup) entry->additional);
                                393                 :             }
                                394                 : 
                                395                 :             /* Advance the counts */
                                396          190557 :             advance_counts((SetOpStatePerGroup) entry->additional, flag);
                                397                 :         }
                                398                 :         else
                                399                 :         {
                                400                 :             /* reached second relation */
 5358 tgl                       401          190458 :             in_first_rel = false;
                                402                 : 
                                403                 :             /* For tuples not seen previously, do not make hashtable entry */
 2368 andres                    404          190458 :             entry = LookupTupleHashEntry(setopstate->hashtable, outerslot,
                                405                 :                                          NULL, NULL);
                                406                 : 
                                407                 :             /* Advance the counts if entry is already present */
 5358 tgl                       408          190458 :             if (entry)
 2368 andres                    409          175035 :                 advance_counts((SetOpStatePerGroup) entry->additional, flag);
                                410                 :         }
                                411                 : 
                                412                 :         /* Must reset expression context after each hashtable lookup */
 1879                           413          381015 :         ResetExprContext(econtext);
                                414                 :     }
                                415                 : 
 5358 tgl                       416             513 :     setopstate->table_filled = true;
                                417                 :     /* Initialize to walk the hash table */
                                418             513 :     ResetTupleHashIterator(setopstate->hashtable, &setopstate->hashiter);
                                419             513 : }
                                420                 : 
                                421                 : /*
                                422                 :  * ExecSetOp for hashed case: phase 2, retrieving groups from hash table
                                423                 :  */
                                424                 : static TupleTableSlot *
                                425           16044 : setop_retrieve_hash_table(SetOpState *setopstate)
                                426                 : {
                                427                 :     TupleHashEntryData *entry;
                                428                 :     TupleTableSlot *resultTupleSlot;
                                429                 : 
                                430                 :     /*
                                431                 :      * get state info from node
                                432                 :      */
                                433           16044 :     resultTupleSlot = setopstate->ps.ps_ResultTupleSlot;
                                434                 : 
                                435                 :     /*
                                436                 :      * We loop retrieving groups until we find one we should return
                                437                 :      */
                                438          175903 :     while (!setopstate->setop_done)
                                439                 :     {
 2084 andres                    440          175903 :         CHECK_FOR_INTERRUPTS();
                                441                 : 
                                442                 :         /*
                                443                 :          * Find the next entry in the hash table
                                444                 :          */
 2368                           445          175903 :         entry = ScanTupleHashTable(setopstate->hashtable, &setopstate->hashiter);
 5358 tgl                       446          175903 :         if (entry == NULL)
                                447                 :         {
                                448                 :             /* No more entries in hashtable, so done */
                                449             513 :             setopstate->setop_done = true;
                                450             513 :             return NULL;
                                451                 :         }
                                452                 : 
                                453                 :         /*
                                454                 :          * See if we should emit any copies of this tuple, and if so return
                                455                 :          * the first copy.
                                456                 :          */
 2368 andres                    457          175390 :         set_output_count(setopstate, (SetOpStatePerGroup) entry->additional);
                                458                 : 
 5358 tgl                       459          175390 :         if (setopstate->numOutput > 0)
                                460                 :         {
                                461           15531 :             setopstate->numOutput--;
 2368 andres                    462           15531 :             return ExecStoreMinimalTuple(entry->firstTuple,
                                463                 :                                          resultTupleSlot,
                                464                 :                                          false);
                                465                 :         }
                                466                 :     }
                                467                 : 
                                468                 :     /* No more groups */
 5358 tgl                       469 UBC           0 :     ExecClearTuple(resultTupleSlot);
                                470               0 :     return NULL;
                                471                 : }
                                472                 : 
                                473                 : /* ----------------------------------------------------------------
                                474                 :  *      ExecInitSetOp
                                475                 :  *
                                476                 :  *      This initializes the setop node state structures and
                                477                 :  *      the node's subplan.
                                478                 :  * ----------------------------------------------------------------
                                479                 :  */
                                480                 : SetOpState *
 6249 tgl                       481 CBC         303 : ExecInitSetOp(SetOp *node, EState *estate, int eflags)
                                482                 : {
                                483                 :     SetOpState *setopstate;
                                484                 :     TupleDesc   outerDesc;
                                485                 : 
                                486                 :     /* check for unsupported flags */
                                487             303 :     Assert(!(eflags & (EXEC_FLAG_BACKWARD | EXEC_FLAG_MARK)));
                                488                 : 
                                489                 :     /*
                                490                 :      * create state structure
                                491                 :      */
 8221                           492             303 :     setopstate = makeNode(SetOpState);
 7430                           493             303 :     setopstate->ps.plan = (Plan *) node;
                                494             303 :     setopstate->ps.state = estate;
 2092 andres                    495             303 :     setopstate->ps.ExecProcNode = ExecSetOp;
                                496                 : 
 1879                           497             303 :     setopstate->eqfuncoids = NULL;
 5358 tgl                       498             303 :     setopstate->hashfunctions = NULL;
                                499             303 :     setopstate->setop_done = false;
 8221                           500             303 :     setopstate->numOutput = 0;
 5358                           501             303 :     setopstate->pergroup = NULL;
                                502             303 :     setopstate->grp_firstTuple = NULL;
                                503             303 :     setopstate->hashtable = NULL;
                                504             303 :     setopstate->tableContext = NULL;
                                505                 : 
                                506                 :     /*
                                507                 :      * create expression context
                                508                 :      */
 1879 andres                    509             303 :     ExecAssignExprContext(estate, &setopstate->ps);
                                510                 : 
                                511                 :     /*
                                512                 :      * If hashing, we also need a longer-lived context to store the hash
                                513                 :      * table.  The table can't just be kept in the per-query context because
                                514                 :      * we want to be able to throw it away in ExecReScanSetOp.
                                515                 :      */
 5358 tgl                       516             303 :     if (node->strategy == SETOP_HASHED)
                                517             234 :         setopstate->tableContext =
                                518             234 :             AllocSetContextCreate(CurrentMemoryContext,
                                519                 :                                   "SetOp hash table",
                                520                 :                                   ALLOCSET_DEFAULT_SIZES);
                                521                 : 
                                522                 :     /*
                                523                 :      * initialize child nodes
                                524                 :      *
                                525                 :      * If we are hashing then the child plan does not need to handle REWIND
                                526                 :      * efficiently; see ExecReScanSetOp.
                                527                 :      */
                                528             303 :     if (node->strategy == SETOP_HASHED)
                                529             234 :         eflags &= ~EXEC_FLAG_REWIND;
 6249                           530             303 :     outerPlanState(setopstate) = ExecInitNode(outerPlan(node), estate, eflags);
 1879 andres                    531             303 :     outerDesc = ExecGetResultType(outerPlanState(setopstate));
                                532                 : 
                                533                 :     /*
                                534                 :      * Initialize result slot and type. Setop nodes do no projections, so
                                535                 :      * initialize projection info for this node appropriately.
                                536                 :      */
 1606                           537             303 :     ExecInitResultTupleSlotTL(&setopstate->ps,
                                538             303 :                               node->strategy == SETOP_HASHED ?
                                539                 :                               &TTSOpsMinimalTuple : &TTSOpsHeapTuple);
 7430 tgl                       540             303 :     setopstate->ps.ps_ProjInfo = NULL;
                                541                 : 
                                542                 :     /*
                                543                 :      * Precompute fmgr lookup data for inner loop. We need both equality and
                                544                 :      * hashing functions to do it by hashing, but only equality if not
                                545                 :      * hashing.
                                546                 :      */
 5358                           547             303 :     if (node->strategy == SETOP_HASHED)
                                548             234 :         execTuplesHashPrepare(node->numCols,
                                549             234 :                               node->dupOperators,
                                550                 :                               &setopstate->eqfuncoids,
                                551                 :                               &setopstate->hashfunctions);
                                552                 :     else
 1879 andres                    553              69 :         setopstate->eqfunction =
                                554              69 :             execTuplesMatchPrepare(outerDesc,
                                555                 :                                    node->numCols,
                                556              69 :                                    node->dupColIdx,
                                557              69 :                                    node->dupOperators,
 1479 peter                     558              69 :                                    node->dupCollations,
                                559                 :                                    &setopstate->ps);
                                560                 : 
 5358 tgl                       561             303 :     if (node->strategy == SETOP_HASHED)
                                562                 :     {
                                563             234 :         build_hash_table(setopstate);
                                564             234 :         setopstate->table_filled = false;
                                565                 :     }
                                566                 :     else
                                567                 :     {
                                568              69 :         setopstate->pergroup =
                                569              69 :             (SetOpStatePerGroup) palloc0(sizeof(SetOpStatePerGroupData));
                                570                 :     }
                                571                 : 
 7430                           572             303 :     return setopstate;
                                573                 : }
                                574                 : 
                                575                 : /* ----------------------------------------------------------------
                                576                 :  *      ExecEndSetOp
                                577                 :  *
                                578                 :  *      This shuts down the subplan and frees resources allocated
                                579                 :  *      to this node.
                                580                 :  * ----------------------------------------------------------------
                                581                 :  */
                                582                 : void
                                583             303 : ExecEndSetOp(SetOpState *node)
                                584                 : {
                                585                 :     /* clean up tuple table */
                                586             303 :     ExecClearTuple(node->ps.ps_ResultTupleSlot);
                                587                 : 
                                588                 :     /* free subsidiary stuff including hashtable */
 5358                           589             303 :     if (node->tableContext)
                                590             234 :         MemoryContextDelete(node->tableContext);
 1879 andres                    591             303 :     ExecFreeExprContext(&node->ps);
                                592                 : 
 7420 tgl                       593             303 :     ExecEndNode(outerPlanState(node));
 8221                           594             303 : }
                                595                 : 
                                596                 : 
                                597                 : void
 4654                           598             300 : ExecReScanSetOp(SetOpState *node)
                                599                 : {
  276 tgl                       600 GNC         300 :     PlanState  *outerPlan = outerPlanState(node);
                                601                 : 
 7430 tgl                       602 CBC         300 :     ExecClearTuple(node->ps.ps_ResultTupleSlot);
 5358 tgl                       603 GIC         300 :     node->setop_done = false;
 7430 tgl                       604 CBC         300 :     node->numOutput = 0;
 8221 tgl                       605 ECB             : 
 5358 tgl                       606 CBC         300 :     if (((SetOp *) node->ps.plan)->strategy == SETOP_HASHED)
                                607                 :     {
 5358 tgl                       608 ECB             :         /*
                                609                 :          * In the hashed case, if we haven't yet built the hash table then we
                                610                 :          * can just return; nothing done yet, so nothing to undo. If subnode's
                                611                 :          * chgParam is not NULL then it will be re-scanned by ExecProcNode,
                                612                 :          * else no reason to re-scan it at all.
                                613                 :          */
 5358 tgl                       614 GIC         300 :         if (!node->table_filled)
                                615               3 :             return;
 5358 tgl                       616 ECB             : 
                                617                 :         /*
                                618                 :          * If we do have the hash table and the subplan does not have any
                                619                 :          * parameter changes, then we can just rescan the existing hash table;
                                620                 :          * no need to build it again.
                                621                 :          */
  276 tgl                       622 GNC         297 :         if (outerPlan->chgParam == NULL)
                                623                 :         {
 5358 tgl                       624 LBC           0 :             ResetTupleHashIterator(node->hashtable, &node->hashiter);
 5358 tgl                       625 UIC           0 :             return;
 5358 tgl                       626 EUB             :         }
                                627                 :     }
                                628                 : 
                                629                 :     /* Release first tuple of group, if we have made a copy */
 5358 tgl                       630 GIC         297 :     if (node->grp_firstTuple != NULL)
                                631                 :     {
 5358 tgl                       632 LBC           0 :         heap_freetuple(node->grp_firstTuple);
 5358 tgl                       633 UIC           0 :         node->grp_firstTuple = NULL;
 5358 tgl                       634 EUB             :     }
                                635                 : 
                                636                 :     /* Release any hashtable storage */
 5358 tgl                       637 GIC         297 :     if (node->tableContext)
                                638             297 :         MemoryContextResetAndDeleteChildren(node->tableContext);
 5358 tgl                       639 ECB             : 
                                640                 :     /* And rebuild empty hashtable if needed */
 5358 tgl                       641 GIC         297 :     if (((SetOp *) node->ps.plan)->strategy == SETOP_HASHED)
                                642                 :     {
 1520 andres                    643 CBC         297 :         ResetTupleHashTable(node->hashtable);
 5358 tgl                       644 GIC         297 :         node->table_filled = false;
 5358 tgl                       645 ECB             :     }
                                646                 : 
                                647                 :     /*
                                648                 :      * if chgParam of subnode is not null then plan will be re-scanned by
                                649                 :      * first ExecProcNode.
                                650                 :      */
  276 tgl                       651 GNC         297 :     if (outerPlan->chgParam == NULL)
  276 tgl                       652 UNC           0 :         ExecReScan(outerPlan);
 8221 tgl                       653 ECB             : }
        

Generated by: LCOV version v1.16-55-g56c0a2a