execUtils.c 28.6 KB
Newer Older
1 2
/*-------------------------------------------------------------------------
 *
3
 * execUtils.c
4
 *	  miscellaneous executor utility routines
5
 *
P
 
PostgreSQL Daemon 已提交
6
 * Portions Copyright (c) 1996-2005, PostgreSQL Global Development Group
B
Add:  
Bruce Momjian 已提交
7
 * Portions Copyright (c) 1994, Regents of the University of California
8 9 10
 *
 *
 * IDENTIFICATION
11
 *	  $PostgreSQL: pgsql/src/backend/executor/execUtils.c,v 1.121 2005/04/14 20:03:24 tgl Exp $
12 13 14 15 16
 *
 *-------------------------------------------------------------------------
 */
/*
 * INTERFACE ROUTINES
17 18 19 20
 *		CreateExecutorState		Create/delete executor working state
 *		FreeExecutorState
 *		CreateExprContext
 *		FreeExprContext
21
 *		ReScanExprContext
22
 *
23
 *		ExecAssignExprContext	Common code for plan node init routines.
24 25
 *		ExecAssignResultType
 *		etc
26
 *
27 28
 *		ExecOpenIndices			\
 *		ExecCloseIndices		 | referenced by InitPlan, EndPlan,
29
 *		ExecInsertIndexTuples	/  ExecInsert, ExecUpdate
30
 *
31 32 33
 *		RegisterExprContextCallback    Register function shutdown callback
 *		UnregisterExprContextCallback  Deregister function shutdown callback
 *
34 35 36
 *	 NOTES
 *		This file has traditionally been the place to stick misc.
 *		executor support stuff that doesn't really go anyplace else.
37 38
 */

39 40
#include "postgres.h"

41
#include "access/genam.h"
B
Bruce Momjian 已提交
42 43
#include "access/heapam.h"
#include "catalog/index.h"
H
Hiroshi Inoue 已提交
44
#include "catalog/catalog.h"
B
Bruce Momjian 已提交
45
#include "catalog/pg_index.h"
B
Bruce Momjian 已提交
46
#include "executor/execdebug.h"
H
Hiroshi Inoue 已提交
47
#include "miscadmin.h"
48 49
#include "utils/builtins.h"
#include "utils/fmgroids.h"
50
#include "utils/memutils.h"
51 52
#include "utils/relcache.h"
#include "utils/syscache.h"
53

54

55
/* ----------------------------------------------------------------
56 57
 *		global counters for number of tuples processed, retrieved,
 *		appended, replaced, deleted.
58 59
 * ----------------------------------------------------------------
 */
60 61 62 63 64 65 66
int			NTupleProcessed;
int			NTupleRetrieved;
int			NTupleReplaced;
int			NTupleAppended;
int			NTupleDeleted;
int			NIndexTupleInserted;
extern int	NIndexTupleProcessed;		/* have to be defined in the
67 68
										 * access method level so that the
										 * cinterface.a will link ok. */
69

70 71 72

static void ShutdownExprContext(ExprContext *econtext);

73

74
/* ----------------------------------------------------------------
75
 *						statistic functions
76 77 78 79
 * ----------------------------------------------------------------
 */

/* ----------------------------------------------------------------
80
 *		ResetTupleCount
81 82
 * ----------------------------------------------------------------
 */
83
#ifdef NOT_USED
84
void
85
ResetTupleCount(void)
86
{
87 88 89 90 91 92
	NTupleProcessed = 0;
	NTupleRetrieved = 0;
	NTupleAppended = 0;
	NTupleDeleted = 0;
	NTupleReplaced = 0;
	NIndexTupleProcessed = 0;
93
}
94
#endif
95 96

/* ----------------------------------------------------------------
97
 *		PrintTupleCount
98 99
 * ----------------------------------------------------------------
 */
100
#ifdef NOT_USED
101
void
102
DisplayTupleCount(FILE *statfp)
103
{
104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130
	if (NTupleProcessed > 0)
		fprintf(statfp, "!\t%d tuple%s processed, ", NTupleProcessed,
				(NTupleProcessed == 1) ? "" : "s");
	else
	{
		fprintf(statfp, "!\tno tuples processed.\n");
		return;
	}
	if (NIndexTupleProcessed > 0)
		fprintf(statfp, "%d indextuple%s processed, ", NIndexTupleProcessed,
				(NIndexTupleProcessed == 1) ? "" : "s");
	if (NIndexTupleInserted > 0)
		fprintf(statfp, "%d indextuple%s inserted, ", NIndexTupleInserted,
				(NIndexTupleInserted == 1) ? "" : "s");
	if (NTupleRetrieved > 0)
		fprintf(statfp, "%d tuple%s retrieved. ", NTupleRetrieved,
				(NTupleRetrieved == 1) ? "" : "s");
	if (NTupleAppended > 0)
		fprintf(statfp, "%d tuple%s appended. ", NTupleAppended,
				(NTupleAppended == 1) ? "" : "s");
	if (NTupleDeleted > 0)
		fprintf(statfp, "%d tuple%s deleted. ", NTupleDeleted,
				(NTupleDeleted == 1) ? "" : "s");
	if (NTupleReplaced > 0)
		fprintf(statfp, "%d tuple%s replaced. ", NTupleReplaced,
				(NTupleReplaced == 1) ? "" : "s");
	fprintf(statfp, "\n");
131
}
132
#endif
133

