execMain.c 50.2 KB
Newer Older
1 2
/*-------------------------------------------------------------------------
 *
3
 * execMain.c
4
 *	  top level executor interface routines
5 6
 *
 * INTERFACE ROUTINES
7 8 9
 *	ExecutorStart()
 *	ExecutorRun()
 *	ExecutorEnd()
10
 *
11 12 13 14 15 16 17 18 19 20 21 22 23
 *	The old ExecutorMain() has been replaced by ExecutorStart(),
 *	ExecutorRun() and ExecutorEnd()
 *
 *	These three procedures are the external interfaces to the executor.
 *	In each case, the query descriptor and the execution state is required
 *	 as arguments
 *
 *	ExecutorStart() must be called at the beginning of any execution of any
 *	query plan and ExecutorEnd() should always be called at the end of
 *	execution of a plan.
 *
 *	ExecutorRun accepts 'feature' and 'count' arguments that specify whether
 *	the plan is to be executed forwards, backwards, and for how many tuples.
24
 *
25
 * Portions Copyright (c) 1996-2001, PostgreSQL Global Development Group
B
Add:  
Bruce Momjian 已提交
26
 * Portions Copyright (c) 1994, Regents of the University of California
27 28 29
 *
 *
 * IDENTIFICATION
30
 *	  $Header: /cvsroot/pgsql/src/backend/executor/execMain.c,v 1.146 2001/09/08 16:15:28 inoue Exp $
31 32 33
 *
 *-------------------------------------------------------------------------
 */
34 35
#include "postgres.h"

36 37
#include "access/heapam.h"
#include "catalog/heap.h"
38
#include "commands/command.h"
39
#include "commands/trigger.h"
B
Bruce Momjian 已提交
40 41 42 43 44 45
#include "executor/execdebug.h"
#include "executor/execdefs.h"
#include "miscadmin.h"
#include "optimizer/var.h"
#include "parser/parsetree.h"
#include "utils/acl.h"
46

47 48

/* decls for local routines only used within this module */
49
static TupleDesc InitPlan(CmdType operation,
B
Bruce Momjian 已提交
50 51 52
		 Query *parseTree,
		 Plan *plan,
		 EState *estate);
53
static void initResultRelInfo(ResultRelInfo *resultRelInfo,
B
Bruce Momjian 已提交
54 55 56
				  Index resultRelationIndex,
				  List *rangeTable,
				  CmdType operation);
57
static void EndPlan(Plan *plan, EState *estate);
58
static TupleTableSlot *ExecutePlan(EState *estate, Plan *plan,
B
Bruce Momjian 已提交
59 60 61 62
			CmdType operation,
			long numberTuples,
			ScanDirection direction,
			DestReceiver *destfunc);
63
static void ExecRetrieve(TupleTableSlot *slot,
64
			 DestReceiver *destfunc,
B
Bruce Momjian 已提交
65
			 EState *estate);
66
static void ExecAppend(TupleTableSlot *slot, ItemPointer tupleid,
B
Bruce Momjian 已提交
67
		   EState *estate);
68
static void ExecDelete(TupleTableSlot *slot, ItemPointer tupleid,
B
Bruce Momjian 已提交
69
		   EState *estate);
70
static void ExecReplace(TupleTableSlot *slot, ItemPointer tupleid,
B
Bruce Momjian 已提交
71
			EState *estate);
72
static TupleTableSlot *EvalPlanQualNext(EState *estate);
73
static void EndEvalPlanQual(EState *estate);
74
static void ExecCheckQueryPerms(CmdType operation, Query *parseTree,
B
Bruce Momjian 已提交
75
					Plan *plan);
76
static void ExecCheckPlanPerms(Plan *plan, List *rangeTable,
B
Bruce Momjian 已提交
77
				   CmdType operation);
78 79
static void ExecCheckRTPerms(List *rangeTable, CmdType operation);
static void ExecCheckRTEPerms(RangeTblEntry *rte, CmdType operation);
80

81 82
/* end of local decls */

83

84
/* ----------------------------------------------------------------
85 86 87 88 89
 *		ExecutorStart
 *
 *		This routine must be called at the beginning of any execution of any
 *		query plan
 *
90
 *		returns a TupleDesc which describes the attributes of the tuples to
91
 *		be returned by the query.
92
 *
93
 * NB: the CurrentMemoryContext when this is called must be the context
B
Bruce Momjian 已提交
94
 * to be used as the per-query context for the query plan.	ExecutorRun()
95
 * and ExecutorEnd() must be called in this same memory context.
96 97 98
 * ----------------------------------------------------------------
 */
TupleDesc
99
ExecutorStart(QueryDesc *queryDesc, EState *estate)
100
{
101
	TupleDesc	result;
102 103 104

	/* sanity checks */
	Assert(queryDesc != NULL);
105

V
Vadim B. Mikheev 已提交
106 107
	if (queryDesc->plantree->nParamExec > 0)
	{
108 109
		estate->es_param_exec_vals = (ParamExecData *)
			palloc(queryDesc->plantree->nParamExec * sizeof(ParamExecData));
110 111
		MemSet(estate->es_param_exec_vals, 0,
			   queryDesc->plantree->nParamExec * sizeof(ParamExecData));
V
Vadim B. Mikheev 已提交
112
	}
113

114 115 116
	/*
	 * Make our own private copy of the current queries snapshot data
	 */
117
	if (QuerySnapshot == NULL)
J
Jan Wieck 已提交
118
		estate->es_snapshot = NULL;
119
	else
120
	{
B
Bruce Momjian 已提交
121
		estate->es_snapshot = (Snapshot) palloc(sizeof(SnapshotData));
122 123 124 125
		memcpy(estate->es_snapshot, QuerySnapshot, sizeof(SnapshotData));
		if (estate->es_snapshot->xcnt > 0)
		{
			estate->es_snapshot->xip = (TransactionId *)
B
Bruce Momjian 已提交
126
				palloc(estate->es_snapshot->xcnt * sizeof(TransactionId));
127
			memcpy(estate->es_snapshot->xip, QuerySnapshot->xip,
B
Bruce Momjian 已提交
128
				   estate->es_snapshot->xcnt * sizeof(TransactionId));
129
		}
130
	}
131

132 133 134
	/*
	 * Initialize the plan
	 */
135 136 137 138 139 140
	result = InitPlan(queryDesc->operation,
					  queryDesc->parsetree,
					  queryDesc->plantree,
					  estate);

	return result;
141 142 143
}

/* ----------------------------------------------------------------
144 145 146 147 148 149 150
 *		ExecutorRun
 *
 *		This is the main routine of the executor module. It accepts
 *		the query descriptor from the traffic cop and executes the
 *		query plan.
 *
 *		ExecutorStart must have been called already.
151
 *
152 153 154 155 156 157
 *		the different features supported are:
 *			 EXEC_RUN:	retrieve all tuples in the forward direction
 *			 EXEC_FOR:	retrieve 'count' number of tuples in the forward dir
 *			 EXEC_BACK: retrieve 'count' number of tuples in the backward dir
 *			 EXEC_RETONE: return one tuple but don't 'retrieve' it
 *						   used in postquel function processing
158
 *
159 160
 *		Note: count = 0 is interpreted as "no limit".
 *
161 162
 * ----------------------------------------------------------------
 */
163
TupleTableSlot *
164
ExecutorRun(QueryDesc *queryDesc, EState *estate, int feature, long count)
165
{
B
Bruce Momjian 已提交
166 167
	CmdType		operation;
	Plan	   *plan;
168
	TupleTableSlot *result;
B
Bruce Momjian 已提交
169 170
	CommandDest dest;
	DestReceiver *destfunc;
171

B
Bruce Momjian 已提交
172
	/*
B
Bruce Momjian 已提交
173
	 * sanity checks
174
	 */
175 176
	Assert(queryDesc != NULL);

B
Bruce Momjian 已提交
177
	/*
B
Bruce Momjian 已提交
178 179
	 * extract information from the query descriptor and the query
	 * feature.
180
	 */
181 182 183
	operation = queryDesc->operation;
	plan = queryDesc->plantree;
	dest = queryDesc->dest;
184
	destfunc = DestToFunction(dest);
185 186 187
	estate->es_processed = 0;
	estate->es_lastoid = InvalidOid;

B
Bruce Momjian 已提交
188
	/*
B
Bruce Momjian 已提交
189 190 191 192
	 * FIXME: the dest setup function ought to be handed the tuple desc
	 * for the tuples to be output, but I'm not quite sure how to get that
	 * info at this point.	For now, passing NULL is OK because no
	 * existing dest setup function actually uses the pointer.
193 194 195
	 */
	(*destfunc->setup) (destfunc, (TupleDesc) NULL);

196 197
	switch (feature)
	{
198 199 200 201
		case EXEC_RUN:
			result = ExecutePlan(estate,
								 plan,
								 operation,
202
								 count,
203
								 ForwardScanDirection,
204
								 destfunc);
205
			break;
206

207 208 209 210 211 212
		case EXEC_FOR:
			result = ExecutePlan(estate,
								 plan,
								 operation,
								 count,
								 ForwardScanDirection,
213
								 destfunc);
214
			break;
215

B
Bruce Momjian 已提交
216
			/*
B
Bruce Momjian 已提交
217
			 * retrieve next n "backward" tuples
218 219 220 221 222 223 224
			 */
		case EXEC_BACK:
			result = ExecutePlan(estate,
								 plan,
								 operation,
								 count,
								 BackwardScanDirection,
225
								 destfunc);
226
			break;
227

B
Bruce Momjian 已提交
228
			/*
B
Bruce Momjian 已提交
229 230
			 * return one tuple but don't "retrieve" it. (this is used by
			 * the rule manager..) -cim 9/14/89
231 232 233 234 235 236 237
			 */
		case EXEC_RETONE:
			result = ExecutePlan(estate,
								 plan,
								 operation,
								 ONE_TUPLE,
								 ForwardScanDirection,
238
								 destfunc);
239
			break;
240

241 242
		default:
			elog(DEBUG, "ExecutorRun: Unknown feature %d", feature);
243
			result = NULL;
244
			break;
245 246
	}

247 248
	(*destfunc->cleanup) (destfunc);

249
	return result;
250 251 252
}

