diff --git a/doc/src/sgml/ref/create_trigger.sgml b/doc/src/sgml/ref/create_trigger.sgml index 3339a4b..65aeea3 100644 --- a/doc/src/sgml/ref/create_trigger.sgml +++ b/doc/src/sgml/ref/create_trigger.sgml @@ -26,7 +26,7 @@ PostgreSQL documentation -CREATE [ CONSTRAINT ] TRIGGER name { BEFORE | AFTER | INSTEAD OF } { event [ OR ... ] } +CREATE [ OR REPLACE ] [ CONSTRAINT ] TRIGGER name { BEFORE | AFTER | INSTEAD OF } { event [ OR ... ] } ON table_name [ FROM referenced_table_name ] [ NOT DEFERRABLE | [ DEFERRABLE ] [ INITIALLY IMMEDIATE | INITIALLY DEFERRED ] ] diff --git a/src/backend/catalog/heap.c b/src/backend/catalog/heap.c index 032fab9..0bf65cd 100644 --- a/src/backend/catalog/heap.c +++ b/src/backend/catalog/heap.c @@ -2399,7 +2399,8 @@ StoreRelCheck(Relation rel, const char *ccname, Node *expr, is_local, /* conislocal */ inhcount, /* coninhcount */ is_no_inherit, /* connoinherit */ - is_internal); /* internally constructed? */ + is_internal, /* internally constructed? */ + InvalidOid); pfree(ccbin); diff --git a/src/backend/catalog/index.c b/src/backend/catalog/index.c index 9234e93..dbe4a70 100644 --- a/src/backend/catalog/index.c +++ b/src/backend/catalog/index.c @@ -1770,7 +1770,8 @@ index_constraint_create(Relation heapRelation, islocal, inhcount, noinherit, - is_internal); + is_internal, + InvalidOid); /* * Register the index as internally dependent on the constraint. diff --git a/src/backend/catalog/pg_constraint.c b/src/backend/catalog/pg_constraint.c index b614559..4d742f5 100644 --- a/src/backend/catalog/pg_constraint.c +++ b/src/backend/catalog/pg_constraint.c @@ -76,10 +76,11 @@ CreateConstraintEntry(const char *constraintName, bool conIsLocal, int conInhCount, bool conNoInherit, - bool is_internal) + bool is_internal, + Oid existing_constraint_oid) { Relation conDesc; - Oid conOid; + Oid conOid = InvalidOid; HeapTuple tup; bool nulls[Natts_pg_constraint]; Datum values[Natts_pg_constraint]; @@ -92,7 +93,12 @@ CreateConstraintEntry(const char *constraintName, NameData cname; int i; ObjectAddress conobject; - + SysScanDesc conscan; + ScanKeyData skey[2]; + HeapTuple tuple; + bool replaces[Natts_pg_constraint]; + Form_pg_constraint constrForm; + conDesc = table_open(ConstraintRelationId, RowExclusiveLock); Assert(constraintName); @@ -164,9 +170,12 @@ CreateConstraintEntry(const char *constraintName, values[i] = (Datum) NULL; } - conOid = GetNewOidWithIndex(conDesc, ConstraintOidIndexId, - Anum_pg_constraint_oid); - values[Anum_pg_constraint_oid - 1] = ObjectIdGetDatum(conOid); + if(!existing_constraint_oid) + { + conOid = GetNewOidWithIndex(conDesc, ConstraintOidIndexId, + Anum_pg_constraint_oid); + values[Anum_pg_constraint_oid - 1] = ObjectIdGetDatum(conOid); + } values[Anum_pg_constraint_conname - 1] = NameGetDatum(&cname); values[Anum_pg_constraint_connamespace - 1] = ObjectIdGetDatum(constraintNamespace); values[Anum_pg_constraint_contype - 1] = CharGetDatum(constraintType); @@ -220,9 +229,47 @@ CreateConstraintEntry(const char *constraintName, else nulls[Anum_pg_constraint_conbin - 1] = true; - tup = heap_form_tuple(RelationGetDescr(conDesc), values, nulls); + if (OidIsValid(existing_constraint_oid)) + { + ScanKeyInit(&skey[0], + Anum_pg_constraint_oid, + BTEqualStrategyNumber, F_OIDEQ, + ObjectIdGetDatum(existing_constraint_oid)); + + conscan = systable_beginscan(conDesc, + ConstraintOidIndexId, + true, + NULL, + 1, + skey); + + tuple = systable_getnext(conscan); + Assert (HeapTupleIsValid(tuple)); + constrForm = (Form_pg_constraint) GETSTRUCT(tuple); + conOid = constrForm->oid; + Assert(conOid == existing_constraint_oid); + + memset(replaces, true, sizeof(replaces)); + replaces[Anum_pg_constraint_oid - 1] = false; /* skip updating Oid data */ + replaces[Anum_pg_constraint_conname - 1] = false; + replaces[Anum_pg_constraint_confrelid - 1] = false; + + /* Modify the existing constraint entry */ + tup = heap_modify_tuple(tuple, RelationGetDescr(conDesc), values, nulls, replaces); + CatalogTupleUpdate(conDesc, &tuple->t_self, tup); + heap_freetuple(tup); - CatalogTupleInsert(conDesc, tup); + /* Remove all old dependencies before registering new ones */ + deleteDependencyRecordsFor(ConstraintRelationId, conOid, true); + + systable_endscan(conscan); + } + else + { + tup = heap_form_tuple(RelationGetDescr(conDesc), values, nulls); + CatalogTupleInsert(conDesc, tup); + heap_freetuple(tup); + } conobject.classId = ConstraintRelationId; conobject.objectId = conOid; diff --git a/src/backend/commands/tablecmds.c b/src/backend/commands/tablecmds.c index fc1c4df..e8faac5 100644 --- a/src/backend/commands/tablecmds.c +++ b/src/backend/commands/tablecmds.c @@ -8081,7 +8081,8 @@ addFkRecurseReferenced(List **wqueue, Constraint *fkconstraint, Relation rel, conislocal, /* islocal */ coninhcount, /* inhcount */ connoinherit, /* conNoInherit */ - false); /* is_internal */ + false, /* is_internal */ + InvalidOid); ObjectAddressSet(address, ConstraintRelationId, constrOid); @@ -8351,7 +8352,8 @@ addFkRecurseReferencing(List **wqueue, Constraint *fkconstraint, Relation rel, false, 1, false, - false); + false, + InvalidOid); /* * Give this constraint partition-type dependencies on the parent @@ -8748,7 +8750,8 @@ CloneFkReferencing(List **wqueue, Relation parentRel, Relation partRel) false, /* islocal */ 1, /* inhcount */ false, /* conNoInherit */ - true); + true, + InvalidOid); /* Set up partition dependencies for the new constraint */ ObjectAddressSet(address, ConstraintRelationId, constrOid); diff --git a/src/backend/commands/trigger.c b/src/backend/commands/trigger.c index ee878d7..5188a6d 100644 --- a/src/backend/commands/trigger.c +++ b/src/backend/commands/trigger.c @@ -31,6 +31,7 @@ #include "catalog/pg_proc.h" #include "catalog/pg_trigger.h" #include "catalog/pg_type.h" +#include "catalog/pg_depend.h" #include "commands/dbcommands.h" #include "commands/defrem.h" #include "commands/trigger.h" @@ -181,7 +182,7 @@ CreateTrigger(CreateTrigStmt *stmt, const char *queryString, HeapTuple tuple; Oid fargtypes[1]; /* dummy */ Oid funcrettype; - Oid trigoid; + Oid trigoid = InvalidOid; char internaltrigname[NAMEDATALEN]; char *trigname; Oid constrrelid = InvalidOid; @@ -190,6 +191,12 @@ CreateTrigger(CreateTrigStmt *stmt, const char *queryString, char *oldtablename = NULL; char *newtablename = NULL; bool partition_recurse; + bool is_update = false; + HeapTuple newtup; + TupleDesc tupDesc; + bool replaces[Natts_pg_trigger]; + Oid existing_constraint_oid = InvalidOid; + bool trigger_exists = false; if (OidIsValid(relOid)) rel = table_open(relOid, ShareRowExclusiveLock); @@ -687,6 +694,58 @@ CreateTrigger(CreateTrigStmt *stmt, const char *queryString, } /* + * Generate the trigger's OID now, so that we can use it in the name if + * needed. + */ + tgrel = heap_open(TriggerRelationId, RowExclusiveLock); + + ScanKeyInit(&key, + Anum_pg_trigger_tgrelid, + BTEqualStrategyNumber, F_OIDEQ, + ObjectIdGetDatum(RelationGetRelid(rel))); + tgscan = systable_beginscan(tgrel, TriggerRelidNameIndexId, true, + NULL, 1, &key); + while (HeapTupleIsValid(tuple = systable_getnext(tgscan))) + { + Form_pg_trigger pg_trigger = (Form_pg_trigger) GETSTRUCT(tuple); + + if (namestrcmp(&(pg_trigger->tgname), stmt->trigname) == 0) + { + trigoid = pg_trigger->oid; + existing_constraint_oid = pg_trigger->tgconstraint; + trigger_exists = true; + break; + } + } + systable_endscan(tgscan); + + if (!trigger_exists) + trigoid = GetNewOidWithIndex(tgrel, TriggerOidIndexId, + Anum_pg_trigger_oid); + + /* Replacement between normal triggers and constraint triggers are restricted */ + if (stmt->replace && trigger_exists) + { + if (stmt->isconstraint && !OidIsValid(existing_constraint_oid)) + ereport(ERROR, + (errcode(ERRCODE_DUPLICATE_OBJECT), + errmsg("Trigger \"%s\" for relation \"%s\" cannot be replaced with constraint trigger", + stmt->trigname, RelationGetRelationName(rel)))); + else if (!stmt->isconstraint && OidIsValid(existing_constraint_oid)) + ereport(ERROR, + (errcode(ERRCODE_DUPLICATE_OBJECT), + errmsg("Constraint trigger \"%s\" for relation \"%s\" cannot be replaced with non-constraint trigger", + stmt->trigname, RelationGetRelationName(rel)))); + } + else if (trigger_exists && !isInternal) + { + ereport(ERROR, + (errcode(ERRCODE_DUPLICATE_OBJECT), + errmsg("trigger \"%s\" for relation \"%s\" already exists", + stmt->trigname, RelationGetRelationName(rel)))); + } + + /* * Find and validate the trigger function. */ if (!OidIsValid(funcoid)) @@ -776,19 +835,11 @@ CreateTrigger(CreateTrigStmt *stmt, const char *queryString, true, /* islocal */ 0, /* inhcount */ true, /* isnoinherit */ - isInternal); /* is_internal */ + isInternal, /* is_internal */ + existing_constraint_oid); } /* - * Generate the trigger's OID now, so that we can use it in the name if - * needed. - */ - tgrel = table_open(TriggerRelationId, RowExclusiveLock); - - trigoid = GetNewOidWithIndex(tgrel, TriggerOidIndexId, - Anum_pg_trigger_oid); - - /* * 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.) @@ -806,37 +857,6 @@ CreateTrigger(CreateTrigStmt *stmt, const char *queryString, } /* - * Scan pg_trigger for existing triggers on relation. We do this only to - * give a nice error message if there's already a trigger of the same - * name. (The unique index on tgrelid/tgname would complain anyway.) We - * can skip this for internally generated triggers, since the name - * modification above should be sufficient. - * - * NOTE that this is cool only because we have ShareRowExclusiveLock on - * the relation, so the trigger set won't be changing underneath us. - */ - if (!isInternal) - { - ScanKeyInit(&key, - Anum_pg_trigger_tgrelid, - BTEqualStrategyNumber, F_OIDEQ, - ObjectIdGetDatum(RelationGetRelid(rel))); - tgscan = systable_beginscan(tgrel, TriggerRelidNameIndexId, true, - NULL, 1, &key); - while (HeapTupleIsValid(tuple = systable_getnext(tgscan))) - { - Form_pg_trigger pg_trigger = (Form_pg_trigger) GETSTRUCT(tuple); - - if (namestrcmp(&(pg_trigger->tgname), trigname) == 0) - ereport(ERROR, - (errcode(ERRCODE_DUPLICATE_OBJECT), - errmsg("trigger \"%s\" for relation \"%s\" already exists", - trigname, RelationGetRelationName(rel)))); - } - systable_endscan(tgscan); - } - - /* * Build the new pg_trigger tuple. * * When we're creating a trigger in a partition, we mark it as internal, @@ -963,12 +983,51 @@ CreateTrigger(CreateTrigStmt *stmt, const char *queryString, tuple = heap_form_tuple(tgrel->rd_att, values, nulls); - /* - * Insert tuple into pg_trigger. - */ - CatalogTupleInsert(tgrel, tuple); + if (!trigger_exists) + { + tuple = heap_form_tuple(tgrel->rd_att, values, nulls); + + /* + * Insert tuple into pg_trigger. + */ + CatalogTupleInsert(tgrel, tuple); + + heap_freetuple(tuple); + } + else + { + memset(replaces, true, sizeof(replaces)); + + ScanKeyInit(&key, + Anum_pg_trigger_tgrelid, + BTEqualStrategyNumber, F_OIDEQ, + ObjectIdGetDatum(RelationGetRelid(rel))); + tgscan = systable_beginscan(tgrel, TriggerRelidNameIndexId, true, + NULL, 1, &key); + while (HeapTupleIsValid(tuple = systable_getnext(tgscan))) + { + Form_pg_trigger pg_trigger = (Form_pg_trigger) GETSTRUCT(tuple); + + if (namestrcmp(&(pg_trigger->tgname), trigname) == 0) + { + tupDesc = RelationGetDescr(tgrel); + replaces[Anum_pg_trigger_oid - 1] = false; /* skip updating Oid data */ + replaces[Anum_pg_trigger_tgrelid - 1] = false; + replaces[Anum_pg_trigger_tgname - 1] = false; + trigoid = pg_trigger->oid; + newtup = heap_modify_tuple(tuple, tupDesc, values, nulls, replaces); + + /* Update tuple in pg_trigger */ + CatalogTupleUpdate(tgrel, &tuple->t_self, newtup); + + heap_freetuple(newtup); + is_update = true; + break; + } + } + systable_endscan(tgscan); + } - heap_freetuple(tuple); table_close(tgrel, RowExclusiveLock); pfree(DatumGetPointer(values[Anum_pg_trigger_tgname - 1])); @@ -1011,6 +1070,15 @@ CreateTrigger(CreateTrigStmt *stmt, const char *queryString, myself.objectId = trigoid; myself.objectSubId = 0; + /* + * In case of replace trigger, trigger should no-more dependent on old + * referenced objects. Always remove the old dependencies and then + * register new ones. In that way, even if the old referenced object gets + * dropped, trigger will remain in the database. + */ + if (is_update) + deleteDependencyRecordsFor(myself.classId, myself.objectId, true); + referenced.classId = ProcedureRelationId; referenced.objectId = funcoid; referenced.objectSubId = 0; diff --git a/src/backend/commands/typecmds.c b/src/backend/commands/typecmds.c index 89887b8..3be6e61 100644 --- a/src/backend/commands/typecmds.c +++ b/src/backend/commands/typecmds.c @@ -3174,7 +3174,8 @@ domainAddConstraint(Oid domainOid, Oid domainNamespace, Oid baseTypeOid, true, /* is local */ 0, /* inhcount */ false, /* connoinherit */ - false); /* is_internal */ + false, /* is_internal */ + InvalidOid); if (constrAddr) ObjectAddressSet(*constrAddr, ConstraintRelationId, ccoid); diff --git a/src/backend/parser/gram.y b/src/backend/parser/gram.y index c97bb36..d662b03 100644 --- a/src/backend/parser/gram.y +++ b/src/backend/parser/gram.y @@ -5322,48 +5322,50 @@ am_type: *****************************************************************************/ CreateTrigStmt: - CREATE TRIGGER name TriggerActionTime TriggerEvents ON + CREATE opt_or_replace TRIGGER name TriggerActionTime TriggerEvents ON qualified_name TriggerReferencing TriggerForSpec TriggerWhen EXECUTE FUNCTION_or_PROCEDURE func_name '(' TriggerFuncArgs ')' { CreateTrigStmt *n = makeNode(CreateTrigStmt); - n->trigname = $3; - n->relation = $7; - n->funcname = $13; - n->args = $15; - n->row = $9; - n->timing = $4; - n->events = intVal(linitial($5)); - n->columns = (List *) lsecond($5); - n->whenClause = $10; - n->transitionRels = $8; + n->replace = $2; + n->trigname = $4; + n->relation = $8; + n->funcname = $14; + n->args = $16; + n->row = $10; + n->timing = $5; + n->events = intVal(linitial($6)); + n->columns = (List *) lsecond($6); + n->whenClause = $11; + n->transitionRels = $9; n->isconstraint = false; n->deferrable = false; n->initdeferred = false; n->constrrel = NULL; $$ = (Node *)n; } - | CREATE CONSTRAINT TRIGGER name AFTER TriggerEvents ON + | CREATE opt_or_replace CONSTRAINT TRIGGER name AFTER TriggerEvents ON qualified_name OptConstrFromTable ConstraintAttributeSpec FOR EACH ROW TriggerWhen EXECUTE FUNCTION_or_PROCEDURE func_name '(' TriggerFuncArgs ')' { CreateTrigStmt *n = makeNode(CreateTrigStmt); - n->trigname = $4; - n->relation = $8; - n->funcname = $17; - n->args = $19; + n->replace = $2; + n->trigname = $5; + n->relation = $9; + n->funcname = $18; + n->args = $20; n->row = true; n->timing = TRIGGER_TYPE_AFTER; - n->events = intVal(linitial($6)); - n->columns = (List *) lsecond($6); - n->whenClause = $14; + n->events = intVal(linitial($7)); + n->columns = (List *) lsecond($7); + n->whenClause = $15; n->transitionRels = NIL; n->isconstraint = true; - processCASbits($10, @10, "TRIGGER", + processCASbits($11, @11, "TRIGGER", &n->deferrable, &n->initdeferred, NULL, NULL, yyscanner); - n->constrrel = $9; + n->constrrel = $10; $$ = (Node *)n; } ; diff --git a/src/include/catalog/pg_constraint.h b/src/include/catalog/pg_constraint.h index c1e60c7..e48a6ce 100644 --- a/src/include/catalog/pg_constraint.h +++ b/src/include/catalog/pg_constraint.h @@ -211,7 +211,8 @@ extern Oid CreateConstraintEntry(const char *constraintName, bool conIsLocal, int conInhCount, bool conNoInherit, - bool is_internal); + bool is_internal, + Oid existing_constraint_oid); extern void RemoveConstraintById(Oid conId); extern void RenameConstraintById(Oid conId, const char *newname); diff --git a/src/include/nodes/parsenodes.h b/src/include/nodes/parsenodes.h index 94ded3c..e2da9f6 100644 --- a/src/include/nodes/parsenodes.h +++ b/src/include/nodes/parsenodes.h @@ -2418,6 +2418,7 @@ typedef struct CreateTrigStmt bool isconstraint; /* This is a constraint trigger */ /* explicitly named transition data */ List *transitionRels; /* TriggerTransition nodes, or NIL if none */ + bool replace; /* T => replace if already exists */ /* The remaining fields are only used for constraint triggers */ bool deferrable; /* [NOT] DEFERRABLE */ bool initdeferred; /* INITIALLY {DEFERRED|IMMEDIATE} */ diff --git a/src/test/regress/expected/triggers.out b/src/test/regress/expected/triggers.out index c64151b..16eedf2 100644 --- a/src/test/regress/expected/triggers.out +++ b/src/test/regress/expected/triggers.out @@ -2848,3 +2848,49 @@ drop table self_ref; drop function dump_insert(); drop function dump_update(); drop function dump_delete(); +-- +-- Test case for CREATE OR REPLACE TRIGGER +-- +\h CREATE TRIGGER; +Command: CREATE TRIGGER +Description: define a new trigger +Syntax: +CREATE [ OR REPLACE ] [ CONSTRAINT ] TRIGGER name { BEFORE | AFTER | INSTEAD OF } { event [ OR ... ] } + ON table_name + [ FROM referenced_table_name ] + [ NOT DEFERRABLE | [ DEFERRABLE ] [ INITIALLY IMMEDIATE | INITIALLY DEFERRED ] ] + [ REFERENCING { { OLD | NEW } TABLE [ AS ] transition_relation_name } [ ... ] ] + [ FOR [ EACH ] { ROW | STATEMENT } ] + [ WHEN ( condition ) ] + EXECUTE { FUNCTION | PROCEDURE } function_name ( arguments ) + +where event can be one of: + + INSERT + UPDATE [ OF column_name [, ... ] ] + DELETE + TRUNCATE + +URL: https://www.postgresql.org/docs/devel/sql-createtrigger.html + +create table my_table (id integer); +create function before_replacement() returns trigger as $$ +begin +raise notice 'function replaced by another function'; +return null; +end; $$ language plpgsql; +create function after_replacement() returns trigger as $$ +begin +raise notice 'function to replace the initial function'; +return null; +end; $$ language plpgsql; +create or replace trigger my_trig before insert on my_table for each row execute procedure before_replacement(); +insert into my_table (id) values (1); +NOTICE: function replaced by another function +create or replace trigger my_trig before insert on my_table for each row execute procedure after_replacement(); +insert into my_table (id) values (2); +NOTICE: function to replace the initial function +drop trigger my_trig on my_table; +drop function before_replacement(); +drop function after_replacement(); +drop table my_table; diff --git a/src/test/regress/sql/triggers.sql b/src/test/regress/sql/triggers.sql index 4534dc9..607c420 100644 --- a/src/test/regress/sql/triggers.sql +++ b/src/test/regress/sql/triggers.sql @@ -2174,3 +2174,31 @@ drop table self_ref; drop function dump_insert(); drop function dump_update(); drop function dump_delete(); + +-- +-- Test case for CREATE OR REPLACE TRIGGER +-- +\h CREATE TRIGGER; +create table my_table (id integer); +create function before_replacement() returns trigger as $$ +begin +raise notice 'function replaced by another function'; +return null; +end; $$ language plpgsql; + +create function after_replacement() returns trigger as $$ +begin +raise notice 'function to replace the initial function'; +return null; +end; $$ language plpgsql; + +create or replace trigger my_trig before insert on my_table for each row execute procedure before_replacement(); +insert into my_table (id) values (1); + +create or replace trigger my_trig before insert on my_table for each row execute procedure after_replacement(); +insert into my_table (id) values (2); + +drop trigger my_trig on my_table; +drop function before_replacement(); +drop function after_replacement(); +drop table my_table;