134

135
/* ----------------------------------------------------------------
136
 *				 Executor state and memory management functions
137 138 139 140
 * ----------------------------------------------------------------
 */

/* ----------------
141
 *		CreateExecutorState
142
 *
143 144
 *		Create and initialize an EState node, which is the root of
 *		working storage for an entire Executor invocation.
145
 *
146 147 148 149
 * Principally, this creates the per-query memory context that will be
 * used to hold all working data that lives till the end of the query.
 * Note that the per-query context will become a child of the caller's
 * CurrentMemoryContext.
150 151
 * ----------------
 */
152 153
EState *
CreateExecutorState(void)
154
{
155 156 157
	EState	   *estate;
	MemoryContext qcontext;
	MemoryContext oldcontext;
158

159 160 161 162 163 164 165 166
	/*
	 * Create the per-query context for this Executor run.
	 */
	qcontext = AllocSetContextCreate(CurrentMemoryContext,
									 "ExecutorState",
									 ALLOCSET_DEFAULT_MINSIZE,
									 ALLOCSET_DEFAULT_INITSIZE,
									 ALLOCSET_DEFAULT_MAXSIZE);
B
Bruce Momjian 已提交
167

168
	/*
B
Bruce Momjian 已提交
169 170
	 * Make the EState node within the per-query context.  This way, we
	 * don't need a separate pfree() operation for it at shutdown.
171
	 */
172 173 174 175 176 177 178 179 180
	oldcontext = MemoryContextSwitchTo(qcontext);

	estate = makeNode(EState);

	/*
	 * Initialize all fields of the Executor State structure
	 */
	estate->es_direction = ForwardScanDirection;
	estate->es_snapshot = SnapshotNow;
181
	estate->es_crosscheck_snapshot = InvalidSnapshot;	/* no crosscheck */
182 183 184 185 186 187 188 189 190 191 192 193 194 195 196
	estate->es_range_table = NIL;

	estate->es_result_relations = NULL;
	estate->es_num_result_relations = 0;
	estate->es_result_relation_info = NULL;

	estate->es_junkFilter = NULL;
	estate->es_into_relation_descriptor = NULL;

	estate->es_param_list_info = NULL;
	estate->es_param_exec_vals = NULL;

	estate->es_query_cxt = qcontext;

	estate->es_tupleTable = NULL;
197

198 199 200 201 202
	estate->es_processed = 0;
	estate->es_lastoid = InvalidOid;
	estate->es_rowMark = NIL;

	estate->es_instrument = false;
203 204
	estate->es_select_into = false;
	estate->es_into_oids = false;
205 206 207 208 209

	estate->es_exprcontexts = NIL;

	estate->es_per_tuple_exprcontext = NULL;

210
	estate->es_topPlan = NULL;
211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249
	estate->es_evalPlanQual = NULL;
	estate->es_evTupleNull = NULL;
	estate->es_evTuple = NULL;
	estate->es_useEvalPlan = false;

	/*
	 * Return the executor state structure
	 */
	MemoryContextSwitchTo(oldcontext);

	return estate;
}

/* ----------------
 *		FreeExecutorState
 *
 *		Release an EState along with all remaining working storage.
 *
 * Note: this is not responsible for releasing non-memory resources,
 * such as open relations or buffer pins.  But it will shut down any
 * still-active ExprContexts within the EState.  That is sufficient
 * cleanup for situations where the EState has only been used for expression
 * evaluation, and not to run a complete Plan.
 *
 * This can be called in any memory context ... so long as it's not one
 * of the ones to be freed.
 * ----------------
 */
void
FreeExecutorState(EState *estate)
{
	/*
	 * Shut down and free any remaining ExprContexts.  We do this
	 * explicitly to ensure that any remaining shutdown callbacks get
	 * called (since they might need to release resources that aren't
	 * simply memory within the per-query memory context).
	 */
	while (estate->es_exprcontexts)
	{
B
Bruce Momjian 已提交
250 251
		/*
		 * XXX: seems there ought to be a faster way to implement this
252
		 * than repeated list_delete(), no?
253 254
		 */
		FreeExprContext((ExprContext *) linitial(estate->es_exprcontexts));
255 256
		/* FreeExprContext removed the list link for us */
	}
B
Bruce Momjian 已提交
257

258 259 260 261 262
	/*
	 * Free the per-query memory context, thereby releasing all working
	 * memory, including the EState node itself.
	 */
	MemoryContextDelete(estate->es_query_cxt);
263 264 265
}