/* ----------------------------------------------------------------
253 254
 *		ExecutorEnd
 *
255
 *		This routine must be called at the end of execution of any
256
 *		query plan
257 258 259
 * ----------------------------------------------------------------
 */
void
260
ExecutorEnd(QueryDesc *queryDesc, EState *estate)
261
{
262 263
	/* sanity checks */
	Assert(queryDesc != NULL);
264

265
	EndPlan(queryDesc->plantree, estate);
266

267
	if (estate->es_snapshot != NULL)
B
Bruce Momjian 已提交
268 269 270 271
	{
		if (estate->es_snapshot->xcnt > 0)
			pfree(estate->es_snapshot->xip);
		pfree(estate->es_snapshot);
272
		estate->es_snapshot = NULL;
273 274
	}

275
	if (estate->es_param_exec_vals != NULL)
B
Bruce Momjian 已提交
276 277 278
	{
		pfree(estate->es_param_exec_vals);
		estate->es_param_exec_vals = NULL;
279
	}
280 281
}

282 283 284 285 286 287 288

/*
 * ExecCheckQueryPerms
 *		Check access permissions for all relations referenced in a query.
 */
static void
ExecCheckQueryPerms(CmdType operation, Query *parseTree, Plan *plan)
289
{
B
Bruce Momjian 已提交
290

291 292 293
	/*
	 * Check RTEs in the query's primary rangetable.
	 */
294
	ExecCheckRTPerms(parseTree->rtable, operation);
295

296 297 298
	/*
	 * Search for subplans and APPEND nodes to check their rangetables.
	 */
299
	ExecCheckPlanPerms(plan, parseTree->rtable, operation);
300 301 302 303 304 305 306 307
}

/*
 * ExecCheckPlanPerms
 *		Recursively scan the plan tree to check access permissions in
 *		subplans.
 */
static void
308
ExecCheckPlanPerms(Plan *plan, List *rangeTable, CmdType operation)
309 310 311 312 313 314 315 316 317 318
{
	List	   *subp;

	if (plan == NULL)
		return;

	/* Check subplans, which we assume are plain SELECT queries */

	foreach(subp, plan->initPlan)
	{
319
		SubPlan    *subplan = (SubPlan *) lfirst(subp);
320

321 322
		ExecCheckRTPerms(subplan->rtable, CMD_SELECT);
		ExecCheckPlanPerms(subplan->plan, subplan->rtable, CMD_SELECT);
323 324 325
	}
	foreach(subp, plan->subPlan)
	{
326
		SubPlan    *subplan = (SubPlan *) lfirst(subp);
M
Marc G. Fournier 已提交
327

328 329
		ExecCheckRTPerms(subplan->rtable, CMD_SELECT);
		ExecCheckPlanPerms(subplan->plan, subplan->rtable, CMD_SELECT);
330 331 332 333
	}

	/* Check lower plan nodes */

334 335
	ExecCheckPlanPerms(plan->lefttree, rangeTable, operation);
	ExecCheckPlanPerms(plan->righttree, rangeTable, operation);
336 337 338 339 340

	/* Do node-type-specific checks */

	switch (nodeTag(plan))
	{
341 342
		case T_SubqueryScan:
			{
B
Bruce Momjian 已提交
343
				SubqueryScan *scan = (SubqueryScan *) plan;
344 345 346 347 348 349 350 351
				RangeTblEntry *rte;

				/* Recursively check the subquery */
				rte = rt_fetch(scan->scan.scanrelid, rangeTable);
				Assert(rte->subquery != NULL);
				ExecCheckQueryPerms(operation, rte->subquery, scan->subplan);
				break;
			}
352
		case T_Append:
353
			{
354 355
				Append	   *app = (Append *) plan;
				List	   *appendplans;
356

357
				foreach(appendplans, app->appendplans)
358
				{
359 360 361
					ExecCheckPlanPerms((Plan *) lfirst(appendplans),
									   rangeTable,
									   operation);
362
				}
363
				break;
364 365 366
			}

		default:
367
			break;
368
	}
369
}
370

371 372 373 374 375
/*
 * ExecCheckRTPerms
 *		Check access permissions for all relations listed in a range table.
 */
static void
376
ExecCheckRTPerms(List *rangeTable, CmdType operation)
377 378 379 380
{
	List	   *lp;

	foreach(lp, rangeTable)
381
	{
382 383
		RangeTblEntry *rte = lfirst(lp);

384
		ExecCheckRTEPerms(rte, operation);
385 386 387 388 389 390 391 392
	}
}

/*
 * ExecCheckRTEPerms
 *		Check access permissions for a single RTE.
 */
static void
393
ExecCheckRTEPerms(RangeTblEntry *rte, CmdType operation)
394 395
{
	char	   *relName;
396
	Oid			userid;
397 398
	int32		aclcheck_result;

399 400 401 402 403
	/*
	 * If it's a subquery RTE, ignore it --- it will be checked when
	 * ExecCheckPlanPerms finds the SubqueryScan node for it.
	 */
	if (rte->subquery)
404 405 406 407 408
		return;

	relName = rte->relname;

	/*
B
Bruce Momjian 已提交
409 410
	 * userid to check as: current user unless we have a setuid
	 * indication.
411
	 *
B
Bruce Momjian 已提交
412 413 414 415
	 * Note: GetUserId() is presently fast enough that there's no harm in
	 * calling it separately for each RTE.	If that stops being true, we
	 * could call it once in ExecCheckQueryPerms and pass the userid down
	 * from there.	But for now, no need for the extra clutter.
416
	 */
417
	userid = rte->checkAsUser ? rte->checkAsUser : GetUserId();
418

419
#define CHECK(MODE)		pg_aclcheck(relName, userid, MODE)
420

421
	if (rte->checkForRead)
422
	{
423
		aclcheck_result = CHECK(ACL_SELECT);
424 425 426 427 428 429 430
		if (aclcheck_result != ACLCHECK_OK)
			elog(ERROR, "%s: %s",
				 relName, aclcheck_error_strings[aclcheck_result]);
	}

	if (rte->checkForWrite)
	{
B
Bruce Momjian 已提交
431

432 433 434 435 436
		/*
		 * Note: write access in a SELECT context means SELECT FOR UPDATE.
		 * Right now we don't distinguish that from true update as far as
		 * permissions checks are concerned.
		 */
437 438 439
		switch (operation)
		{
			case CMD_INSERT:
440
				aclcheck_result = CHECK(ACL_INSERT);
441
				break;
442
			case CMD_SELECT:
443
			case CMD_UPDATE:
444 445 446 447
				aclcheck_result = CHECK(ACL_UPDATE);
				break;
			case CMD_DELETE:
				aclcheck_result = CHECK(ACL_DELETE);
448 449 450 451
				break;
			default:
				elog(ERROR, "ExecCheckRTEPerms: bogus operation %d",
					 operation);
452
				aclcheck_result = ACLCHECK_OK;	/* keep compiler quiet */
453 454
				break;
		}
455 456 457
		if (aclcheck_result != ACLCHECK_OK)
			elog(ERROR, "%s: %s",
				 relName, aclcheck_error_strings[aclcheck_result]);
458
	}
459 460
}

461

462 463 464 465 466 467 468
/* ===============================================================
 * ===============================================================
						 static routines follow
 * ===============================================================
 * ===============================================================
 */

469 470 471
typedef struct execRowMark
{
	Relation	relation;
472
	Index		rti;
473
	char		resname[32];
474
} execRowMark;
475

476 477
typedef struct evalPlanQual
{
B
Bruce Momjian 已提交
478 479 480 481
	Plan	   *plan;
	Index		rti;
	EState		estate;
	struct evalPlanQual *free;
482
} evalPlanQual;
483

484
/* ----------------------------------------------------------------
485 486 487 488
 *		InitPlan
 *
 *		Initializes the query plan: open files, allocate storage
 *		and start up the rule manager
489 490
 * ----------------------------------------------------------------
 */
