trigger.c File Reference

#include "postgres.h"
#include "access/genam.h"
#include "access/htup_details.h"
#include "access/relation.h"
#include "access/sysattr.h"
#include "access/table.h"
#include "access/tableam.h"
#include "access/xact.h"
#include "catalog/catalog.h"
#include "catalog/dependency.h"
#include "catalog/index.h"
#include "catalog/indexing.h"
#include "catalog/objectaccess.h"
#include "catalog/partition.h"
#include "catalog/pg_constraint.h"
#include "catalog/pg_inherits.h"
#include "catalog/pg_proc.h"
#include "catalog/pg_trigger.h"
#include "catalog/pg_type.h"
#include "commands/dbcommands.h"
#include "commands/defrem.h"
#include "commands/trigger.h"
#include "executor/executor.h"
#include "executor/execPartition.h"
#include "miscadmin.h"
#include "nodes/bitmapset.h"
#include "nodes/makefuncs.h"
#include "optimizer/optimizer.h"
#include "parser/parse_clause.h"
#include "parser/parse_collate.h"
#include "parser/parse_func.h"
#include "parser/parse_relation.h"
#include "parser/parsetree.h"
#include "partitioning/partdesc.h"
#include "pgstat.h"
#include "rewrite/rewriteManip.h"
#include "storage/bufmgr.h"
#include "storage/lmgr.h"
#include "tcop/utility.h"
#include "utils/acl.h"
#include "utils/builtins.h"
#include "utils/bytea.h"
#include "utils/fmgroids.h"
#include "utils/guc_hooks.h"
#include "utils/inval.h"
#include "utils/lsyscache.h"
#include "utils/memutils.h"
#include "utils/plancache.h"
#include "utils/rel.h"
#include "utils/snapmgr.h"
#include "utils/syscache.h"
#include "utils/tuplestore.h"

Include dependency graph for trigger.c:

Go to the source code of this file.

Data Structures
struct	SetConstraintTriggerData
struct	SetConstraintStateData
struct	AfterTriggerSharedData
struct	AfterTriggerEventData
struct	AfterTriggerEventDataNoOids
struct	AfterTriggerEventDataOneCtid
struct	AfterTriggerEventDataZeroCtids
struct	AfterTriggerEventChunk
struct	AfterTriggerEventList
struct	AfterTriggersData
struct	AfterTriggersQueryData
struct	AfterTriggersTransData
struct	AfterTriggersTableData

Macros
#define	AFTER_TRIGGER_OFFSET   0x07FFFFFF /* must be low-order bits */
#define	AFTER_TRIGGER_DONE   0x80000000
#define	AFTER_TRIGGER_IN_PROGRESS   0x40000000
#define	AFTER_TRIGGER_FDW_REUSE   0x00000000
#define	AFTER_TRIGGER_FDW_FETCH   0x20000000
#define	AFTER_TRIGGER_1CTID   0x10000000
#define	AFTER_TRIGGER_2CTID   0x30000000
#define	AFTER_TRIGGER_CP_UPDATE   0x08000000
#define	AFTER_TRIGGER_TUP_BITS   0x38000000
#define	SizeofTriggerEvent(evt)
#define	GetTriggerSharedData(evt)    ((AfterTriggerShared) ((char *) (evt) + ((evt)->ate_flags & AFTER_TRIGGER_OFFSET)))
#define	CHUNK_DATA_START(cptr)   ((char *) (cptr) + MAXALIGN(sizeof(AfterTriggerEventChunk)))
#define	for_each_chunk(cptr, evtlist)    for (cptr = (evtlist).head; cptr != NULL; cptr = cptr->next)
#define	for_each_event(eptr, cptr)
#define	for_each_event_chunk(eptr, cptr, evtlist)    for_each_chunk(cptr, evtlist) for_each_event(eptr, cptr)
#define	for_each_chunk_from(cptr)    for (; cptr != NULL; cptr = cptr->next)
#define	for_each_event_from(eptr, cptr)
#define	MIN_CHUNK_SIZE   1024
#define	MAX_CHUNK_SIZE   (1024*1024)

Typedefs
typedef struct SetConstraintTriggerData	SetConstraintTriggerData
typedef struct SetConstraintTriggerData *	SetConstraintTrigger
typedef struct SetConstraintStateData	SetConstraintStateData
typedef SetConstraintStateData *	SetConstraintState
typedef uint32	TriggerFlags
typedef struct AfterTriggerSharedData *	AfterTriggerShared
typedef struct AfterTriggerSharedData	AfterTriggerSharedData
typedef struct AfterTriggerEventData *	AfterTriggerEvent
typedef struct AfterTriggerEventData	AfterTriggerEventData
typedef struct AfterTriggerEventDataNoOids	AfterTriggerEventDataNoOids
typedef struct AfterTriggerEventDataOneCtid	AfterTriggerEventDataOneCtid
typedef struct AfterTriggerEventDataZeroCtids	AfterTriggerEventDataZeroCtids
typedef struct AfterTriggerEventChunk	AfterTriggerEventChunk
typedef struct AfterTriggerEventList	AfterTriggerEventList
typedef struct AfterTriggersQueryData	AfterTriggersQueryData
typedef struct AfterTriggersTransData	AfterTriggersTransData
typedef struct AfterTriggersTableData	AfterTriggersTableData
typedef struct AfterTriggersData	AfterTriggersData

Functions
static void	renametrig_internal (Relation tgrel, Relation targetrel, HeapTuple trigtup, const char *newname, const char *expected_name)
static void	renametrig_partition (Relation tgrel, Oid partitionId, Oid parentTriggerOid, const char *newname, const char *expected_name)
static void	SetTriggerFlags (TriggerDesc *trigdesc, Trigger *trigger)
static bool	GetTupleForTrigger (EState *estate, EPQState *epqstate, ResultRelInfo *relinfo, ItemPointer tid, LockTupleMode lockmode, TupleTableSlot *oldslot, TupleTableSlot **epqslot, TM_Result *tmresultp, TM_FailureData *tmfdp)
static bool	TriggerEnabled (EState *estate, ResultRelInfo *relinfo, Trigger *trigger, TriggerEvent event, Bitmapset *modifiedCols, TupleTableSlot *oldslot, TupleTableSlot *newslot)
static HeapTuple	ExecCallTriggerFunc (TriggerData *trigdata, int tgindx, FmgrInfo *finfo, Instrumentation *instr, MemoryContext per_tuple_context)
static void	AfterTriggerSaveEvent (EState *estate, ResultRelInfo *relinfo, ResultRelInfo *src_partinfo, ResultRelInfo *dst_partinfo, int event, bool row_trigger, TupleTableSlot *oldslot, TupleTableSlot *newslot, List *recheckIndexes, Bitmapset *modifiedCols, TransitionCaptureState *transition_capture, bool is_crosspart_update)
static void	AfterTriggerEnlargeQueryState (void)
static bool	before_stmt_triggers_fired (Oid relid, CmdType cmdType)
ObjectAddress	CreateTrigger (CreateTrigStmt *stmt, const char *queryString, Oid relOid, Oid refRelOid, Oid constraintOid, Oid indexOid, Oid funcoid, Oid parentTriggerOid, Node *whenClause, bool isInternal, bool in_partition)
ObjectAddress	CreateTriggerFiringOn (CreateTrigStmt *stmt, const char *queryString, Oid relOid, Oid refRelOid, Oid constraintOid, Oid indexOid, Oid funcoid, Oid parentTriggerOid, Node *whenClause, bool isInternal, bool in_partition, char trigger_fires_when)
void	TriggerSetParentTrigger (Relation trigRel, Oid childTrigId, Oid parentTrigId, Oid childTableId)
void	RemoveTriggerById (Oid trigOid)
Oid	get_trigger_oid (Oid relid, const char *trigname, bool missing_ok)
static void	RangeVarCallbackForRenameTrigger (const RangeVar *rv, Oid relid, Oid oldrelid, void *arg)
ObjectAddress	renametrig (RenameStmt *stmt)
void	EnableDisableTrigger (Relation rel, const char *tgname, Oid tgparent, char fires_when, bool skip_system, bool recurse, LOCKMODE lockmode)
void	RelationBuildTriggers (Relation relation)
TriggerDesc *	CopyTriggerDesc (TriggerDesc *trigdesc)
void	FreeTriggerDesc (TriggerDesc *trigdesc)
const char *	FindTriggerIncompatibleWithInheritance (TriggerDesc *trigdesc)
void	ExecBSInsertTriggers (EState *estate, ResultRelInfo *relinfo)
void	ExecASInsertTriggers (EState *estate, ResultRelInfo *relinfo, TransitionCaptureState *transition_capture)
bool	ExecBRInsertTriggers (EState *estate, ResultRelInfo *relinfo, TupleTableSlot *slot)
void	ExecARInsertTriggers (EState *estate, ResultRelInfo *relinfo, TupleTableSlot *slot, List *recheckIndexes, TransitionCaptureState *transition_capture)
bool	ExecIRInsertTriggers (EState *estate, ResultRelInfo *relinfo, TupleTableSlot *slot)
void	ExecBSDeleteTriggers (EState *estate, ResultRelInfo *relinfo)
void	ExecASDeleteTriggers (EState *estate, ResultRelInfo *relinfo, TransitionCaptureState *transition_capture)
bool	ExecBRDeleteTriggers (EState *estate, EPQState *epqstate, ResultRelInfo *relinfo, ItemPointer tupleid, HeapTuple fdw_trigtuple, TupleTableSlot **epqslot, TM_Result *tmresult, TM_FailureData *tmfd)
void	ExecARDeleteTriggers (EState *estate, ResultRelInfo *relinfo, ItemPointer tupleid, HeapTuple fdw_trigtuple, TransitionCaptureState *transition_capture, bool is_crosspart_update)
bool	ExecIRDeleteTriggers (EState *estate, ResultRelInfo *relinfo, HeapTuple trigtuple)
void	ExecBSUpdateTriggers (EState *estate, ResultRelInfo *relinfo)
void	ExecASUpdateTriggers (EState *estate, ResultRelInfo *relinfo, TransitionCaptureState *transition_capture)
bool	ExecBRUpdateTriggers (EState *estate, EPQState *epqstate, ResultRelInfo *relinfo, ItemPointer tupleid, HeapTuple fdw_trigtuple, TupleTableSlot *newslot, TM_Result *tmresult, TM_FailureData *tmfd)
void	ExecARUpdateTriggers (EState *estate, ResultRelInfo *relinfo, ResultRelInfo *src_partinfo, ResultRelInfo *dst_partinfo, ItemPointer tupleid, HeapTuple fdw_trigtuple, TupleTableSlot *newslot, List *recheckIndexes, TransitionCaptureState *transition_capture, bool is_crosspart_update)
bool	ExecIRUpdateTriggers (EState *estate, ResultRelInfo *relinfo, HeapTuple trigtuple, TupleTableSlot *newslot)
void	ExecBSTruncateTriggers (EState *estate, ResultRelInfo *relinfo)
void	ExecASTruncateTriggers (EState *estate, ResultRelInfo *relinfo)
static void	AfterTriggerExecute (EState *estate, AfterTriggerEvent event, ResultRelInfo *relInfo, ResultRelInfo *src_relInfo, ResultRelInfo *dst_relInfo, TriggerDesc *trigdesc, FmgrInfo *finfo, Instrumentation *instr, MemoryContext per_tuple_context, TupleTableSlot *trig_tuple_slot1, TupleTableSlot *trig_tuple_slot2)
static AfterTriggersTableData *	GetAfterTriggersTableData (Oid relid, CmdType cmdType)
static TupleTableSlot *	GetAfterTriggersStoreSlot (AfterTriggersTableData *table, TupleDesc tupdesc)
static Tuplestorestate *	GetAfterTriggersTransitionTable (int event, TupleTableSlot *oldslot, TupleTableSlot *newslot, TransitionCaptureState *transition_capture)
static void	TransitionTableAddTuple (EState *estate, TransitionCaptureState *transition_capture, ResultRelInfo *relinfo, TupleTableSlot *slot, TupleTableSlot *original_insert_tuple, Tuplestorestate *tuplestore)
static void	AfterTriggerFreeQuery (AfterTriggersQueryData *qs)
static SetConstraintState	SetConstraintStateCreate (int numalloc)
static SetConstraintState	SetConstraintStateCopy (SetConstraintState origstate)
static SetConstraintState	SetConstraintStateAddItem (SetConstraintState state, Oid tgoid, bool tgisdeferred)
static void	cancel_prior_stmt_triggers (Oid relid, CmdType cmdType, int tgevent)
static Tuplestorestate *	GetCurrentFDWTuplestore (void)
static bool	afterTriggerCheckState (AfterTriggerShared evtshared)
static void	afterTriggerAddEvent (AfterTriggerEventList *events, AfterTriggerEvent event, AfterTriggerShared evtshared)
static void	afterTriggerFreeEventList (AfterTriggerEventList *events)
static void	afterTriggerRestoreEventList (AfterTriggerEventList *events, const AfterTriggerEventList *old_events)
static void	afterTriggerDeleteHeadEventChunk (AfterTriggersQueryData *qs)
static bool	afterTriggerMarkEvents (AfterTriggerEventList *events, AfterTriggerEventList *move_list, bool immediate_only)
static bool	afterTriggerInvokeEvents (AfterTriggerEventList *events, CommandId firing_id, EState *estate, bool delete_ok)
TransitionCaptureState *	MakeTransitionCaptureState (TriggerDesc *trigdesc, Oid relid, CmdType cmdType)
void	AfterTriggerBeginXact (void)
void	AfterTriggerBeginQuery (void)
void	AfterTriggerEndQuery (EState *estate)
void	AfterTriggerFireDeferred (void)
void	AfterTriggerEndXact (bool isCommit)
void	AfterTriggerBeginSubXact (void)
void	AfterTriggerEndSubXact (bool isCommit)
void	AfterTriggerSetState (ConstraintsSetStmt *stmt)
bool	AfterTriggerPendingOnRel (Oid relid)
void	assign_session_replication_role (int newval, void *extra)
Datum	pg_trigger_depth (PG_FUNCTION_ARGS)

Variables
int	SessionReplicationRole = SESSION_REPLICATION_ROLE_ORIGIN
static int	MyTriggerDepth = 0
static AfterTriggersData	afterTriggers

Macro Definition Documentation

AFTER_TRIGGER_1CTID

#define AFTER_TRIGGER_1CTID   0x10000000

Definition at line 3618 of file trigger.c.

AFTER_TRIGGER_2CTID

#define AFTER_TRIGGER_2CTID   0x30000000

Definition at line 3619 of file trigger.c.

AFTER_TRIGGER_CP_UPDATE

#define AFTER_TRIGGER_CP_UPDATE   0x08000000

Definition at line 3620 of file trigger.c.

AFTER_TRIGGER_DONE

#define AFTER_TRIGGER_DONE   0x80000000

Definition at line 3613 of file trigger.c.

AFTER_TRIGGER_FDW_FETCH

#define AFTER_TRIGGER_FDW_FETCH   0x20000000

Definition at line 3617 of file trigger.c.

AFTER_TRIGGER_FDW_REUSE

#define AFTER_TRIGGER_FDW_REUSE   0x00000000

Definition at line 3616 of file trigger.c.

AFTER_TRIGGER_IN_PROGRESS

#define AFTER_TRIGGER_IN_PROGRESS   0x40000000

Definition at line 3614 of file trigger.c.

AFTER_TRIGGER_OFFSET

#define AFTER_TRIGGER_OFFSET   0x07FFFFFF /* must be low-order bits */

Definition at line 3612 of file trigger.c.

AFTER_TRIGGER_TUP_BITS

#define AFTER_TRIGGER_TUP_BITS   0x38000000

Definition at line 3621 of file trigger.c.

CHUNK_DATA_START

#define CHUNK_DATA_START

(

cptr

)

((char *) (cptr) + MAXALIGN(sizeof(AfterTriggerEventChunk)))

Definition at line 3700 of file trigger.c.

for_each_chunk

#define for_each_chunk	(		cptr,
			evtlist
	)		for (cptr = (evtlist).head; cptr != NULL; cptr = cptr->next)

Definition at line 3711 of file trigger.c.

for_each_chunk_from

#define for_each_chunk_from

(

cptr

)

for (; cptr != NULL; cptr = cptr->next)

Definition at line 3722 of file trigger.c.

for_each_event

#define for_each_event	(		eptr,
			cptr
	)

Value:

    for (eptr = (AfterTriggerEvent) CHUNK_DATA_START(cptr); \
         (char *) eptr < (cptr)->freeptr; \
         eptr = (AfterTriggerEvent) (((char *) eptr) + SizeofTriggerEvent(eptr)))

Definition at line 3713 of file trigger.c.

for_each_event_chunk

#define for_each_event_chunk	(		eptr,
			cptr,
			evtlist
	)		for_each_chunk(cptr, evtlist) for_each_event(eptr, cptr)

Definition at line 3718 of file trigger.c.

for_each_event_from

#define for_each_event_from	(		eptr,
			cptr
	)

Value:

    for (; \
         (char *) eptr < (cptr)->freeptr; \
         eptr = (AfterTriggerEvent) (((char *) eptr) + SizeofTriggerEvent(eptr)))

Definition at line 3724 of file trigger.c.

GetTriggerSharedData

#define GetTriggerSharedData

(

evt

)

((AfterTriggerShared) ((char *) (evt) + ((evt)->ate_flags & AFTER_TRIGGER_OFFSET)))

Definition at line 3681 of file trigger.c.

MAX_CHUNK_SIZE

#define MAX_CHUNK_SIZE   (1024*1024)

MIN_CHUNK_SIZE

#define MIN_CHUNK_SIZE   1024

SizeofTriggerEvent

#define SizeofTriggerEvent

(

evt

)

Value:

    (((evt)->ate_flags & AFTER_TRIGGER_TUP_BITS) == AFTER_TRIGGER_CP_UPDATE ? \
     sizeof(AfterTriggerEventData) : \
     (((evt)->ate_flags & AFTER_TRIGGER_TUP_BITS) == AFTER_TRIGGER_2CTID ? \
      sizeof(AfterTriggerEventDataNoOids) : \
      (((evt)->ate_flags & AFTER_TRIGGER_TUP_BITS) == AFTER_TRIGGER_1CTID ? \
       sizeof(AfterTriggerEventDataOneCtid) : \
       sizeof(AfterTriggerEventDataZeroCtids))))

Definition at line 3672 of file trigger.c.

Typedef Documentation

AfterTriggerEvent

typedef struct AfterTriggerEventData* AfterTriggerEvent

Definition at line 3634 of file trigger.c.

AfterTriggerEventChunk

typedef struct AfterTriggerEventChunk AfterTriggerEventChunk

AfterTriggerEventData

typedef struct AfterTriggerEventData AfterTriggerEventData

AfterTriggerEventDataNoOids

typedef struct AfterTriggerEventDataNoOids AfterTriggerEventDataNoOids

AfterTriggerEventDataOneCtid

typedef struct AfterTriggerEventDataOneCtid AfterTriggerEventDataOneCtid

AfterTriggerEventDataZeroCtids

typedef struct AfterTriggerEventDataZeroCtids AfterTriggerEventDataZeroCtids

AfterTriggerEventList

typedef struct AfterTriggerEventList AfterTriggerEventList

AfterTriggersData

typedef struct AfterTriggersData AfterTriggersData

AfterTriggerShared

typedef struct AfterTriggerSharedData* AfterTriggerShared

Definition at line 3622 of file trigger.c.

AfterTriggerSharedData

typedef struct AfterTriggerSharedData AfterTriggerSharedData

AfterTriggersQueryData

typedef struct AfterTriggersQueryData AfterTriggersQueryData

Definition at line 3634 of file trigger.c.

AfterTriggersTableData

typedef struct AfterTriggersTableData AfterTriggersTableData

Definition at line 3634 of file trigger.c.

AfterTriggersTransData

typedef struct AfterTriggersTransData AfterTriggersTransData

Definition at line 3634 of file trigger.c.

SetConstraintState

typedef SetConstraintStateData* SetConstraintState

Definition at line 3565 of file trigger.c.

SetConstraintStateData

typedef struct SetConstraintStateData SetConstraintStateData

SetConstraintTrigger

typedef struct SetConstraintTriggerData* SetConstraintTrigger

Definition at line 3544 of file trigger.c.

SetConstraintTriggerData

typedef struct SetConstraintTriggerData SetConstraintTriggerData

TriggerFlags

typedef uint32 TriggerFlags

Definition at line 3610 of file trigger.c.

Function Documentation

afterTriggerAddEvent()

static void afterTriggerAddEvent ( AfterTriggerEventList *  events, AfterTriggerEvent  event, AfterTriggerShared  evtshared  )

static

Definition at line 3993 of file trigger.c.