/* ----------------
266 267 268 269 270 271 272 273
 *		CreateExprContext
 *
 *		Create a context for expression evaluation within an EState.
 *
 * An executor run may require multiple ExprContexts (we usually make one
 * for each Plan node, and a separate one for per-output-tuple processing
 * such as constraint checking).  Each ExprContext has its own "per-tuple"
 * memory context.
274
 *
275
 * Note we make no assumption about the caller's memory context.
276 277
 * ----------------
 */
278
ExprContext *
279
CreateExprContext(EState *estate)
280
{
281 282
	ExprContext *econtext;
	MemoryContext oldcontext;
283

284 285 286 287 288 289 290
	/* Create the ExprContext node within the per-query memory context */
	oldcontext = MemoryContextSwitchTo(estate->es_query_cxt);

	econtext = makeNode(ExprContext);

	/* Initialize fields of ExprContext */
	econtext->ecxt_scantuple = NULL;
291 292
	econtext->ecxt_innertuple = NULL;
	econtext->ecxt_outertuple = NULL;
293 294

	econtext->ecxt_per_query_memory = estate->es_query_cxt;
B
Bruce Momjian 已提交
295

296
	/*
297
	 * Create working memory for expression evaluation in this context.
298 299
	 */
	econtext->ecxt_per_tuple_memory =
300 301 302
		AllocSetContextCreate(estate->es_query_cxt,
							  "ExprContext",
							  ALLOCSET_DEFAULT_MINSIZE,
303 304
							  ALLOCSET_DEFAULT_INITSIZE,
							  ALLOCSET_DEFAULT_MAXSIZE);
305 306 307 308

	econtext->ecxt_param_exec_vals = estate->es_param_exec_vals;
	econtext->ecxt_param_list_info = estate->es_param_list_info;

309 310
	econtext->ecxt_aggvalues = NULL;
	econtext->ecxt_aggnulls = NULL;
311

312 313 314
	econtext->caseValue_datum = (Datum) 0;
	econtext->caseValue_isNull = true;

315 316 317 318 319
	econtext->domainValue_datum = (Datum) 0;
	econtext->domainValue_isNull = true;

	econtext->ecxt_estate = estate;

320
	econtext->ecxt_callbacks = NULL;
321

322
	/*
B
Bruce Momjian 已提交
323 324 325 326
	 * Link the ExprContext into the EState to ensure it is shut down when
	 * the EState is freed.  Because we use lcons(), shutdowns will occur
	 * in reverse order of creation, which may not be essential but can't
	 * hurt.
327 328 329 330 331
	 */
	estate->es_exprcontexts = lcons(econtext, estate->es_exprcontexts);

	MemoryContextSwitchTo(oldcontext);

332 333
	return econtext;
}
334

335 336 337 338 339 340 341 342 343 344 345
/* ----------------
 *		FreeExprContext
 *
 *		Free an expression context, including calling any remaining
 *		shutdown callbacks.
 *
 * Since we free the temporary context used for expression evaluation,
 * any previously computed pass-by-reference expression result will go away!
 *
 * Note we make no assumption about the caller's memory context.
 * ----------------
346 347 348 349
 */
void
FreeExprContext(ExprContext *econtext)
{
350 351
	EState	   *estate;

352 353 354
	/* Call any registered callbacks */
	ShutdownExprContext(econtext);
	/* And clean up the memory used */
355
	MemoryContextDelete(econtext->ecxt_per_tuple_memory);
356 357
	/* Unlink self from owning EState */
	estate = econtext->ecxt_estate;
358
	estate->es_exprcontexts = list_delete_ptr(estate->es_exprcontexts, econtext);
359
	/* And delete the ExprContext node */
360
	pfree(econtext);
361 362
}

363 364 365 366
/*
 * ReScanExprContext
 *
 *		Reset an expression context in preparation for a rescan of its
B
Bruce Momjian 已提交
367
 *		plan node.	This requires calling any registered shutdown callbacks,
368 369 370 371 372 373 374 375 376 377 378 379 380
 *		since any partially complete set-returning-functions must be canceled.
 *
 * Note we make no assumption about the caller's memory context.
 */
void
ReScanExprContext(ExprContext *econtext)
{
	/* Call any registered callbacks */
	ShutdownExprContext(econtext);
	/* And clean up the memory used */
	MemoryContextReset(econtext->ecxt_per_tuple_memory);
}

381 382 383
/*
 * Build a per-output-tuple ExprContext for an EState.
 *
384 385
 * This is normally invoked via GetPerTupleExprContext() macro,
 * not directly.
386 387 388 389 390
 */