491
static TupleDesc
492
InitPlan(CmdType operation, Query *parseTree, Plan *plan, EState *estate)
493
{
B
Bruce Momjian 已提交
494 495 496
	List	   *rangeTable;
	Relation	intoRelationDesc;
	TupleDesc	tupType;
497

498 499 500 501 502
	/*
	 * Do permissions checks.
	 */
	ExecCheckQueryPerms(operation, parseTree, plan);

B
Bruce Momjian 已提交
503
	/*
B
Bruce Momjian 已提交
504
	 * get information from query descriptor
505
	 */
506
	rangeTable = parseTree->rtable;
507

B
Bruce Momjian 已提交
508
	/*
B
Bruce Momjian 已提交
509
	 * initialize the node's execution state
510
	 */
511 512
	estate->es_range_table = rangeTable;

B
Bruce Momjian 已提交
513
	/*
514
	 * if there is a result relation, initialize result relation stuff
515
	 */
516
	if (parseTree->resultRelation != 0 && operation != CMD_SELECT)
517
	{
518 519 520
		List	   *resultRelations = parseTree->resultRelations;
		int			numResultRelations;
		ResultRelInfo *resultRelInfos;
B
Bruce Momjian 已提交
521

522 523
		if (resultRelations != NIL)
		{
B
Bruce Momjian 已提交
524

525 526 527 528 529
			/*
			 * Multiple result relations (due to inheritance)
			 * parseTree->resultRelations identifies them all
			 */
			ResultRelInfo *resultRelInfo;
530

531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546
			numResultRelations = length(resultRelations);
			resultRelInfos = (ResultRelInfo *)
				palloc(numResultRelations * sizeof(ResultRelInfo));
			resultRelInfo = resultRelInfos;
			while (resultRelations != NIL)
			{
				initResultRelInfo(resultRelInfo,
								  lfirsti(resultRelations),
								  rangeTable,
								  operation);
				resultRelInfo++;
				resultRelations = lnext(resultRelations);
			}
		}
		else
		{
B
Bruce Momjian 已提交
547

548
			/*
B
Bruce Momjian 已提交
549 550
			 * Single result relation identified by
			 * parseTree->resultRelation
551 552 553 554 555 556 557 558
			 */
			numResultRelations = 1;
			resultRelInfos = (ResultRelInfo *) palloc(sizeof(ResultRelInfo));
			initResultRelInfo(resultRelInfos,
							  parseTree->resultRelation,
							  rangeTable,
							  operation);
		}
559

560 561 562 563
		estate->es_result_relations = resultRelInfos;
		estate->es_num_result_relations = numResultRelations;
		/* Initialize to first or only result rel */
		estate->es_result_relation_info = resultRelInfos;
564
	}
565 566
	else
	{
B
Bruce Momjian 已提交
567

B
Bruce Momjian 已提交
568
		/*
B
Bruce Momjian 已提交
569
		 * if no result relation, then set state appropriately
570
		 */
571 572
		estate->es_result_relations = NULL;
		estate->es_num_result_relations = 0;
573 574 575
		estate->es_result_relation_info = NULL;
	}

576 577 578
	/*
	 * Have to lock relations selected for update
	 */
579 580
	estate->es_rowMark = NIL;
	if (parseTree->rowMarks != NIL)
581
	{
B
Bruce Momjian 已提交
582
		List	   *l;
583

584
		foreach(l, parseTree->rowMarks)
585
		{
586 587
			Index		rti = lfirsti(l);
			Oid			relid = getrelid(rti, rangeTable);
588 589 590 591
			Relation	relation;
			execRowMark *erm;

			relation = heap_open(relid, RowShareLock);
B
Bruce Momjian 已提交
592
			erm = (execRowMark *) palloc(sizeof(execRowMark));
593
			erm->relation = relation;
594 595
			erm->rti = rti;
			sprintf(erm->resname, "ctid%u", rti);
596 597 598
			estate->es_rowMark = lappend(estate->es_rowMark, erm);
		}
	}
599

B
Bruce Momjian 已提交
600
	/*
601 602 603 604
	 * initialize the executor "tuple" table.  We need slots for all the
	 * plan nodes, plus possibly output slots for the junkfilter(s).
	 * At this point we aren't sure if we need junkfilters, so just add
	 * slots for them unconditionally.
605 606
	 */
	{
607
		int			nSlots = ExecCountSlotsNode(plan);
608

609 610 611 612 613
		if (parseTree->resultRelations != NIL)
			nSlots += length(parseTree->resultRelations);
		else
			nSlots += 1;
		estate->es_tupleTable = ExecCreateTupleTable(nSlots);
614
	}
615

616 617 618 619 620 621 622
	/* mark EvalPlanQual not active */
	estate->es_origPlan = plan;
	estate->es_evalPlanQual = NULL;
	estate->es_evTuple = NULL;
	estate->es_evTupleNull = NULL;
	estate->es_useEvalPlan = false;

B
Bruce Momjian 已提交
623
	/*
B
Bruce Momjian 已提交
624 625
	 * initialize the private state information for all the nodes in the
	 * query tree.	This opens files, allocates storage and leaves us
626
	 * ready to start processing tuples.
627 628 629
	 */
	ExecInitNode(plan, estate, NULL);

B
Bruce Momjian 已提交
630
	/*
631
	 * Get the tuple descriptor describing the type of tuples to return.
B
Bruce Momjian 已提交
632 633
	 * (this is especially important if we are creating a relation with
	 * "retrieve into")
634 635 636
	 */
	tupType = ExecGetTupType(plan);		/* tuple descriptor */

B
Bruce Momjian 已提交
637
	/*
B
Bruce Momjian 已提交
638 639 640 641
	 * Initialize the junk filter if needed. SELECT and INSERT queries
	 * need a filter if there are any junk attrs in the tlist.	UPDATE and
	 * DELETE always need one, since there's always a junk 'ctid'
	 * attribute present --- no need to look first.
642 643
	 */
	{
644 645 646
		bool		junk_filter_needed = false;
		List	   *tlist;

647
		switch (operation)
648
		{
649 650
			case CMD_SELECT:
			case CMD_INSERT:
651
				foreach(tlist, plan->targetlist)
652
				{
653 654 655 656 657 658 659
					TargetEntry *tle = (TargetEntry *) lfirst(tlist);

					if (tle->resdom->resjunk)
					{
						junk_filter_needed = true;
						break;
					}
660
				}
661 662 663 664 665 666 667
				break;
			case CMD_UPDATE:
			case CMD_DELETE:
				junk_filter_needed = true;
				break;
			default:
				break;
668 669
		}

670
		if (junk_filter_needed)
671
		{
B
Bruce Momjian 已提交
672

673
			/*
B
Bruce Momjian 已提交
674 675 676 677
			 * If there are multiple result relations, each one needs its
			 * own junk filter.  Note this is only possible for
			 * UPDATE/DELETE, so we can't be fooled by some needing a
			 * filter and some not.
678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695
			 */
			if (parseTree->resultRelations != NIL)
			{
				List	   *subplans;
				ResultRelInfo *resultRelInfo;

				/* Top plan had better be an Append here. */
				Assert(IsA(plan, Append));
				Assert(((Append *) plan)->isTarget);
				subplans = ((Append *) plan)->appendplans;
				Assert(length(subplans) == estate->es_num_result_relations);
				resultRelInfo = estate->es_result_relations;
				while (subplans != NIL)
				{
					Plan	   *subplan = (Plan *) lfirst(subplans);
					JunkFilter *j;

					j = ExecInitJunkFilter(subplan->targetlist,
696 697
										   ExecGetTupType(subplan),
										   ExecAllocTableSlot(estate->es_tupleTable));
698 699 700 701
					resultRelInfo->ri_junkFilter = j;
					resultRelInfo++;
					subplans = lnext(subplans);
				}
B
Bruce Momjian 已提交
702

703 704 705 706 707 708 709 710 711 712
				/*
				 * Set active junkfilter too; at this point ExecInitAppend
				 * has already selected an active result relation...
				 */
				estate->es_junkFilter =
					estate->es_result_relation_info->ri_junkFilter;
			}
			else
			{
				/* Normal case with just one JunkFilter */
713
				JunkFilter *j;
714

715 716 717
				j = ExecInitJunkFilter(plan->targetlist,
									   tupType,
									   ExecAllocTableSlot(estate->es_tupleTable));
718 719 720
				estate->es_junkFilter = j;
				if (estate->es_result_relation_info)
					estate->es_result_relation_info->ri_junkFilter = j;
721

722 723 724 725
				/* For SELECT, want to return the cleaned tuple type */
				if (operation == CMD_SELECT)
					tupType = j->jf_cleanTupType;
			}
726 727 728 729
		}
		else
			estate->es_junkFilter = NULL;
	}
730

B
Bruce Momjian 已提交
731
	/*
B
Bruce Momjian 已提交
732
	 * initialize the "into" relation
733 734 735 736 737
	 */
	intoRelationDesc = (Relation) NULL;

	if (operation == CMD_SELECT)
	{
738 739 740
		char	   *intoName;
		Oid			intoRelationId;
		TupleDesc	tupdesc;
741 742 743 744 745 746 747 748 749

		if (!parseTree->isPortal)
		{

			/*
			 * a select into table
			 */
			if (parseTree->into != NULL)
			{
B
Bruce Momjian 已提交
750

B
Bruce Momjian 已提交
751
				/*
B
Bruce Momjian 已提交
752
				 * create the "into" relation
753 754 755 756 757 758 759 760
				 */
				intoName = parseTree->into;

				/*
				 * have to copy tupType to get rid of constraints
				 */
				tupdesc = CreateTupleDescCopy(tupType);

761 762 763
				intoRelationId =
					heap_create_with_catalog(intoName,
											 tupdesc,
764
											 RELKIND_RELATION, true,
765 766
											 parseTree->isTemp,
											 allowSystemTableMods);
767

768 769
				FreeTupleDesc(tupdesc);

B
Bruce Momjian 已提交
770
				/*
771 772
				 * Advance command counter so that the newly-created
				 * relation's catalog tuples will be visible to heap_open.
773
				 */
774
				CommandCounterIncrement();
775

776
				/*
B
Bruce Momjian 已提交
777 778 779 780
				 * If necessary, create a TOAST table for the into
				 * relation. Note that AlterTableCreateToastTable ends
				 * with CommandCounterIncrement(), so that the TOAST table
				 * will be visible for insertion.
781 782 783
				 */
				AlterTableCreateToastTable(intoName, true);

784 785
				intoRelationDesc = heap_open(intoRelationId,
											 AccessExclusiveLock);
786 787 788 789 790 791 792
			}
		}
	}

	estate->es_into_relation_descriptor = intoRelationDesc;

	return tupType;
793 794
}

795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832
/*
 * Initialize ResultRelInfo data for one result relation
 */