{
    Size        eventsize = SizeofTriggerEvent(event);
    Size        needed = eventsize + sizeof(AfterTriggerSharedData);
    AfterTriggerEventChunk *chunk;
    AfterTriggerShared newshared;
    AfterTriggerEvent newevent;
 
    /*
     * If empty list or not enough room in the tail chunk, make a new chunk.
     * We assume here that a new shared record will always be needed.
     */
    chunk = events->tail;
    if (chunk == NULL ||
        chunk->endfree - chunk->freeptr < needed)
    {
        Size        chunksize;
 
        /* Create event context if we didn't already */
        if (afterTriggers.event_cxt == NULL)
            afterTriggers.event_cxt =
                AllocSetContextCreate(TopTransactionContext,
                                      "AfterTriggerEvents",
                                      ALLOCSET_DEFAULT_SIZES);
 
        /*
         * Chunk size starts at 1KB and is allowed to increase up to 1MB.
         * These numbers are fairly arbitrary, though there is a hard limit at
         * AFTER_TRIGGER_OFFSET; else we couldn't link event records to their
         * shared records using the available space in ate_flags.  Another
         * constraint is that if the chunk size gets too huge, the search loop
         * below would get slow given a (not too common) usage pattern with
         * many distinct event types in a chunk.  Therefore, we double the
         * preceding chunk size only if there weren't too many shared records
         * in the preceding chunk; otherwise we halve it.  This gives us some
         * ability to adapt to the actual usage pattern of the current query
         * while still having large chunk sizes in typical usage.  All chunk
         * sizes used should be MAXALIGN multiples, to ensure that the shared
         * records will be aligned safely.
         */
#define MIN_CHUNK_SIZE 1024
#define MAX_CHUNK_SIZE (1024*1024)
 
#if MAX_CHUNK_SIZE > (AFTER_TRIGGER_OFFSET+1)
#error MAX_CHUNK_SIZE must not exceed AFTER_TRIGGER_OFFSET
#endif
 
        if (chunk == NULL)
            chunksize = MIN_CHUNK_SIZE;
        else
        {
            /* preceding chunk size... */
            chunksize = chunk->endptr - (char *) chunk;
            /* check number of shared records in preceding chunk */
            if ((chunk->endptr - chunk->endfree) <=
                (100 * sizeof(AfterTriggerSharedData)))
                chunksize *= 2; /* okay, double it */
            else
                chunksize /= 2; /* too many shared records */
            chunksize = Min(chunksize, MAX_CHUNK_SIZE);
        }
        chunk = MemoryContextAlloc(afterTriggers.event_cxt, chunksize);
        chunk->next = NULL;
        chunk->freeptr = CHUNK_DATA_START(chunk);
        chunk->endptr = chunk->endfree = (char *) chunk + chunksize;
        Assert(chunk->endfree - chunk->freeptr >= needed);
 
        if (events->head == NULL)
            events->head = chunk;
        else
            events->tail->next = chunk;
        events->tail = chunk;
        /* events->tailfree is now out of sync, but we'll fix it below */
    }
 
    /*
     * Try to locate a matching shared-data record already in the chunk. If
     * none, make a new one.
     */
    for (newshared = ((AfterTriggerShared) chunk->endptr) - 1;
         (char *) newshared >= chunk->endfree;
         newshared--)
    {
        if (newshared->ats_tgoid == evtshared->ats_tgoid &&
            newshared->ats_relid == evtshared->ats_relid &&
            newshared->ats_event == evtshared->ats_event &&
            newshared->ats_table == evtshared->ats_table &&
            newshared->ats_firing_id == 0)
            break;
    }
    if ((char *) newshared < chunk->endfree)
    {
        *newshared = *evtshared;
        newshared->ats_firing_id = 0;   /* just to be sure */
        chunk->endfree = (char *) newshared;
    }
 
    /* Insert the data */
    newevent = (AfterTriggerEvent) chunk->freeptr;
    memcpy(newevent, event, eventsize);
    /* ... and link the new event to its shared record */
    newevent->ate_flags &= ~AFTER_TRIGGER_OFFSET;
    newevent->ate_flags |= (char *) newshared - (char *) newevent;
 
    chunk->freeptr += eventsize;
    events->tailfree = chunk->freeptr;
}

References AFTER_TRIGGER_OFFSET, afterTriggers, ALLOCSET_DEFAULT_SIZES, AllocSetContextCreate, Assert(), AfterTriggerEventData::ate_flags, AfterTriggerSharedData::ats_event, AfterTriggerSharedData::ats_firing_id, AfterTriggerSharedData::ats_relid, AfterTriggerSharedData::ats_table, AfterTriggerSharedData::ats_tgoid, CHUNK_DATA_START, AfterTriggersData::event_cxt, AfterTriggerEventList::head, MAX_CHUNK_SIZE, MemoryContextAlloc(), Min, MIN_CHUNK_SIZE, AfterTriggerEventChunk::next, SizeofTriggerEvent, AfterTriggerEventList::tail, AfterTriggerEventList::tailfree, and TopTransactionContext.

Referenced by afterTriggerMarkEvents(), and AfterTriggerSaveEvent().

AfterTriggerBeginQuery()

void AfterTriggerBeginQuery

(

void

)

Definition at line 4980 of file trigger.c.

{
    /* Increase the query stack depth */
    afterTriggers.query_depth++;
}

References afterTriggers, and AfterTriggersData::query_depth.

Referenced by CopyFrom(), create_edata_for_relation(), ExecuteTruncateGuts(), and standard_ExecutorStart().

AfterTriggerBeginSubXact()

void AfterTriggerBeginSubXact

(

void

)

Definition at line 5263 of file trigger.c.

{
    int         my_level = GetCurrentTransactionNestLevel();
 
    /*
     * Allocate more space in the trans_stack if needed.  (Note: because the
     * minimum nest level of a subtransaction is 2, we waste the first couple
     * entries of the array; not worth the notational effort to avoid it.)
     */
    while (my_level >= afterTriggers.maxtransdepth)
    {
        if (afterTriggers.maxtransdepth == 0)
        {
            /* Arbitrarily initialize for max of 8 subtransaction levels */
            afterTriggers.trans_stack = (AfterTriggersTransData *)
                MemoryContextAlloc(TopTransactionContext,
                                   8 * sizeof(AfterTriggersTransData));
            afterTriggers.maxtransdepth = 8;
        }
        else
        {
            /* repalloc will keep the stack in the same context */
            int         new_alloc = afterTriggers.maxtransdepth * 2;
 
            afterTriggers.trans_stack = (AfterTriggersTransData *)
                repalloc(afterTriggers.trans_stack,
                         new_alloc * sizeof(AfterTriggersTransData));
            afterTriggers.maxtransdepth = new_alloc;
        }
    }
 
    /*
     * Push the current information into the stack.  The SET CONSTRAINTS state
     * is not saved until/unless changed.  Likewise, we don't make a
     * per-subtransaction event context until needed.
     */
    afterTriggers.trans_stack[my_level].state = NULL;
    afterTriggers.trans_stack[my_level].events = afterTriggers.events;
    afterTriggers.trans_stack[my_level].query_depth = afterTriggers.query_depth;
    afterTriggers.trans_stack[my_level].firing_counter = afterTriggers.firing_counter;
}

References afterTriggers, AfterTriggersData::events, AfterTriggersTransData::events, AfterTriggersData::firing_counter, AfterTriggersTransData::firing_counter, GetCurrentTransactionNestLevel(), AfterTriggersData::maxtransdepth, MemoryContextAlloc(), AfterTriggersData::query_depth, AfterTriggersTransData::query_depth, repalloc(), AfterTriggersTransData::state, TopTransactionContext, and AfterTriggersData::trans_stack.

Referenced by StartSubTransaction().

AfterTriggerBeginXact()

void AfterTriggerBeginXact

(

void

)

Definition at line 4948 of file trigger.c.

{
    /*
     * Initialize after-trigger state structure to empty
     */
    afterTriggers.firing_counter = (CommandId) 1;   /* mustn't be 0 */
    afterTriggers.query_depth = -1;
 
    /*
     * Verify that there is no leftover state remaining.  If these assertions
     * trip, it means that AfterTriggerEndXact wasn't called or didn't clean
     * up properly.
     */
    Assert(afterTriggers.state == NULL);
    Assert(afterTriggers.query_stack == NULL);
    Assert(afterTriggers.maxquerydepth == 0);
    Assert(afterTriggers.event_cxt == NULL);
    Assert(afterTriggers.events.head == NULL);
    Assert(afterTriggers.trans_stack == NULL);
    Assert(afterTriggers.maxtransdepth == 0);
}

References afterTriggers, Assert(), AfterTriggersData::event_cxt, AfterTriggersData::events, AfterTriggersData::firing_counter, AfterTriggerEventList::head, AfterTriggersData::maxquerydepth, AfterTriggersData::maxtransdepth, AfterTriggersData::query_depth, AfterTriggersData::query_stack, AfterTriggersData::state, and AfterTriggersData::trans_stack.

Referenced by StartTransaction().

afterTriggerCheckState()

static bool afterTriggerCheckState ( AfterTriggerShared  evtshared )

static

Definition at line 3947 of file trigger.c.

{
    Oid         tgoid = evtshared->ats_tgoid;
    SetConstraintState state = afterTriggers.state;
    int         i;
 
    /*
     * For not-deferrable triggers (i.e. normal AFTER ROW triggers and
     * constraints declared NOT DEFERRABLE), the state is always false.
     */
    if ((evtshared->ats_event & AFTER_TRIGGER_DEFERRABLE) == 0)
        return false;
 
    /*
     * If constraint state exists, SET CONSTRAINTS might have been executed
     * either for this trigger or for all triggers.
     */
    if (state != NULL)
    {
        /* Check for SET CONSTRAINTS for this specific trigger. */
        for (i = 0; i < state->numstates; i++)
        {
            if (state->trigstates[i].sct_tgoid == tgoid)
                return state->trigstates[i].sct_tgisdeferred;
        }
 
        /* Check for SET CONSTRAINTS ALL. */
        if (state->all_isset)
            return state->all_isdeferred;
    }
 
    /*
     * Otherwise return the default state for the trigger.
     */
    return ((evtshared->ats_event & AFTER_TRIGGER_INITDEFERRED) != 0);
}

References AFTER_TRIGGER_DEFERRABLE, AFTER_TRIGGER_INITDEFERRED, afterTriggers, AfterTriggerSharedData::ats_event, AfterTriggerSharedData::ats_tgoid, i, and AfterTriggersData::state.

Referenced by afterTriggerMarkEvents().

afterTriggerDeleteHeadEventChunk()

static void afterTriggerDeleteHeadEventChunk ( AfterTriggersQueryData *  qs )

static

Definition at line 4170 of file trigger.c.

{
    AfterTriggerEventChunk *target = qs->events.head;
    ListCell   *lc;
 
    Assert(target && target->next);
 
    /*
     * First, update any pointers in the per-table data, so that they won't be
     * dangling.  Resetting obsoleted pointers to NULL will make
     * cancel_prior_stmt_triggers start from the list head, which is fine.
     */
    foreach(lc, qs->tables)
    {
        AfterTriggersTableData *table = (AfterTriggersTableData *) lfirst(lc);
 
        if (table->after_trig_done &&
            table->after_trig_events.tail == target)
        {
            table->after_trig_events.head = NULL;
            table->after_trig_events.tail = NULL;
            table->after_trig_events.tailfree = NULL;
        }
    }
 
    /* Now we can flush the head chunk */
    qs->events.head = target->next;
    pfree(target);
}

References AfterTriggersTableData::after_trig_done, AfterTriggersTableData::after_trig_events, Assert(), AfterTriggersQueryData::events, AfterTriggerEventList::head, lfirst, AfterTriggerEventChunk::next, pfree(), AfterTriggersQueryData::tables, AfterTriggerEventList::tail, and AfterTriggerEventList::tailfree.

Referenced by AfterTriggerEndQuery().

AfterTriggerEndQuery()

void AfterTriggerEndQuery

(

EState *

estate

)

Definition at line 5000 of file trigger.c.

{
    AfterTriggersQueryData *qs;
 
    /* Must be inside a query, too */
    Assert(afterTriggers.query_depth >= 0);
 
    /*
     * If we never even got as far as initializing the event stack, there
     * certainly won't be any events, so exit quickly.
     */
    if (afterTriggers.query_depth >= afterTriggers.maxquerydepth)
    {
        afterTriggers.query_depth--;
        return;
    }
 
    /*
     * Process all immediate-mode triggers queued by the query, and move the
     * deferred ones to the main list of deferred events.
     *
     * Notice that we decide which ones will be fired, and put the deferred
     * ones on the main list, before anything is actually fired.  This ensures
     * reasonably sane behavior if a trigger function does SET CONSTRAINTS ...
     * IMMEDIATE: all events we have decided to defer will be available for it
     * to fire.
     *
     * We loop in case a trigger queues more events at the same query level.
     * Ordinary trigger functions, including all PL/pgSQL trigger functions,
     * will instead fire any triggers in a dedicated query level.  Foreign key
     * enforcement triggers do add to the current query level, thanks to their
     * passing fire_triggers = false to SPI_execute_snapshot().  Other
     * C-language triggers might do likewise.
     *
     * If we find no firable events, we don't have to increment
     * firing_counter.
     */
    qs = &afterTriggers.query_stack[afterTriggers.query_depth];
 
    for (;;)
    {
        if (afterTriggerMarkEvents(&qs->events, &afterTriggers.events, true))
        {
            CommandId   firing_id = afterTriggers.firing_counter++;
            AfterTriggerEventChunk *oldtail = qs->events.tail;
 
            if (afterTriggerInvokeEvents(&qs->events, firing_id, estate, false))
                break;          /* all fired */
 
            /*
             * Firing a trigger could result in query_stack being repalloc'd,
             * so we must recalculate qs after each afterTriggerInvokeEvents
             * call.  Furthermore, it's unsafe to pass delete_ok = true here,
             * because that could cause afterTriggerInvokeEvents to try to
             * access qs->events after the stack has been repalloc'd.
             */
            qs = &afterTriggers.query_stack[afterTriggers.query_depth];
 
            /*
             * We'll need to scan the events list again.  To reduce the cost
             * of doing so, get rid of completely-fired chunks.  We know that
             * all events were marked IN_PROGRESS or DONE at the conclusion of
             * afterTriggerMarkEvents, so any still-interesting events must
             * have been added after that, and so must be in the chunk that
             * was then the tail chunk, or in later chunks.  So, zap all
             * chunks before oldtail.  This is approximately the same set of
             * events we would have gotten rid of by passing delete_ok = true.
             */
            Assert(oldtail != NULL);
            while (qs->events.head != oldtail)
                afterTriggerDeleteHeadEventChunk(qs);
        }
        else
            break;
    }
 
    /* Release query-level-local storage, including tuplestores if any */
    AfterTriggerFreeQuery(&afterTriggers.query_stack[afterTriggers.query_depth]);
 
    afterTriggers.query_depth--;
}

References afterTriggerDeleteHeadEventChunk(), AfterTriggerFreeQuery(), afterTriggerInvokeEvents(), afterTriggerMarkEvents(), afterTriggers, Assert(), AfterTriggersData::events, AfterTriggersQueryData::events, AfterTriggersData::firing_counter, AfterTriggerEventList::head, AfterTriggersData::maxquerydepth, AfterTriggersData::query_depth, AfterTriggersData::query_stack, and AfterTriggerEventList::tail.

Referenced by CopyFrom(), ExecuteTruncateGuts(), finish_edata(), and standard_ExecutorFinish().

AfterTriggerEndSubXact()

void AfterTriggerEndSubXact

(

bool

isCommit

)

Definition at line 5311 of file trigger.c.

{
    int         my_level = GetCurrentTransactionNestLevel();
    SetConstraintState state;
    AfterTriggerEvent event;
    AfterTriggerEventChunk *chunk;
    CommandId   subxact_firing_id;
 
    /*
     * Pop the prior state if needed.
     */
    if (isCommit)
    {
        Assert(my_level < afterTriggers.maxtransdepth);
        /* If we saved a prior state, we don't need it anymore */
        state = afterTriggers.trans_stack[my_level].state;
        if (state != NULL)
            pfree(state);
        /* this avoids double pfree if error later: */
        afterTriggers.trans_stack[my_level].state = NULL;
        Assert(afterTriggers.query_depth ==
               afterTriggers.trans_stack[my_level].query_depth);
    }
    else
    {
        /*
         * Aborting.  It is possible subxact start failed before calling
         * AfterTriggerBeginSubXact, in which case we mustn't risk touching
         * trans_stack levels that aren't there.
         */
        if (my_level >= afterTriggers.maxtransdepth)
            return;
 
        /*
         * Release query-level storage for queries being aborted, and restore
         * query_depth to its pre-subxact value.  This assumes that a
         * subtransaction will not add events to query levels started in a
         * earlier transaction state.
         */
        while (afterTriggers.query_depth > afterTriggers.trans_stack[my_level].query_depth)
        {
            if (afterTriggers.query_depth < afterTriggers.maxquerydepth)
                AfterTriggerFreeQuery(&afterTriggers.query_stack[afterTriggers.query_depth]);
            afterTriggers.query_depth--;
        }
        Assert(afterTriggers.query_depth ==
               afterTriggers.trans_stack[my_level].query_depth);
 
        /*
         * Restore the global deferred-event list to its former length,
         * discarding any events queued by the subxact.
         */
        afterTriggerRestoreEventList(&afterTriggers.events,
                                     &afterTriggers.trans_stack[my_level].events);
 
        /*
         * Restore the trigger state.  If the saved state is NULL, then this
         * subxact didn't save it, so it doesn't need restoring.
         */
        state = afterTriggers.trans_stack[my_level].state;
        if (state != NULL)
        {
            pfree(afterTriggers.state);
            afterTriggers.state = state;
        }
        /* this avoids double pfree if error later: */
        afterTriggers.trans_stack[my_level].state = NULL;
 
        /*
         * Scan for any remaining deferred events that were marked DONE or IN
         * PROGRESS by this subxact or a child, and un-mark them. We can
         * recognize such events because they have a firing ID greater than or
         * equal to the firing_counter value we saved at subtransaction start.
         * (This essentially assumes that the current subxact includes all
         * subxacts started after it.)
         */
        subxact_firing_id = afterTriggers.trans_stack[my_level].firing_counter;
        for_each_event_chunk(event, chunk, afterTriggers.events)
        {
            AfterTriggerShared evtshared = GetTriggerSharedData(event);
 
            if (event->ate_flags &
                (AFTER_TRIGGER_DONE | AFTER_TRIGGER_IN_PROGRESS))
            {
                if (evtshared->ats_firing_id >= subxact_firing_id)
                    event->ate_flags &=
                        ~(AFTER_TRIGGER_DONE | AFTER_TRIGGER_IN_PROGRESS);
            }
        }
    }
}

References AFTER_TRIGGER_DONE, AFTER_TRIGGER_IN_PROGRESS, AfterTriggerFreeQuery(), afterTriggerRestoreEventList(), afterTriggers, Assert(), AfterTriggerEventData::ate_flags, AfterTriggerSharedData::ats_firing_id, AfterTriggersData::events, AfterTriggersTransData::events, AfterTriggersTransData::firing_counter, for_each_event_chunk, GetCurrentTransactionNestLevel(), GetTriggerSharedData, AfterTriggersData::maxquerydepth, AfterTriggersData::maxtransdepth, pfree(), AfterTriggersData::query_depth, AfterTriggersTransData::query_depth, AfterTriggersData::query_stack, AfterTriggersData::state, AfterTriggersTransData::state, and AfterTriggersData::trans_stack.

Referenced by AbortSubTransaction(), and CommitSubTransaction().

AfterTriggerEndXact()

void AfterTriggerEndXact

(

bool

isCommit

)

Definition at line 5215 of file trigger.c.

{
    /*
     * Forget the pending-events list.
     *
     * Since all the info is in TopTransactionContext or children thereof, we
     * don't really need to do anything to reclaim memory.  However, the
     * pending-events list could be large, and so it's useful to discard it as
     * soon as possible --- especially if we are aborting because we ran out
     * of memory for the list!
     */
    if (afterTriggers.event_cxt)
    {
        MemoryContextDelete(afterTriggers.event_cxt);
        afterTriggers.event_cxt = NULL;
        afterTriggers.events.head = NULL;
        afterTriggers.events.tail = NULL;
        afterTriggers.events.tailfree = NULL;
    }
 
    /*
     * Forget any subtransaction state as well.  Since this can't be very
     * large, we let the eventual reset of TopTransactionContext free the
     * memory instead of doing it here.
     */
    afterTriggers.trans_stack = NULL;
    afterTriggers.maxtransdepth = 0;
 
 
    /*
     * Forget the query stack and constraint-related state information.  As
     * with the subtransaction state information, we don't bother freeing the
     * memory here.
     */
    afterTriggers.query_stack = NULL;
    afterTriggers.maxquerydepth = 0;
    afterTriggers.state = NULL;
 
    /* No more afterTriggers manipulation until next transaction starts. */
    afterTriggers.query_depth = -1;
}

References afterTriggers, AfterTriggersData::event_cxt, AfterTriggersData::events, AfterTriggerEventList::head, AfterTriggersData::maxquerydepth, AfterTriggersData::maxtransdepth, MemoryContextDelete(), AfterTriggersData::query_depth, AfterTriggersData::query_stack, AfterTriggersData::state, AfterTriggerEventList::tail, AfterTriggerEventList::tailfree, and AfterTriggersData::trans_stack.

Referenced by AbortTransaction(), CommitTransaction(), and PrepareTransaction().

AfterTriggerEnlargeQueryState()

static void AfterTriggerEnlargeQueryState ( void  )

static

Definition at line 5499 of file trigger.c.

{
    int         init_depth = afterTriggers.maxquerydepth;
 
    Assert(afterTriggers.query_depth >= afterTriggers.maxquerydepth);
 
    if (afterTriggers.maxquerydepth == 0)
    {
        int         new_alloc = Max(afterTriggers.query_depth + 1, 8);
 
        afterTriggers.query_stack = (AfterTriggersQueryData *)
            MemoryContextAlloc(TopTransactionContext,
                               new_alloc * sizeof(AfterTriggersQueryData));
        afterTriggers.maxquerydepth = new_alloc;
    }
    else
    {
        /* repalloc will keep the stack in the same context */
        int         old_alloc = afterTriggers.maxquerydepth;
        int         new_alloc = Max(afterTriggers.query_depth + 1,
                                    old_alloc * 2);
 
        afterTriggers.query_stack = (AfterTriggersQueryData *)
            repalloc(afterTriggers.query_stack,
                     new_alloc * sizeof(AfterTriggersQueryData));
        afterTriggers.maxquerydepth = new_alloc;
    }
 
    /* Initialize new array entries to empty */
    while (init_depth < afterTriggers.maxquerydepth)
    {
        AfterTriggersQueryData *qs = &afterTriggers.query_stack[init_depth];
 
        qs->events.head = NULL;
        qs->events.tail = NULL;
        qs->events.tailfree = NULL;
        qs->fdw_tuplestore = NULL;
        qs->tables = NIL;
 
        ++init_depth;
    }
}

References afterTriggers, Assert(), AfterTriggersQueryData::events, AfterTriggersQueryData::fdw_tuplestore, AfterTriggerEventList::head, Max, AfterTriggersData::maxquerydepth, MemoryContextAlloc(), NIL, AfterTriggersData::query_depth, AfterTriggersData::query_stack, repalloc(), AfterTriggersQueryData::tables, AfterTriggerEventList::tail, AfterTriggerEventList::tailfree, and TopTransactionContext.