ExprContext *
MakePerTupleExprContext(EState *estate)
{
	if (estate->es_per_tuple_exprcontext == NULL)
391
		estate->es_per_tuple_exprcontext = CreateExprContext(estate);
392 393 394 395

	return estate->es_per_tuple_exprcontext;
}

396

397
/* ----------------------------------------------------------------
398 399 400 401
 *				 miscellaneous node-init support functions
 *
 * Note: all of these are expected to be called with CurrentMemoryContext
 * equal to the per-query memory context.
402 403 404
 * ----------------------------------------------------------------
 */

405 406 407
/* ----------------
 *		ExecAssignExprContext
 *
B
Bruce Momjian 已提交
408
 *		This initializes the ps_ExprContext field.	It is only necessary
409
 *		to do this for nodes which use ExecQual or ExecProject
B
Bruce Momjian 已提交
410
 *		because those routines require an econtext. Other nodes that
411 412 413 414
 *		don't have to evaluate expressions don't need to do this.
 * ----------------
 */
void
415
ExecAssignExprContext(EState *estate, PlanState *planstate)
416 417 418 419
{
	planstate->ps_ExprContext = CreateExprContext(estate);
}

420
/* ----------------
421
 *		ExecAssignResultType
422 423 424
 * ----------------
 */
void
425
ExecAssignResultType(PlanState *planstate,
426
					 TupleDesc tupDesc, bool shouldFree)
427
{
428
	TupleTableSlot *slot = planstate->ps_ResultTupleSlot;
429

430
	ExecSetSlotDescriptor(slot, tupDesc, shouldFree);
431 432 433
}

/* ----------------
434
 *		ExecAssignResultTypeFromOuterPlan
435 436 437
 * ----------------
 */
void
438
ExecAssignResultTypeFromOuterPlan(PlanState *planstate)
439
{
440
	PlanState  *outerPlan;
441
	TupleDesc	tupDesc;
442

443
	outerPlan = outerPlanState(planstate);
444
	tupDesc = ExecGetResultType(outerPlan);
445

446
	ExecAssignResultType(planstate, tupDesc, false);
447 448 449
}

/* ----------------
450
 *		ExecAssignResultTypeFromTL
451 452 453
 * ----------------
 */
void
454
ExecAssignResultTypeFromTL(PlanState *planstate)
455
{
456
	bool		hasoid;
457
	TupleDesc	tupDesc;
458

459 460 461 462
	if (ExecContextForcesOids(planstate, &hasoid))
	{
		/* context forces OID choice; hasoid is now set correctly */
	}
463
	else
464
	{
465 466
		/* given free choice, don't leave space for OIDs in result tuples */
		hasoid = false;
467
	}
B
Bruce Momjian 已提交
468

469
	/*
B
Bruce Momjian 已提交
470 471 472
	 * ExecTypeFromTL needs the parse-time representation of the tlist,
	 * not a list of ExprStates.  This is good because some plan nodes
	 * don't bother to set up planstate->targetlist ...
473 474 475
	 */
	tupDesc = ExecTypeFromTL(planstate->plan->targetlist, hasoid);
	ExecAssignResultType(planstate, tupDesc, true);
476 477 478
}

/* ----------------
479
 *		ExecGetResultType
480 481 482
 * ----------------
 */
TupleDesc
483
ExecGetResultType(PlanState *planstate)
484
{
485
	TupleTableSlot *slot = planstate->ps_ResultTupleSlot;
486

487
	return slot->tts_tupleDescriptor;
488 489 490
}

/* ----------------
491 492 493 494 495 496
 *		ExecBuildProjectionInfo
 *
 * Build a ProjectionInfo node for evaluating the given tlist in the given
 * econtext, and storing the result into the tuple slot.  (Caller must have
 * ensured that tuple slot has a descriptor matching the tlist!)  Note that
 * the given tlist should be a list of ExprState nodes, not Expr nodes.
497 498
 * ----------------
 */
499 500 501 502
ProjectionInfo *
ExecBuildProjectionInfo(List *targetList,
						ExprContext *econtext,
						TupleTableSlot *slot)
