From 54a277ff018ce801bce649a89601b22a1a98fafa Mon Sep 17 00:00:00 2001
From: Metin Doslu <metin@citusdata.com>
Date: Wed, 29 Mar 2017 16:45:10 +0300
Subject: [PATCH] Add disable/enable trigger all support

---
 .../distributed/executor/multi_utility.c      |  4 +-
 .../input/multi_alter_table_statements.source | 39 ++++++++++++++
 .../multi_alter_table_statements.source       | 53 +++++++++++++++++++
 3 files changed, 95 insertions(+), 1 deletion(-)

diff --git a/src/backend/distributed/executor/multi_utility.c b/src/backend/distributed/executor/multi_utility.c
index e569232aa..745b440ef 100644
--- a/src/backend/distributed/executor/multi_utility.c
+++ b/src/backend/distributed/executor/multi_utility.c
@@ -1650,8 +1650,10 @@ ErrorIfUnsupportedAlterTableStmt(AlterTableStmt *alterTableStatement)
 			}
 
 			case AT_DropConstraint:
+			case AT_EnableTrigAll:
+			case AT_DisableTrigAll:
 			{
-				/* we will no perform any special check for ALTER TABLE DROP CONSTRAINT */
+				/* we will not perform any special checks for these ALTER TABLE types */
 				break;
 			}
 
diff --git a/src/test/regress/input/multi_alter_table_statements.source b/src/test/regress/input/multi_alter_table_statements.source
index 710abf9c0..931129866 100644
--- a/src/test/regress/input/multi_alter_table_statements.source
+++ b/src/test/regress/input/multi_alter_table_statements.source
@@ -380,6 +380,45 @@ END;
 
 DROP TABLE sequence_deadlock_test;
 
+-- verify enable/disable trigger all works
+
+SET citus.shard_replication_factor TO 1;
+SET citus.shard_count TO 1;
+
+CREATE TABLE trigger_table (
+	id int,
+	value text
+);
+
+SELECT create_distributed_table('trigger_table', 'id');
+
+-- first set a trigger on a shard
+\c - - - :worker_1_port
+
+CREATE FUNCTION update_value() RETURNS trigger AS $up$
+    BEGIN
+		NEW.value := 'trigger enabled';
+		RETURN NEW;
+    END;
+$up$ LANGUAGE plpgsql;
+
+CREATE TRIGGER update_value
+BEFORE INSERT ON trigger_table_220056
+FOR EACH ROW EXECUTE PROCEDURE update_value();
+
+\c - - - :master_port
+INSERT INTO trigger_table VALUES (1, 'trigger disabled');
+SELECT value, count(*) FROM trigger_table GROUP BY value ORDER BY value;
+
+ALTER TABLE trigger_table DISABLE TRIGGER ALL;
+INSERT INTO trigger_table VALUES (1, 'trigger disabled');
+SELECT value, count(*) FROM trigger_table GROUP BY value ORDER BY value;
+
+ALTER TABLE trigger_table ENABLE TRIGGER ALL;
+INSERT INTO trigger_table VALUES (1, 'trigger disabled');
+SELECT value, count(*) FROM trigger_table GROUP BY value ORDER BY value;
+
+DROP TABLE trigger_table;
 
 -- test ALTER TABLE ALL IN TABLESPACE
 -- we expect that it will warn out
diff --git a/src/test/regress/output/multi_alter_table_statements.source b/src/test/regress/output/multi_alter_table_statements.source
index 4f9395709..2fdb0bb32 100644
--- a/src/test/regress/output/multi_alter_table_statements.source
+++ b/src/test/regress/output/multi_alter_table_statements.source
@@ -780,6 +780,59 @@ DROP SEQUENCE sequence_deadlock_test_b_seq CASCADE;
 NOTICE:  drop cascades to default for table sequence_deadlock_test column b
 END;
 DROP TABLE sequence_deadlock_test;
+-- verify enable/disable trigger all works
+SET citus.shard_replication_factor TO 1;
+SET citus.shard_count TO 1;
+CREATE TABLE trigger_table (
+	id int,
+	value text
+);
+SELECT create_distributed_table('trigger_table', 'id');
+ create_distributed_table 
+--------------------------
+ 
+(1 row)
+
+-- first set a trigger on a shard
+\c - - - :worker_1_port
+CREATE FUNCTION update_value() RETURNS trigger AS $up$
+    BEGIN
+		NEW.value := 'trigger enabled';
+		RETURN NEW;
+    END;
+$up$ LANGUAGE plpgsql;
+CREATE TRIGGER update_value
+BEFORE INSERT ON trigger_table_220056
+FOR EACH ROW EXECUTE PROCEDURE update_value();
+\c - - - :master_port
+INSERT INTO trigger_table VALUES (1, 'trigger disabled');
+SELECT value, count(*) FROM trigger_table GROUP BY value ORDER BY value;
+      value      | count 
+-----------------+-------
+ trigger enabled |     1
+(1 row)
+
+ALTER TABLE trigger_table DISABLE TRIGGER ALL;
+NOTICE:  using one-phase commit for distributed DDL commands
+HINT:  You can enable two-phase commit for extra safety with: SET citus.multi_shard_commit_protocol TO '2pc'
+INSERT INTO trigger_table VALUES (1, 'trigger disabled');
+SELECT value, count(*) FROM trigger_table GROUP BY value ORDER BY value;
+      value       | count 
+------------------+-------
+ trigger disabled |     1
+ trigger enabled  |     1
+(2 rows)
+
+ALTER TABLE trigger_table ENABLE TRIGGER ALL;
+INSERT INTO trigger_table VALUES (1, 'trigger disabled');
+SELECT value, count(*) FROM trigger_table GROUP BY value ORDER BY value;
+      value       | count 
+------------------+-------
+ trigger disabled |     1
+ trigger enabled  |     2
+(2 rows)
+
+DROP TABLE trigger_table;
 -- test ALTER TABLE ALL IN TABLESPACE
 -- we expect that it will warn out
 CREATE TABLESPACE super_fast_ssd LOCATION '@abs_srcdir@/data';