Referenced by AfterTriggerSaveEvent(), before_stmt_triggers_fired(), and MakeTransitionCaptureState().

AfterTriggerExecute()

static void AfterTriggerExecute ( EState *  estate, AfterTriggerEvent  event, ResultRelInfo *  relInfo, ResultRelInfo *  src_relInfo, ResultRelInfo *  dst_relInfo, TriggerDesc *  trigdesc, FmgrInfo *  finfo, Instrumentation *  instr, MemoryContext  per_tuple_context, TupleTableSlot *  trig_tuple_slot1, TupleTableSlot *  trig_tuple_slot2  )

static

Definition at line 4232 of file trigger.c.

{
    Relation    rel = relInfo->ri_RelationDesc;
    Relation    src_rel = src_relInfo->ri_RelationDesc;
    Relation    dst_rel = dst_relInfo->ri_RelationDesc;
    AfterTriggerShared evtshared = GetTriggerSharedData(event);
    Oid         tgoid = evtshared->ats_tgoid;
    TriggerData LocTriggerData = {0};
    HeapTuple   rettuple;
    int         tgindx;
    bool        should_free_trig = false;
    bool        should_free_new = false;
 
    /*
     * Locate trigger in trigdesc.
     */
    for (tgindx = 0; tgindx < trigdesc->numtriggers; tgindx++)
    {
        if (trigdesc->triggers[tgindx].tgoid == tgoid)
        {
            LocTriggerData.tg_trigger = &(trigdesc->triggers[tgindx]);
            break;
        }
    }
    if (LocTriggerData.tg_trigger == NULL)
        elog(ERROR, "could not find trigger %u", tgoid);
 
    /*
     * If doing EXPLAIN ANALYZE, start charging time to this trigger. We want
     * to include time spent re-fetching tuples in the trigger cost.
     */
    if (instr)
        InstrStartNode(instr + tgindx);
 
    /*
     * Fetch the required tuple(s).
     */
    switch (event->ate_flags & AFTER_TRIGGER_TUP_BITS)
    {
        case AFTER_TRIGGER_FDW_FETCH:
            {
                Tuplestorestate *fdw_tuplestore = GetCurrentFDWTuplestore();
 
                if (!tuplestore_gettupleslot(fdw_tuplestore, true, false,
                                             trig_tuple_slot1))
                    elog(ERROR, "failed to fetch tuple1 for AFTER trigger");
 
                if ((evtshared->ats_event & TRIGGER_EVENT_OPMASK) ==
                    TRIGGER_EVENT_UPDATE &&
                    !tuplestore_gettupleslot(fdw_tuplestore, true, false,
                                             trig_tuple_slot2))
                    elog(ERROR, "failed to fetch tuple2 for AFTER trigger");
            }
            /* fall through */
        case AFTER_TRIGGER_FDW_REUSE:
 
            /*
             * Store tuple in the slot so that tg_trigtuple does not reference
             * tuplestore memory.  (It is formally possible for the trigger
             * function to queue trigger events that add to the same
             * tuplestore, which can push other tuples out of memory.)  The
             * distinction is academic, because we start with a minimal tuple
             * that is stored as a heap tuple, constructed in different memory
             * context, in the slot anyway.
             */
            LocTriggerData.tg_trigslot = trig_tuple_slot1;
            LocTriggerData.tg_trigtuple =
                ExecFetchSlotHeapTuple(trig_tuple_slot1, true, &should_free_trig);
 
            if ((evtshared->ats_event & TRIGGER_EVENT_OPMASK) ==
                TRIGGER_EVENT_UPDATE)
            {
                LocTriggerData.tg_newslot = trig_tuple_slot2;
                LocTriggerData.tg_newtuple =
                    ExecFetchSlotHeapTuple(trig_tuple_slot2, true, &should_free_new);
            }
            else
            {
                LocTriggerData.tg_newtuple = NULL;
            }
            break;
 
        default:
            if (ItemPointerIsValid(&(event->ate_ctid1)))
            {
                TupleTableSlot *src_slot = ExecGetTriggerOldSlot(estate,
                                                                 src_relInfo);
 
                if (!table_tuple_fetch_row_version(src_rel,
                                                   &(event->ate_ctid1),
                                                   SnapshotAny,
                                                   src_slot))
                    elog(ERROR, "failed to fetch tuple1 for AFTER trigger");
 
                /*
                 * Store the tuple fetched from the source partition into the
                 * target (root partitioned) table slot, converting if needed.
                 */
                if (src_relInfo != relInfo)
                {
                    TupleConversionMap *map = ExecGetChildToRootMap(src_relInfo);
 
                    LocTriggerData.tg_trigslot = ExecGetTriggerOldSlot(estate, relInfo);
                    if (map)
                    {
                        execute_attr_map_slot(map->attrMap,
                                              src_slot,
                                              LocTriggerData.tg_trigslot);
                    }
                    else
                        ExecCopySlot(LocTriggerData.tg_trigslot, src_slot);
                }
                else
                    LocTriggerData.tg_trigslot = src_slot;
                LocTriggerData.tg_trigtuple =
                    ExecFetchSlotHeapTuple(LocTriggerData.tg_trigslot, false, &should_free_trig);
            }
            else
            {
                LocTriggerData.tg_trigtuple = NULL;
            }
 
            /* don't touch ctid2 if not there */
            if (((event->ate_flags & AFTER_TRIGGER_TUP_BITS) == AFTER_TRIGGER_2CTID ||
                 (event->ate_flags & AFTER_TRIGGER_CP_UPDATE)) &&
                ItemPointerIsValid(&(event->ate_ctid2)))
            {
                TupleTableSlot *dst_slot = ExecGetTriggerNewSlot(estate,
                                                                 dst_relInfo);
 
                if (!table_tuple_fetch_row_version(dst_rel,
                                                   &(event->ate_ctid2),
                                                   SnapshotAny,
                                                   dst_slot))
                    elog(ERROR, "failed to fetch tuple2 for AFTER trigger");
 
                /*
                 * Store the tuple fetched from the destination partition into
                 * the target (root partitioned) table slot, converting if
                 * needed.
                 */
                if (dst_relInfo != relInfo)
                {
                    TupleConversionMap *map = ExecGetChildToRootMap(dst_relInfo);
 
                    LocTriggerData.tg_newslot = ExecGetTriggerNewSlot(estate, relInfo);
                    if (map)
                    {
                        execute_attr_map_slot(map->attrMap,
                                              dst_slot,
                                              LocTriggerData.tg_newslot);
                    }
                    else
                        ExecCopySlot(LocTriggerData.tg_newslot, dst_slot);
                }
                else
                    LocTriggerData.tg_newslot = dst_slot;
                LocTriggerData.tg_newtuple =
                    ExecFetchSlotHeapTuple(LocTriggerData.tg_newslot, false, &should_free_new);
            }
            else
            {
                LocTriggerData.tg_newtuple = NULL;
            }
    }
 
    /*
     * Set up the tuplestore information to let the trigger have access to
     * transition tables.  When we first make a transition table available to
     * a trigger, mark it "closed" so that it cannot change anymore.  If any
     * additional events of the same type get queued in the current trigger
     * query level, they'll go into new transition tables.
     */
    LocTriggerData.tg_oldtable = LocTriggerData.tg_newtable = NULL;
    if (evtshared->ats_table)
    {
        if (LocTriggerData.tg_trigger->tgoldtable)
        {
            if (TRIGGER_FIRED_BY_UPDATE(evtshared->ats_event))
                LocTriggerData.tg_oldtable = evtshared->ats_table->old_upd_tuplestore;
            else
                LocTriggerData.tg_oldtable = evtshared->ats_table->old_del_tuplestore;
            evtshared->ats_table->closed = true;
        }
 
        if (LocTriggerData.tg_trigger->tgnewtable)
        {
            if (TRIGGER_FIRED_BY_INSERT(evtshared->ats_event))
                LocTriggerData.tg_newtable = evtshared->ats_table->new_ins_tuplestore;
            else
                LocTriggerData.tg_newtable = evtshared->ats_table->new_upd_tuplestore;
            evtshared->ats_table->closed = true;
        }
    }
 
    /*
     * Setup the remaining trigger information
     */
    LocTriggerData.type = T_TriggerData;
    LocTriggerData.tg_event =
        evtshared->ats_event & (TRIGGER_EVENT_OPMASK | TRIGGER_EVENT_ROW);
    LocTriggerData.tg_relation = rel;
    if (TRIGGER_FOR_UPDATE(LocTriggerData.tg_trigger->tgtype))
        LocTriggerData.tg_updatedcols = evtshared->ats_modifiedcols;
 
    MemoryContextReset(per_tuple_context);
 
    /*
     * Call the trigger and throw away any possibly returned updated tuple.
     * (Don't let ExecCallTriggerFunc measure EXPLAIN time.)
     */
    rettuple = ExecCallTriggerFunc(&LocTriggerData,
                                   tgindx,
                                   finfo,
                                   NULL,
                                   per_tuple_context);
    if (rettuple != NULL &&
        rettuple != LocTriggerData.tg_trigtuple &&
        rettuple != LocTriggerData.tg_newtuple)
        heap_freetuple(rettuple);
 
    /*
     * Release resources
     */
    if (should_free_trig)
        heap_freetuple(LocTriggerData.tg_trigtuple);
    if (should_free_new)
        heap_freetuple(LocTriggerData.tg_newtuple);
 
    /* don't clear slots' contents if foreign table */
    if (trig_tuple_slot1 == NULL)
    {
        if (LocTriggerData.tg_trigslot)
            ExecClearTuple(LocTriggerData.tg_trigslot);
        if (LocTriggerData.tg_newslot)
            ExecClearTuple(LocTriggerData.tg_newslot);
    }
 
    /*
     * If doing EXPLAIN ANALYZE, stop charging time to this trigger, and count
     * one "tuple returned" (really the number of firings).
     */
    if (instr)
        InstrStopNode(instr + tgindx, 1);
}

References AFTER_TRIGGER_2CTID, AFTER_TRIGGER_CP_UPDATE, AFTER_TRIGGER_FDW_FETCH, AFTER_TRIGGER_FDW_REUSE, AFTER_TRIGGER_TUP_BITS, AfterTriggerEventData::ate_ctid1, AfterTriggerEventData::ate_ctid2, AfterTriggerEventData::ate_flags, AfterTriggerSharedData::ats_event, AfterTriggerSharedData::ats_modifiedcols, AfterTriggerSharedData::ats_table, AfterTriggerSharedData::ats_tgoid, TupleConversionMap::attrMap, AfterTriggersTableData::closed, elog(), ERROR, ExecCallTriggerFunc(), ExecClearTuple(), ExecCopySlot(), ExecFetchSlotHeapTuple(), ExecGetChildToRootMap(), ExecGetTriggerNewSlot(), ExecGetTriggerOldSlot(), execute_attr_map_slot(), GetCurrentFDWTuplestore(), GetTriggerSharedData, heap_freetuple(), InstrStartNode(), InstrStopNode(), ItemPointerIsValid(), MemoryContextReset(), AfterTriggersTableData::new_ins_tuplestore, AfterTriggersTableData::new_upd_tuplestore, TriggerDesc::numtriggers, AfterTriggersTableData::old_del_tuplestore, AfterTriggersTableData::old_upd_tuplestore, ResultRelInfo::ri_RelationDesc, SnapshotAny, table_tuple_fetch_row_version(), TriggerData::tg_event, TriggerData::tg_newslot, TriggerData::tg_newtable, TriggerData::tg_newtuple, TriggerData::tg_oldtable, TriggerData::tg_relation, TriggerData::tg_trigger, TriggerData::tg_trigslot, TriggerData::tg_trigtuple, TriggerData::tg_updatedcols, Trigger::tgnewtable, Trigger::tgoid, Trigger::tgoldtable, Trigger::tgtype, TRIGGER_EVENT_OPMASK, TRIGGER_EVENT_ROW, TRIGGER_EVENT_UPDATE, TRIGGER_FIRED_BY_INSERT, TRIGGER_FIRED_BY_UPDATE, TriggerDesc::triggers, tuplestore_gettupleslot(), and TriggerData::type.

Referenced by afterTriggerInvokeEvents().

AfterTriggerFireDeferred()

void AfterTriggerFireDeferred

(

void

)

Definition at line 5159 of file trigger.c.

{
    AfterTriggerEventList *events;
    bool        snap_pushed = false;
 
    /* Must not be inside a query */
    Assert(afterTriggers.query_depth == -1);
 
    /*
     * If there are any triggers to fire, make sure we have set a snapshot for
     * them to use.  (Since PortalRunUtility doesn't set a snap for COMMIT, we
     * can't assume ActiveSnapshot is valid on entry.)
     */
    events = &afterTriggers.events;
    if (events->head != NULL)
    {
        PushActiveSnapshot(GetTransactionSnapshot());
        snap_pushed = true;
    }
 
    /*
     * Run all the remaining triggers.  Loop until they are all gone, in case
     * some trigger queues more for us to do.
     */
    while (afterTriggerMarkEvents(events, NULL, false))
    {
        CommandId   firing_id = afterTriggers.firing_counter++;
 
        if (afterTriggerInvokeEvents(events, firing_id, NULL, true))
            break;              /* all fired */
    }
 
    /*
     * We don't bother freeing the event list, since it will go away anyway
     * (and more efficiently than via pfree) in AfterTriggerEndXact.
     */
 
    if (snap_pushed)
        PopActiveSnapshot();
}

References afterTriggerInvokeEvents(), afterTriggerMarkEvents(), afterTriggers, Assert(), AfterTriggersData::events, AfterTriggersData::firing_counter, GetTransactionSnapshot(), AfterTriggerEventList::head, PopActiveSnapshot(), PushActiveSnapshot(), and AfterTriggersData::query_depth.

Referenced by CommitTransaction(), and PrepareTransaction().

afterTriggerFreeEventList()

static void afterTriggerFreeEventList ( AfterTriggerEventList *  events )

static

Definition at line 4109 of file trigger.c.

{
    AfterTriggerEventChunk *chunk;
 
    while ((chunk = events->head) != NULL)
    {
        events->head = chunk->next;
        pfree(chunk);
    }
    events->tail = NULL;
    events->tailfree = NULL;
}

References AfterTriggerEventList::head, AfterTriggerEventChunk::next, pfree(), AfterTriggerEventList::tail, and AfterTriggerEventList::tailfree.

Referenced by AfterTriggerFreeQuery(), and afterTriggerRestoreEventList().

AfterTriggerFreeQuery()

static void AfterTriggerFreeQuery ( AfterTriggersQueryData *  qs )

static

Definition at line 5091 of file trigger.c.

{
    Tuplestorestate *ts;
    List       *tables;
    ListCell   *lc;
 
    /* Drop the trigger events */
    afterTriggerFreeEventList(&qs->events);
 
    /* Drop FDW tuplestore if any */
    ts = qs->fdw_tuplestore;
    qs->fdw_tuplestore = NULL;
    if (ts)
        tuplestore_end(ts);
 
    /* Release per-table subsidiary storage */
    tables = qs->tables;
    foreach(lc, tables)
    {
        AfterTriggersTableData *table = (AfterTriggersTableData *) lfirst(lc);
 
        ts = table->old_upd_tuplestore;
        table->old_upd_tuplestore = NULL;
        if (ts)
            tuplestore_end(ts);
        ts = table->new_upd_tuplestore;
        table->new_upd_tuplestore = NULL;
        if (ts)
            tuplestore_end(ts);
        ts = table->old_del_tuplestore;
        table->old_del_tuplestore = NULL;
        if (ts)
            tuplestore_end(ts);
        ts = table->new_ins_tuplestore;
        table->new_ins_tuplestore = NULL;
        if (ts)
            tuplestore_end(ts);
        if (table->storeslot)
        {
            TupleTableSlot *slot = table->storeslot;
 
            table->storeslot = NULL;
            ExecDropSingleTupleTableSlot(slot);
        }
    }
 
    /*
     * Now free the AfterTriggersTableData structs and list cells.  Reset list
     * pointer first; if list_free_deep somehow gets an error, better to leak
     * that storage than have an infinite loop.
     */
    qs->tables = NIL;
    list_free_deep(tables);
}

References afterTriggerFreeEventList(), AfterTriggersQueryData::events, ExecDropSingleTupleTableSlot(), AfterTriggersQueryData::fdw_tuplestore, lfirst, list_free_deep(), AfterTriggersTableData::new_ins_tuplestore, AfterTriggersTableData::new_upd_tuplestore, NIL, AfterTriggersTableData::old_del_tuplestore, AfterTriggersTableData::old_upd_tuplestore, AfterTriggersTableData::storeslot, AfterTriggersQueryData::tables, and tuplestore_end().

Referenced by AfterTriggerEndQuery(), and AfterTriggerEndSubXact().

afterTriggerInvokeEvents()

static bool afterTriggerInvokeEvents ( AfterTriggerEventList *  events, CommandId  firing_id, EState *  estate, bool  delete_ok  )

static

Definition at line 4587 of file trigger.c.

{
    bool        all_fired = true;
    AfterTriggerEventChunk *chunk;
    MemoryContext per_tuple_context;
    bool        local_estate = false;
    ResultRelInfo *rInfo = NULL;
    Relation    rel = NULL;
    TriggerDesc *trigdesc = NULL;
    FmgrInfo   *finfo = NULL;
    Instrumentation *instr = NULL;
    TupleTableSlot *slot1 = NULL,
               *slot2 = NULL;
 
    /* Make a local EState if need be */
    if (estate == NULL)
    {
        estate = CreateExecutorState();
        local_estate = true;
    }
 
    /* Make a per-tuple memory context for trigger function calls */
    per_tuple_context =
        AllocSetContextCreate(CurrentMemoryContext,
                              "AfterTriggerTupleContext",
                              ALLOCSET_DEFAULT_SIZES);
 
    for_each_chunk(chunk, *events)
    {
        AfterTriggerEvent event;
        bool        all_fired_in_chunk = true;
 
        for_each_event(event, chunk)
        {
            AfterTriggerShared evtshared = GetTriggerSharedData(event);
 
            /*
             * Is it one for me to fire?
             */
            if ((event->ate_flags & AFTER_TRIGGER_IN_PROGRESS) &&
                evtshared->ats_firing_id == firing_id)
            {
                ResultRelInfo *src_rInfo,
                           *dst_rInfo;
 
                /*
                 * So let's fire it... but first, find the correct relation if
                 * this is not the same relation as before.
                 */
                if (rel == NULL || RelationGetRelid(rel) != evtshared->ats_relid)
                {
                    rInfo = ExecGetTriggerResultRel(estate, evtshared->ats_relid,
                                                    NULL);
                    rel = rInfo->ri_RelationDesc;
                    /* Catch calls with insufficient relcache refcounting */
                    Assert(!RelationHasReferenceCountZero(rel));
                    trigdesc = rInfo->ri_TrigDesc;
                    finfo = rInfo->ri_TrigFunctions;
                    instr = rInfo->ri_TrigInstrument;
                    if (slot1 != NULL)
                    {
                        ExecDropSingleTupleTableSlot(slot1);
                        ExecDropSingleTupleTableSlot(slot2);
                        slot1 = slot2 = NULL;
                    }
                    if (rel->rd_rel->relkind == RELKIND_FOREIGN_TABLE)
                    {
                        slot1 = MakeSingleTupleTableSlot(rel->rd_att,
                                                         &TTSOpsMinimalTuple);
                        slot2 = MakeSingleTupleTableSlot(rel->rd_att,
                                                         &TTSOpsMinimalTuple);
                    }
                    if (trigdesc == NULL)   /* should not happen */
                        elog(ERROR, "relation %u has no triggers",
                             evtshared->ats_relid);
                }
 
                /*
                 * Look up source and destination partition result rels of a
                 * cross-partition update event.
                 */
                if ((event->ate_flags & AFTER_TRIGGER_TUP_BITS) ==
                    AFTER_TRIGGER_CP_UPDATE)
                {
                    Assert(OidIsValid(event->ate_src_part) &&
                           OidIsValid(event->ate_dst_part));
                    src_rInfo = ExecGetTriggerResultRel(estate,
                                                        event->ate_src_part,
                                                        rInfo);
                    dst_rInfo = ExecGetTriggerResultRel(estate,
                                                        event->ate_dst_part,
                                                        rInfo);
                }
                else
                    src_rInfo = dst_rInfo = rInfo;
 
                /*
                 * Fire it.  Note that the AFTER_TRIGGER_IN_PROGRESS flag is
                 * still set, so recursive examinations of the event list
                 * won't try to re-fire it.
                 */
                AfterTriggerExecute(estate, event, rInfo,
                                    src_rInfo, dst_rInfo,
                                    trigdesc, finfo, instr,
                                    per_tuple_context, slot1, slot2);
 
                /*
                 * Mark the event as done.
                 */
                event->ate_flags &= ~AFTER_TRIGGER_IN_PROGRESS;
                event->ate_flags |= AFTER_TRIGGER_DONE;
            }
            else if (!(event->ate_flags & AFTER_TRIGGER_DONE))
            {
                /* something remains to be done */
                all_fired = all_fired_in_chunk = false;
            }
        }
 
        /* Clear the chunk if delete_ok and nothing left of interest */
        if (delete_ok && all_fired_in_chunk)
        {
            chunk->freeptr = CHUNK_DATA_START(chunk);
            chunk->endfree = chunk->endptr;
 
            /*
             * If it's last chunk, must sync event list's tailfree too.  Note
             * that delete_ok must NOT be passed as true if there could be
             * additional AfterTriggerEventList values pointing at this event
             * list, since we'd fail to fix their copies of tailfree.
             */
            if (chunk == events->tail)
                events->tailfree = chunk->freeptr;
        }
    }
    if (slot1 != NULL)
    {
        ExecDropSingleTupleTableSlot(slot1);
        ExecDropSingleTupleTableSlot(slot2);
    }
 
    /* Release working resources */
    MemoryContextDelete(per_tuple_context);
 
    if (local_estate)
    {
        ExecCloseResultRelations(estate);
        ExecResetTupleTable(estate->es_tupleTable, false);
        FreeExecutorState(estate);
    }
 
    return all_fired;
}