503
{
504
	ProjectionInfo *projInfo = makeNode(ProjectionInfo);
505
	int			len;
506 507
	bool		isVarList;
	ListCell   *tl;
508 509 510 511

	len = ExecTargetListLength(targetList);

	projInfo->pi_targetlist = targetList;
512 513
	projInfo->pi_exprContext = econtext;
	projInfo->pi_slot = slot;
514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562

	/*
	 * Determine whether the target list consists entirely of simple Var
	 * references (ie, references to non-system attributes).  If so,
	 * we can use the simpler ExecVariableList instead of ExecTargetList.
	 */
	isVarList = true;
	foreach(tl, targetList)
	{
		GenericExprState *gstate = (GenericExprState *) lfirst(tl);
		Var		   *variable = (Var *) gstate->arg->expr;

		if (variable == NULL ||
			!IsA(variable, Var) ||
			variable->varattno <= 0)
		{
			isVarList = false;
			break;
		}
	}
	projInfo->pi_isVarList = isVarList;

	if (isVarList)
	{
		int		   *varSlotOffsets;
		int		   *varNumbers;
		AttrNumber	lastInnerVar = 0;
		AttrNumber	lastOuterVar = 0;
		AttrNumber	lastScanVar = 0;

		projInfo->pi_itemIsDone = NULL;	/* not needed */
		projInfo->pi_varSlotOffsets = varSlotOffsets = (int *)
			palloc0(len * sizeof(int));
		projInfo->pi_varNumbers = varNumbers = (int *)
			palloc0(len * sizeof(int));

		/*
		 * Set up the data needed by ExecVariableList.  The slots in which
		 * the variables can be found at runtime are denoted by the offsets
		 * of their slot pointers within the econtext.  This rather grotty
		 * representation is needed because the caller may not have given
		 * us the real econtext yet (see hacks in nodeSubplan.c).
		 */
		foreach(tl, targetList)
		{
			GenericExprState *gstate = (GenericExprState *) lfirst(tl);
			Var		   *variable = (Var *) gstate->arg->expr;
			AttrNumber	attnum = variable->varattno;
			TargetEntry *tle = (TargetEntry *) gstate->xprstate.expr;
563
			AttrNumber	resind = tle->resno - 1;
564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593

			Assert(resind >= 0 && resind < len);
			varNumbers[resind] = attnum;

			switch (variable->varno)
			{
				case INNER:
					varSlotOffsets[resind] = offsetof(ExprContext,
													  ecxt_innertuple);
					lastInnerVar = Max(lastInnerVar, attnum);
					break;

				case OUTER:
					varSlotOffsets[resind] = offsetof(ExprContext,
													  ecxt_outertuple);
					lastOuterVar = Max(lastOuterVar, attnum);
					break;

				default:
					varSlotOffsets[resind] = offsetof(ExprContext,
													  ecxt_scantuple);
					lastScanVar = Max(lastScanVar, attnum);
					break;
			}
		}
		projInfo->pi_lastInnerVar = lastInnerVar;
		projInfo->pi_lastOuterVar = lastOuterVar;
		projInfo->pi_lastScanVar = lastScanVar;
	}
	else
594
	{
595 596 597 598
		projInfo->pi_itemIsDone = (ExprDoneCond *)
			palloc(len * sizeof(ExprDoneCond));
		projInfo->pi_varSlotOffsets = NULL;
		projInfo->pi_varNumbers = NULL;
599 600 601 602
	}

	return projInfo;
}
603

604 605 606 607 608 609 610
/* ----------------
 *		ExecAssignProjectionInfo
 *
 * forms the projection information from the node's targetlist
 * ----------------
 */
void
611
ExecAssignProjectionInfo(PlanState *planstate)
612 613 614 615 616
{
	planstate->ps_ProjInfo =
		ExecBuildProjectionInfo(planstate->targetlist,
								planstate->ps_ExprContext,
								planstate->ps_ResultTupleSlot);
617 618 619
}


620 621
/* ----------------
 *		ExecFreeExprContext
622 623 624 625 626
 *
 * A plan node's ExprContext should be freed explicitly during ExecEndNode
 * because there may be shutdown callbacks to call.  (Other resources made
 * by the above routines, such as projection info, don't need to be freed
 * explicitly because they're just memory in the per-query memory context.)
627 628 629
 * ----------------
 */
void
630
ExecFreeExprContext(PlanState *planstate)
631 632 633
{
	ExprContext *econtext;

634 635 636
	/*
	 * get expression context.	if NULL then this node has none so we just
	 * return.
637
	 */
638
	econtext = planstate->ps_ExprContext;
639 640 641
	if (econtext == NULL)
		return;

642
	FreeExprContext(econtext);
643

644
	planstate->ps_ExprContext = NULL;
645 646
}

647
/* ----------------------------------------------------------------
648 649 650 651 652 653
 *		the following scan type support functions are for
 *		those nodes which are stubborn and return tuples in
 *		their Scan tuple slot instead of their Result tuple
 *		slot..	luck fur us, these nodes do not do projections
 *		so we don't have to worry about getting the ProjectionInfo
 *		right for them...  -cim 6/3/91
654 655 656 657
 * ----------------------------------------------------------------
 */

/* ----------------
658
 *		ExecGetScanType
659 660 661
 * ----------------
 */
TupleDesc
662
ExecGetScanType(ScanState *scanstate)
663
{
664
	TupleTableSlot *slot = scanstate->ss_ScanTupleSlot;
665

666
	return slot->tts_tupleDescriptor;
667 668 669
}