static void
initResultRelInfo(ResultRelInfo *resultRelInfo,
				  Index resultRelationIndex,
				  List *rangeTable,
				  CmdType operation)
{
	Oid			resultRelationOid;
	Relation	resultRelationDesc;

	resultRelationOid = getrelid(resultRelationIndex, rangeTable);
	resultRelationDesc = heap_open(resultRelationOid, RowExclusiveLock);

	switch (resultRelationDesc->rd_rel->relkind)
	{
		case RELKIND_SEQUENCE:
			elog(ERROR, "You can't change sequence relation %s",
				 RelationGetRelationName(resultRelationDesc));
			break;
		case RELKIND_TOASTVALUE:
			elog(ERROR, "You can't change toast relation %s",
				 RelationGetRelationName(resultRelationDesc));
			break;
		case RELKIND_VIEW:
			elog(ERROR, "You can't change view relation %s",
				 RelationGetRelationName(resultRelationDesc));
			break;
	}

	MemSet(resultRelInfo, 0, sizeof(ResultRelInfo));
	resultRelInfo->type = T_ResultRelInfo;
	resultRelInfo->ri_RangeTableIndex = resultRelationIndex;
	resultRelInfo->ri_RelationDesc = resultRelationDesc;
	resultRelInfo->ri_NumIndices = 0;
	resultRelInfo->ri_IndexRelationDescs = NULL;
	resultRelInfo->ri_IndexRelationInfo = NULL;
833 834
	resultRelInfo->ri_TrigDesc = resultRelationDesc->trigdesc;
	resultRelInfo->ri_TrigFunctions = NULL;
835 836 837 838 839 840
	resultRelInfo->ri_ConstraintExprs = NULL;
	resultRelInfo->ri_junkFilter = NULL;

	/*
	 * If there are indices on the result relation, open them and save
	 * descriptors in the result relation info, so that we can add new
B
Bruce Momjian 已提交
841 842
	 * index entries for the tuples we add/update.	We need not do this
	 * for a DELETE, however, since deletion doesn't affect indexes.
843 844 845 846 847 848
	 */
	if (resultRelationDesc->rd_rel->relhasindex &&
		operation != CMD_DELETE)
		ExecOpenIndices(resultRelInfo);
}

849
/* ----------------------------------------------------------------
850 851 852
 *		EndPlan
 *
 *		Cleans up the query plan -- closes files and free up storages
853 854 855
 * ----------------------------------------------------------------
 */
static void
856
EndPlan(Plan *plan, EState *estate)
857
{
858 859
	ResultRelInfo *resultRelInfo;
	int			i;
860
	List	   *l;
861

862 863 864 865 866 867
	/*
	 * shut down any PlanQual processing we were doing
	 */
	if (estate->es_evalPlanQual != NULL)
		EndEvalPlanQual(estate);

B
Bruce Momjian 已提交
868
	/*
869
	 * shut down the node-type-specific query processing
870 871 872
	 */
	ExecEndNode(plan, plan);

B
Bruce Momjian 已提交
873
	/*
B
Bruce Momjian 已提交
874
	 * destroy the executor "tuple" table.
875
	 */
876 877
	ExecDropTupleTable(estate->es_tupleTable, true);
	estate->es_tupleTable = NULL;
878

B
Bruce Momjian 已提交
879
	/*
B
Bruce Momjian 已提交
880 881
	 * close the result relation(s) if any, but hold locks until xact
	 * commit.	Also clean up junkfilters if present.
882
	 */
883 884
	resultRelInfo = estate->es_result_relations;
	for (i = estate->es_num_result_relations; i > 0; i--)
885
	{
886 887 888
		/* Close indices and then the relation itself */
		ExecCloseIndices(resultRelInfo);
		heap_close(resultRelInfo->ri_RelationDesc, NoLock);
889 890 891
		/* Delete the junkfilter if any */
		if (resultRelInfo->ri_junkFilter != NULL)
			ExecFreeJunkFilter(resultRelInfo->ri_junkFilter);
892
		resultRelInfo++;
893 894
	}

B
Bruce Momjian 已提交
895
	/*
896
	 * close the "into" relation if necessary, again keeping lock
897
	 */
898 899
	if (estate->es_into_relation_descriptor != NULL)
		heap_close(estate->es_into_relation_descriptor, NoLock);
900

901 902 903 904 905 906 907 908 909 910
	/*
	 * There might be a junkfilter without a result relation.
	 */
	if (estate->es_num_result_relations == 0 &&
		estate->es_junkFilter != NULL)
	{
		ExecFreeJunkFilter(estate->es_junkFilter);
		estate->es_junkFilter = NULL;
	}

911 912 913 914 915 916 917 918 919
	/*
	 * close any relations selected FOR UPDATE, again keeping locks
	 */
	foreach(l, estate->es_rowMark)
	{
		execRowMark *erm = lfirst(l);

		heap_close(erm->relation, NoLock);
	}
920 921 922
}

/* ----------------------------------------------------------------
923 924
 *		ExecutePlan
 *
925
 *		processes the query plan to retrieve 'numberTuples' tuples in the
926 927 928
 *		direction specified.
 *		Retrieves all tuples if tupleCount is 0
 *
929
 *		result is either a slot containing the last tuple in the case
930
 *		of a RETRIEVE or NULL otherwise.
931
 *
932 933
 * Note: the ctid attribute is a 'junk' attribute that is removed before the
 * user can see it
934 935 936
 * ----------------------------------------------------------------
 */
static TupleTableSlot *
937 938
ExecutePlan(EState *estate,
			Plan *plan,
939
			CmdType operation,
940
			long numberTuples,
941
			ScanDirection direction,
942
			DestReceiver *destfunc)
943
{
944
	JunkFilter *junkfilter;
945
	TupleTableSlot *slot;
946
	ItemPointer tupleid = NULL;
947
	ItemPointerData tuple_ctid;
948
	long		current_tuple_count;
949 950
	TupleTableSlot *result;

B
Bruce Momjian 已提交
951
	/*
B
Bruce Momjian 已提交
952
	 * initialize local variables
953
	 */
954 955 956 957
	slot = NULL;
	current_tuple_count = 0;
	result = NULL;

B
Bruce Momjian 已提交
958 959
	/*
	 * Set the direction.
960
	 */
961 962
	estate->es_direction = direction;

B
Bruce Momjian 已提交
963
	/*
B
Bruce Momjian 已提交
964
	 * Loop until we've processed the proper number of tuples from the
965
	 * plan.
966 967 968 969
	 */

	for (;;)
	{
970 971
		/* Reset the per-output-tuple exprcontext */
		ResetPerTupleExprContext(estate);
B
Bruce Momjian 已提交
972

B
Bruce Momjian 已提交
973
		/*
B
Bruce Momjian 已提交
974
		 * Execute the plan and obtain a tuple
975 976
		 */
		/* at the top level, the parent of a plan (2nd arg) is itself */
B
Bruce Momjian 已提交
977
lnext:	;
978 979 980 981 982 983 984 985
		if (estate->es_useEvalPlan)
		{
			slot = EvalPlanQualNext(estate);
			if (TupIsNull(slot))
				slot = ExecProcNode(plan, plan);
		}
		else
			slot = ExecProcNode(plan, plan);
986

B
Bruce Momjian 已提交
987
		/*
B
Bruce Momjian 已提交
988 989
		 * if the tuple is null, then we assume there is nothing more to
		 * process so we just return null...
990 991 992 993 994
		 */
		if (TupIsNull(slot))
		{
			result = NULL;
			break;
995 996
		}

B
Bruce Momjian 已提交
997
		/*
B
Bruce Momjian 已提交
998 999
		 * if we have a junk filter, then project a new tuple with the
		 * junk removed.
1000
		 *
1001 1002 1003
		 * Store this new "clean" tuple in the junkfilter's resultSlot.
		 * (Formerly, we stored it back over the "dirty" tuple, which is
		 * WRONG because that tuple slot has the wrong descriptor.)
1004
		 *
B
Bruce Momjian 已提交
1005
		 * Also, extract all the junk information we need.
1006 1007 1008
		 */
		if ((junkfilter = estate->es_junkFilter) != (JunkFilter *) NULL)
		{
1009 1010 1011
			Datum		datum;
			HeapTuple	newTuple;
			bool		isNull;
1012

B
Bruce Momjian 已提交
1013
			/*
1014 1015 1016 1017 1018 1019 1020 1021 1022
			 * extract the 'ctid' junk attribute.
			 */
			if (operation == CMD_UPDATE || operation == CMD_DELETE)
			{
				if (!ExecGetJunkAttribute(junkfilter,
										  slot,
										  "ctid",
										  &datum,
										  &isNull))
1023
					elog(ERROR, "ExecutePlan: NO (junk) `ctid' was found!");
1024

1025
				/* shouldn't ever get a null result... */
1026
				if (isNull)
1027
					elog(ERROR, "ExecutePlan: (junk) `ctid' is NULL!");
1028 1029 1030 1031 1032 1033

				tupleid = (ItemPointer) DatumGetPointer(datum);
				tuple_ctid = *tupleid;	/* make sure we don't free the
										 * ctid!! */
				tupleid = &tuple_ctid;
			}
1034
			else if (estate->es_rowMark != NIL)
1035
			{
B
Bruce Momjian 已提交
1036
				List	   *l;
1037

B
Bruce Momjian 已提交
1038 1039
		lmark:	;
				foreach(l, estate->es_rowMark)
1040
				{
1041 1042 1043 1044 1045 1046
					execRowMark *erm = lfirst(l);
					Buffer		buffer;
					HeapTupleData tuple;
					TupleTableSlot *newSlot;
					int			test;

1047 1048 1049 1050 1051
					if (!ExecGetJunkAttribute(junkfilter,
											  slot,
											  erm->resname,
											  &datum,
											  &isNull))
1052 1053
						elog(ERROR, "ExecutePlan: NO (junk) `%s' was found!",
							 erm->resname);
1054

1055
					/* shouldn't ever get a null result... */
1056
					if (isNull)
1057 1058
						elog(ERROR, "ExecutePlan: (junk) `%s' is NULL!",
							 erm->resname);
1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071

					tuple.t_self = *((ItemPointer) DatumGetPointer(datum));
					test = heap_mark4update(erm->relation, &tuple, &buffer);
					ReleaseBuffer(buffer);
					switch (test)
					{
						case HeapTupleSelfUpdated:
						case HeapTupleMayBeUpdated:
							break;

						case HeapTupleUpdated:
							if (XactIsoLevel == XACT_SERIALIZABLE)
								elog(ERROR, "Can't serialize access due to concurrent update");
1072
							if (!(ItemPointerEquals(&(tuple.t_self),
B
Bruce Momjian 已提交
1073
								  (ItemPointer) DatumGetPointer(datum))))
1074
							{
B
Bruce Momjian 已提交
1075
								newSlot = EvalPlanQual(estate, erm->rti, &(tuple.t_self));
1076 1077 1078 1079 1080 1081 1082
								if (!(TupIsNull(newSlot)))
								{
									slot = newSlot;
									estate->es_useEvalPlan = true;
									goto lmark;
								}
							}
B
Bruce Momjian 已提交
1083 1084 1085

							/*
							 * if tuple was deleted or PlanQual failed for
1086
							 * updated tuple - we must not return this
B
Bruce Momjian 已提交
1087
							 * tuple!
1088 1089
							 */
							goto lnext;
1090 1091 1092

						default:
							elog(ERROR, "Unknown status %u from heap_mark4update", test);
B
Bruce Momjian 已提交
1093
							return (NULL);
1094 1095 1096
					}
				}
			}
1097

B
Bruce Momjian 已提交
1098
			/*
1099 1100 1101 1102 1103 1104
			 * Finally create a new "clean" tuple with all junk attributes
			 * removed
			 */
			newTuple = ExecRemoveJunk(junkfilter, slot);

			slot = ExecStoreTuple(newTuple,		/* tuple to store */
1105
								  junkfilter->jf_resultSlot, /* dest slot */
1106 1107 1108 1109 1110
								  InvalidBuffer,		/* this tuple has no
														 * buffer */
								  true);		/* tuple should be pfreed */
		}						/* if (junkfilter... */

B
Bruce Momjian 已提交
1111
		/*
B
Bruce Momjian 已提交
1112 1113
		 * now that we have a tuple, do the appropriate thing with it..
		 * either return it to the user, add it to a relation someplace,
B
Bruce Momjian 已提交
1114
		 * delete it from a relation, or modify some of its attributes.
1115 1116 1117 1118
		 */

		switch (operation)
		{
1119 1120
			case CMD_SELECT:
				ExecRetrieve(slot,		/* slot containing tuple */
B
Bruce Momjian 已提交
1121 1122
							 destfunc,	/* destination's tuple-receiver
										 * obj */
1123 1124 1125
							 estate);	/* */
				result = slot;
				break;
1126

1127 1128 1129 1130
			case CMD_INSERT:
				ExecAppend(slot, tupleid, estate);
				result = NULL;
				break;
1131

1132 1133 1134 1135
			case CMD_DELETE:
				ExecDelete(slot, tupleid, estate);
				result = NULL;
				break;
1136

1137
			case CMD_UPDATE:
1138
				ExecReplace(slot, tupleid, estate);
1139 1140
				result = NULL;
				break;
1141

1142 1143
			default:
				elog(DEBUG, "ExecutePlan: unknown operation in queryDesc");
1144
				result = NULL;
1145
				break;
1146
		}
B
Bruce Momjian 已提交
1147

B
Bruce Momjian 已提交
1148
		/*
1149 1150
		 * check our tuple count.. if we've processed the proper number
		 * then quit, else loop again and process more tuples..
1151
		 */
1152
		current_tuple_count++;
1153 1154
		if (numberTuples == current_tuple_count)
			break;
1155
	}
1156

B
Bruce Momjian 已提交
1157
	/*
B
Bruce Momjian 已提交
1158 1159
	 * here, result is either a slot containing a tuple in the case of a
	 * RETRIEVE or NULL otherwise.
1160
	 */
1161
	return result;
1162 1163 1164
}