References AFTER_TRIGGER_CP_UPDATE, AFTER_TRIGGER_DONE, AFTER_TRIGGER_IN_PROGRESS, AFTER_TRIGGER_TUP_BITS, AfterTriggerExecute(), ALLOCSET_DEFAULT_SIZES, AllocSetContextCreate, Assert(), AfterTriggerEventData::ate_dst_part, AfterTriggerEventData::ate_flags, AfterTriggerEventData::ate_src_part, AfterTriggerSharedData::ats_firing_id, AfterTriggerSharedData::ats_relid, CHUNK_DATA_START, CreateExecutorState(), CurrentMemoryContext, elog(), AfterTriggerEventChunk::endfree, AfterTriggerEventChunk::endptr, ERROR, EState::es_tupleTable, ExecCloseResultRelations(), ExecDropSingleTupleTableSlot(), ExecGetTriggerResultRel(), ExecResetTupleTable(), for_each_chunk, for_each_event, FreeExecutorState(), AfterTriggerEventChunk::freeptr, GetTriggerSharedData, MakeSingleTupleTableSlot(), MemoryContextDelete(), OidIsValid, RelationData::rd_att, RelationData::rd_rel, RelationGetRelid, RelationHasReferenceCountZero, ResultRelInfo::ri_RelationDesc, ResultRelInfo::ri_TrigDesc, ResultRelInfo::ri_TrigFunctions, ResultRelInfo::ri_TrigInstrument, AfterTriggerEventList::tail, AfterTriggerEventList::tailfree, and TTSOpsMinimalTuple.

Referenced by AfterTriggerEndQuery(), AfterTriggerFireDeferred(), and AfterTriggerSetState().

afterTriggerMarkEvents()

static bool afterTriggerMarkEvents ( AfterTriggerEventList *  events, AfterTriggerEventList *  move_list, bool  immediate_only  )

static

Definition at line 4503 of file trigger.c.

{
    bool        found = false;
    bool        deferred_found = false;
    AfterTriggerEvent event;
    AfterTriggerEventChunk *chunk;
 
    for_each_event_chunk(event, chunk, *events)
    {
        AfterTriggerShared evtshared = GetTriggerSharedData(event);
        bool        defer_it = false;
 
        if (!(event->ate_flags &
              (AFTER_TRIGGER_DONE | AFTER_TRIGGER_IN_PROGRESS)))
        {
            /*
             * This trigger hasn't been called or scheduled yet. Check if we
             * should call it now.
             */
            if (immediate_only && afterTriggerCheckState(evtshared))
            {
                defer_it = true;
            }
            else
            {
                /*
                 * Mark it as to be fired in this firing cycle.
                 */
                evtshared->ats_firing_id = afterTriggers.firing_counter;
                event->ate_flags |= AFTER_TRIGGER_IN_PROGRESS;
                found = true;
            }
        }
 
        /*
         * If it's deferred, move it to move_list, if requested.
         */
        if (defer_it && move_list != NULL)
        {
            deferred_found = true;
            /* add it to move_list */
            afterTriggerAddEvent(move_list, event, evtshared);
            /* mark original copy "done" so we don't do it again */
            event->ate_flags |= AFTER_TRIGGER_DONE;
        }
    }
 
    /*
     * We could allow deferred triggers if, before the end of the
     * security-restricted operation, we were to verify that a SET CONSTRAINTS
     * ... IMMEDIATE has fired all such triggers.  For now, don't bother.
     */
    if (deferred_found && InSecurityRestrictedOperation())
        ereport(ERROR,
                (errcode(ERRCODE_INSUFFICIENT_PRIVILEGE),
                 errmsg("cannot fire deferred trigger within security-restricted operation")));
 
    return found;
}

References AFTER_TRIGGER_DONE, AFTER_TRIGGER_IN_PROGRESS, afterTriggerAddEvent(), afterTriggerCheckState(), afterTriggers, AfterTriggerEventData::ate_flags, AfterTriggerSharedData::ats_firing_id, ereport, errcode(), errmsg(), ERROR, AfterTriggersData::firing_counter, for_each_event_chunk, GetTriggerSharedData, and InSecurityRestrictedOperation().

Referenced by AfterTriggerEndQuery(), AfterTriggerFireDeferred(), and AfterTriggerSetState().

AfterTriggerPendingOnRel()

bool AfterTriggerPendingOnRel

(

Oid

relid

)

Definition at line 5936 of file trigger.c.

{
    AfterTriggerEvent event;
    AfterTriggerEventChunk *chunk;
    int         depth;
 
    /* Scan queued events */
    for_each_event_chunk(event, chunk, afterTriggers.events)
    {
        AfterTriggerShared evtshared = GetTriggerSharedData(event);
 
        /*
         * We can ignore completed events.  (Even if a DONE flag is rolled
         * back by subxact abort, it's OK because the effects of the TRUNCATE
         * or whatever must get rolled back too.)
         */
        if (event->ate_flags & AFTER_TRIGGER_DONE)
            continue;
 
        if (evtshared->ats_relid == relid)
            return true;
    }
 
    /*
     * Also scan events queued by incomplete queries.  This could only matter
     * if TRUNCATE/etc is executed by a function or trigger within an updating
     * query on the same relation, which is pretty perverse, but let's check.
     */
    for (depth = 0; depth <= afterTriggers.query_depth && depth < afterTriggers.maxquerydepth; depth++)
    {
        for_each_event_chunk(event, chunk, afterTriggers.query_stack[depth].events)
        {
            AfterTriggerShared evtshared = GetTriggerSharedData(event);
 
            if (event->ate_flags & AFTER_TRIGGER_DONE)
                continue;
 
            if (evtshared->ats_relid == relid)
                return true;
        }
    }
 
    return false;
}

References AFTER_TRIGGER_DONE, afterTriggers, AfterTriggerEventData::ate_flags, AfterTriggerSharedData::ats_relid, AfterTriggersData::events, AfterTriggersQueryData::events, for_each_event_chunk, GetTriggerSharedData, AfterTriggersData::maxquerydepth, AfterTriggersData::query_depth, and AfterTriggersData::query_stack.

Referenced by CheckTableNotInUse().

afterTriggerRestoreEventList()

static void afterTriggerRestoreEventList ( AfterTriggerEventList *  events, const AfterTriggerEventList *  old_events  )

static

Definition at line 4130 of file trigger.c.

{
    AfterTriggerEventChunk *chunk;
    AfterTriggerEventChunk *next_chunk;
 
    if (old_events->tail == NULL)
    {
        /* restoring to a completely empty state, so free everything */
        afterTriggerFreeEventList(events);
    }
    else
    {
        *events = *old_events;
        /* free any chunks after the last one we want to keep */
        for (chunk = events->tail->next; chunk != NULL; chunk = next_chunk)
        {
            next_chunk = chunk->next;
            pfree(chunk);
        }
        /* and clean up the tail chunk to be the right length */
        events->tail->next = NULL;
        events->tail->freeptr = events->tailfree;
 
        /*
         * We don't make any effort to remove now-unused shared data records.
         * They might still be useful, anyway.
         */
    }
}

References afterTriggerFreeEventList(), AfterTriggerEventChunk::freeptr, AfterTriggerEventChunk::next, pfree(), AfterTriggerEventList::tail, and AfterTriggerEventList::tailfree.

Referenced by AfterTriggerEndSubXact().

AfterTriggerSaveEvent()

static void AfterTriggerSaveEvent ( EState *  estate, ResultRelInfo *  relinfo, ResultRelInfo *  src_partinfo, ResultRelInfo *  dst_partinfo, int  event, bool  row_trigger, TupleTableSlot *  oldslot, TupleTableSlot *  newslot, List *  recheckIndexes, Bitmapset *  modifiedCols, TransitionCaptureState *  transition_capture, bool  is_crosspart_update  )

static

Definition at line 6023 of file trigger.c.

{
    Relation    rel = relinfo->ri_RelationDesc;
    TriggerDesc *trigdesc = relinfo->ri_TrigDesc;
    AfterTriggerEventData new_event;
    AfterTriggerSharedData new_shared;
    char        relkind = rel->rd_rel->relkind;
    int         tgtype_event;
    int         tgtype_level;
    int         i;
    Tuplestorestate *fdw_tuplestore = NULL;
 
    /*
     * Check state.  We use a normal test not Assert because it is possible to
     * reach here in the wrong state given misconfigured RI triggers, in
     * particular deferring a cascade action trigger.
     */
    if (afterTriggers.query_depth < 0)
        elog(ERROR, "AfterTriggerSaveEvent() called outside of query");
 
    /* Be sure we have enough space to record events at this query depth. */
    if (afterTriggers.query_depth >= afterTriggers.maxquerydepth)
        AfterTriggerEnlargeQueryState();
 
    /*
     * If the directly named relation has any triggers with transition tables,
     * then we need to capture transition tuples.
     */
    if (row_trigger && transition_capture != NULL)
    {
        TupleTableSlot *original_insert_tuple = transition_capture->tcs_original_insert_tuple;
 
        /*
         * Capture the old tuple in the appropriate transition table based on
         * the event.
         */
        if (!TupIsNull(oldslot))
        {
            Tuplestorestate *old_tuplestore;
 
            old_tuplestore = GetAfterTriggersTransitionTable(event,
                                                             oldslot,
                                                             NULL,
                                                             transition_capture);
            TransitionTableAddTuple(estate, transition_capture, relinfo,
                                    oldslot, NULL, old_tuplestore);
        }
 
        /*
         * Capture the new tuple in the appropriate transition table based on
         * the event.
         */
        if (!TupIsNull(newslot))
        {
            Tuplestorestate *new_tuplestore;
 
            new_tuplestore = GetAfterTriggersTransitionTable(event,
                                                             NULL,
                                                             newslot,
                                                             transition_capture);
            TransitionTableAddTuple(estate, transition_capture, relinfo,
                                    newslot, original_insert_tuple, new_tuplestore);
        }
 
        /*
         * If transition tables are the only reason we're here, return. As
         * mentioned above, we can also be here during update tuple routing in
         * presence of transition tables, in which case this function is
         * called separately for OLD and NEW, so we expect exactly one of them
         * to be NULL.
         */
        if (trigdesc == NULL ||
            (event == TRIGGER_EVENT_DELETE && !trigdesc->trig_delete_after_row) ||
            (event == TRIGGER_EVENT_INSERT && !trigdesc->trig_insert_after_row) ||
            (event == TRIGGER_EVENT_UPDATE && !trigdesc->trig_update_after_row) ||
            (event == TRIGGER_EVENT_UPDATE && (TupIsNull(oldslot) ^ TupIsNull(newslot))))
            return;
    }
 
    /*
     * We normally don't see partitioned tables here for row level triggers
     * except in the special case of a cross-partition update.  In that case,
     * nodeModifyTable.c:ExecCrossPartitionUpdateForeignKey() calls here to
     * queue an update event on the root target partitioned table, also
     * passing the source and destination partitions and their tuples.
     */
    Assert(!row_trigger ||
           rel->rd_rel->relkind != RELKIND_PARTITIONED_TABLE ||
           (is_crosspart_update &&
            TRIGGER_FIRED_BY_UPDATE(event) &&
            src_partinfo != NULL && dst_partinfo != NULL));
 
    /*
     * Validate the event code and collect the associated tuple CTIDs.
     *
     * The event code will be used both as a bitmask and an array offset, so
     * validation is important to make sure we don't walk off the edge of our
     * arrays.
     *
     * Also, if we're considering statement-level triggers, check whether we
     * already queued a set of them for this event, and cancel the prior set
     * if so.  This preserves the behavior that statement-level triggers fire
     * just once per statement and fire after row-level triggers.
     */
    switch (event)
    {
        case TRIGGER_EVENT_INSERT:
            tgtype_event = TRIGGER_TYPE_INSERT;
            if (row_trigger)
            {
                Assert(oldslot == NULL);
                Assert(newslot != NULL);
                ItemPointerCopy(&(newslot->tts_tid), &(new_event.ate_ctid1));
                ItemPointerSetInvalid(&(new_event.ate_ctid2));
            }
            else
            {
                Assert(oldslot == NULL);
                Assert(newslot == NULL);
                ItemPointerSetInvalid(&(new_event.ate_ctid1));
                ItemPointerSetInvalid(&(new_event.ate_ctid2));
                cancel_prior_stmt_triggers(RelationGetRelid(rel),
                                           CMD_INSERT, event);
            }
            break;
        case TRIGGER_EVENT_DELETE:
            tgtype_event = TRIGGER_TYPE_DELETE;
            if (row_trigger)
            {
                Assert(oldslot != NULL);
                Assert(newslot == NULL);
                ItemPointerCopy(&(oldslot->tts_tid), &(new_event.ate_ctid1));
                ItemPointerSetInvalid(&(new_event.ate_ctid2));
            }
            else
            {
                Assert(oldslot == NULL);
                Assert(newslot == NULL);
                ItemPointerSetInvalid(&(new_event.ate_ctid1));
                ItemPointerSetInvalid(&(new_event.ate_ctid2));
                cancel_prior_stmt_triggers(RelationGetRelid(rel),
                                           CMD_DELETE, event);
            }
            break;
        case TRIGGER_EVENT_UPDATE:
            tgtype_event = TRIGGER_TYPE_UPDATE;
            if (row_trigger)
            {
                Assert(oldslot != NULL);
                Assert(newslot != NULL);
                ItemPointerCopy(&(oldslot->tts_tid), &(new_event.ate_ctid1));
                ItemPointerCopy(&(newslot->tts_tid), &(new_event.ate_ctid2));
 
                /*
                 * Also remember the OIDs of partitions to fetch these tuples
                 * out of later in AfterTriggerExecute().
                 */
                if (rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE)
                {
                    Assert(src_partinfo != NULL && dst_partinfo != NULL);
                    new_event.ate_src_part =
                        RelationGetRelid(src_partinfo->ri_RelationDesc);
                    new_event.ate_dst_part =
                        RelationGetRelid(dst_partinfo->ri_RelationDesc);
                }
            }
            else
            {
                Assert(oldslot == NULL);
                Assert(newslot == NULL);
                ItemPointerSetInvalid(&(new_event.ate_ctid1));
                ItemPointerSetInvalid(&(new_event.ate_ctid2));
                cancel_prior_stmt_triggers(RelationGetRelid(rel),
                                           CMD_UPDATE, event);
            }
            break;
        case TRIGGER_EVENT_TRUNCATE:
            tgtype_event = TRIGGER_TYPE_TRUNCATE;
            Assert(oldslot == NULL);
            Assert(newslot == NULL);
            ItemPointerSetInvalid(&(new_event.ate_ctid1));
            ItemPointerSetInvalid(&(new_event.ate_ctid2));
            break;
        default:
            elog(ERROR, "invalid after-trigger event code: %d", event);
            tgtype_event = 0;   /* keep compiler quiet */
            break;
    }
 
    /* Determine flags */
    if (!(relkind == RELKIND_FOREIGN_TABLE && row_trigger))
    {
        if (row_trigger && event == TRIGGER_EVENT_UPDATE)
        {
            if (relkind == RELKIND_PARTITIONED_TABLE)
                new_event.ate_flags = AFTER_TRIGGER_CP_UPDATE;
            else
                new_event.ate_flags = AFTER_TRIGGER_2CTID;
        }
        else
            new_event.ate_flags = AFTER_TRIGGER_1CTID;
    }
 
    /* else, we'll initialize ate_flags for each trigger */
 
    tgtype_level = (row_trigger ? TRIGGER_TYPE_ROW : TRIGGER_TYPE_STATEMENT);
 
    /*
     * Must convert/copy the source and destination partition tuples into the
     * root partitioned table's format/slot, because the processing in the
     * loop below expects both oldslot and newslot tuples to be in that form.
     */
    if (row_trigger && rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE)
    {
        TupleTableSlot *rootslot;
        TupleConversionMap *map;
 
        rootslot = ExecGetTriggerOldSlot(estate, relinfo);
        map = ExecGetChildToRootMap(src_partinfo);
        if (map)
            oldslot = execute_attr_map_slot(map->attrMap,
                                            oldslot,
                                            rootslot);
        else
            oldslot = ExecCopySlot(rootslot, oldslot);
 
        rootslot = ExecGetTriggerNewSlot(estate, relinfo);
        map = ExecGetChildToRootMap(dst_partinfo);
        if (map)
            newslot = execute_attr_map_slot(map->attrMap,
                                            newslot,
                                            rootslot);
        else
            newslot = ExecCopySlot(rootslot, newslot);
    }
 
    for (i = 0; i < trigdesc->numtriggers; i++)
    {
        Trigger    *trigger = &trigdesc->triggers[i];
 
        if (!TRIGGER_TYPE_MATCHES(trigger->tgtype,
                                  tgtype_level,
                                  TRIGGER_TYPE_AFTER,
                                  tgtype_event))
            continue;
        if (!TriggerEnabled(estate, relinfo, trigger, event,
                            modifiedCols, oldslot, newslot))
            continue;
 
        if (relkind == RELKIND_FOREIGN_TABLE && row_trigger)
        {
            if (fdw_tuplestore == NULL)
            {
                fdw_tuplestore = GetCurrentFDWTuplestore();
                new_event.ate_flags = AFTER_TRIGGER_FDW_FETCH;
            }
            else
                /* subsequent event for the same tuple */
                new_event.ate_flags = AFTER_TRIGGER_FDW_REUSE;
        }
 
        /*
         * If the trigger is a foreign key enforcement trigger, there are
         * certain cases where we can skip queueing the event because we can
         * tell by inspection that the FK constraint will still pass. There
         * are also some cases during cross-partition updates of a partitioned
         * table where queuing the event can be skipped.
         */
        if (TRIGGER_FIRED_BY_UPDATE(event) || TRIGGER_FIRED_BY_DELETE(event))
        {
            switch (RI_FKey_trigger_type(trigger->tgfoid))
            {
                case RI_TRIGGER_PK:
 
                    /*
                     * For cross-partitioned updates of partitioned PK table,
                     * skip the event fired by the component delete on the
                     * source leaf partition unless the constraint originates
                     * in the partition itself (!tgisclone), because the
                     * update event that will be fired on the root
                     * (partitioned) target table will be used to perform the
                     * necessary foreign key enforcement action.
                     */
                    if (is_crosspart_update &&
                        TRIGGER_FIRED_BY_DELETE(event) &&
                        trigger->tgisclone)
                        continue;
 
                    /* Update or delete on trigger's PK table */
                    if (!RI_FKey_pk_upd_check_required(trigger, rel,
                                                       oldslot, newslot))
                    {
                        /* skip queuing this event */
                        continue;
                    }
                    break;
 
                case RI_TRIGGER_FK:
 
                    /*
                     * Update on trigger's FK table.  We can skip the update
                     * event fired on a partitioned table during a
                     * cross-partition of that table, because the insert event
                     * that is fired on the destination leaf partition would
                     * suffice to perform the necessary foreign key check.
                     * Moreover, RI_FKey_fk_upd_check_required() expects to be
                     * passed a tuple that contains system attributes, most of
                     * which are not present in the virtual slot belonging to
                     * a partitioned table.
                     */
                    if (rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE ||
                        !RI_FKey_fk_upd_check_required(trigger, rel,
                                                       oldslot, newslot))
                    {
                        /* skip queuing this event */
                        continue;
                    }
                    break;
 
                case RI_TRIGGER_NONE:
 
                    /*
                     * Not an FK trigger.  No need to queue the update event
                     * fired during a cross-partitioned update of a
                     * partitioned table, because the same row trigger must be
                     * present in the leaf partition(s) that are affected as
                     * part of this update and the events fired on them are
                     * queued instead.
                     */
                    if (row_trigger &&
                        rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE)
                        continue;
                    break;
            }
        }
 
        /*
         * If the trigger is a deferred unique constraint check trigger, only
         * queue it if the unique constraint was potentially violated, which
         * we know from index insertion time.
         */
        if (trigger->tgfoid == F_UNIQUE_KEY_RECHECK)
        {
            if (!list_member_oid(recheckIndexes, trigger->tgconstrindid))
                continue;       /* Uniqueness definitely not violated */
        }
 
        /*
         * Fill in event structure and add it to the current query's queue.
         * Note we set ats_table to NULL whenever this trigger doesn't use
         * transition tables, to improve sharability of the shared event data.
         */
        new_shared.ats_event =
            (event & TRIGGER_EVENT_OPMASK) |
            (row_trigger ? TRIGGER_EVENT_ROW : 0) |
            (trigger->tgdeferrable ? AFTER_TRIGGER_DEFERRABLE : 0) |
            (trigger->tginitdeferred ? AFTER_TRIGGER_INITDEFERRED : 0);
        new_shared.ats_tgoid = trigger->tgoid;
        new_shared.ats_relid = RelationGetRelid(rel);
        new_shared.ats_firing_id = 0;
        if ((trigger->tgoldtable || trigger->tgnewtable) &&
            transition_capture != NULL)
            new_shared.ats_table = transition_capture->tcs_private;
        else
            new_shared.ats_table = NULL;
        new_shared.ats_modifiedcols = modifiedCols;
 
        afterTriggerAddEvent(&afterTriggers.query_stack[afterTriggers.query_depth].events,
                             &new_event, &new_shared);
    }
 
    /*
     * Finally, spool any foreign tuple(s).  The tuplestore squashes them to
     * minimal tuples, so this loses any system columns.  The executor lost
     * those columns before us, for an unrelated reason, so this is fine.
     */
    if (fdw_tuplestore)
    {
        if (oldslot != NULL)
            tuplestore_puttupleslot(fdw_tuplestore, oldslot);
        if (newslot != NULL)
            tuplestore_puttupleslot(fdw_tuplestore, newslot);
    }
}