/* ----------------
670
 *		ExecAssignScanType
671 672 673
 * ----------------
 */
void
674
ExecAssignScanType(ScanState *scanstate,
675
				   TupleDesc tupDesc, bool shouldFree)
676
{
677
	TupleTableSlot *slot = scanstate->ss_ScanTupleSlot;
678

679
	ExecSetSlotDescriptor(slot, tupDesc, shouldFree);
680 681 682
}

/* ----------------
683
 *		ExecAssignScanTypeFromOuterPlan
684 685 686
 * ----------------
 */
void
687
ExecAssignScanTypeFromOuterPlan(ScanState *scanstate)
688
{
689
	PlanState  *outerPlan;
690
	TupleDesc	tupDesc;
691

692
	outerPlan = outerPlanState(scanstate);
693
	tupDesc = ExecGetResultType(outerPlan);
694

695
	ExecAssignScanType(scanstate, tupDesc, false);
696 697 698 699
}


/* ----------------------------------------------------------------
700
 *				  ExecInsertIndexTuples support
701 702
 * ----------------------------------------------------------------
 */
703 704 705 706

/* ----------------------------------------------------------------
 *		ExecOpenIndices
 *
707
 *		Find the indices associated with a result relation, open them,
708
 *		and save information about them in the result ResultRelInfo.
709
 *
710
 *		At entry, caller has already opened and locked
711
 *		resultRelInfo->ri_RelationDesc.
712
 *
713
 *		This used to be horribly ugly code, and slow too because it
B
Bruce Momjian 已提交
714
 *		did a sequential scan of pg_index.	Now we rely on the relcache
715 716 717
 *		to cache a list of the OIDs of the indices associated with any
 *		specific relation, and we use the pg_index syscache to get the
 *		entries we need from pg_index.
718 719 720
 * ----------------------------------------------------------------
 */
void
721
ExecOpenIndices(ResultRelInfo *resultRelInfo)
722
{
723
	Relation	resultRelation = resultRelInfo->ri_RelationDesc;
724 725
	List	   *indexoidlist;
	ListCell   *l;
726 727
	int			len,
				i;
728 729
	RelationPtr relationDescs;
	IndexInfo **indexInfoArray;
730

731
	resultRelInfo->ri_NumIndices = 0;
732

733
	/* fast path if no indexes */
B
Bruce Momjian 已提交
734
	if (!RelationGetForm(resultRelation)->relhasindex)
H
Hiroshi Inoue 已提交
735
		return;
736

737 738
	/*
	 * Get cached list of index OIDs
739
	 */
740
	indexoidlist = RelationGetIndexList(resultRelation);
741
	len = list_length(indexoidlist);
742 743
	if (len == 0)
		return;
744

745 746
	/*
	 * allocate space for result arrays
747
	 */
748 749 750
	relationDescs = (RelationPtr) palloc(len * sizeof(Relation));
	indexInfoArray = (IndexInfo **) palloc(len * sizeof(IndexInfo *));

751 752 753
	resultRelInfo->ri_NumIndices = len;
	resultRelInfo->ri_IndexRelationDescs = relationDescs;
	resultRelInfo->ri_IndexRelationInfo = indexInfoArray;
754

755 756
	/*
	 * For each index, open the index relation and save pg_index info.
757
	 */
758
	i = 0;
759
	foreach(l, indexoidlist)
760
	{
761
		Oid			indexOid = lfirst_oid(l);
762 763
		Relation	indexDesc;
		IndexInfo  *ii;
764

765
		/*
766 767 768 769 770 771
		 * Open and lock the index relation
		 *
		 * If the index AM supports concurrent updates, obtain RowExclusiveLock
		 * to signify that we are updating the index.  This locks out only
		 * operations that need exclusive access, such as relocating the index
		 * to a new tablespace.
772
		 *
773 774
		 * If the index AM is not safe for concurrent updates, obtain an
		 * exclusive lock on the index to lock out other updaters as well
775
		 * as readers (index_beginscan places AccessShareLock).
776 777
		 *
		 * If there are multiple not-concurrent-safe indexes, all backends
778 779 780 781 782
		 * must lock the indexes in the same order or we will get deadlocks
		 * here.  This is guaranteed by RelationGetIndexList(), which promises
		 * to return the index list in OID order.
		 *
		 * The locks will be released in ExecCloseIndices.
783
		 */
784 785
		indexDesc = index_open(indexOid);

786 787 788
		if (indexDesc->rd_am->amconcurrent)
			LockRelation(indexDesc, RowExclusiveLock);
		else
789
			LockRelation(indexDesc, AccessExclusiveLock);
790

791 792
		/* extract index key information from the index's pg_index info */
		ii = BuildIndexInfo(indexDesc);
793

794 795 796
		relationDescs[i] = indexDesc;
		indexInfoArray[i] = ii;
		i++;
797
	}
798

799
	list_free(indexoidlist);
800 801 802
}