/* ----------------------------------------------------------------
1165
 *		ExecRetrieve
1166
 *
1167 1168 1169 1170 1171
 *		RETRIEVEs are easy.. we just pass the tuple to the appropriate
 *		print function.  The only complexity is when we do a
 *		"retrieve into", in which case we insert the tuple into
 *		the appropriate relation (note: this is a newly created relation
 *		so we don't need to worry about indices or locks.)
1172 1173 1174
 * ----------------------------------------------------------------
 */
static void
1175
ExecRetrieve(TupleTableSlot *slot,
1176
			 DestReceiver *destfunc,
1177
			 EState *estate)
1178
{
1179 1180
	HeapTuple	tuple;
	TupleDesc	attrtype;
1181

B
Bruce Momjian 已提交
1182
	/*
B
Bruce Momjian 已提交
1183
	 * get the heap tuple out of the tuple table slot
1184 1185 1186 1187
	 */
	tuple = slot->val;
	attrtype = slot->ttc_tupleDescriptor;

B
Bruce Momjian 已提交
1188
	/*
B
Bruce Momjian 已提交
1189
	 * insert the tuple into the "into relation"
1190 1191 1192 1193 1194 1195 1196
	 */
	if (estate->es_into_relation_descriptor != NULL)
	{
		heap_insert(estate->es_into_relation_descriptor, tuple);
		IncrAppended();
	}

B
Bruce Momjian 已提交
1197
	/*
B
Bruce Momjian 已提交
1198
	 * send the tuple to the front end (or the screen)
1199
	 */
1200
	(*destfunc->receiveTuple) (tuple, attrtype, destfunc);
1201 1202
	IncrRetrieved();
	(estate->es_processed)++;
1203 1204 1205
}

/* ----------------------------------------------------------------
1206
 *		ExecAppend
1207
 *
1208 1209 1210
 *		APPENDs are trickier.. we have to insert the tuple into
 *		the base relation and insert appropriate tuples into the
 *		index relations.
1211 1212 1213 1214
 * ----------------------------------------------------------------
 */

static void
1215
ExecAppend(TupleTableSlot *slot,
1216
		   ItemPointer tupleid,
1217
		   EState *estate)
1218
{
1219
	HeapTuple	tuple;
1220
	ResultRelInfo *resultRelInfo;
1221 1222 1223
	Relation	resultRelationDesc;
	int			numIndices;
	Oid			newId;
1224

B
Bruce Momjian 已提交
1225
	/*
B
Bruce Momjian 已提交
1226
	 * get the heap tuple out of the tuple table slot
1227 1228 1229
	 */
	tuple = slot->val;

B
Bruce Momjian 已提交
1230
	/*
1231
	 * get information on the (current) result relation
1232
	 */
1233 1234
	resultRelInfo = estate->es_result_relation_info;
	resultRelationDesc = resultRelInfo->ri_RelationDesc;
1235 1236

	/* BEFORE ROW INSERT Triggers */
1237 1238
	if (resultRelInfo->ri_TrigDesc &&
		resultRelInfo->ri_TrigDesc->n_before_row[TRIGGER_EVENT_INSERT] > 0)
1239
	{
1240
		HeapTuple	newtuple;
1241

1242
		newtuple = ExecBRInsertTriggers(estate, resultRelInfo, tuple);
1243 1244 1245 1246 1247 1248

		if (newtuple == NULL)	/* "do nothing" */
			return;

		if (newtuple != tuple)	/* modified by Trigger(s) */
		{
B
Bruce Momjian 已提交
1249

1250 1251 1252
			/*
			 * Insert modified tuple into tuple table slot, replacing the
			 * original.  We assume that it was allocated in per-tuple
B
Bruce Momjian 已提交
1253 1254
			 * memory context, and therefore will go away by itself. The
			 * tuple table slot should not try to clear it.
1255 1256 1257
			 */
			ExecStoreTuple(newtuple, slot, InvalidBuffer, false);
			tuple = newtuple;
1258 1259 1260
		}
	}

B
Bruce Momjian 已提交
1261
	/*
1262
	 * Check the constraints of the tuple
1263 1264
	 */
	if (resultRelationDesc->rd_att->constr)
1265
		ExecConstraints("ExecAppend", resultRelInfo, slot, estate);
1266

B
Bruce Momjian 已提交
1267
	/*
B
Bruce Momjian 已提交
1268
	 * insert the tuple
1269
	 */
1270
	newId = heap_insert(resultRelationDesc, tuple);
1271
	setLastTid(&(tuple->t_self));
1272

1273
	IncrAppended();
1274 1275
	(estate->es_processed)++;
	estate->es_lastoid = newId;
1276

B
Bruce Momjian 已提交
1277
	/*
B
Bruce Momjian 已提交
1278
	 * process indices
1279
	 *
B
Bruce Momjian 已提交
1280 1281 1282
	 * Note: heap_insert adds a new tuple to a relation.  As a side effect,
	 * the tupleid of the new tuple is placed in the new tuple's t_ctid
	 * field.
1283
	 */
1284
	numIndices = resultRelInfo->ri_NumIndices;
1285
	if (numIndices > 0)
1286
		ExecInsertIndexTuples(slot, &(tuple->t_self), estate, false);
1287 1288

	/* AFTER ROW INSERT Triggers */
1289 1290
	if (resultRelInfo->ri_TrigDesc)
		ExecARInsertTriggers(estate, resultRelInfo, tuple);
1291 1292 1293
}

/* ----------------------------------------------------------------
1294
 *		ExecDelete
1295
 *
1296 1297
 *		DELETE is like append, we delete the tuple and its
 *		index tuples.
1298 1299 1300
 * ----------------------------------------------------------------
 */