References AFTER_TRIGGER_1CTID, AFTER_TRIGGER_2CTID, AFTER_TRIGGER_CP_UPDATE, AFTER_TRIGGER_DEFERRABLE, AFTER_TRIGGER_FDW_FETCH, AFTER_TRIGGER_FDW_REUSE, AFTER_TRIGGER_INITDEFERRED, afterTriggerAddEvent(), AfterTriggerEnlargeQueryState(), afterTriggers, Assert(), AfterTriggerEventData::ate_ctid1, AfterTriggerEventData::ate_ctid2, AfterTriggerEventData::ate_dst_part, AfterTriggerEventData::ate_flags, AfterTriggerEventData::ate_src_part, AfterTriggerSharedData::ats_event, AfterTriggerSharedData::ats_firing_id, AfterTriggerSharedData::ats_modifiedcols, AfterTriggerSharedData::ats_relid, AfterTriggerSharedData::ats_table, AfterTriggerSharedData::ats_tgoid, TupleConversionMap::attrMap, cancel_prior_stmt_triggers(), CMD_DELETE, CMD_INSERT, CMD_UPDATE, elog(), ERROR, AfterTriggersQueryData::events, ExecCopySlot(), ExecGetChildToRootMap(), ExecGetTriggerNewSlot(), ExecGetTriggerOldSlot(), execute_attr_map_slot(), GetAfterTriggersTransitionTable(), GetCurrentFDWTuplestore(), i, ItemPointerCopy(), ItemPointerSetInvalid(), list_member_oid(), AfterTriggersData::maxquerydepth, TriggerDesc::numtriggers, AfterTriggersData::query_depth, AfterTriggersData::query_stack, RelationData::rd_rel, RelationGetRelid, RI_FKey_fk_upd_check_required(), RI_FKey_pk_upd_check_required(), RI_FKey_trigger_type(), ResultRelInfo::ri_RelationDesc, ResultRelInfo::ri_TrigDesc, RI_TRIGGER_FK, RI_TRIGGER_NONE, RI_TRIGGER_PK, TransitionCaptureState::tcs_original_insert_tuple, TransitionCaptureState::tcs_private, Trigger::tgconstrindid, Trigger::tgdeferrable, Trigger::tgfoid, Trigger::tginitdeferred, Trigger::tgisclone, Trigger::tgnewtable, Trigger::tgoid, Trigger::tgoldtable, Trigger::tgtype, TransitionTableAddTuple(), TriggerDesc::trig_delete_after_row, TriggerDesc::trig_insert_after_row, TriggerDesc::trig_update_after_row, TRIGGER_EVENT_DELETE, TRIGGER_EVENT_INSERT, TRIGGER_EVENT_OPMASK, TRIGGER_EVENT_ROW, TRIGGER_EVENT_TRUNCATE, TRIGGER_EVENT_UPDATE, TRIGGER_FIRED_BY_DELETE, TRIGGER_FIRED_BY_UPDATE, TriggerEnabled(), TriggerDesc::triggers, TupleTableSlot::tts_tid, TupIsNull, and tuplestore_puttupleslot().

Referenced by ExecARDeleteTriggers(), ExecARInsertTriggers(), ExecARUpdateTriggers(), ExecASDeleteTriggers(), ExecASInsertTriggers(), ExecASTruncateTriggers(), and ExecASUpdateTriggers().

AfterTriggerSetState()

void AfterTriggerSetState

(

ConstraintsSetStmt *

stmt

)

Definition at line 5621 of file trigger.c.

{
    int         my_level = GetCurrentTransactionNestLevel();
 
    /* If we haven't already done so, initialize our state. */
    if (afterTriggers.state == NULL)
        afterTriggers.state = SetConstraintStateCreate(8);
 
    /*
     * If in a subtransaction, and we didn't save the current state already,
     * save it so it can be restored if the subtransaction aborts.
     */
    if (my_level > 1 &&
        afterTriggers.trans_stack[my_level].state == NULL)
    {
        afterTriggers.trans_stack[my_level].state =
            SetConstraintStateCopy(afterTriggers.state);
    }
 
    /*
     * Handle SET CONSTRAINTS ALL ...
     */
    if (stmt->constraints == NIL)
    {
        /*
         * Forget any previous SET CONSTRAINTS commands in this transaction.
         */
        afterTriggers.state->numstates = 0;
 
        /*
         * Set the per-transaction ALL state to known.
         */
        afterTriggers.state->all_isset = true;
        afterTriggers.state->all_isdeferred = stmt->deferred;
    }
    else
    {
        Relation    conrel;
        Relation    tgrel;
        List       *conoidlist = NIL;
        List       *tgoidlist = NIL;
        ListCell   *lc;
 
        /*
         * Handle SET CONSTRAINTS constraint-name [, ...]
         *
         * First, identify all the named constraints and make a list of their
         * OIDs.  Since, unlike the SQL spec, we allow multiple constraints of
         * the same name within a schema, the specifications are not
         * necessarily unique.  Our strategy is to target all matching
         * constraints within the first search-path schema that has any
         * matches, but disregard matches in schemas beyond the first match.
         * (This is a bit odd but it's the historical behavior.)
         *
         * A constraint in a partitioned table may have corresponding
         * constraints in the partitions.  Grab those too.
         */
        conrel = table_open(ConstraintRelationId, AccessShareLock);
 
        foreach(lc, stmt->constraints)
        {
            RangeVar   *constraint = lfirst(lc);
            bool        found;
            List       *namespacelist;
            ListCell   *nslc;
 
            if (constraint->catalogname)
            {
                if (strcmp(constraint->catalogname, get_database_name(MyDatabaseId)) != 0)
                    ereport(ERROR,
                            (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                             errmsg("cross-database references are not implemented: \"%s.%s.%s\"",
                                    constraint->catalogname, constraint->schemaname,
                                    constraint->relname)));
            }
 
            /*
             * If we're given the schema name with the constraint, look only
             * in that schema.  If given a bare constraint name, use the
             * search path to find the first matching constraint.
             */
            if (constraint->schemaname)
            {
                Oid         namespaceId = LookupExplicitNamespace(constraint->schemaname,
                                                                  false);
 
                namespacelist = list_make1_oid(namespaceId);
            }
            else
            {
                namespacelist = fetch_search_path(true);
            }
 
            found = false;
            foreach(nslc, namespacelist)
            {
                Oid         namespaceId = lfirst_oid(nslc);
                SysScanDesc conscan;
                ScanKeyData skey[2];
                HeapTuple   tup;
 
                ScanKeyInit(&skey[0],
                            Anum_pg_constraint_conname,
                            BTEqualStrategyNumber, F_NAMEEQ,
                            CStringGetDatum(constraint->relname));
                ScanKeyInit(&skey[1],
                            Anum_pg_constraint_connamespace,
                            BTEqualStrategyNumber, F_OIDEQ,
                            ObjectIdGetDatum(namespaceId));
 
                conscan = systable_beginscan(conrel, ConstraintNameNspIndexId,
                                             true, NULL, 2, skey);
 
                while (HeapTupleIsValid(tup = systable_getnext(conscan)))
                {
                    Form_pg_constraint con = (Form_pg_constraint) GETSTRUCT(tup);
 
                    if (con->condeferrable)
                        conoidlist = lappend_oid(conoidlist, con->oid);
                    else if (stmt->deferred)
                        ereport(ERROR,
                                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                                 errmsg("constraint \"%s\" is not deferrable",
                                        constraint->relname)));
                    found = true;
                }
 
                systable_endscan(conscan);
 
                /*
                 * Once we've found a matching constraint we do not search
                 * later parts of the search path.
                 */
                if (found)
                    break;
            }
 
            list_free(namespacelist);
 
            /*
             * Not found ?
             */
            if (!found)
                ereport(ERROR,
                        (errcode(ERRCODE_UNDEFINED_OBJECT),
                         errmsg("constraint \"%s\" does not exist",
                                constraint->relname)));
        }
 
        /*
         * Scan for any possible descendants of the constraints.  We append
         * whatever we find to the same list that we're scanning; this has the
         * effect that we create new scans for those, too, so if there are
         * further descendents, we'll also catch them.
         */
        foreach(lc, conoidlist)
        {
            Oid         parent = lfirst_oid(lc);
            ScanKeyData key;
            SysScanDesc scan;
            HeapTuple   tuple;
 
            ScanKeyInit(&key,
                        Anum_pg_constraint_conparentid,
                        BTEqualStrategyNumber, F_OIDEQ,
                        ObjectIdGetDatum(parent));
 
            scan = systable_beginscan(conrel, ConstraintParentIndexId, true, NULL, 1, &key);
 
            while (HeapTupleIsValid(tuple = systable_getnext(scan)))
            {
                Form_pg_constraint con = (Form_pg_constraint) GETSTRUCT(tuple);
 
                conoidlist = lappend_oid(conoidlist, con->oid);
            }
 
            systable_endscan(scan);
        }
 
        table_close(conrel, AccessShareLock);
 
        /*
         * Now, locate the trigger(s) implementing each of these constraints,
         * and make a list of their OIDs.
         */
        tgrel = table_open(TriggerRelationId, AccessShareLock);
 
        foreach(lc, conoidlist)
        {
            Oid         conoid = lfirst_oid(lc);
            ScanKeyData skey;
            SysScanDesc tgscan;
            HeapTuple   htup;
 
            ScanKeyInit(&skey,
                        Anum_pg_trigger_tgconstraint,
                        BTEqualStrategyNumber, F_OIDEQ,
                        ObjectIdGetDatum(conoid));
 
            tgscan = systable_beginscan(tgrel, TriggerConstraintIndexId, true,
                                        NULL, 1, &skey);
 
            while (HeapTupleIsValid(htup = systable_getnext(tgscan)))
            {
                Form_pg_trigger pg_trigger = (Form_pg_trigger) GETSTRUCT(htup);
 
                /*
                 * Silently skip triggers that are marked as non-deferrable in
                 * pg_trigger.  This is not an error condition, since a
                 * deferrable RI constraint may have some non-deferrable
                 * actions.
                 */
                if (pg_trigger->tgdeferrable)
                    tgoidlist = lappend_oid(tgoidlist, pg_trigger->oid);
            }
 
            systable_endscan(tgscan);
        }
 
        table_close(tgrel, AccessShareLock);
 
        /*
         * Now we can set the trigger states of individual triggers for this
         * xact.
         */
        foreach(lc, tgoidlist)
        {
            Oid         tgoid = lfirst_oid(lc);
            SetConstraintState state = afterTriggers.state;
            bool        found = false;
            int         i;
 
            for (i = 0; i < state->numstates; i++)
            {
                if (state->trigstates[i].sct_tgoid == tgoid)
                {
                    state->trigstates[i].sct_tgisdeferred = stmt->deferred;
                    found = true;
                    break;
                }
            }
            if (!found)
            {
                afterTriggers.state =
                    SetConstraintStateAddItem(state, tgoid, stmt->deferred);
            }
        }
    }
 
    /*
     * SQL99 requires that when a constraint is set to IMMEDIATE, any deferred
     * checks against that constraint must be made when the SET CONSTRAINTS
     * command is executed -- i.e. the effects of the SET CONSTRAINTS command
     * apply retroactively.  We've updated the constraints state, so scan the
     * list of previously deferred events to fire any that have now become
     * immediate.
     *
     * Obviously, if this was SET ... DEFERRED then it can't have converted
     * any unfired events to immediate, so we need do nothing in that case.
     */
    if (!stmt->deferred)
    {
        AfterTriggerEventList *events = &afterTriggers.events;
        bool        snapshot_set = false;
 
        while (afterTriggerMarkEvents(events, NULL, true))
        {
            CommandId   firing_id = afterTriggers.firing_counter++;
 
            /*
             * Make sure a snapshot has been established in case trigger
             * functions need one.  Note that we avoid setting a snapshot if
             * we don't find at least one trigger that has to be fired now.
             * This is so that BEGIN; SET CONSTRAINTS ...; SET TRANSACTION
             * ISOLATION LEVEL SERIALIZABLE; ... works properly.  (If we are
             * at the start of a transaction it's not possible for any trigger
             * events to be queued yet.)
             */
            if (!snapshot_set)
            {
                PushActiveSnapshot(GetTransactionSnapshot());
                snapshot_set = true;
            }
 
            /*
             * We can delete fired events if we are at top transaction level,
             * but we'd better not if inside a subtransaction, since the
             * subtransaction could later get rolled back.
             */
            if (afterTriggerInvokeEvents(events, firing_id, NULL,
                                         !IsSubTransaction()))
                break;          /* all fired */
        }
 
        if (snapshot_set)
            PopActiveSnapshot();
    }
}

References AccessShareLock, afterTriggerInvokeEvents(), afterTriggerMarkEvents(), afterTriggers, SetConstraintStateData::all_isdeferred, SetConstraintStateData::all_isset, BTEqualStrategyNumber, RangeVar::catalogname, CStringGetDatum(), ereport, errcode(), errmsg(), ERROR, AfterTriggersData::events, fetch_search_path(), AfterTriggersData::firing_counter, get_database_name(), GetCurrentTransactionNestLevel(), GETSTRUCT, GetTransactionSnapshot(), HeapTupleIsValid, i, IsSubTransaction(), sort-test::key, lappend_oid(), lfirst, lfirst_oid, list_free(), list_make1_oid, LookupExplicitNamespace(), MyDatabaseId, NIL, SetConstraintStateData::numstates, ObjectIdGetDatum(), PopActiveSnapshot(), PushActiveSnapshot(), RangeVar::relname, ScanKeyInit(), RangeVar::schemaname, SetConstraintStateAddItem(), SetConstraintStateCopy(), SetConstraintStateCreate(), AfterTriggersData::state, AfterTriggersTransData::state, stmt, systable_beginscan(), systable_endscan(), systable_getnext(), table_close(), table_open(), and AfterTriggersData::trans_stack.

Referenced by standard_ProcessUtility().

assign_session_replication_role()

void assign_session_replication_role	(	int	newval,
		void *	extra
	)

Definition at line 6541 of file trigger.c.

{
    /*
     * Must flush the plan cache when changing replication role; but don't
     * flush unnecessarily.
     */
    if (SessionReplicationRole != newval)
        ResetPlanCache();
}

References newval, ResetPlanCache(), and SessionReplicationRole.

before_stmt_triggers_fired()

static bool before_stmt_triggers_fired ( Oid  relid, CmdType  cmdType  )

static

Definition at line 6420 of file trigger.c.

{
    bool        result;
    AfterTriggersTableData *table;
 
    /* Check state, like AfterTriggerSaveEvent. */
    if (afterTriggers.query_depth < 0)
        elog(ERROR, "before_stmt_triggers_fired() called outside of query");
 
    /* Be sure we have enough space to record events at this query depth. */
    if (afterTriggers.query_depth >= afterTriggers.maxquerydepth)
        AfterTriggerEnlargeQueryState();
 
    /*
     * We keep this state in the AfterTriggersTableData that also holds
     * transition tables for the relation + operation.  In this way, if we are
     * forced to make a new set of transition tables because more tuples get
     * entered after we've already fired triggers, we will allow a new set of
     * statement triggers to get queued.
     */
    table = GetAfterTriggersTableData(relid, cmdType);
    result = table->before_trig_done;
    table->before_trig_done = true;
    return result;
}

References AfterTriggerEnlargeQueryState(), afterTriggers, AfterTriggersTableData::before_trig_done, elog(), ERROR, GetAfterTriggersTableData(), AfterTriggersData::maxquerydepth, and AfterTriggersData::query_depth.

Referenced by ExecBSDeleteTriggers(), ExecBSInsertTriggers(), and ExecBSUpdateTriggers().

cancel_prior_stmt_triggers()

static void cancel_prior_stmt_triggers ( Oid  relid, CmdType  cmdType, int  tgevent  )

static

Definition at line 6466 of file trigger.c.

{
    AfterTriggersTableData *table;
    AfterTriggersQueryData *qs = &afterTriggers.query_stack[afterTriggers.query_depth];
 
    /*
     * We keep this state in the AfterTriggersTableData that also holds
     * transition tables for the relation + operation.  In this way, if we are
     * forced to make a new set of transition tables because more tuples get
     * entered after we've already fired triggers, we will allow a new set of
     * statement triggers to get queued without canceling the old ones.
     */
    table = GetAfterTriggersTableData(relid, cmdType);
 
    if (table->after_trig_done)
    {
        /*
         * We want to start scanning from the tail location that existed just
         * before we inserted any statement triggers.  But the events list
         * might've been entirely empty then, in which case scan from the
         * current head.
         */
        AfterTriggerEvent event;
        AfterTriggerEventChunk *chunk;
 
        if (table->after_trig_events.tail)
        {
            chunk = table->after_trig_events.tail;
            event = (AfterTriggerEvent) table->after_trig_events.tailfree;
        }
        else
        {
            chunk = qs->events.head;
            event = NULL;
        }
 
        for_each_chunk_from(chunk)
        {
            if (event == NULL)
                event = (AfterTriggerEvent) CHUNK_DATA_START(chunk);
            for_each_event_from(event, chunk)
            {
                AfterTriggerShared evtshared = GetTriggerSharedData(event);
 
                /*
                 * Exit loop when we reach events that aren't AS triggers for
                 * the target relation.
                 */
                if (evtshared->ats_relid != relid)
                    goto done;
                if ((evtshared->ats_event & TRIGGER_EVENT_OPMASK) != tgevent)
                    goto done;
                if (!TRIGGER_FIRED_FOR_STATEMENT(evtshared->ats_event))
                    goto done;
                if (!TRIGGER_FIRED_AFTER(evtshared->ats_event))
                    goto done;
                /* OK, mark it DONE */
                event->ate_flags &= ~AFTER_TRIGGER_IN_PROGRESS;
                event->ate_flags |= AFTER_TRIGGER_DONE;
            }
            /* signal we must reinitialize event ptr for next chunk */
            event = NULL;
        }
    }
done:
 
    /* In any case, save current insertion point for next time */
    table->after_trig_done = true;
    table->after_trig_events = qs->events;
}

References AfterTriggersTableData::after_trig_done, AfterTriggersTableData::after_trig_events, AFTER_TRIGGER_DONE, AFTER_TRIGGER_IN_PROGRESS, afterTriggers, AfterTriggerSharedData::ats_event, AfterTriggerSharedData::ats_relid, CHUNK_DATA_START, AfterTriggersQueryData::events, for_each_chunk_from, for_each_event_from, GetAfterTriggersTableData(), GetTriggerSharedData, AfterTriggerEventList::head, AfterTriggersData::query_depth, AfterTriggersData::query_stack, AfterTriggerEventList::tail, AfterTriggerEventList::tailfree, TRIGGER_EVENT_OPMASK, TRIGGER_FIRED_AFTER, and TRIGGER_FIRED_FOR_STATEMENT.

Referenced by AfterTriggerSaveEvent().

CopyTriggerDesc()

TriggerDesc* CopyTriggerDesc

(

TriggerDesc *

trigdesc

)

Definition at line 2096 of file trigger.c.

{
    TriggerDesc *newdesc;
    Trigger    *trigger;
    int         i;
 
    if (trigdesc == NULL || trigdesc->numtriggers <= 0)
        return NULL;
 
    newdesc = (TriggerDesc *) palloc(sizeof(TriggerDesc));
    memcpy(newdesc, trigdesc, sizeof(TriggerDesc));
 
    trigger = (Trigger *) palloc(trigdesc->numtriggers * sizeof(Trigger));
    memcpy(trigger, trigdesc->triggers,
           trigdesc->numtriggers * sizeof(Trigger));
    newdesc->triggers = trigger;
 
    for (i = 0; i < trigdesc->numtriggers; i++)
    {
        trigger->tgname = pstrdup(trigger->tgname);
        if (trigger->tgnattr > 0)
        {
            int16      *newattr;
 
            newattr = (int16 *) palloc(trigger->tgnattr * sizeof(int16));
            memcpy(newattr, trigger->tgattr,
                   trigger->tgnattr * sizeof(int16));
            trigger->tgattr = newattr;
        }
        if (trigger->tgnargs > 0)
        {
            char      **newargs;
            int16       j;
 
            newargs = (char **) palloc(trigger->tgnargs * sizeof(char *));
            for (j = 0; j < trigger->tgnargs; j++)
                newargs[j] = pstrdup(trigger->tgargs[j]);
            trigger->tgargs = newargs;
        }
        if (trigger->tgqual)
            trigger->tgqual = pstrdup(trigger->tgqual);
        if (trigger->tgoldtable)
            trigger->tgoldtable = pstrdup(trigger->tgoldtable);
        if (trigger->tgnewtable)
            trigger->tgnewtable = pstrdup(trigger->tgnewtable);
        trigger++;
    }
 
    return newdesc;
}

References i, j, TriggerDesc::numtriggers, palloc(), pstrdup(), Trigger::tgargs, Trigger::tgattr, Trigger::tgname, Trigger::tgnargs, Trigger::tgnattr, Trigger::tgnewtable, Trigger::tgoldtable, Trigger::tgqual, and TriggerDesc::triggers.

Referenced by InitResultRelInfo(), and RelationBuildTriggers().

CreateTrigger()

ObjectAddress CreateTrigger	(	CreateTrigStmt *	stmt,
		const char *	queryString,
		Oid	relOid,
		Oid	refRelOid,
		Oid	constraintOid,
		Oid	indexOid,
		Oid	funcoid,
		Oid	parentTriggerOid,
		Node *	whenClause,
		bool	isInternal,
		bool	in_partition
	)

Definition at line 165 of file trigger.c.

{
    return
        CreateTriggerFiringOn(stmt, queryString, relOid, refRelOid,
                              constraintOid, indexOid, funcoid,
                              parentTriggerOid, whenClause, isInternal,
                              in_partition, TRIGGER_FIRES_ON_ORIGIN);
}

References CreateTriggerFiringOn(), stmt, and TRIGGER_FIRES_ON_ORIGIN.

Referenced by CreateFKCheckTrigger(), createForeignKeyActionTriggers(), index_constraint_create(), and ProcessUtilitySlow().

CreateTriggerFiringOn()

ObjectAddress CreateTriggerFiringOn	(	CreateTrigStmt *	stmt,
		const char *	queryString,
		Oid	relOid,
		Oid	refRelOid,
		Oid	constraintOid,
		Oid	indexOid,
		Oid	funcoid,
		Oid	parentTriggerOid,
		Node *	whenClause,
		bool	isInternal,
		bool	in_partition,
		char	trigger_fires_when
	)

Definition at line 182 of file trigger.c.