/* ----------------------------------------------------------------
803
 *		ExecCloseIndices
804
 *
805
 *		Close the index relations stored in resultRelInfo
806 807 808
 * ----------------------------------------------------------------
 */
void
809
ExecCloseIndices(ResultRelInfo *resultRelInfo)
810
{
811 812
	int			i;
	int			numIndices;
813
	RelationPtr indexDescs;
814

815
	numIndices = resultRelInfo->ri_NumIndices;
816
	indexDescs = resultRelInfo->ri_IndexRelationDescs;
817 818

	for (i = 0; i < numIndices; i++)
V
Vadim B. Mikheev 已提交
819
	{
820
		if (indexDescs[i] == NULL)
821
			continue;			/* shouldn't happen? */
B
Bruce Momjian 已提交
822

823 824 825 826
		/* Drop lock acquired by ExecOpenIndices */
		if (indexDescs[i]->rd_am->amconcurrent)
			UnlockRelation(indexDescs[i], RowExclusiveLock);
		else
827
			UnlockRelation(indexDescs[i], AccessExclusiveLock);
828

829
		index_close(indexDescs[i]);
V
Vadim B. Mikheev 已提交
830
	}
B
Bruce Momjian 已提交
831

832
	/*
833 834
	 * XXX should free indexInfo array here too?  Currently we assume that
	 * such stuff will be cleaned up automatically in FreeExecutorState.
835
	 */
836 837 838
}

/* ----------------------------------------------------------------
839
 *		ExecInsertIndexTuples
840
 *
841 842 843 844 845 846 847
 *		This routine takes care of inserting index tuples
 *		into all the relations indexing the result relation
 *		when a heap tuple is inserted into the result relation.
 *		Much of this code should be moved into the genam
 *		stuff as it only exists here because the genam stuff
 *		doesn't provide the functionality needed by the
 *		executor.. -cim 9/27/89
848 849 850
 * ----------------------------------------------------------------
 */
void
851
ExecInsertIndexTuples(TupleTableSlot *slot,
852
					  ItemPointer tupleid,
853
					  EState *estate,
854
					  bool is_vacuum)
855
{
856
	ResultRelInfo *resultRelInfo;
857 858 859 860 861 862
	int			i;
	int			numIndices;
	RelationPtr relationDescs;
	Relation	heapRelation;
	IndexInfo **indexInfoArray;
	ExprContext *econtext;
863 864
	Datum		values[INDEX_MAX_KEYS];
	bool		isnull[INDEX_MAX_KEYS];
865

866 867
	/*
	 * Get information from the result relation info structure.
868
	 */
869 870 871 872 873
	resultRelInfo = estate->es_result_relation_info;
	numIndices = resultRelInfo->ri_NumIndices;
	relationDescs = resultRelInfo->ri_IndexRelationDescs;
	indexInfoArray = resultRelInfo->ri_IndexRelationInfo;
	heapRelation = resultRelInfo->ri_RelationDesc;
874

875
	/*
B
Bruce Momjian 已提交
876
	 * We will use the EState's per-tuple context for evaluating
B
Bruce Momjian 已提交
877 878
	 * predicates and index expressions (creating it if it's not already
	 * there).
879
	 */
880
	econtext = GetPerTupleExprContext(estate);
881 882 883

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

885 886
	/*
	 * for each index, form and insert the index tuple
887
	 */
888 889
	for (i = 0; i < numIndices; i++)
	{
890 891
		IndexInfo  *indexInfo;

892 893 894 895
		if (relationDescs[i] == NULL)
			continue;

		indexInfo = indexInfoArray[i];
896 897 898

		/* Check for partial index */
		if (indexInfo->ii_Predicate != NIL)
899
		{
900 901 902 903 904 905 906 907 908 909 910 911 912 913 914
			List	   *predicate;

			/*
			 * If predicate state not set up yet, create it (in the
			 * estate's per-query context)
			 */
			predicate = indexInfo->ii_PredicateState;
			if (predicate == NIL)
			{
				predicate = (List *)
					ExecPrepareExpr((Expr *) indexInfo->ii_Predicate,
									estate);
				indexInfo->ii_PredicateState = predicate;
			}

915
			/* Skip this index-update if the predicate isn't satisfied */
916
			if (!ExecQual(predicate, econtext, false))
917 918 919
				continue;
		}

920
		/*
921 922
		 * FormIndexDatum fills in its values and isnull parameters with
		 * the appropriate values for the column(s) of the index.
923
		 */
924
		FormIndexDatum(indexInfo,
925
					   slot,
926
					   estate,
927 928
					   values,
					   isnull);
929

930
		/*
B
Bruce Momjian 已提交
931
		 * The index AM does the rest.	Note we suppress unique-index
932 933 934
		 * checks if we are being called from VACUUM, since VACUUM may
		 * need to move dead tuples that have the same keys as live ones.
		 */
935 936 937 938 939 940
		index_insert(relationDescs[i],	/* index relation */
					 values,			/* array of index Datums */
					 isnull,			/* null flags */
					 tupleid,			/* tid of heap tuple */
					 heapRelation,
					 relationDescs[i]->rd_index->indisunique && !is_vacuum);
941

942 943
		/*
		 * keep track of index inserts for debugging
944 945 946
		 */
		IncrIndexInserted();
	}
947
}
V
Vadim B. Mikheev 已提交
948

