1 | /*------------------------------------------------------------------------- |
2 | * |
3 | * trigger.h |
4 | * Declarations for trigger handling. |
5 | * |
6 | * Portions Copyright (c) 1996-2019, PostgreSQL Global Development Group |
7 | * Portions Copyright (c) 1994, Regents of the University of California |
8 | * |
9 | * src/include/commands/trigger.h |
10 | * |
11 | *------------------------------------------------------------------------- |
12 | */ |
13 | #ifndef TRIGGER_H |
14 | #define TRIGGER_H |
15 | |
16 | #include "catalog/objectaddress.h" |
17 | #include "nodes/execnodes.h" |
18 | #include "nodes/parsenodes.h" |
19 | |
20 | /* |
21 | * TriggerData is the node type that is passed as fmgr "context" info |
22 | * when a function is called by the trigger manager. |
23 | */ |
24 | |
25 | #define CALLED_AS_TRIGGER(fcinfo) \ |
26 | ((fcinfo)->context != NULL && IsA((fcinfo)->context, TriggerData)) |
27 | |
28 | typedef uint32 TriggerEvent; |
29 | |
30 | typedef struct TriggerData |
31 | { |
32 | NodeTag type; |
33 | TriggerEvent tg_event; |
34 | Relation tg_relation; |
35 | HeapTuple tg_trigtuple; |
36 | HeapTuple tg_newtuple; |
37 | Trigger *tg_trigger; |
38 | TupleTableSlot *tg_trigslot; |
39 | TupleTableSlot *tg_newslot; |
40 | Tuplestorestate *tg_oldtable; |
41 | Tuplestorestate *tg_newtable; |
42 | } TriggerData; |
43 | |
44 | /* |
45 | * The state for capturing old and new tuples into transition tables for a |
46 | * single ModifyTable node (or other operation source, e.g. copy.c). |
47 | * |
48 | * This is per-caller to avoid conflicts in setting tcs_map or |
49 | * tcs_original_insert_tuple. Note, however, that the pointed-to |
50 | * private data may be shared across multiple callers. |
51 | */ |
52 | struct AfterTriggersTableData; /* private in trigger.c */ |
53 | |
54 | typedef struct TransitionCaptureState |
55 | { |
56 | /* |
57 | * Is there at least one trigger specifying each transition relation on |
58 | * the relation explicitly named in the DML statement or COPY command? |
59 | * Note: in current usage, these flags could be part of the private state, |
60 | * but it seems possibly useful to let callers see them. |
61 | */ |
62 | bool tcs_delete_old_table; |
63 | bool tcs_update_old_table; |
64 | bool tcs_update_new_table; |
65 | bool tcs_insert_new_table; |
66 | |
67 | /* |
68 | * For UPDATE and DELETE, AfterTriggerSaveEvent may need to convert the |
69 | * new and old tuples from a child table's format to the format of the |
70 | * relation named in a query so that it is compatible with the transition |
71 | * tuplestores. The caller must store the conversion map here if so. |
72 | */ |
73 | TupleConversionMap *tcs_map; |
74 | |
75 | /* |
76 | * For INSERT and COPY, it would be wasteful to convert tuples from child |
77 | * format to parent format after they have already been converted in the |
78 | * opposite direction during routing. In that case we bypass conversion |
79 | * and allow the inserting code (copy.c and nodeModifyTable.c) to provide |
80 | * a slot containing the original tuple directly. |
81 | */ |
82 | TupleTableSlot *tcs_original_insert_tuple; |
83 | |
84 | /* |
85 | * Private data including the tuplestore(s) into which to insert tuples. |
86 | */ |
87 | struct AfterTriggersTableData *tcs_private; |
88 | } TransitionCaptureState; |
89 | |
90 | /* |
91 | * TriggerEvent bit flags |
92 | * |
93 | * Note that we assume different event types (INSERT/DELETE/UPDATE/TRUNCATE) |
94 | * can't be OR'd together in a single TriggerEvent. This is unlike the |
95 | * situation for pg_trigger rows, so pg_trigger.tgtype uses a different |
96 | * representation! |
97 | */ |
98 | #define TRIGGER_EVENT_INSERT 0x00000000 |
99 | #define TRIGGER_EVENT_DELETE 0x00000001 |
100 | #define TRIGGER_EVENT_UPDATE 0x00000002 |
101 | #define TRIGGER_EVENT_TRUNCATE 0x00000003 |
102 | #define TRIGGER_EVENT_OPMASK 0x00000003 |
103 | |
104 | #define TRIGGER_EVENT_ROW 0x00000004 |
105 | |
106 | #define TRIGGER_EVENT_BEFORE 0x00000008 |
107 | #define TRIGGER_EVENT_AFTER 0x00000000 |
108 | #define TRIGGER_EVENT_INSTEAD 0x00000010 |
109 | #define TRIGGER_EVENT_TIMINGMASK 0x00000018 |
110 | |
111 | /* More TriggerEvent flags, used only within trigger.c */ |
112 | |
113 | #define AFTER_TRIGGER_DEFERRABLE 0x00000020 |
114 | #define AFTER_TRIGGER_INITDEFERRED 0x00000040 |
115 | |
116 | #define TRIGGER_FIRED_BY_INSERT(event) \ |
117 | (((event) & TRIGGER_EVENT_OPMASK) == TRIGGER_EVENT_INSERT) |
118 | |
119 | #define TRIGGER_FIRED_BY_DELETE(event) \ |
120 | (((event) & TRIGGER_EVENT_OPMASK) == TRIGGER_EVENT_DELETE) |
121 | |
122 | #define TRIGGER_FIRED_BY_UPDATE(event) \ |
123 | (((event) & TRIGGER_EVENT_OPMASK) == TRIGGER_EVENT_UPDATE) |
124 | |
125 | #define TRIGGER_FIRED_BY_TRUNCATE(event) \ |
126 | (((event) & TRIGGER_EVENT_OPMASK) == TRIGGER_EVENT_TRUNCATE) |
127 | |
128 | #define TRIGGER_FIRED_FOR_ROW(event) \ |
129 | ((event) & TRIGGER_EVENT_ROW) |
130 | |
131 | #define TRIGGER_FIRED_FOR_STATEMENT(event) \ |
132 | (!TRIGGER_FIRED_FOR_ROW(event)) |
133 | |
134 | #define TRIGGER_FIRED_BEFORE(event) \ |
135 | (((event) & TRIGGER_EVENT_TIMINGMASK) == TRIGGER_EVENT_BEFORE) |
136 | |
137 | #define TRIGGER_FIRED_AFTER(event) \ |
138 | (((event) & TRIGGER_EVENT_TIMINGMASK) == TRIGGER_EVENT_AFTER) |
139 | |
140 | #define TRIGGER_FIRED_INSTEAD(event) \ |
141 | (((event) & TRIGGER_EVENT_TIMINGMASK) == TRIGGER_EVENT_INSTEAD) |
142 | |
143 | /* |
144 | * Definitions for replication role based firing. |
145 | */ |
146 | #define SESSION_REPLICATION_ROLE_ORIGIN 0 |
147 | #define SESSION_REPLICATION_ROLE_REPLICA 1 |
148 | #define SESSION_REPLICATION_ROLE_LOCAL 2 |
149 | extern PGDLLIMPORT int SessionReplicationRole; |
150 | |
151 | /* |
152 | * States at which a trigger can be fired. These are the |
153 | * possible values for pg_trigger.tgenabled. |
154 | */ |
155 | #define TRIGGER_FIRES_ON_ORIGIN 'O' |
156 | #define TRIGGER_FIRES_ALWAYS 'A' |
157 | #define TRIGGER_FIRES_ON_REPLICA 'R' |
158 | #define TRIGGER_DISABLED 'D' |
159 | |
160 | extern ObjectAddress CreateTrigger(CreateTrigStmt *stmt, const char *queryString, |
161 | Oid relOid, Oid refRelOid, Oid constraintOid, Oid indexOid, |
162 | Oid funcoid, Oid parentTriggerOid, Node *whenClause, |
163 | bool isInternal, bool in_partition); |
164 | |
165 | extern void RemoveTriggerById(Oid trigOid); |
166 | extern Oid get_trigger_oid(Oid relid, const char *name, bool missing_ok); |
167 | |
168 | extern ObjectAddress renametrig(RenameStmt *stmt); |
169 | |
170 | extern void EnableDisableTrigger(Relation rel, const char *tgname, |
171 | char fires_when, bool skip_system, LOCKMODE lockmode); |
172 | |
173 | extern void RelationBuildTriggers(Relation relation); |
174 | |
175 | extern TriggerDesc *CopyTriggerDesc(TriggerDesc *trigdesc); |
176 | |
177 | extern const char *FindTriggerIncompatibleWithInheritance(TriggerDesc *trigdesc); |
178 | |
179 | extern TransitionCaptureState *MakeTransitionCaptureState(TriggerDesc *trigdesc, |
180 | Oid relid, CmdType cmdType); |
181 | |
182 | extern void FreeTriggerDesc(TriggerDesc *trigdesc); |
183 | |
184 | extern void ExecBSInsertTriggers(EState *estate, |
185 | ResultRelInfo *relinfo); |
186 | extern void ExecASInsertTriggers(EState *estate, |
187 | ResultRelInfo *relinfo, |
188 | TransitionCaptureState *transition_capture); |
189 | extern bool ExecBRInsertTriggers(EState *estate, |
190 | ResultRelInfo *relinfo, |
191 | TupleTableSlot *slot); |
192 | extern void ExecARInsertTriggers(EState *estate, |
193 | ResultRelInfo *relinfo, |
194 | TupleTableSlot *slot, |
195 | List *recheckIndexes, |
196 | TransitionCaptureState *transition_capture); |
197 | extern bool ExecIRInsertTriggers(EState *estate, |
198 | ResultRelInfo *relinfo, |
199 | TupleTableSlot *slot); |
200 | extern void ExecBSDeleteTriggers(EState *estate, |
201 | ResultRelInfo *relinfo); |
202 | extern void ExecASDeleteTriggers(EState *estate, |
203 | ResultRelInfo *relinfo, |
204 | TransitionCaptureState *transition_capture); |
205 | extern bool ExecBRDeleteTriggers(EState *estate, |
206 | EPQState *epqstate, |
207 | ResultRelInfo *relinfo, |
208 | ItemPointer tupleid, |
209 | HeapTuple fdw_trigtuple, |
210 | TupleTableSlot **epqslot); |
211 | extern void ExecARDeleteTriggers(EState *estate, |
212 | ResultRelInfo *relinfo, |
213 | ItemPointer tupleid, |
214 | HeapTuple fdw_trigtuple, |
215 | TransitionCaptureState *transition_capture); |
216 | extern bool ExecIRDeleteTriggers(EState *estate, |
217 | ResultRelInfo *relinfo, |
218 | HeapTuple trigtuple); |
219 | extern void ExecBSUpdateTriggers(EState *estate, |
220 | ResultRelInfo *relinfo); |
221 | extern void ExecASUpdateTriggers(EState *estate, |
222 | ResultRelInfo *relinfo, |
223 | TransitionCaptureState *transition_capture); |
224 | extern bool ExecBRUpdateTriggers(EState *estate, |
225 | EPQState *epqstate, |
226 | ResultRelInfo *relinfo, |
227 | ItemPointer tupleid, |
228 | HeapTuple fdw_trigtuple, |
229 | TupleTableSlot *slot); |
230 | extern void ExecARUpdateTriggers(EState *estate, |
231 | ResultRelInfo *relinfo, |
232 | ItemPointer tupleid, |
233 | HeapTuple fdw_trigtuple, |
234 | TupleTableSlot *slot, |
235 | List *recheckIndexes, |
236 | TransitionCaptureState *transition_capture); |
237 | extern bool ExecIRUpdateTriggers(EState *estate, |
238 | ResultRelInfo *relinfo, |
239 | HeapTuple trigtuple, |
240 | TupleTableSlot *slot); |
241 | extern void ExecBSTruncateTriggers(EState *estate, |
242 | ResultRelInfo *relinfo); |
243 | extern void ExecASTruncateTriggers(EState *estate, |
244 | ResultRelInfo *relinfo); |
245 | |
246 | extern void AfterTriggerBeginXact(void); |
247 | extern void AfterTriggerBeginQuery(void); |
248 | extern void AfterTriggerEndQuery(EState *estate); |
249 | extern void AfterTriggerFireDeferred(void); |
250 | extern void AfterTriggerEndXact(bool isCommit); |
251 | extern void AfterTriggerBeginSubXact(void); |
252 | extern void AfterTriggerEndSubXact(bool isCommit); |
253 | extern void AfterTriggerSetState(ConstraintsSetStmt *stmt); |
254 | extern bool AfterTriggerPendingOnRel(Oid relid); |
255 | |
256 | |
257 | /* |
258 | * in utils/adt/ri_triggers.c |
259 | */ |
260 | extern bool RI_FKey_pk_upd_check_required(Trigger *trigger, Relation pk_rel, |
261 | TupleTableSlot *old_slot, TupleTableSlot *new_slot); |
262 | extern bool RI_FKey_fk_upd_check_required(Trigger *trigger, Relation fk_rel, |
263 | TupleTableSlot *old_slot, TupleTableSlot *new_slot); |
264 | extern bool RI_Initial_Check(Trigger *trigger, |
265 | Relation fk_rel, Relation pk_rel); |
266 | extern void RI_PartitionRemove_Check(Trigger *trigger, Relation fk_rel, |
267 | Relation pk_rel); |
268 | |
269 | /* result values for RI_FKey_trigger_type: */ |
270 | #define RI_TRIGGER_PK 1 /* is a trigger on the PK relation */ |
271 | #define RI_TRIGGER_FK 2 /* is a trigger on the FK relation */ |
272 | #define RI_TRIGGER_NONE 0 /* is not an RI trigger function */ |
273 | |
274 | extern int RI_FKey_trigger_type(Oid tgfoid); |
275 | |
276 | #endif /* TRIGGER_H */ |
277 | |