mirror of
https://github.com/postgres/postgres.git
synced 2025-06-11 20:28:21 +03:00
Refactor the executor's API to support data-modifying CTEs better.
The originally committed patch for modifying CTEs didn't interact well with EXPLAIN, as noted by myself, and also had corner-case problems with triggers, as noted by Dean Rasheed. Those problems show it is really not practical for ExecutorEnd to call any user-defined code; so split the cleanup duties out into a new function ExecutorFinish, which must be called between the last ExecutorRun call and ExecutorEnd. Some Asserts have been added to these functions to help verify correct usage. It is no longer necessary for callers of the executor to call AfterTriggerBeginQuery/AfterTriggerEndQuery for themselves, as this is now done by ExecutorStart/ExecutorFinish respectively. If you really need to suppress that and do it for yourself, pass EXEC_FLAG_SKIP_TRIGGERS to ExecutorStart. Also, refactor portal commit processing to allow for the possibility that PortalDrop will invoke user-defined code. I think this is not actually necessary just yet, since the portal-execution-strategy logic forces any non-pure-SELECT query to be run to completion before we will consider committing. But it seems like good future-proofing.
This commit is contained in:
@ -2041,6 +2041,7 @@ static int
|
||||
_SPI_pquery(QueryDesc *queryDesc, bool fire_triggers, long tcount)
|
||||
{
|
||||
int operation = queryDesc->operation;
|
||||
int eflags;
|
||||
int res;
|
||||
|
||||
switch (operation)
|
||||
@ -2084,10 +2085,13 @@ _SPI_pquery(QueryDesc *queryDesc, bool fire_triggers, long tcount)
|
||||
ResetUsage();
|
||||
#endif
|
||||
|
||||
/* Select execution options */
|
||||
if (fire_triggers)
|
||||
AfterTriggerBeginQuery();
|
||||
eflags = 0; /* default run-to-completion flags */
|
||||
else
|
||||
eflags = EXEC_FLAG_SKIP_TRIGGERS;
|
||||
|
||||
ExecutorStart(queryDesc, 0);
|
||||
ExecutorStart(queryDesc, eflags);
|
||||
|
||||
ExecutorRun(queryDesc, ForwardScanDirection, tcount);
|
||||
|
||||
@ -2101,10 +2105,7 @@ _SPI_pquery(QueryDesc *queryDesc, bool fire_triggers, long tcount)
|
||||
elog(ERROR, "consistency check on SPI tuple count failed");
|
||||
}
|
||||
|
||||
/* Take care of any queued AFTER triggers */
|
||||
if (fire_triggers)
|
||||
AfterTriggerEndQuery(queryDesc->estate);
|
||||
|
||||
ExecutorFinish(queryDesc);
|
||||
ExecutorEnd(queryDesc);
|
||||
/* FreeQueryDesc is done by the caller */
|
||||
|
||||
|
Reference in New Issue
Block a user