949 950 951 952
/*
 * UpdateChangedParamSet
 *		Add changed parameters to a plan node's chgParam set
 */
953
void
954
UpdateChangedParamSet(PlanState *node, Bitmapset *newchg)
V
Vadim B. Mikheev 已提交
955
{
956
	Bitmapset  *parmset;
957

958 959 960 961 962
	/*
	 * The plan node only depends on params listed in its allParam set.
	 * Don't include anything else into its chgParam set.
	 */
	parmset = bms_intersect(node->plan->allParam, newchg);
B
Bruce Momjian 已提交
963

964 965 966 967 968 969 970 971
	/*
	 * Keep node->chgParam == NULL if there's not actually any members;
	 * this allows the simplest possible tests in executor node files.
	 */
	if (!bms_is_empty(parmset))
		node->chgParam = bms_join(node->chgParam, parmset);
	else
		bms_free(parmset);
V
Vadim B. Mikheev 已提交
972
}
973 974 975 976 977 978 979 980 981 982 983 984 985 986 987 988

/*
 * Register a shutdown callback in an ExprContext.
 *
 * Shutdown callbacks will be called (in reverse order of registration)
 * when the ExprContext is deleted or rescanned.  This provides a hook
 * for functions called in the context to do any cleanup needed --- it's
 * particularly useful for functions returning sets.  Note that the
 * callback will *not* be called in the event that execution is aborted
 * by an error.
 */
void
RegisterExprContextCallback(ExprContext *econtext,
							ExprContextCallbackFunction function,
							Datum arg)
{
B
Bruce Momjian 已提交
989
	ExprContext_CB *ecxt_callback;
990 991 992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014

	/* Save the info in appropriate memory context */
	ecxt_callback = (ExprContext_CB *)
		MemoryContextAlloc(econtext->ecxt_per_query_memory,
						   sizeof(ExprContext_CB));

	ecxt_callback->function = function;
	ecxt_callback->arg = arg;

	/* link to front of list for appropriate execution order */
	ecxt_callback->next = econtext->ecxt_callbacks;
	econtext->ecxt_callbacks = ecxt_callback;
}

/*
 * Deregister a shutdown callback in an ExprContext.
 *
 * Any list entries matching the function and arg will be removed.
 * This can be used if it's no longer necessary to call the callback.
 */
void
UnregisterExprContextCallback(ExprContext *econtext,
							  ExprContextCallbackFunction function,
							  Datum arg)
{
B
Bruce Momjian 已提交
1015 1016
	ExprContext_CB **prev_callback;
	ExprContext_CB *ecxt_callback;
1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040

	prev_callback = &econtext->ecxt_callbacks;

	while ((ecxt_callback = *prev_callback) != NULL)
	{
		if (ecxt_callback->function == function && ecxt_callback->arg == arg)
		{
			*prev_callback = ecxt_callback->next;
			pfree(ecxt_callback);
		}
		else
			prev_callback = &ecxt_callback->next;
	}
}

/*
 * Call all the shutdown callbacks registered in an ExprContext.
 *
 * The callback list is emptied (important in case this is only a rescan
 * reset, and not deletion of the ExprContext).
 */
static void
ShutdownExprContext(ExprContext *econtext)
{
B
Bruce Momjian 已提交
1041
	ExprContext_CB *ecxt_callback;
1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052
	MemoryContext oldcontext;

	/* Fast path in normal case where there's nothing to do. */
	if (econtext->ecxt_callbacks == NULL)
		return;

	/*
	 * Call the callbacks in econtext's per-tuple context.  This ensures
	 * that any memory they might leak will get cleaned up.
	 */
	oldcontext = MemoryContextSwitchTo(econtext->ecxt_per_tuple_memory);
1053 1054 1055 1056 1057 1058 1059 1060 1061 1062

	/*
	 * Call each callback function in reverse registration order.
	 */
	while ((ecxt_callback = econtext->ecxt_callbacks) != NULL)
	{
		econtext->ecxt_callbacks = ecxt_callback->next;
		(*ecxt_callback->function) (ecxt_callback->arg);
		pfree(ecxt_callback);
	}
1063 1064

	MemoryContextSwitchTo(oldcontext);
1065
}