diff options
Diffstat (limited to 'src/backend/commands')
-rw-r--r-- | src/backend/commands/trigger.c | 50 |
1 files changed, 41 insertions, 9 deletions
diff --git a/src/backend/commands/trigger.c b/src/backend/commands/trigger.c index 6b360afd19e..e05917e278c 100644 --- a/src/backend/commands/trigger.c +++ b/src/backend/commands/trigger.c @@ -1620,18 +1620,19 @@ ExecASUpdateTriggers(EState *estate, ResultRelInfo *relinfo) false, NULL, NULL); } -HeapTuple +TupleTableSlot * ExecBRUpdateTriggers(EState *estate, ResultRelInfo *relinfo, - ItemPointer tupleid, HeapTuple newtuple, + ItemPointer tupleid, TupleTableSlot *slot, CommandId cid) { TriggerDesc *trigdesc = relinfo->ri_TrigDesc; int ntrigs = trigdesc->n_before_row[TRIGGER_EVENT_UPDATE]; int *tgindx = trigdesc->tg_before_row[TRIGGER_EVENT_UPDATE]; + HeapTuple slottuple = ExecMaterializeSlot(slot); + HeapTuple newtuple = slottuple; TriggerData LocTriggerData; HeapTuple trigtuple; HeapTuple oldtuple; - HeapTuple intuple = newtuple; TupleTableSlot *newSlot; int i; @@ -1640,11 +1641,22 @@ ExecBRUpdateTriggers(EState *estate, ResultRelInfo *relinfo, return NULL; /* - * In READ COMMITTED isolation level it's possible that newtuple was - * changed due to concurrent update. + * In READ COMMITTED isolation level it's possible that target tuple was + * changed due to concurrent update. In that case we have a raw subplan + * output tuple in newSlot, and need to run it through the junk filter to + * produce an insertable tuple. + * + * Caution: more than likely, the passed-in slot is the same as the + * junkfilter's output slot, so we are clobbering the original value of + * slottuple by doing the filtering. This is OK since neither we nor our + * caller have any more interest in the prior contents of that slot. */ if (newSlot != NULL) - intuple = newtuple = ExecRemoveJunk(estate->es_junkFilter, newSlot); + { + slot = ExecFilterJunk(estate->es_junkFilter, newSlot); + slottuple = ExecMaterializeSlot(slot); + newtuple = slottuple; + } LocTriggerData.type = T_TriggerData; LocTriggerData.tg_event = TRIGGER_EVENT_UPDATE | @@ -1667,13 +1679,33 @@ ExecBRUpdateTriggers(EState *estate, ResultRelInfo *relinfo, relinfo->ri_TrigFunctions, relinfo->ri_TrigInstrument, GetPerTupleMemoryContext(estate)); - if (oldtuple != newtuple && oldtuple != intuple) + if (oldtuple != newtuple && oldtuple != slottuple) heap_freetuple(oldtuple); if (newtuple == NULL) - break; + { + heap_freetuple(trigtuple); + return NULL; /* "do nothing" */ + } } heap_freetuple(trigtuple); - return newtuple; + + if (newtuple != slottuple) + { + /* + * Return the modified tuple using the es_trig_tuple_slot. We assume + * the tuple was allocated in per-tuple memory context, and therefore + * will go away by itself. The tuple table slot should not try to + * clear it. + */ + TupleTableSlot *newslot = estate->es_trig_tuple_slot; + TupleDesc tupdesc = RelationGetDescr(relinfo->ri_RelationDesc); + + if (newslot->tts_tupleDescriptor != tupdesc) + ExecSetSlotDescriptor(newslot, tupdesc); + ExecStoreTuple(newtuple, newslot, InvalidBuffer, false); + slot = newslot; + } + return slot; } void |