{
    int16       tgtype;
    int         ncolumns;
    int16      *columns;
    int2vector *tgattr;
    List       *whenRtable;
    char       *qual;
    Datum       values[Natts_pg_trigger];
    bool        nulls[Natts_pg_trigger];
    Relation    rel;
    AclResult   aclresult;
    Relation    tgrel;
    Relation    pgrel;
    HeapTuple   tuple = NULL;
    Oid         funcrettype;
    Oid         trigoid = InvalidOid;
    char        internaltrigname[NAMEDATALEN];
    char       *trigname;
    Oid         constrrelid = InvalidOid;
    ObjectAddress myself,
                referenced;
    char       *oldtablename = NULL;
    char       *newtablename = NULL;
    bool        partition_recurse;
    bool        trigger_exists = false;
    Oid         existing_constraint_oid = InvalidOid;
    bool        existing_isInternal = false;
    bool        existing_isClone = false;
 
    if (OidIsValid(relOid))
        rel = table_open(relOid, ShareRowExclusiveLock);
    else
        rel = table_openrv(stmt->relation, ShareRowExclusiveLock);
 
    /*
     * Triggers must be on tables or views, and there are additional
     * relation-type-specific restrictions.
     */
    if (rel->rd_rel->relkind == RELKIND_RELATION)
    {
        /* Tables can't have INSTEAD OF triggers */
        if (stmt->timing != TRIGGER_TYPE_BEFORE &&
            stmt->timing != TRIGGER_TYPE_AFTER)
            ereport(ERROR,
                    (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                     errmsg("\"%s\" is a table",
                            RelationGetRelationName(rel)),
                     errdetail("Tables cannot have INSTEAD OF triggers.")));
    }
    else if (rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE)
    {
        /* Partitioned tables can't have INSTEAD OF triggers */
        if (stmt->timing != TRIGGER_TYPE_BEFORE &&
            stmt->timing != TRIGGER_TYPE_AFTER)
            ereport(ERROR,
                    (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                     errmsg("\"%s\" is a table",
                            RelationGetRelationName(rel)),
                     errdetail("Tables cannot have INSTEAD OF triggers.")));
 
        /*
         * FOR EACH ROW triggers have further restrictions
         */
        if (stmt->row)
        {
            /*
             * Disallow use of transition tables.
             *
             * Note that we have another restriction about transition tables
             * in partitions; search for 'has_superclass' below for an
             * explanation.  The check here is just to protect from the fact
             * that if we allowed it here, the creation would succeed for a
             * partitioned table with no partitions, but would be blocked by
             * the other restriction when the first partition was created,
             * which is very unfriendly behavior.
             */
            if (stmt->transitionRels != NIL)
                ereport(ERROR,
                        (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                         errmsg("\"%s\" is a partitioned table",
                                RelationGetRelationName(rel)),
                         errdetail("ROW triggers with transition tables are not supported on partitioned tables.")));
        }
    }
    else if (rel->rd_rel->relkind == RELKIND_VIEW)
    {
        /*
         * Views can have INSTEAD OF triggers (which we check below are
         * row-level), or statement-level BEFORE/AFTER triggers.
         */
        if (stmt->timing != TRIGGER_TYPE_INSTEAD && stmt->row)
            ereport(ERROR,
                    (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                     errmsg("\"%s\" is a view",
                            RelationGetRelationName(rel)),
                     errdetail("Views cannot have row-level BEFORE or AFTER triggers.")));
        /* Disallow TRUNCATE triggers on VIEWs */
        if (TRIGGER_FOR_TRUNCATE(stmt->events))
            ereport(ERROR,
                    (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                     errmsg("\"%s\" is a view",
                            RelationGetRelationName(rel)),
                     errdetail("Views cannot have TRUNCATE triggers.")));
    }
    else if (rel->rd_rel->relkind == RELKIND_FOREIGN_TABLE)
    {
        if (stmt->timing != TRIGGER_TYPE_BEFORE &&
            stmt->timing != TRIGGER_TYPE_AFTER)
            ereport(ERROR,
                    (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                     errmsg("\"%s\" is a foreign table",
                            RelationGetRelationName(rel)),
                     errdetail("Foreign tables cannot have INSTEAD OF triggers.")));
 
        /*
         * We disallow constraint triggers to protect the assumption that
         * triggers on FKs can't be deferred.  See notes with AfterTriggers
         * data structures, below.
         */
        if (stmt->isconstraint)
            ereport(ERROR,
                    (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                     errmsg("\"%s\" is a foreign table",
                            RelationGetRelationName(rel)),
                     errdetail("Foreign tables cannot have constraint triggers.")));
    }
    else
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("relation \"%s\" cannot have triggers",
                        RelationGetRelationName(rel)),
                 errdetail_relkind_not_supported(rel->rd_rel->relkind)));
 
    if (!allowSystemTableMods && IsSystemRelation(rel))
        ereport(ERROR,
                (errcode(ERRCODE_INSUFFICIENT_PRIVILEGE),
                 errmsg("permission denied: \"%s\" is a system catalog",
                        RelationGetRelationName(rel))));
 
    if (stmt->isconstraint)
    {
        /*
         * We must take a lock on the target relation to protect against
         * concurrent drop.  It's not clear that AccessShareLock is strong
         * enough, but we certainly need at least that much... otherwise, we
         * might end up creating a pg_constraint entry referencing a
         * nonexistent table.
         */
        if (OidIsValid(refRelOid))
        {
            LockRelationOid(refRelOid, AccessShareLock);
            constrrelid = refRelOid;
        }
        else if (stmt->constrrel != NULL)
            constrrelid = RangeVarGetRelid(stmt->constrrel, AccessShareLock,
                                           false);
    }
 
    /* permission checks */
    if (!isInternal)
    {
        aclresult = pg_class_aclcheck(RelationGetRelid(rel), GetUserId(),
                                      ACL_TRIGGER);
        if (aclresult != ACLCHECK_OK)
            aclcheck_error(aclresult, get_relkind_objtype(rel->rd_rel->relkind),
                           RelationGetRelationName(rel));
 
        if (OidIsValid(constrrelid))
        {
            aclresult = pg_class_aclcheck(constrrelid, GetUserId(),
                                          ACL_TRIGGER);
            if (aclresult != ACLCHECK_OK)
                aclcheck_error(aclresult, get_relkind_objtype(get_rel_relkind(constrrelid)),
                               get_rel_name(constrrelid));
        }
    }
 
    /*
     * When called on a partitioned table to create a FOR EACH ROW trigger
     * that's not internal, we create one trigger for each partition, too.
     *
     * For that, we'd better hold lock on all of them ahead of time.
     */
    partition_recurse = !isInternal && stmt->row &&
        rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE;
    if (partition_recurse)
        list_free(find_all_inheritors(RelationGetRelid(rel),
                                      ShareRowExclusiveLock, NULL));
 
    /* Compute tgtype */
    TRIGGER_CLEAR_TYPE(tgtype);
    if (stmt->row)
        TRIGGER_SETT_ROW(tgtype);
    tgtype |= stmt->timing;
    tgtype |= stmt->events;
 
    /* Disallow ROW-level TRUNCATE triggers */
    if (TRIGGER_FOR_ROW(tgtype) && TRIGGER_FOR_TRUNCATE(tgtype))
        ereport(ERROR,
                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                 errmsg("TRUNCATE FOR EACH ROW triggers are not supported")));
 
    /* INSTEAD triggers must be row-level, and can't have WHEN or columns */
    if (TRIGGER_FOR_INSTEAD(tgtype))
    {
        if (!TRIGGER_FOR_ROW(tgtype))
            ereport(ERROR,
                    (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                     errmsg("INSTEAD OF triggers must be FOR EACH ROW")));
        if (stmt->whenClause)
            ereport(ERROR,
                    (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                     errmsg("INSTEAD OF triggers cannot have WHEN conditions")));
        if (stmt->columns != NIL)
            ereport(ERROR,
                    (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                     errmsg("INSTEAD OF triggers cannot have column lists")));
    }
 
    /*
     * We don't yet support naming ROW transition variables, but the parser
     * recognizes the syntax so we can give a nicer message here.
     *
     * Per standard, REFERENCING TABLE names are only allowed on AFTER
     * triggers.  Per standard, REFERENCING ROW names are not allowed with FOR
     * EACH STATEMENT.  Per standard, each OLD/NEW, ROW/TABLE permutation is
     * only allowed once.  Per standard, OLD may not be specified when
     * creating a trigger only for INSERT, and NEW may not be specified when
     * creating a trigger only for DELETE.
     *
     * Notice that the standard allows an AFTER ... FOR EACH ROW trigger to
     * reference both ROW and TABLE transition data.
     */
    if (stmt->transitionRels != NIL)
    {
        List       *varList = stmt->transitionRels;
        ListCell   *lc;
 
        foreach(lc, varList)
        {
            TriggerTransition *tt = lfirst_node(TriggerTransition, lc);
 
            if (!(tt->isTable))
                ereport(ERROR,
                        (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                         errmsg("ROW variable naming in the REFERENCING clause is not supported"),
                         errhint("Use OLD TABLE or NEW TABLE for naming transition tables.")));
 
            /*
             * Because of the above test, we omit further ROW-related testing
             * below.  If we later allow naming OLD and NEW ROW variables,
             * adjustments will be needed below.
             */
 
            if (rel->rd_rel->relkind == RELKIND_FOREIGN_TABLE)
                ereport(ERROR,
                        (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                         errmsg("\"%s\" is a foreign table",
                                RelationGetRelationName(rel)),
                         errdetail("Triggers on foreign tables cannot have transition tables.")));
 
            if (rel->rd_rel->relkind == RELKIND_VIEW)
                ereport(ERROR,
                        (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                         errmsg("\"%s\" is a view",
                                RelationGetRelationName(rel)),
                         errdetail("Triggers on views cannot have transition tables.")));
 
            /*
             * We currently don't allow row-level triggers with transition
             * tables on partition or inheritance children.  Such triggers
             * would somehow need to see tuples converted to the format of the
             * table they're attached to, and it's not clear which subset of
             * tuples each child should see.  See also the prohibitions in
             * ATExecAttachPartition() and ATExecAddInherit().
             */
            if (TRIGGER_FOR_ROW(tgtype) && has_superclass(rel->rd_id))
            {
                /* Use appropriate error message. */
                if (rel->rd_rel->relispartition)
                    ereport(ERROR,
                            (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                             errmsg("ROW triggers with transition tables are not supported on partitions")));
                else
                    ereport(ERROR,
                            (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                             errmsg("ROW triggers with transition tables are not supported on inheritance children")));
            }
 
            if (stmt->timing != TRIGGER_TYPE_AFTER)
                ereport(ERROR,
                        (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
                         errmsg("transition table name can only be specified for an AFTER trigger")));
 
            if (TRIGGER_FOR_TRUNCATE(tgtype))
                ereport(ERROR,
                        (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                         errmsg("TRUNCATE triggers with transition tables are not supported")));
 
            /*
             * We currently don't allow multi-event triggers ("INSERT OR
             * UPDATE") with transition tables, because it's not clear how to
             * handle INSERT ... ON CONFLICT statements which can fire both
             * INSERT and UPDATE triggers.  We show the inserted tuples to
             * INSERT triggers and the updated tuples to UPDATE triggers, but
             * it's not yet clear what INSERT OR UPDATE trigger should see.
             * This restriction could be lifted if we can decide on the right
             * semantics in a later release.
             */
            if (((TRIGGER_FOR_INSERT(tgtype) ? 1 : 0) +
                 (TRIGGER_FOR_UPDATE(tgtype) ? 1 : 0) +
                 (TRIGGER_FOR_DELETE(tgtype) ? 1 : 0)) != 1)
                ereport(ERROR,
                        (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                         errmsg("transition tables cannot be specified for triggers with more than one event")));
 
            /*
             * We currently don't allow column-specific triggers with
             * transition tables.  Per spec, that seems to require
             * accumulating separate transition tables for each combination of
             * columns, which is a lot of work for a rather marginal feature.
             */
            if (stmt->columns != NIL)
                ereport(ERROR,
                        (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                         errmsg("transition tables cannot be specified for triggers with column lists")));
 
            /*
             * We disallow constraint triggers with transition tables, to
             * protect the assumption that such triggers can't be deferred.
             * See notes with AfterTriggers data structures, below.
             *
             * Currently this is enforced by the grammar, so just Assert here.
             */
            Assert(!stmt->isconstraint);
 
            if (tt->isNew)
            {
                if (!(TRIGGER_FOR_INSERT(tgtype) ||
                      TRIGGER_FOR_UPDATE(tgtype)))
                    ereport(ERROR,
                            (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
                             errmsg("NEW TABLE can only be specified for an INSERT or UPDATE trigger")));
 
                if (newtablename != NULL)
                    ereport(ERROR,
                            (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
                             errmsg("NEW TABLE cannot be specified multiple times")));
 
                newtablename = tt->name;
            }
            else
            {
                if (!(TRIGGER_FOR_DELETE(tgtype) ||
                      TRIGGER_FOR_UPDATE(tgtype)))
                    ereport(ERROR,
                            (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
                             errmsg("OLD TABLE can only be specified for a DELETE or UPDATE trigger")));
 
                if (oldtablename != NULL)
                    ereport(ERROR,
                            (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
                             errmsg("OLD TABLE cannot be specified multiple times")));
 
                oldtablename = tt->name;
            }
        }
 
        if (newtablename != NULL && oldtablename != NULL &&
            strcmp(newtablename, oldtablename) == 0)
            ereport(ERROR,
                    (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
                     errmsg("OLD TABLE name and NEW TABLE name cannot be the same")));
    }
 
    /*
     * Parse the WHEN clause, if any and we weren't passed an already
     * transformed one.
     *
     * Note that as a side effect, we fill whenRtable when parsing.  If we got
     * an already parsed clause, this does not occur, which is what we want --
     * no point in adding redundant dependencies below.
     */
    if (!whenClause && stmt->whenClause)
    {
        ParseState *pstate;
        ParseNamespaceItem *nsitem;
        List       *varList;
        ListCell   *lc;
 
        /* Set up a pstate to parse with */
        pstate = make_parsestate(NULL);
        pstate->p_sourcetext = queryString;
 
        /*
         * Set up nsitems for OLD and NEW references.
         *
         * 'OLD' must always have varno equal to 1 and 'NEW' equal to 2.
         */
        nsitem = addRangeTableEntryForRelation(pstate, rel,
                                               AccessShareLock,
                                               makeAlias("old", NIL),
                                               false, false);
        addNSItemToQuery(pstate, nsitem, false, true, true);
        nsitem = addRangeTableEntryForRelation(pstate, rel,
                                               AccessShareLock,
                                               makeAlias("new", NIL),
                                               false, false);
        addNSItemToQuery(pstate, nsitem, false, true, true);
 
        /* Transform expression.  Copy to be sure we don't modify original */
        whenClause = transformWhereClause(pstate,
                                          copyObject(stmt->whenClause),
                                          EXPR_KIND_TRIGGER_WHEN,
                                          "WHEN");
        /* we have to fix its collations too */
        assign_expr_collations(pstate, whenClause);
 
        /*
         * Check for disallowed references to OLD/NEW.
         *
         * NB: pull_var_clause is okay here only because we don't allow
         * subselects in WHEN clauses; it would fail to examine the contents
         * of subselects.
         */
        varList = pull_var_clause(whenClause, 0);
        foreach(lc, varList)
        {
            Var        *var = (Var *) lfirst(lc);
 
            switch (var->varno)
            {
                case PRS2_OLD_VARNO:
                    if (!TRIGGER_FOR_ROW(tgtype))
                        ereport(ERROR,
                                (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
                                 errmsg("statement trigger's WHEN condition cannot reference column values"),
                                 parser_errposition(pstate, var->location)));
                    if (TRIGGER_FOR_INSERT(tgtype))
                        ereport(ERROR,
                                (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
                                 errmsg("INSERT trigger's WHEN condition cannot reference OLD values"),
                                 parser_errposition(pstate, var->location)));
                    /* system columns are okay here */
                    break;
                case PRS2_NEW_VARNO:
                    if (!TRIGGER_FOR_ROW(tgtype))
                        ereport(ERROR,
                                (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
                                 errmsg("statement trigger's WHEN condition cannot reference column values"),
                                 parser_errposition(pstate, var->location)));
                    if (TRIGGER_FOR_DELETE(tgtype))
                        ereport(ERROR,
                                (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
                                 errmsg("DELETE trigger's WHEN condition cannot reference NEW values"),
                                 parser_errposition(pstate, var->location)));
                    if (var->varattno < 0 && TRIGGER_FOR_BEFORE(tgtype))
                        ereport(ERROR,
                                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                                 errmsg("BEFORE trigger's WHEN condition cannot reference NEW system columns"),
                                 parser_errposition(pstate, var->location)));
                    if (TRIGGER_FOR_BEFORE(tgtype) &&
                        var->varattno == 0 &&
                        RelationGetDescr(rel)->constr &&
                        RelationGetDescr(rel)->constr->has_generated_stored)
                        ereport(ERROR,
                                (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
                                 errmsg("BEFORE trigger's WHEN condition cannot reference NEW generated columns"),
                                 errdetail("A whole-row reference is used and the table contains generated columns."),
                                 parser_errposition(pstate, var->location)));
                    if (TRIGGER_FOR_BEFORE(tgtype) &&
                        var->varattno > 0 &&
                        TupleDescAttr(RelationGetDescr(rel), var->varattno - 1)->attgenerated)
                        ereport(ERROR,
                                (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
                                 errmsg("BEFORE trigger's WHEN condition cannot reference NEW generated columns"),
                                 errdetail("Column \"%s\" is a generated column.",
                                           NameStr(TupleDescAttr(RelationGetDescr(rel), var->varattno - 1)->attname)),
                                 parser_errposition(pstate, var->location)));
                    break;
                default:
                    /* can't happen without add_missing_from, so just elog */
                    elog(ERROR, "trigger WHEN condition cannot contain references to other relations");
                    break;
            }
        }
 
        /* we'll need the rtable for recordDependencyOnExpr */
        whenRtable = pstate->p_rtable;
 
        qual = nodeToString(whenClause);
 
        free_parsestate(pstate);
    }
    else if (!whenClause)
    {
        whenClause = NULL;
        whenRtable = NIL;
        qual = NULL;
    }
    else
    {
        qual = nodeToString(whenClause);
        whenRtable = NIL;
    }
 
    /*
     * Find and validate the trigger function.
     */
    if (!OidIsValid(funcoid))
        funcoid = LookupFuncName(stmt->funcname, 0, NULL, false);
    if (!isInternal)
    {
        aclresult = object_aclcheck(ProcedureRelationId, funcoid, GetUserId(), ACL_EXECUTE);
        if (aclresult != ACLCHECK_OK)
            aclcheck_error(aclresult, OBJECT_FUNCTION,
                           NameListToString(stmt->funcname));
    }
    funcrettype = get_func_rettype(funcoid);
    if (funcrettype != TRIGGEROID)
        ereport(ERROR,
                (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
                 errmsg("function %s must return type %s",
                        NameListToString(stmt->funcname), "trigger")));
 
    /*
     * Scan pg_trigger to see if there is already a trigger of the same name.
     * Skip this for internally generated triggers, since we'll modify the
     * name to be unique below.
     *
     * NOTE that this is cool only because we have ShareRowExclusiveLock on
     * the relation, so the trigger set won't be changing underneath us.
     */
    tgrel = table_open(TriggerRelationId, RowExclusiveLock);
    if (!isInternal)
    {
        ScanKeyData skeys[2];
        SysScanDesc tgscan;
 
        ScanKeyInit(&skeys[0],
                    Anum_pg_trigger_tgrelid,
                    BTEqualStrategyNumber, F_OIDEQ,
                    ObjectIdGetDatum(RelationGetRelid(rel)));
 
        ScanKeyInit(&skeys[1],
                    Anum_pg_trigger_tgname,
                    BTEqualStrategyNumber, F_NAMEEQ,
                    CStringGetDatum(stmt->trigname));
 
        tgscan = systable_beginscan(tgrel, TriggerRelidNameIndexId, true,
                                    NULL, 2, skeys);
 
        /* There should be at most one matching tuple */
        if (HeapTupleIsValid(tuple = systable_getnext(tgscan)))
        {
            Form_pg_trigger oldtrigger = (Form_pg_trigger) GETSTRUCT(tuple);
 
            trigoid = oldtrigger->oid;
            existing_constraint_oid = oldtrigger->tgconstraint;
            existing_isInternal = oldtrigger->tgisinternal;
            existing_isClone = OidIsValid(oldtrigger->tgparentid);
            trigger_exists = true;
            /* copy the tuple to use in CatalogTupleUpdate() */
            tuple = heap_copytuple(tuple);
        }
        systable_endscan(tgscan);
    }
 
    if (!trigger_exists)
    {
        /* Generate the OID for the new trigger. */
        trigoid = GetNewOidWithIndex(tgrel, TriggerOidIndexId,
                                     Anum_pg_trigger_oid);
    }
    else
    {
        /*
         * If OR REPLACE was specified, we'll replace the old trigger;
         * otherwise complain about the duplicate name.
         */
        if (!stmt->replace)
            ereport(ERROR,
                    (errcode(ERRCODE_DUPLICATE_OBJECT),
                     errmsg("trigger \"%s\" for relation \"%s\" already exists",
                            stmt->trigname, RelationGetRelationName(rel))));
 
        /*
         * An internal trigger or a child trigger (isClone) cannot be replaced
         * by a user-defined trigger.  However, skip this test when
         * in_partition, because then we're recursing from a partitioned table
         * and the check was made at the parent level.
         */
        if ((existing_isInternal || existing_isClone) &&
            !isInternal && !in_partition)
            ereport(ERROR,
                    (errcode(ERRCODE_DUPLICATE_OBJECT),
                     errmsg("trigger \"%s\" for relation \"%s\" is an internal or a child trigger",
                            stmt->trigname, RelationGetRelationName(rel))));
 
        /*
         * It is not allowed to replace with a constraint trigger; gram.y
         * should have enforced this already.
         */
        Assert(!stmt->isconstraint);
 
        /*
         * It is not allowed to replace an existing constraint trigger,
         * either.  (The reason for these restrictions is partly that it seems
         * difficult to deal with pending trigger events in such cases, and
         * partly that the command might imply changing the constraint's
         * properties as well, which doesn't seem nice.)
         */
        if (OidIsValid(existing_constraint_oid))
            ereport(ERROR,
                    (errcode(ERRCODE_DUPLICATE_OBJECT),
                     errmsg("trigger \"%s\" for relation \"%s\" is a constraint trigger",
                            stmt->trigname, RelationGetRelationName(rel))));
    }
 
    /*
     * If it's a user-entered CREATE CONSTRAINT TRIGGER command, make a
     * corresponding pg_constraint entry.
     */
    if (stmt->isconstraint && !OidIsValid(constraintOid))
    {
        /* Internal callers should have made their own constraints */
        Assert(!isInternal);
        constraintOid = CreateConstraintEntry(stmt->trigname,
                                              RelationGetNamespace(rel),
                                              CONSTRAINT_TRIGGER,
                                              stmt->deferrable,
                                              stmt->initdeferred,
                                              true,
                                              InvalidOid,   /* no parent */
                                              RelationGetRelid(rel),
                                              NULL, /* no conkey */
                                              0,
                                              0,
                                              InvalidOid,   /* no domain */
                                              InvalidOid,   /* no index */
                                              InvalidOid,   /* no foreign key */
                                              NULL,
                                              NULL,
                                              NULL,
                                              NULL,
                                              0,
                                              ' ',
                                              ' ',
                                              NULL,
                                              0,
                                              ' ',
                                              NULL, /* no exclusion */
                                              NULL, /* no check constraint */
                                              NULL,
                                              true, /* islocal */
                                              0,    /* inhcount */
                                              true, /* noinherit */
                                              isInternal);  /* is_internal */
    }
 
    /*
     * If trigger is internally generated, modify the provided trigger name to
     * ensure uniqueness by appending the trigger OID.  (Callers will usually
     * supply a simple constant trigger name in these cases.)
     */
    if (isInternal)
    {
        snprintf(internaltrigname, sizeof(internaltrigname),
                 "%s_%u", stmt->trigname, trigoid);
        trigname = internaltrigname;
    }
    else
    {
        /* user-defined trigger; use the specified trigger name as-is */
        trigname = stmt->trigname;
    }
 
    /*
     * Build the new pg_trigger tuple.
     *
     * When we're creating a trigger in a partition, we mark it as internal,
     * even though we don't do the isInternal magic in this function.  This
     * makes the triggers in partitions identical to the ones in the
     * partitioned tables, except that they are marked internal.
     */
    memset(nulls, false, sizeof(nulls));
 
    values[Anum_pg_trigger_oid - 1] = ObjectIdGetDatum(trigoid);
    values[Anum_pg_trigger_tgrelid - 1] = ObjectIdGetDatum(RelationGetRelid(rel));
    values[Anum_pg_trigger_tgparentid - 1] = ObjectIdGetDatum(parentTriggerOid);
    values[Anum_pg_trigger_tgname - 1] = DirectFunctionCall1(namein,
                                                             CStringGetDatum(trigname));
    values[Anum_pg_trigger_tgfoid - 1] = ObjectIdGetDatum(funcoid);
    values[Anum_pg_trigger_tgtype - 1] = Int16GetDatum(tgtype);
    values[Anum_pg_trigger_tgenabled - 1] = trigger_fires_when;
    values[Anum_pg_trigger_tgisinternal - 1] = BoolGetDatum(isInternal);
    values[Anum_pg_trigger_tgconstrrelid - 1] = ObjectIdGetDatum(constrrelid);
    values[Anum_pg_trigger_tgconstrindid - 1] = ObjectIdGetDatum(indexOid);
    values[Anum_pg_trigger_tgconstraint - 1] = ObjectIdGetDatum(constraintOid);
    values[Anum_pg_trigger_tgdeferrable - 1] = BoolGetDatum(stmt->deferrable);
    values[Anum_pg_trigger_tginitdeferred - 1] = BoolGetDatum(stmt->initdeferred);
 
    if (stmt->args)
    {
        ListCell   *le;
        char       *args;
        int16       nargs = list_length(stmt->args);
        int         len = 0;
 
        foreach(le, stmt->args)
        {
            char       *ar = strVal(lfirst(le));
 
            len += strlen(ar) + 4;
            for (; *ar; ar++)
            {
                if (*ar == '\\')
                    len++;
            }
        }
        args = (char *) palloc(len + 1);
        args[0] = '\0';
        foreach(le, stmt->args)
        {
            char       *s = strVal(lfirst(le));
            char       *d = args + strlen(args);
 
            while (*s)
            {
                if (*s == '\\')
                    *d++ = '\\';
                *d++ = *s++;
            }
            strcpy(d, "\\000");
        }
        values[Anum_pg_trigger_tgnargs - 1] = Int16GetDatum(nargs);
        values[Anum_pg_trigger_tgargs - 1] = DirectFunctionCall1(byteain,
                                                                 CStringGetDatum(args));
    }
    else
    {
        values[Anum_pg_trigger_tgnargs - 1] = Int16GetDatum(0);
        values[Anum_pg_trigger_tgargs - 1] = DirectFunctionCall1(byteain,
                                                                 CStringGetDatum(""));
    }
 
    /* build column number array if it's a column-specific trigger */
    ncolumns = list_length(stmt->columns);
    if (ncolumns == 0)
        columns = NULL;
    else
    {
        ListCell   *cell;
        int         i = 0;
 
        columns = (int16 *) palloc(ncolumns * sizeof(int16));
        foreach(cell, stmt->columns)
        {
            char       *name = strVal(lfirst(cell));
            int16       attnum;
            int         j;
 
            /* Lookup column name.  System columns are not allowed */
            attnum = attnameAttNum(rel, name, false);
            if (attnum == InvalidAttrNumber)
                ereport(ERROR,
                        (errcode(ERRCODE_UNDEFINED_COLUMN),
                         errmsg("column \"%s\" of relation \"%s\" does not exist",
                                name, RelationGetRelationName(rel))));
 
            /* Check for duplicates */
            for (j = i - 1; j >= 0; j--)
            {
                if (columns[j] == attnum)
                    ereport(ERROR,
                            (errcode(ERRCODE_DUPLICATE_COLUMN),
                             errmsg("column \"%s\" specified more than once",
                                    name)));
            }
 
            columns[i++] = attnum;
        }
    }
    tgattr = buildint2vector(columns, ncolumns);
    values[Anum_pg_trigger_tgattr - 1] = PointerGetDatum(tgattr);
 
    /* set tgqual if trigger has WHEN clause */
    if (qual)
        values[Anum_pg_trigger_tgqual - 1] = CStringGetTextDatum(qual);
    else
        nulls[Anum_pg_trigger_tgqual - 1] = true;
 
    if (oldtablename)
        values[Anum_pg_trigger_tgoldtable - 1] = DirectFunctionCall1(namein,
                                                                     CStringGetDatum(oldtablename));
    else
        nulls[Anum_pg_trigger_tgoldtable - 1] = true;
    if (newtablename)
        values[Anum_pg_trigger_tgnewtable - 1] = DirectFunctionCall1(namein,
                                                                     CStringGetDatum(newtablename));
    else
        nulls[Anum_pg_trigger_tgnewtable - 1] = true;
 
    /*
     * Insert or replace tuple in pg_trigger.
     */
    if (!trigger_exists)
    {
        tuple = heap_form_tuple(tgrel->rd_att, values, nulls);
        CatalogTupleInsert(tgrel, tuple);
    }
    else
    {
        HeapTuple   newtup;
 
        newtup = heap_form_tuple(tgrel->rd_att, values, nulls);
        CatalogTupleUpdate(tgrel, &tuple->t_self, newtup);
        heap_freetuple(newtup);
    }
 
    heap_freetuple(tuple);      /* free either original or new tuple */
    table_close(tgrel, RowExclusiveLock);
 
    pfree(DatumGetPointer(values[Anum_pg_trigger_tgname - 1]));
    pfree(DatumGetPointer(values[Anum_pg_trigger_tgargs - 1]));
    pfree(DatumGetPointer(values[Anum_pg_trigger_tgattr - 1]));
    if (oldtablename)
        pfree(DatumGetPointer(values[Anum_pg_trigger_tgoldtable - 1]));
    if (newtablename)
        pfree(DatumGetPointer(values[Anum_pg_trigger_tgnewtable - 1]));
 
    /*
     * Update relation's pg_class entry; if necessary; and if not, send an SI
     * message to make other backends (and this one) rebuild relcache entries.
     */
    pgrel = table_open(RelationRelationId, RowExclusiveLock);
    tuple = SearchSysCacheCopy1(RELOID,
                                ObjectIdGetDatum(RelationGetRelid(rel)));
    if (!HeapTupleIsValid(tuple))
        elog(ERROR, "cache lookup failed for relation %u",
             RelationGetRelid(rel));
    if (!((Form_pg_class) GETSTRUCT(tuple))->relhastriggers)
    {
        ((Form_pg_class) GETSTRUCT(tuple))->relhastriggers = true;
 
        CatalogTupleUpdate(pgrel, &tuple->t_self, tuple);
 
        CommandCounterIncrement();
    }
    else
        CacheInvalidateRelcacheByTuple(tuple);
 
    heap_freetuple(tuple);
    table_close(pgrel, RowExclusiveLock);
 
    /*
     * If we're replacing a trigger, flush all the old dependencies before
     * recording new ones.
     */
    if (trigger_exists)
        deleteDependencyRecordsFor(TriggerRelationId, trigoid, true);
 
    /*
     * Record dependencies for trigger.  Always place a normal dependency on
     * the function.
     */
    myself.classId = TriggerRelationId;
    myself.objectId = trigoid;
    myself.objectSubId = 0;
 
    referenced.classId = ProcedureRelationId;
    referenced.objectId = funcoid;
    referenced.objectSubId = 0;
    recordDependencyOn(&myself, &referenced, DEPENDENCY_NORMAL);
 
    if (isInternal && OidIsValid(constraintOid))
    {
        /*
         * Internally-generated trigger for a constraint, so make it an
         * internal dependency of the constraint.  We can skip depending on
         * the relation(s), as there'll be an indirect dependency via the
         * constraint.
         */
        referenced.classId = ConstraintRelationId;
        referenced.objectId = constraintOid;
        referenced.objectSubId = 0;
        recordDependencyOn(&myself, &referenced, DEPENDENCY_INTERNAL);
    }
    else
    {
        /*
         * User CREATE TRIGGER, so place dependencies.  We make trigger be
         * auto-dropped if its relation is dropped or if the FK relation is
         * dropped.  (Auto drop is compatible with our pre-7.3 behavior.)
         */
        referenced.classId = RelationRelationId;
        referenced.objectId = RelationGetRelid(rel);
        referenced.objectSubId = 0;
        recordDependencyOn(&myself, &referenced, DEPENDENCY_AUTO);
 
        if (OidIsValid(constrrelid))
        {
            referenced.classId = RelationRelationId;
            referenced.objectId = constrrelid;
            referenced.objectSubId = 0;
            recordDependencyOn(&myself, &referenced, DEPENDENCY_AUTO);
        }
        /* Not possible to have an index dependency in this case */
        Assert(!OidIsValid(indexOid));
 
        /*
         * If it's a user-specified constraint trigger, make the constraint
         * internally dependent on the trigger instead of vice versa.
         */
        if (OidIsValid(constraintOid))
        {
            referenced.classId = ConstraintRelationId;
            referenced.objectId = constraintOid;
            referenced.objectSubId = 0;
            recordDependencyOn(&referenced, &myself, DEPENDENCY_INTERNAL);
        }
 
        /*
         * If it's a partition trigger, create the partition dependencies.
         */
        if (OidIsValid(parentTriggerOid))
        {
            ObjectAddressSet(referenced, TriggerRelationId, parentTriggerOid);
            recordDependencyOn(&myself, &referenced, DEPENDENCY_PARTITION_PRI);
            ObjectAddressSet(referenced, RelationRelationId, RelationGetRelid(rel));
            recordDependencyOn(&myself, &referenced, DEPENDENCY_PARTITION_SEC);
        }
    }
 
    /* If column-specific trigger, add normal dependencies on columns */
    if (columns != NULL)
    {
        int         i;
 
        referenced.classId = RelationRelationId;
        referenced.objectId = RelationGetRelid(rel);
        for (i = 0; i < ncolumns; i++)
        {
            referenced.objectSubId = columns[i];
            recordDependencyOn(&myself, &referenced, DEPENDENCY_NORMAL);
        }
    }
 
    /*
     * If it has a WHEN clause, add dependencies on objects mentioned in the
     * expression (eg, functions, as well as any columns used).
     */
    if (whenRtable != NIL)
        recordDependencyOnExpr(&myself, whenClause, whenRtable,
                               DEPENDENCY_NORMAL);
 
    /* Post creation hook for new trigger */
    InvokeObjectPostCreateHookArg(TriggerRelationId, trigoid, 0,
                                  isInternal);
 
    /*
     * Lastly, create the trigger on child relations, if needed.
     */
    if (partition_recurse)
    {
        PartitionDesc partdesc = RelationGetPartitionDesc(rel, true);
        int         i;
        MemoryContext oldcxt,
                    perChildCxt;
 
        perChildCxt = AllocSetContextCreate(CurrentMemoryContext,
                                            "part trig clone",
                                            ALLOCSET_SMALL_SIZES);
 
        /*
         * We don't currently expect to be called with a valid indexOid.  If
         * that ever changes then we'll need to write code here to find the
         * corresponding child index.
         */
        Assert(!OidIsValid(indexOid));
 
        oldcxt = MemoryContextSwitchTo(perChildCxt);
 
        /* Iterate to create the trigger on each existing partition */
        for (i = 0; i < partdesc->nparts; i++)
        {
            CreateTrigStmt *childStmt;
            Relation    childTbl;
            Node       *qual;
 
            childTbl = table_open(partdesc->oids[i], ShareRowExclusiveLock);
 
            /*
             * Initialize our fabricated parse node by copying the original
             * one, then resetting fields that we pass separately.
             */
            childStmt = (CreateTrigStmt *) copyObject(stmt);
            childStmt->funcname = NIL;
            childStmt->whenClause = NULL;
 
            /* If there is a WHEN clause, create a modified copy of it */
            qual = copyObject(whenClause);
            qual = (Node *)
                map_partition_varattnos((List *) qual, PRS2_OLD_VARNO,
                                        childTbl, rel);
            qual = (Node *)
                map_partition_varattnos((List *) qual, PRS2_NEW_VARNO,
                                        childTbl, rel);
 
            CreateTriggerFiringOn(childStmt, queryString,
                                  partdesc->oids[i], refRelOid,
                                  InvalidOid, InvalidOid,
                                  funcoid, trigoid, qual,
                                  isInternal, true, trigger_fires_when);
 
            table_close(childTbl, NoLock);
 
            MemoryContextReset(perChildCxt);
        }
 
        MemoryContextSwitchTo(oldcxt);
        MemoryContextDelete(perChildCxt);
    }
 
    /* Keep lock on target rel until end of xact */
    table_close(rel, NoLock);
 
    return myself;
}

