mirror of
https://github.com/postgres/postgres.git
synced 2025-06-30 21:42:05 +03:00
Represent columns requiring insert and update privileges indentently.
Previously, relation range table entries used a single Bitmapset field representing which columns required either UPDATE or INSERT privileges, despite the fact that INSERT and UPDATE privileges are separately cataloged, and may be independently held. As statements so far required either insert or update privileges but never both, that was sufficient. The required permission could be inferred from the top level statement run. The upcoming INSERT ... ON CONFLICT UPDATE feature needs to independently check for both privileges in one statement though, so that is not sufficient anymore. Bumps catversion as stored rules change. Author: Peter Geoghegan Reviewed-By: Andres Freund
This commit is contained in:
@ -66,13 +66,13 @@ int SessionReplicationRole = SESSION_REPLICATION_ROLE_ORIGIN;
|
||||
static int MyTriggerDepth = 0;
|
||||
|
||||
/*
|
||||
* Note that this macro also exists in executor/execMain.c. There does not
|
||||
* appear to be any good header to put it into, given the structures that
|
||||
* it uses, so we let them be duplicated. Be sure to update both if one needs
|
||||
* to be changed, however.
|
||||
* Note that similar macros also exists in executor/execMain.c. There does not
|
||||
* appear to be any good header to put it into, given the structures that it
|
||||
* uses, so we let them be duplicated. Be sure to update both if one needs to
|
||||
* be changed, however.
|
||||
*/
|
||||
#define GetModifiedColumns(relinfo, estate) \
|
||||
(rt_fetch((relinfo)->ri_RangeTableIndex, (estate)->es_range_table)->modifiedCols)
|
||||
#define GetUpdatedColumns(relinfo, estate) \
|
||||
(rt_fetch((relinfo)->ri_RangeTableIndex, (estate)->es_range_table)->updatedCols)
|
||||
|
||||
/* Local function prototypes */
|
||||
static void ConvertTriggerToFK(CreateTrigStmt *stmt, Oid funcoid);
|
||||
@ -2347,7 +2347,7 @@ ExecBSUpdateTriggers(EState *estate, ResultRelInfo *relinfo)
|
||||
TriggerDesc *trigdesc;
|
||||
int i;
|
||||
TriggerData LocTriggerData;
|
||||
Bitmapset *modifiedCols;
|
||||
Bitmapset *updatedCols;
|
||||
|
||||
trigdesc = relinfo->ri_TrigDesc;
|
||||
|
||||
@ -2356,7 +2356,7 @@ ExecBSUpdateTriggers(EState *estate, ResultRelInfo *relinfo)
|
||||
if (!trigdesc->trig_update_before_statement)
|
||||
return;
|
||||
|
||||
modifiedCols = GetModifiedColumns(relinfo, estate);
|
||||
updatedCols = GetUpdatedColumns(relinfo, estate);
|
||||
|
||||
LocTriggerData.type = T_TriggerData;
|
||||
LocTriggerData.tg_event = TRIGGER_EVENT_UPDATE |
|
||||
@ -2377,7 +2377,7 @@ ExecBSUpdateTriggers(EState *estate, ResultRelInfo *relinfo)
|
||||
TRIGGER_TYPE_UPDATE))
|
||||
continue;
|
||||
if (!TriggerEnabled(estate, relinfo, trigger, LocTriggerData.tg_event,
|
||||
modifiedCols, NULL, NULL))
|
||||
updatedCols, NULL, NULL))
|
||||
continue;
|
||||
|
||||
LocTriggerData.tg_trigger = trigger;
|
||||
@ -2402,7 +2402,7 @@ ExecASUpdateTriggers(EState *estate, ResultRelInfo *relinfo)
|
||||
if (trigdesc && trigdesc->trig_update_after_statement)
|
||||
AfterTriggerSaveEvent(estate, relinfo, TRIGGER_EVENT_UPDATE,
|
||||
false, NULL, NULL, NIL,
|
||||
GetModifiedColumns(relinfo, estate));
|
||||
GetUpdatedColumns(relinfo, estate));
|
||||
}
|
||||
|
||||
TupleTableSlot *
|
||||
@ -2420,7 +2420,7 @@ ExecBRUpdateTriggers(EState *estate, EPQState *epqstate,
|
||||
HeapTuple oldtuple;
|
||||
TupleTableSlot *newSlot;
|
||||
int i;
|
||||
Bitmapset *modifiedCols;
|
||||
Bitmapset *updatedCols;
|
||||
Bitmapset *keyCols;
|
||||
LockTupleMode lockmode;
|
||||
|
||||
@ -2429,10 +2429,10 @@ ExecBRUpdateTriggers(EState *estate, EPQState *epqstate,
|
||||
* been modified, then we can use a weaker lock, allowing for better
|
||||
* concurrency.
|
||||
*/
|
||||
modifiedCols = GetModifiedColumns(relinfo, estate);
|
||||
updatedCols = GetUpdatedColumns(relinfo, estate);
|
||||
keyCols = RelationGetIndexAttrBitmap(relinfo->ri_RelationDesc,
|
||||
INDEX_ATTR_BITMAP_KEY);
|
||||
if (bms_overlap(keyCols, modifiedCols))
|
||||
if (bms_overlap(keyCols, updatedCols))
|
||||
lockmode = LockTupleExclusive;
|
||||
else
|
||||
lockmode = LockTupleNoKeyExclusive;
|
||||
@ -2486,7 +2486,7 @@ ExecBRUpdateTriggers(EState *estate, EPQState *epqstate,
|
||||
TRIGGER_TYPE_UPDATE))
|
||||
continue;
|
||||
if (!TriggerEnabled(estate, relinfo, trigger, LocTriggerData.tg_event,
|
||||
modifiedCols, trigtuple, newtuple))
|
||||
updatedCols, trigtuple, newtuple))
|
||||
continue;
|
||||
|
||||
LocTriggerData.tg_trigtuple = trigtuple;
|
||||
@ -2556,7 +2556,7 @@ ExecARUpdateTriggers(EState *estate, ResultRelInfo *relinfo,
|
||||
|
||||
AfterTriggerSaveEvent(estate, relinfo, TRIGGER_EVENT_UPDATE,
|
||||
true, trigtuple, newtuple, recheckIndexes,
|
||||
GetModifiedColumns(relinfo, estate));
|
||||
GetUpdatedColumns(relinfo, estate));
|
||||
if (trigtuple != fdw_trigtuple)
|
||||
heap_freetuple(trigtuple);
|
||||
}
|
||||
|
Reference in New Issue
Block a user