static void
1301
ExecDelete(TupleTableSlot *slot,
1302
		   ItemPointer tupleid,
1303
		   EState *estate)
1304
{
1305
	ResultRelInfo *resultRelInfo;
B
Bruce Momjian 已提交
1306 1307 1308
	Relation	resultRelationDesc;
	ItemPointerData ctid;
	int			result;
1309

B
Bruce Momjian 已提交
1310
	/*
1311
	 * get information on the (current) result relation
1312
	 */
1313 1314
	resultRelInfo = estate->es_result_relation_info;
	resultRelationDesc = resultRelInfo->ri_RelationDesc;
1315 1316

	/* BEFORE ROW DELETE Triggers */
1317 1318
	if (resultRelInfo->ri_TrigDesc &&
		resultRelInfo->ri_TrigDesc->n_before_row[TRIGGER_EVENT_DELETE] > 0)
1319
	{
1320
		bool		dodelete;
1321

1322
		dodelete = ExecBRDeleteTriggers(estate, resultRelInfo, tupleid);
1323 1324 1325 1326 1327

		if (!dodelete)			/* "do nothing" */
			return;
	}

V
Vadim B. Mikheev 已提交
1328
	/*
B
Bruce Momjian 已提交
1329
	 * delete the tuple
1330
	 */
1331
ldelete:;
V
Vadim B. Mikheev 已提交
1332 1333 1334 1335 1336 1337 1338 1339 1340 1341
	result = heap_delete(resultRelationDesc, tupleid, &ctid);
	switch (result)
	{
		case HeapTupleSelfUpdated:
			return;

		case HeapTupleMayBeUpdated:
			break;

		case HeapTupleUpdated:
1342 1343
			if (XactIsoLevel == XACT_SERIALIZABLE)
				elog(ERROR, "Can't serialize access due to concurrent update");
1344 1345
			else if (!(ItemPointerEquals(tupleid, &ctid)))
			{
B
Bruce Momjian 已提交
1346
				TupleTableSlot *epqslot = EvalPlanQual(estate,
B
Bruce Momjian 已提交
1347
							   resultRelInfo->ri_RangeTableIndex, &ctid);
1348

V
Vadim B. Mikheev 已提交
1349
				if (!TupIsNull(epqslot))
1350 1351 1352 1353 1354
				{
					*tupleid = ctid;
					goto ldelete;
				}
			}
1355
			/* tuple already deleted; nothing to do */
V
Vadim B. Mikheev 已提交
1356 1357 1358 1359 1360 1361
			return;

		default:
			elog(ERROR, "Unknown status %u from heap_delete", result);
			return;
	}
1362 1363 1364 1365

	IncrDeleted();
	(estate->es_processed)++;

B
Bruce Momjian 已提交
1366
	/*
B
Bruce Momjian 已提交
1367 1368
	 * Note: Normally one would think that we have to delete index tuples
	 * associated with the heap tuple now..
1369
	 *
B
Bruce Momjian 已提交
1370 1371 1372
	 * ... but in POSTGRES, we have no need to do this because the vacuum
	 * daemon automatically opens an index scan and deletes index tuples
	 * when it finds deleted heap tuples. -cim 9/27/89
1373 1374 1375
	 */

	/* AFTER ROW DELETE Triggers */
1376 1377
	if (resultRelInfo->ri_TrigDesc)
		ExecARDeleteTriggers(estate, resultRelInfo, tupleid);
1378 1379 1380
}

/* ----------------------------------------------------------------
1381
 *		ExecReplace
1382
 *
1383 1384 1385 1386 1387 1388
 *		note: we can't run replace queries with transactions
 *		off because replaces are actually appends and our
 *		scan will mistakenly loop forever, replacing the tuple
 *		it just appended..	This should be fixed but until it
 *		is, we don't want to get stuck in an infinite loop
 *		which corrupts your database..
1389 1390 1391
 * ----------------------------------------------------------------
 */
static void
1392
ExecReplace(TupleTableSlot *slot,
1393
			ItemPointer tupleid,
1394
			EState *estate)
1395
{
B
Bruce Momjian 已提交
1396
	HeapTuple	tuple;
1397
	ResultRelInfo *resultRelInfo;
B
Bruce Momjian 已提交
1398 1399 1400 1401
	Relation	resultRelationDesc;
	ItemPointerData ctid;
	int			result;
	int			numIndices;
1402

B
Bruce Momjian 已提交
1403
	/*
B
Bruce Momjian 已提交
1404
	 * abort the operation if not running transactions
1405 1406 1407
	 */
	if (IsBootstrapProcessingMode())
	{
1408
		elog(NOTICE, "ExecReplace: replace can't run without transactions");
1409 1410 1411
		return;
	}

B
Bruce Momjian 已提交
1412
	/*
B
Bruce Momjian 已提交
1413
	 * get the heap tuple out of the tuple table slot
1414 1415 1416
	 */
	tuple = slot->val;

B
Bruce Momjian 已提交
1417
	/*
1418
	 * get information on the (current) result relation
1419
	 */
1420 1421
	resultRelInfo = estate->es_result_relation_info;
	resultRelationDesc = resultRelInfo->ri_RelationDesc;
1422 1423

	/* BEFORE ROW UPDATE Triggers */
1424 1425
	if (resultRelInfo->ri_TrigDesc &&
		resultRelInfo->ri_TrigDesc->n_before_row[TRIGGER_EVENT_UPDATE] > 0)
1426
	{
1427
		HeapTuple	newtuple;
1428

1429 1430
		newtuple = ExecBRUpdateTriggers(estate, resultRelInfo,
										tupleid, tuple);
1431 1432 1433 1434 1435 1436

		if (newtuple == NULL)	/* "do nothing" */
			return;

		if (newtuple != tuple)	/* modified by Trigger(s) */
		{
B
Bruce Momjian 已提交
1437

1438 1439 1440
			/*
			 * Insert modified tuple into tuple table slot, replacing the
			 * original.  We assume that it was allocated in per-tuple
B
Bruce Momjian 已提交
1441 1442
			 * memory context, and therefore will go away by itself. The
			 * tuple table slot should not try to clear it.
1443 1444 1445
			 */
			ExecStoreTuple(newtuple, slot, InvalidBuffer, false);
			tuple = newtuple;
1446 1447 1448
		}
	}

B
Bruce Momjian 已提交
1449
	/*
1450
	 * Check the constraints of the tuple
1451 1452 1453 1454 1455 1456
	 *
	 * If we generate a new candidate tuple after EvalPlanQual testing,
	 * we must loop back here and recheck constraints.  (We don't need to
	 * redo triggers, however.  If there are any BEFORE triggers then
	 * trigger.c will have done mark4update to lock the correct tuple,
	 * so there's no need to do them again.)
1457
	 */
1458
lreplace:;
1459
	if (resultRelationDesc->rd_att->constr)
1460
		ExecConstraints("ExecReplace", resultRelInfo, slot, estate);
1461

V
Vadim B. Mikheev 已提交
1462
	/*
B
Bruce Momjian 已提交
1463
	 * replace the heap tuple
1464
	 */
1465
	result = heap_update(resultRelationDesc, tupleid, tuple, &ctid);
V
Vadim B. Mikheev 已提交
1466 1467 1468 1469 1470 1471 1472 1473 1474
	switch (result)
	{
		case HeapTupleSelfUpdated:
			return;

		case HeapTupleMayBeUpdated:
			break;

		case HeapTupleUpdated:
1475 1476
			if (XactIsoLevel == XACT_SERIALIZABLE)
				elog(ERROR, "Can't serialize access due to concurrent update");
1477 1478
			else if (!(ItemPointerEquals(tupleid, &ctid)))
			{
B
Bruce Momjian 已提交
1479
				TupleTableSlot *epqslot = EvalPlanQual(estate,
B
Bruce Momjian 已提交
1480
							   resultRelInfo->ri_RangeTableIndex, &ctid);
1481

V
Vadim B. Mikheev 已提交
1482
				if (!TupIsNull(epqslot))
1483 1484
				{
					*tupleid = ctid;
V
Vadim B. Mikheev 已提交
1485
					tuple = ExecRemoveJunk(estate->es_junkFilter, epqslot);
1486 1487 1488
					slot = ExecStoreTuple(tuple,
										  estate->es_junkFilter->jf_resultSlot,
										  InvalidBuffer, true);
1489 1490 1491
					goto lreplace;
				}
			}
1492
			/* tuple already deleted; nothing to do */
V
Vadim B. Mikheev 已提交
1493 1494 1495
			return;

		default:
1496
			elog(ERROR, "Unknown status %u from heap_update", result);
V
Vadim B. Mikheev 已提交
1497
			return;
1498 1499 1500 1501 1502
	}

	IncrReplaced();
	(estate->es_processed)++;

B
Bruce Momjian 已提交
1503
	/*
B
Bruce Momjian 已提交
1504
	 * Note: instead of having to update the old index tuples associated
B
Bruce Momjian 已提交
1505 1506 1507 1508
	 * with the heap tuple, all we do is form and insert new index tuples.
	 * This is because replaces are actually deletes and inserts and index
	 * tuple deletion is done automagically by the vacuum daemon. All we
	 * do is insert new index tuples.  -cim 9/27/89
1509 1510
	 */

B
Bruce Momjian 已提交
1511
	/*
B
Bruce Momjian 已提交
1512
	 * process indices
1513
	 *
1514
	 * heap_update updates a tuple in the base relation by invalidating it
B
Bruce Momjian 已提交
1515 1516 1517 1518
	 * and then appending a new tuple to the relation.	As a side effect,
	 * the tupleid of the new tuple is placed in the new tuple's t_ctid
	 * field.  So we now insert index tuples using the new tupleid stored
	 * there.
1519 1520
	 */

1521
	numIndices = resultRelInfo->ri_NumIndices;
1522
	if (numIndices > 0)
1523
		ExecInsertIndexTuples(slot, &(tuple->t_self), estate, true);
1524 1525

	/* AFTER ROW UPDATE Triggers */
1526 1527
	if (resultRelInfo->ri_TrigDesc)
		ExecARUpdateTriggers(estate, resultRelInfo, tupleid, tuple);
1528
}
V
Vadim B. Mikheev 已提交
1529