References AccessShareLock, ACL_EXECUTE, ACL_TRIGGER, aclcheck_error(), ACLCHECK_OK, addNSItemToQuery(), addRangeTableEntryForRelation(), ALLOCSET_SMALL_SIZES, AllocSetContextCreate, allowSystemTableMods, generate_unaccent_rules::args, Assert(), assign_expr_collations(), attnameAttNum(), attnum, BoolGetDatum(), BTEqualStrategyNumber, buildint2vector(), byteain(), CacheInvalidateRelcacheByTuple(), CatalogTupleInsert(), CatalogTupleUpdate(), ObjectAddress::classId, CommandCounterIncrement(), copyObject, CreateConstraintEntry(), CStringGetDatum(), CStringGetTextDatum, CurrentMemoryContext, DatumGetPointer(), deleteDependencyRecordsFor(), DEPENDENCY_AUTO, DEPENDENCY_INTERNAL, DEPENDENCY_NORMAL, DEPENDENCY_PARTITION_PRI, DEPENDENCY_PARTITION_SEC, DirectFunctionCall1, elog(), ereport, errcode(), ERRCODE_DUPLICATE_OBJECT, errdetail(), errdetail_relkind_not_supported(), errhint(), errmsg(), ERROR, EXPR_KIND_TRIGGER_WHEN, find_all_inheritors(), free_parsestate(), CreateTrigStmt::funcname, get_func_rettype(), get_rel_name(), get_rel_relkind(), get_relkind_objtype(), GetNewOidWithIndex(), GETSTRUCT, GetUserId(), has_superclass(), heap_copytuple(), heap_form_tuple(), heap_freetuple(), HeapTupleIsValid, i, Int16GetDatum(), InvalidAttrNumber, InvalidOid, InvokeObjectPostCreateHookArg, TriggerTransition::isNew, IsSystemRelation(), TriggerTransition::isTable, j, len, lfirst, lfirst_node, list_free(), list_length(), Var::location, LockRelationOid(), LookupFuncName(), make_parsestate(), makeAlias(), map_partition_varattnos(), MemoryContextDelete(), MemoryContextReset(), MemoryContextSwitchTo(), name, TriggerTransition::name, NAMEDATALEN, namein(), NameListToString(), NameStr, NIL, nodeToString(), NoLock, PartitionDescData::nparts, object_aclcheck(), OBJECT_FUNCTION, ObjectAddressSet, ObjectAddress::objectId, ObjectIdGetDatum(), ObjectAddress::objectSubId, OidIsValid, PartitionDescData::oids, ParseState::p_rtable, ParseState::p_sourcetext, palloc(), parser_errposition(), pfree(), pg_class_aclcheck(), PointerGetDatum(), PRS2_NEW_VARNO, PRS2_OLD_VARNO, pull_var_clause(), RangeVarGetRelid, RelationData::rd_att, RelationData::rd_id, RelationData::rd_rel, recordDependencyOn(), recordDependencyOnExpr(), RelationGetDescr, RelationGetNamespace, RelationGetPartitionDesc(), RelationGetRelationName, RelationGetRelid, RELOID, RowExclusiveLock, ScanKeyInit(), SearchSysCacheCopy1, ShareRowExclusiveLock, snprintf, stmt, strVal, systable_beginscan(), systable_endscan(), systable_getnext(), HeapTupleData::t_self, table_close(), table_open(), table_openrv(), transformWhereClause(), TupleDescAttr, values, Var::varattno, Var::varno, and CreateTrigStmt::whenClause.

Referenced by CloneRowTriggersToPartition(), and CreateTrigger().

EnableDisableTrigger()

void EnableDisableTrigger	(	Relation	rel,
		const char *	tgname,
		Oid	tgparent,
		char	fires_when,
		bool	skip_system,
		bool	recurse,
		LOCKMODE	lockmode
	)

Definition at line 1732 of file trigger.c.

{
    Relation    tgrel;
    int         nkeys;
    ScanKeyData keys[2];
    SysScanDesc tgscan;
    HeapTuple   tuple;
    bool        found;
    bool        changed;
 
    /* Scan the relevant entries in pg_triggers */
    tgrel = table_open(TriggerRelationId, RowExclusiveLock);
 
    ScanKeyInit(&keys[0],
                Anum_pg_trigger_tgrelid,
                BTEqualStrategyNumber, F_OIDEQ,
                ObjectIdGetDatum(RelationGetRelid(rel)));
    if (tgname)
    {
        ScanKeyInit(&keys[1],
                    Anum_pg_trigger_tgname,
                    BTEqualStrategyNumber, F_NAMEEQ,
                    CStringGetDatum(tgname));
        nkeys = 2;
    }
    else
        nkeys = 1;
 
    tgscan = systable_beginscan(tgrel, TriggerRelidNameIndexId, true,
                                NULL, nkeys, keys);
 
    found = changed = false;
 
    while (HeapTupleIsValid(tuple = systable_getnext(tgscan)))
    {
        Form_pg_trigger oldtrig = (Form_pg_trigger) GETSTRUCT(tuple);
 
        if (OidIsValid(tgparent) && tgparent != oldtrig->tgparentid)
            continue;
 
        if (oldtrig->tgisinternal)
        {
            /* system trigger ... ok to process? */
            if (skip_system)
                continue;
            if (!superuser())
                ereport(ERROR,
                        (errcode(ERRCODE_INSUFFICIENT_PRIVILEGE),
                         errmsg("permission denied: \"%s\" is a system trigger",
                                NameStr(oldtrig->tgname))));
        }
 
        found = true;
 
        if (oldtrig->tgenabled != fires_when)
        {
            /* need to change this one ... make a copy to scribble on */
            HeapTuple   newtup = heap_copytuple(tuple);
            Form_pg_trigger newtrig = (Form_pg_trigger) GETSTRUCT(newtup);
 
            newtrig->tgenabled = fires_when;
 
            CatalogTupleUpdate(tgrel, &newtup->t_self, newtup);
 
            heap_freetuple(newtup);
 
            changed = true;
        }
 
        /*
         * When altering FOR EACH ROW triggers on a partitioned table, do the
         * same on the partitions as well, unless ONLY is specified.
         *
         * Note that we recurse even if we didn't change the trigger above,
         * because the partitions' copy of the trigger may have a different
         * value of tgenabled than the parent's trigger and thus might need to
         * be changed.
         */
        if (recurse &&
            rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE &&
            (TRIGGER_FOR_ROW(oldtrig->tgtype)))
        {
            PartitionDesc partdesc = RelationGetPartitionDesc(rel, true);
            int         i;
 
            for (i = 0; i < partdesc->nparts; i++)
            {
                Relation    part;
 
                part = relation_open(partdesc->oids[i], lockmode);
                /* Match on child triggers' tgparentid, not their name */
                EnableDisableTrigger(part, NULL, oldtrig->oid,
                                     fires_when, skip_system, recurse,
                                     lockmode);
                table_close(part, NoLock);  /* keep lock till commit */
            }
        }
 
        InvokeObjectPostAlterHook(TriggerRelationId,
                                  oldtrig->oid, 0);
    }
 
    systable_endscan(tgscan);
 
    table_close(tgrel, RowExclusiveLock);
 
    if (tgname && !found)
        ereport(ERROR,
                (errcode(ERRCODE_UNDEFINED_OBJECT),
                 errmsg("trigger \"%s\" for table \"%s\" does not exist",
                        tgname, RelationGetRelationName(rel))));
 
    /*
     * If we changed anything, broadcast a SI inval message to force each
     * backend (including our own!) to rebuild relation's relcache entry.
     * Otherwise they will fail to apply the change promptly.
     */
    if (changed)
        CacheInvalidateRelcache(rel);
}