1530
static char *
1531 1532
ExecRelCheck(ResultRelInfo *resultRelInfo,
			 TupleTableSlot *slot, EState *estate)
V
Vadim B. Mikheev 已提交
1533
{
1534
	Relation	rel = resultRelInfo->ri_RelationDesc;
1535 1536
	int			ncheck = rel->rd_att->constr->num_check;
	ConstrCheck *check = rel->rd_att->constr->check;
1537
	ExprContext *econtext;
1538
	MemoryContext oldContext;
1539 1540
	List	   *qual;
	int			i;
1541

1542 1543 1544 1545 1546 1547 1548 1549 1550 1551 1552 1553 1554 1555 1556 1557 1558 1559
	/*
	 * If first time through for this result relation, build expression
	 * nodetrees for rel's constraint expressions.  Keep them in the
	 * per-query memory context so they'll survive throughout the query.
	 */
	if (resultRelInfo->ri_ConstraintExprs == NULL)
	{
		oldContext = MemoryContextSwitchTo(estate->es_query_cxt);
		resultRelInfo->ri_ConstraintExprs =
			(List **) palloc(ncheck * sizeof(List *));
		for (i = 0; i < ncheck; i++)
		{
			qual = (List *) stringToNode(check[i].ccbin);
			resultRelInfo->ri_ConstraintExprs[i] = qual;
		}
		MemoryContextSwitchTo(oldContext);
	}

1560
	/*
B
Bruce Momjian 已提交
1561 1562
	 * We will use the EState's per-tuple context for evaluating
	 * constraint expressions (creating it if it's not already there).
1563
	 */
1564
	econtext = GetPerTupleExprContext(estate);
1565

1566 1567 1568 1569
	/* Arrange for econtext's scan tuple to be the tuple under test */
	econtext->ecxt_scantuple = slot;

	/* And evaluate the constraints */
1570 1571
	for (i = 0; i < ncheck; i++)
	{
1572
		qual = resultRelInfo->ri_ConstraintExprs[i];
1573

1574 1575
		/*
		 * NOTE: SQL92 specifies that a NULL result from a constraint
1576 1577
		 * expression is not to be treated as a failure.  Therefore, tell
		 * ExecQual to return TRUE for NULL.
1578
		 */
1579
		if (!ExecQual(qual, econtext, true))
1580
			return check[i].ccname;
1581 1582
	}

1583
	/* NULL result means no error */
1584
	return (char *) NULL;
V
Vadim B. Mikheev 已提交
1585 1586
}

1587
void
1588
ExecConstraints(char *caller, ResultRelInfo *resultRelInfo,
1589
				TupleTableSlot *slot, EState *estate)
V
Vadim B. Mikheev 已提交
1590
{
1591
	Relation	rel = resultRelInfo->ri_RelationDesc;
1592 1593 1594 1595
	HeapTuple	tuple = slot->val;
	TupleConstr *constr = rel->rd_att->constr;

	Assert(constr);
1596

1597
	if (constr->has_not_null)
V
Vadim B. Mikheev 已提交
1598
	{
1599
		int			natts = rel->rd_att->natts;
1600
		int			attrChk;
1601

1602
		for (attrChk = 1; attrChk <= natts; attrChk++)
1603
		{
B
Bruce Momjian 已提交
1604
			if (rel->rd_att->attrs[attrChk - 1]->attnotnull &&
1605
				heap_attisnull(tuple, attrChk))
1606
				elog(ERROR, "%s: Fail to add null value in not null attribute %s",
B
Bruce Momjian 已提交
1607
					 caller, NameStr(rel->rd_att->attrs[attrChk - 1]->attname));
1608 1609 1610
		}
	}

1611
	if (constr->num_check > 0)
1612
	{
1613
		char	   *failed;
1614

1615
		if ((failed = ExecRelCheck(resultRelInfo, slot, estate)) != NULL)
1616 1617
			elog(ERROR, "%s: rejected due to CHECK constraint %s",
				 caller, failed);
1618
	}
V
Vadim B. Mikheev 已提交
1619
}
1620

1621 1622 1623 1624 1625 1626
/*
 * Check a modified tuple to see if we want to process its updated version
 * under READ COMMITTED rules.
 *
 * See backend/executor/README for some info about how this works.
 */
B
Bruce Momjian 已提交
1627
TupleTableSlot *
1628 1629
EvalPlanQual(EState *estate, Index rti, ItemPointer tid)
{
1630 1631
	evalPlanQual *epq;
	EState	   *epqstate;
B
Bruce Momjian 已提交
1632 1633
	Relation	relation;
	HeapTupleData tuple;
1634 1635 1636
	HeapTuple	copyTuple = NULL;
	int			rtsize;
	bool		endNode;
1637 1638 1639

	Assert(rti != 0);

1640 1641 1642 1643 1644 1645 1646 1647 1648 1649 1650 1651 1652 1653 1654 1655 1656 1657 1658 1659 1660 1661 1662 1663 1664 1665 1666 1667 1668 1669 1670 1671 1672 1673 1674
	/*
	 * find relation containing target tuple
	 */
	if (estate->es_result_relation_info != NULL &&
		estate->es_result_relation_info->ri_RangeTableIndex == rti)
	{
		relation = estate->es_result_relation_info->ri_RelationDesc;
	}
	else
	{
		List	   *l;

		relation = NULL;
		foreach(l, estate->es_rowMark)
		{
			if (((execRowMark *) lfirst(l))->rti == rti)
			{
				relation = ((execRowMark *) lfirst(l))->relation;
				break;
			}
		}
		if (relation == NULL)
			elog(ERROR, "EvalPlanQual: can't find RTE %d", (int) rti);
	}

	/*
	 * fetch tid tuple
	 *
	 * Loop here to deal with updated or busy tuples
	 */
	tuple.t_self = *tid;
	for (;;)
	{
		Buffer		buffer;

1675
		heap_fetch(relation, SnapshotDirty, &tuple, &buffer, NULL);
1676 1677 1678 1679 1680 1681 1682 1683 1684 1685 1686 1687 1688 1689 1690 1691 1692 1693 1694 1695 1696 1697 1698 1699 1700 1701 1702 1703 1704 1705 1706 1707 1708 1709 1710 1711 1712 1713 1714 1715 1716 1717 1718 1719 1720 1721 1722 1723 1724 1725 1726 1727 1728 1729 1730 1731 1732 1733
		if (tuple.t_data != NULL)
		{
			TransactionId xwait = SnapshotDirty->xmax;

			if (TransactionIdIsValid(SnapshotDirty->xmin))
				elog(ERROR, "EvalPlanQual: t_xmin is uncommitted ?!");

			/*
			 * If tuple is being updated by other transaction then we have
			 * to wait for its commit/abort.
			 */
			if (TransactionIdIsValid(xwait))
			{
				ReleaseBuffer(buffer);
				XactLockTableWait(xwait);
				continue;
			}

			/*
			 * We got tuple - now copy it for use by recheck query.
			 */
			copyTuple = heap_copytuple(&tuple);
			ReleaseBuffer(buffer);
			break;
		}

		/*
		 * Oops! Invalid tuple. Have to check is it updated or deleted.
		 * Note that it's possible to get invalid SnapshotDirty->tid if
		 * tuple updated by this transaction. Have we to check this ?
		 */
		if (ItemPointerIsValid(&(SnapshotDirty->tid)) &&
			!(ItemPointerEquals(&(tuple.t_self), &(SnapshotDirty->tid))))
		{
			/* updated, so look at the updated copy */
			tuple.t_self = SnapshotDirty->tid;
			continue;
		}

		/*
		 * Deleted or updated by this transaction; forget it.
		 */
		return NULL;
	}

	/*
	 * For UPDATE/DELETE we have to return tid of actual row we're
	 * executing PQ for.
	 */
	*tid = tuple.t_self;

	/*
	 * Need to run a recheck subquery.  Find or create a PQ stack entry.
	 */
	epq = (evalPlanQual *) estate->es_evalPlanQual;
	rtsize = length(estate->es_range_table);
	endNode = true;

1734 1735
	if (epq != NULL && epq->rti == 0)
	{
1736
		/* Top PQ stack entry is idle, so re-use it */
B
Bruce Momjian 已提交
1737 1738
		Assert(!(estate->es_useEvalPlan) &&
			   epq->estate.es_evalPlanQual == NULL);
1739 1740 1741 1742 1743 1744
		epq->rti = rti;
		endNode = false;
	}

	/*
	 * If this is request for another RTE - Ra, - then we have to check
B
Bruce Momjian 已提交
1745 1746 1747
	 * wasn't PlanQual requested for Ra already and if so then Ra' row was
	 * updated again and we have to re-start old execution for Ra and
	 * forget all what we done after Ra was suspended. Cool? -:))
1748
	 */
B
Bruce Momjian 已提交
1749
	if (epq != NULL && epq->rti != rti &&
1750 1751 1752 1753
		epq->estate.es_evTuple[rti - 1] != NULL)
	{
		do
		{
1754 1755
			evalPlanQual *oldepq;

1756 1757
			/* pop previous PlanQual from the stack */
			epqstate = &(epq->estate);
B
Bruce Momjian 已提交
1758
			oldepq = (evalPlanQual *) epqstate->es_evalPlanQual;
1759 1760 1761
			Assert(oldepq->rti != 0);
			/* stop execution */
			ExecEndNode(epq->plan, epq->plan);
1762 1763
			ExecDropTupleTable(epqstate->es_tupleTable, true);
			epqstate->es_tupleTable = NULL;
1764
			heap_freetuple(epqstate->es_evTuple[epq->rti - 1]);
1765 1766 1767 1768
			epqstate->es_evTuple[epq->rti - 1] = NULL;
			/* push current PQ to freePQ stack */
			oldepq->free = epq;
			epq = oldepq;
1769
			estate->es_evalPlanQual = (Pointer) epq;
1770 1771 1772
		} while (epq->rti != rti);
	}

B
Bruce Momjian 已提交
1773
	/*
1774 1775 1776 1777 1778 1779
	 * If we are requested for another RTE then we have to suspend
	 * execution of current PlanQual and start execution for new one.
	 */
	if (epq == NULL || epq->rti != rti)
	{
		/* try to reuse plan used previously */
B
Bruce Momjian 已提交
1780
		evalPlanQual *newepq = (epq != NULL) ? epq->free : NULL;
1781

1782
		if (newepq == NULL)		/* first call or freePQ stack is empty */
1783
		{
B
Bruce Momjian 已提交
1784
			newepq = (evalPlanQual *) palloc(sizeof(evalPlanQual));
1785 1786 1787 1788 1789 1790 1791 1792 1793 1794 1795 1796 1797 1798 1799
			newepq->free = NULL;
			/*
			 * Each stack level has its own copy of the plan tree.  This
			 * is wasteful, but necessary as long as plan nodes point to
			 * exec state nodes rather than vice versa.  Note that copyfuncs.c
			 * doesn't attempt to copy the exec state nodes, which is a good
			 * thing in this situation.
			 */
			newepq->plan = copyObject(estate->es_origPlan);
			/*
			 * Init stack level's EState.  We share top level's copy of
			 * es_result_relations array and other non-changing status.
			 * We need our own tupletable, es_param_exec_vals, and other
			 * changeable state.
			 */
1800
			epqstate = &(newepq->estate);
1801
			memcpy(epqstate, estate, sizeof(EState));
1802 1803 1804
			epqstate->es_direction = ForwardScanDirection;
			if (estate->es_origPlan->nParamExec > 0)
				epqstate->es_param_exec_vals = (ParamExecData *)
B
Bruce Momjian 已提交
1805 1806
					palloc(estate->es_origPlan->nParamExec *
						   sizeof(ParamExecData));
1807 1808 1809 1810 1811 1812 1813 1814 1815 1816
			epqstate->es_tupleTable = NULL;
			epqstate->es_per_tuple_exprcontext = NULL;
			/*
			 * Each epqstate must have its own es_evTupleNull state,
			 * but all the stack entries share es_evTuple state.  This
			 * allows sub-rechecks to inherit the value being examined by
			 * an outer recheck.
			 */
			epqstate->es_evTupleNull = (bool *) palloc(rtsize * sizeof(bool));
			if (epq == NULL)
1817
			{
1818
				/* first PQ stack entry */
B
Bruce Momjian 已提交
1819
				epqstate->es_evTuple = (HeapTuple *)
1820 1821
					palloc(rtsize * sizeof(HeapTuple));
				memset(epqstate->es_evTuple, 0, rtsize * sizeof(HeapTuple));
1822 1823
			}
			else
1824 1825
			{
				/* later stack entries share the same storage */
1826
				epqstate->es_evTuple = epq->estate.es_evTuple;
1827
			}
1828 1829
		}
		else
1830 1831
		{
			/* recycle previously used EState */
1832
			epqstate = &(newepq->estate);
1833
		}
1834 1835
		/* push current PQ to the stack */
		epqstate->es_evalPlanQual = (Pointer) epq;
1836 1837
		epq = newepq;
		estate->es_evalPlanQual = (Pointer) epq;
1838 1839 1840 1841
		epq->rti = rti;
		endNode = false;
	}

1842
	Assert(epq->rti == rti);
1843 1844 1845
	epqstate = &(epq->estate);

	/*
1846 1847 1848 1849 1850 1851
	 * Ok - we're requested for the same RTE.  Unfortunately we still
	 * have to end and restart execution of the plan, because ExecReScan
	 * wouldn't ensure that upper plan nodes would reset themselves.  We
	 * could make that work if insertion of the target tuple were integrated
	 * with the Param mechanism somehow, so that the upper plan nodes know
	 * that their children's outputs have changed.
1852 1853
	 */
	if (endNode)
1854
	{
1855
		/* stop execution */
1856
		ExecEndNode(epq->plan, epq->plan);
1857 1858
		ExecDropTupleTable(epqstate->es_tupleTable, true);
		epqstate->es_tupleTable = NULL;
1859
	}
1860

1861 1862 1863 1864 1865 1866 1867
	/*
	 * free old RTE' tuple, if any, and store target tuple where relation's
	 * scan node will see it
	 */
	if (epqstate->es_evTuple[rti - 1] != NULL)
		heap_freetuple(epqstate->es_evTuple[rti - 1]);
	epqstate->es_evTuple[rti - 1] = copyTuple;
1868

1869 1870 1871 1872 1873 1874
	/*
	 * Initialize for new recheck query; be careful to copy down state
	 * that might have changed in top EState.
	 */
	epqstate->es_result_relation_info = estate->es_result_relation_info;
	epqstate->es_junkFilter = estate->es_junkFilter;
1875
	if (estate->es_origPlan->nParamExec > 0)
B
Bruce Momjian 已提交
1876 1877
		memset(epqstate->es_param_exec_vals, 0,
			   estate->es_origPlan->nParamExec * sizeof(ParamExecData));
1878 1879 1880 1881 1882
	memset(epqstate->es_evTupleNull, false, rtsize * sizeof(bool));
	epqstate->es_useEvalPlan = false;
	Assert(epqstate->es_tupleTable == NULL);
	epqstate->es_tupleTable =
		ExecCreateTupleTable(estate->es_tupleTable->size);
1883

1884
	ExecInitNode(epq->plan, epqstate, NULL);
1885

1886
	return EvalPlanQualNext(estate);
1887 1888
}

B
Bruce Momjian 已提交
1889
static TupleTableSlot *
1890 1891
EvalPlanQualNext(EState *estate)
{
B
Bruce Momjian 已提交
1892 1893 1894 1895
	evalPlanQual *epq = (evalPlanQual *) estate->es_evalPlanQual;
	EState	   *epqstate = &(epq->estate);
	evalPlanQual *oldepq;
	TupleTableSlot *slot;
1896 1897 1898 1899 1900 1901 1902 1903 1904 1905 1906

	Assert(epq->rti != 0);

lpqnext:;
	slot = ExecProcNode(epq->plan, epq->plan);

	/*
	 * No more tuples for this PQ. Continue previous one.
	 */
	if (TupIsNull(slot))
	{
1907
		/* stop execution */
1908
		ExecEndNode(epq->plan, epq->plan);
1909 1910
		ExecDropTupleTable(epqstate->es_tupleTable, true);
		epqstate->es_tupleTable = NULL;
1911
		heap_freetuple(epqstate->es_evTuple[epq->rti - 1]);
1912 1913
		epqstate->es_evTuple[epq->rti - 1] = NULL;
		/* pop old PQ from the stack */
B
Bruce Momjian 已提交
1914 1915
		oldepq = (evalPlanQual *) epqstate->es_evalPlanQual;
		if (oldepq == (evalPlanQual *) NULL)
1916
		{
1917 1918 1919
			epq->rti = 0;		/* this is the first (oldest) */
			estate->es_useEvalPlan = false;		/* PQ - mark as free and	  */
			return (NULL);		/* continue Query execution   */
1920 1921 1922 1923 1924 1925 1926 1927 1928 1929 1930 1931
		}
		Assert(oldepq->rti != 0);
		/* push current PQ to freePQ stack */
		oldepq->free = epq;
		epq = oldepq;
		epqstate = &(epq->estate);
		estate->es_evalPlanQual = (Pointer) epq;
		goto lpqnext;
	}

	return (slot);
}
1932 1933 1934 1935 1936 1937 1938 1939

static void
EndEvalPlanQual(EState *estate)
{
	evalPlanQual *epq = (evalPlanQual *) estate->es_evalPlanQual;
	EState	   *epqstate = &(epq->estate);
	evalPlanQual *oldepq;

1940 1941 1942
	if (epq->rti == 0)			/* plans already shutdowned */
	{
		Assert(epq->estate.es_evalPlanQual == NULL);
1943
		return;
1944
	}
1945 1946 1947

	for (;;)
	{
1948
		/* stop execution */
1949
		ExecEndNode(epq->plan, epq->plan);
1950 1951
		ExecDropTupleTable(epqstate->es_tupleTable, true);
		epqstate->es_tupleTable = NULL;
1952 1953 1954 1955 1956
		if (epqstate->es_evTuple[epq->rti - 1] != NULL)
		{
			heap_freetuple(epqstate->es_evTuple[epq->rti - 1]);
			epqstate->es_evTuple[epq->rti - 1] = NULL;
		}
1957 1958 1959 1960
		/* pop old PQ from the stack */
		oldepq = (evalPlanQual *) epqstate->es_evalPlanQual;
		if (oldepq == (evalPlanQual *) NULL)
		{
1961 1962
			epq->rti = 0;		/* this is the first (oldest) */
			estate->es_useEvalPlan = false;		/* PQ - mark as free */
1963 1964 1965 1966 1967 1968 1969 1970 1971 1972
			break;
		}
		Assert(oldepq->rti != 0);
		/* push current PQ to freePQ stack */
		oldepq->free = epq;
		epq = oldepq;
		epqstate = &(epq->estate);
		estate->es_evalPlanQual = (Pointer) epq;
	}
}