References BTEqualStrategyNumber, CacheInvalidateRelcache(), CatalogTupleUpdate(), CStringGetDatum(), ereport, errcode(), errmsg(), ERROR, GETSTRUCT, heap_copytuple(), heap_freetuple(), HeapTupleIsValid, i, InvokeObjectPostAlterHook, NameStr, NoLock, PartitionDescData::nparts, ObjectIdGetDatum(), OidIsValid, PartitionDescData::oids, RelationData::rd_rel, relation_open(), RelationGetPartitionDesc(), RelationGetRelationName, RelationGetRelid, RowExclusiveLock, ScanKeyInit(), superuser(), systable_beginscan(), systable_endscan(), systable_getnext(), HeapTupleData::t_self, table_close(), and table_open().

Referenced by ATExecEnableDisableTrigger().

ExecARDeleteTriggers()

void ExecARDeleteTriggers	(	EState *	estate,
		ResultRelInfo *	relinfo,
		ItemPointer	tupleid,
		HeapTuple	fdw_trigtuple,
		TransitionCaptureState *	transition_capture,
		bool	is_crosspart_update
	)

Definition at line 2785 of file trigger.c.

{
    TriggerDesc *trigdesc = relinfo->ri_TrigDesc;
 
    if ((trigdesc && trigdesc->trig_delete_after_row) ||
        (transition_capture && transition_capture->tcs_delete_old_table))
    {
        TupleTableSlot *slot = ExecGetTriggerOldSlot(estate, relinfo);
 
        Assert(HeapTupleIsValid(fdw_trigtuple) ^ ItemPointerIsValid(tupleid));
        if (fdw_trigtuple == NULL)
            GetTupleForTrigger(estate,
                               NULL,
                               relinfo,
                               tupleid,
                               LockTupleExclusive,
                               slot,
                               NULL,
                               NULL,
                               NULL);
        else
            ExecForceStoreHeapTuple(fdw_trigtuple, slot, false);
 
        AfterTriggerSaveEvent(estate, relinfo, NULL, NULL,
                              TRIGGER_EVENT_DELETE,
                              true, slot, NULL, NIL, NULL,
                              transition_capture,
                              is_crosspart_update);
    }
}

References AfterTriggerSaveEvent(), Assert(), ExecForceStoreHeapTuple(), ExecGetTriggerOldSlot(), GetTupleForTrigger(), HeapTupleIsValid, ItemPointerIsValid(), LockTupleExclusive, NIL, ResultRelInfo::ri_TrigDesc, TransitionCaptureState::tcs_delete_old_table, TriggerDesc::trig_delete_after_row, and TRIGGER_EVENT_DELETE.

Referenced by ExecDeleteEpilogue(), and ExecSimpleRelationDelete().

ExecARInsertTriggers()

void ExecARInsertTriggers	(	EState *	estate,
		ResultRelInfo *	relinfo,
		TupleTableSlot *	slot,
		List *	recheckIndexes,
		TransitionCaptureState *	transition_capture
	)

Definition at line 2545 of file trigger.c.

{
    TriggerDesc *trigdesc = relinfo->ri_TrigDesc;
 
    if ((trigdesc && trigdesc->trig_insert_after_row) ||
        (transition_capture && transition_capture->tcs_insert_new_table))
        AfterTriggerSaveEvent(estate, relinfo, NULL, NULL,
                              TRIGGER_EVENT_INSERT,
                              true, NULL, slot,
                              recheckIndexes, NULL,
                              transition_capture,
                              false);
}

References AfterTriggerSaveEvent(), ResultRelInfo::ri_TrigDesc, TransitionCaptureState::tcs_insert_new_table, TriggerDesc::trig_insert_after_row, and TRIGGER_EVENT_INSERT.

Referenced by CopyFrom(), CopyMultiInsertBufferFlush(), ExecBatchInsert(), ExecInsert(), and ExecSimpleRelationInsert().

ExecARUpdateTriggers()

void ExecARUpdateTriggers	(	EState *	estate,
		ResultRelInfo *	relinfo,
		ResultRelInfo *	src_partinfo,
		ResultRelInfo *	dst_partinfo,
		ItemPointer	tupleid,
		HeapTuple	fdw_trigtuple,
		TupleTableSlot *	newslot,
		List *	recheckIndexes,
		TransitionCaptureState *	transition_capture,
		bool	is_crosspart_update
	)

Definition at line 3090 of file trigger.c.

{
    TriggerDesc *trigdesc = relinfo->ri_TrigDesc;
 
    if ((trigdesc && trigdesc->trig_update_after_row) ||
        (transition_capture &&
         (transition_capture->tcs_update_old_table ||
          transition_capture->tcs_update_new_table)))
    {
        /*
         * Note: if the UPDATE is converted into a DELETE+INSERT as part of
         * update-partition-key operation, then this function is also called
         * separately for DELETE and INSERT to capture transition table rows.
         * In such case, either old tuple or new tuple can be NULL.
         */
        TupleTableSlot *oldslot;
        ResultRelInfo *tupsrc;
 
        Assert((src_partinfo != NULL && dst_partinfo != NULL) ||
               !is_crosspart_update);
 
        tupsrc = src_partinfo ? src_partinfo : relinfo;
        oldslot = ExecGetTriggerOldSlot(estate, tupsrc);
 
        if (fdw_trigtuple == NULL && ItemPointerIsValid(tupleid))
            GetTupleForTrigger(estate,
                               NULL,
                               tupsrc,
                               tupleid,
                               LockTupleExclusive,
                               oldslot,
                               NULL,
                               NULL,
                               NULL);
        else if (fdw_trigtuple != NULL)
            ExecForceStoreHeapTuple(fdw_trigtuple, oldslot, false);
        else
            ExecClearTuple(oldslot);
 
        AfterTriggerSaveEvent(estate, relinfo,
                              src_partinfo, dst_partinfo,
                              TRIGGER_EVENT_UPDATE,
                              true,
                              oldslot, newslot, recheckIndexes,
                              ExecGetAllUpdatedCols(relinfo, estate),
                              transition_capture,
                              is_crosspart_update);
    }
}

References AfterTriggerSaveEvent(), Assert(), ExecClearTuple(), ExecForceStoreHeapTuple(), ExecGetAllUpdatedCols(), ExecGetTriggerOldSlot(), GetTupleForTrigger(), ItemPointerIsValid(), LockTupleExclusive, ResultRelInfo::ri_TrigDesc, TransitionCaptureState::tcs_update_new_table, TransitionCaptureState::tcs_update_old_table, TriggerDesc::trig_update_after_row, and TRIGGER_EVENT_UPDATE.

Referenced by ExecCrossPartitionUpdateForeignKey(), ExecDeleteEpilogue(), ExecInsert(), ExecSimpleRelationUpdate(), and ExecUpdateEpilogue().

ExecASDeleteTriggers()

void ExecASDeleteTriggers	(	EState *	estate,
		ResultRelInfo *	relinfo,
		TransitionCaptureState *	transition_capture
	)

Definition at line 2674 of file trigger.c.

{
    TriggerDesc *trigdesc = relinfo->ri_TrigDesc;
 
    if (trigdesc && trigdesc->trig_delete_after_statement)
        AfterTriggerSaveEvent(estate, relinfo, NULL, NULL,
                              TRIGGER_EVENT_DELETE,
                              false, NULL, NULL, NIL, NULL, transition_capture,
                              false);
}

References AfterTriggerSaveEvent(), NIL, ResultRelInfo::ri_TrigDesc, TriggerDesc::trig_delete_after_statement, and TRIGGER_EVENT_DELETE.

Referenced by fireASTriggers().

ExecASInsertTriggers()

void ExecASInsertTriggers	(	EState *	estate,
		ResultRelInfo *	relinfo,
		TransitionCaptureState *	transition_capture
	)

Definition at line 2456 of file trigger.c.

{
    TriggerDesc *trigdesc = relinfo->ri_TrigDesc;
 
    if (trigdesc && trigdesc->trig_insert_after_statement)
        AfterTriggerSaveEvent(estate, relinfo, NULL, NULL,
                              TRIGGER_EVENT_INSERT,
                              false, NULL, NULL, NIL, NULL, transition_capture,
                              false);
}

References AfterTriggerSaveEvent(), NIL, ResultRelInfo::ri_TrigDesc, TriggerDesc::trig_insert_after_statement, and TRIGGER_EVENT_INSERT.

Referenced by CopyFrom(), and fireASTriggers().

ExecASTruncateTriggers()

void ExecASTruncateTriggers	(	EState *	estate,
		ResultRelInfo *	relinfo
	)

Definition at line 3262 of file trigger.c.

{
    TriggerDesc *trigdesc = relinfo->ri_TrigDesc;
 
    if (trigdesc && trigdesc->trig_truncate_after_statement)
        AfterTriggerSaveEvent(estate, relinfo,
                              NULL, NULL,
                              TRIGGER_EVENT_TRUNCATE,
                              false, NULL, NULL, NIL, NULL, NULL,
                              false);
}

References AfterTriggerSaveEvent(), NIL, ResultRelInfo::ri_TrigDesc, TriggerDesc::trig_truncate_after_statement, and TRIGGER_EVENT_TRUNCATE.

Referenced by ExecuteTruncateGuts().

ExecASUpdateTriggers()

void ExecASUpdateTriggers	(	EState *	estate,
		ResultRelInfo *	relinfo,
		TransitionCaptureState *	transition_capture
	)

Definition at line 2927 of file trigger.c.

{
    TriggerDesc *trigdesc = relinfo->ri_TrigDesc;
 
    /* statement-level triggers operate on the parent table */
    Assert(relinfo->ri_RootResultRelInfo == NULL);
 
    if (trigdesc && trigdesc->trig_update_after_statement)
        AfterTriggerSaveEvent(estate, relinfo, NULL, NULL,
                              TRIGGER_EVENT_UPDATE,
                              false, NULL, NULL, NIL,
                              ExecGetAllUpdatedCols(relinfo, estate),
                              transition_capture,
                              false);
}

References AfterTriggerSaveEvent(), Assert(), ExecGetAllUpdatedCols(), NIL, ResultRelInfo::ri_RootResultRelInfo, ResultRelInfo::ri_TrigDesc, TriggerDesc::trig_update_after_statement, and TRIGGER_EVENT_UPDATE.

Referenced by fireASTriggers().

ExecBRDeleteTriggers()

bool ExecBRDeleteTriggers	(	EState *	estate,
		EPQState *	epqstate,
		ResultRelInfo *	relinfo,
		ItemPointer	tupleid,
		HeapTuple	fdw_trigtuple,
		TupleTableSlot **	epqslot,
		TM_Result *	tmresult,
		TM_FailureData *	tmfd
	)

Definition at line 2694 of file trigger.c.

{
    TupleTableSlot *slot = ExecGetTriggerOldSlot(estate, relinfo);
    TriggerDesc *trigdesc = relinfo->ri_TrigDesc;
    bool        result = true;
    TriggerData LocTriggerData = {0};
    HeapTuple   trigtuple;
    bool        should_free = false;
    int         i;
 
    Assert(HeapTupleIsValid(fdw_trigtuple) ^ ItemPointerIsValid(tupleid));
    if (fdw_trigtuple == NULL)
    {
        TupleTableSlot *epqslot_candidate = NULL;
 
        if (!GetTupleForTrigger(estate, epqstate, relinfo, tupleid,
                                LockTupleExclusive, slot, &epqslot_candidate,
                                tmresult, tmfd))
            return false;
 
        /*
         * If the tuple was concurrently updated and the caller of this
         * function requested for the updated tuple, skip the trigger
         * execution.
         */
        if (epqslot_candidate != NULL && epqslot != NULL)
        {
            *epqslot = epqslot_candidate;
            return false;
        }
 
        trigtuple = ExecFetchSlotHeapTuple(slot, true, &should_free);
    }
    else
    {
        trigtuple = fdw_trigtuple;
        ExecForceStoreHeapTuple(trigtuple, slot, false);
    }
 
    LocTriggerData.type = T_TriggerData;
    LocTriggerData.tg_event = TRIGGER_EVENT_DELETE |
        TRIGGER_EVENT_ROW |
        TRIGGER_EVENT_BEFORE;
    LocTriggerData.tg_relation = relinfo->ri_RelationDesc;
    for (i = 0; i < trigdesc->numtriggers; i++)
    {
        HeapTuple   newtuple;
        Trigger    *trigger = &trigdesc->triggers[i];
 
        if (!TRIGGER_TYPE_MATCHES(trigger->tgtype,
                                  TRIGGER_TYPE_ROW,
                                  TRIGGER_TYPE_BEFORE,
                                  TRIGGER_TYPE_DELETE))
            continue;
        if (!TriggerEnabled(estate, relinfo, trigger, LocTriggerData.tg_event,
                            NULL, slot, NULL))
            continue;
 
        LocTriggerData.tg_trigslot = slot;
        LocTriggerData.tg_trigtuple = trigtuple;
        LocTriggerData.tg_trigger = trigger;
        newtuple = ExecCallTriggerFunc(&LocTriggerData,
                                       i,
                                       relinfo->ri_TrigFunctions,
                                       relinfo->ri_TrigInstrument,
                                       GetPerTupleMemoryContext(estate));
        if (newtuple == NULL)
        {
            result = false;     /* tell caller to suppress delete */
            break;
        }
        if (newtuple != trigtuple)
            heap_freetuple(newtuple);
    }
    if (should_free)
        heap_freetuple(trigtuple);
 
    return result;
}

References Assert(), ExecCallTriggerFunc(), ExecFetchSlotHeapTuple(), ExecForceStoreHeapTuple(), ExecGetTriggerOldSlot(), GetPerTupleMemoryContext, GetTupleForTrigger(), heap_freetuple(), HeapTupleIsValid, i, ItemPointerIsValid(), LockTupleExclusive, TriggerDesc::numtriggers, ResultRelInfo::ri_RelationDesc, ResultRelInfo::ri_TrigDesc, ResultRelInfo::ri_TrigFunctions, ResultRelInfo::ri_TrigInstrument, TriggerData::tg_event, TriggerData::tg_relation, TriggerData::tg_trigger, TriggerData::tg_trigslot, TriggerData::tg_trigtuple, Trigger::tgtype, TRIGGER_EVENT_BEFORE, TRIGGER_EVENT_DELETE, TRIGGER_EVENT_ROW, TriggerEnabled(), TriggerDesc::triggers, and TriggerData::type.

Referenced by ExecDeletePrologue(), and ExecSimpleRelationDelete().

ExecBRInsertTriggers()

bool ExecBRInsertTriggers	(	EState *	estate,
		ResultRelInfo *	relinfo,
		TupleTableSlot *	slot
	)

Definition at line 2469 of file trigger.c.

{
    TriggerDesc *trigdesc = relinfo->ri_TrigDesc;
    HeapTuple   newtuple = NULL;
    bool        should_free;
    TriggerData LocTriggerData = {0};
    int         i;
 
    LocTriggerData.type = T_TriggerData;
    LocTriggerData.tg_event = TRIGGER_EVENT_INSERT |
        TRIGGER_EVENT_ROW |
        TRIGGER_EVENT_BEFORE;
    LocTriggerData.tg_relation = relinfo->ri_RelationDesc;
    for (i = 0; i < trigdesc->numtriggers; i++)
    {
        Trigger    *trigger = &trigdesc->triggers[i];
        HeapTuple   oldtuple;
 
        if (!TRIGGER_TYPE_MATCHES(trigger->tgtype,
                                  TRIGGER_TYPE_ROW,
                                  TRIGGER_TYPE_BEFORE,
                                  TRIGGER_TYPE_INSERT))
            continue;
        if (!TriggerEnabled(estate, relinfo, trigger, LocTriggerData.tg_event,
                            NULL, NULL, slot))
            continue;
 
        if (!newtuple)
            newtuple = ExecFetchSlotHeapTuple(slot, true, &should_free);
 
        LocTriggerData.tg_trigslot = slot;
        LocTriggerData.tg_trigtuple = oldtuple = newtuple;
        LocTriggerData.tg_trigger = trigger;
        newtuple = ExecCallTriggerFunc(&LocTriggerData,
                                       i,
                                       relinfo->ri_TrigFunctions,
                                       relinfo->ri_TrigInstrument,
                                       GetPerTupleMemoryContext(estate));
        if (newtuple == NULL)
        {
            if (should_free)
                heap_freetuple(oldtuple);
            return false;       /* "do nothing" */
        }
        else if (newtuple != oldtuple)
        {
            ExecForceStoreHeapTuple(newtuple, slot, false);
 
            /*
             * After a tuple in a partition goes through a trigger, the user
             * could have changed the partition key enough that the tuple no
             * longer fits the partition.  Verify that.
             */
            if (trigger->tgisclone &&
                !ExecPartitionCheck(relinfo, slot, estate, false))
                ereport(ERROR,
                        (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                         errmsg("moving row to another partition during a BEFORE FOR EACH ROW trigger is not supported"),
                         errdetail("Before executing trigger \"%s\", the row was to be in partition \"%s.%s\".",
                                   trigger->tgname,
                                   get_namespace_name(RelationGetNamespace(relinfo->ri_RelationDesc)),
                                   RelationGetRelationName(relinfo->ri_RelationDesc))));
 
            if (should_free)
                heap_freetuple(oldtuple);
 
            /* signal tuple should be re-fetched if used */
            newtuple = NULL;
        }
    }
 
    return true;
}

References ereport, errcode(), errdetail(), errmsg(), ERROR, ExecCallTriggerFunc(), ExecFetchSlotHeapTuple(), ExecForceStoreHeapTuple(), ExecPartitionCheck(), get_namespace_name(), GetPerTupleMemoryContext, heap_freetuple(), i, TriggerDesc::numtriggers, RelationGetNamespace, RelationGetRelationName, ResultRelInfo::ri_RelationDesc, ResultRelInfo::ri_TrigDesc, ResultRelInfo::ri_TrigFunctions, ResultRelInfo::ri_TrigInstrument, TriggerData::tg_event, TriggerData::tg_relation, TriggerData::tg_trigger, TriggerData::tg_trigslot, TriggerData::tg_trigtuple, Trigger::tgisclone, Trigger::tgname, Trigger::tgtype, TRIGGER_EVENT_BEFORE, TRIGGER_EVENT_INSERT, TRIGGER_EVENT_ROW, TriggerEnabled(), TriggerDesc::triggers, and TriggerData::type.

Referenced by CopyFrom(), ExecInsert(), and ExecSimpleRelationInsert().

ExecBRUpdateTriggers()

bool ExecBRUpdateTriggers	(	EState *	estate,
		EPQState *	epqstate,
		ResultRelInfo *	relinfo,
		ItemPointer	tupleid,
		HeapTuple	fdw_trigtuple,
		TupleTableSlot *	newslot,
		TM_Result *	tmresult,
		TM_FailureData *	tmfd
	)

Definition at line 2945 of file trigger.c.

{
    TriggerDesc *trigdesc = relinfo->ri_TrigDesc;
    TupleTableSlot *oldslot = ExecGetTriggerOldSlot(estate, relinfo);
    HeapTuple   newtuple = NULL;
    HeapTuple   trigtuple;
    bool        should_free_trig = false;
    bool        should_free_new = false;
    TriggerData LocTriggerData = {0};
    int         i;
    Bitmapset  *updatedCols;
    LockTupleMode lockmode;
 
    /* Determine lock mode to use */
    lockmode = ExecUpdateLockMode(estate, relinfo);
 
    Assert(HeapTupleIsValid(fdw_trigtuple) ^ ItemPointerIsValid(tupleid));
    if (fdw_trigtuple == NULL)
    {
        TupleTableSlot *epqslot_candidate = NULL;
 
        /* get a copy of the on-disk tuple we are planning to update */
        if (!GetTupleForTrigger(estate, epqstate, relinfo, tupleid,
                                lockmode, oldslot, &epqslot_candidate,
                                tmresult, tmfd))
            return false;       /* cancel the update action */
 
        /*
         * 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 epqslot_candidate, and need to form a new
         * insertable tuple using ExecGetUpdateNewTuple to replace the one we
         * received in newslot.  Neither we nor our callers have any further
         * interest in the passed-in tuple, so it's okay to overwrite newslot
         * with the newer data.
         *
         * (Typically, newslot was also generated by ExecGetUpdateNewTuple, so
         * that epqslot_clean will be that same slot and the copy step below
         * is not needed.)
         */
        if (epqslot_candidate != NULL)
        {
            TupleTableSlot *epqslot_clean;
 
            epqslot_clean = ExecGetUpdateNewTuple(relinfo, epqslot_candidate,
                                                  oldslot);
 
            if (newslot != epqslot_clean)
                ExecCopySlot(newslot, epqslot_clean);
        }
 
        trigtuple = ExecFetchSlotHeapTuple(oldslot, true, &should_free_trig);
    }
    else
    {
        ExecForceStoreHeapTuple(fdw_trigtuple, oldslot, false);
        trigtuple = fdw_trigtuple;
    }
 
    LocTriggerData.type = T_TriggerData;
    LocTriggerData.tg_event = TRIGGER_EVENT_UPDATE |
        TRIGGER_EVENT_ROW |
        TRIGGER_EVENT_BEFORE;
    LocTriggerData.tg_relation = relinfo->ri_RelationDesc;
    updatedCols = ExecGetAllUpdatedCols(relinfo, estate);
    LocTriggerData.tg_updatedcols = updatedCols;
    for (i = 0; i < trigdesc->numtriggers; i++)
    {
        Trigger    *trigger = &trigdesc->triggers[i];
        HeapTuple   oldtuple;
 
        if (!TRIGGER_TYPE_MATCHES(trigger->tgtype,
                                  TRIGGER_TYPE_ROW,
                                  TRIGGER_TYPE_BEFORE,
                                  TRIGGER_TYPE_UPDATE))
            continue;
        if (!TriggerEnabled(estate, relinfo, trigger, LocTriggerData.tg_event,
                            updatedCols, oldslot, newslot))
            continue;
 
        if (!newtuple)
            newtuple =