Add support for DISABLE on ALTER TABLE for pg

Author: tobyhede

src/ast/ddl.rs

@@ -45,6 +45,18 @@ pub enum AlterTableOperation {
         /// <column_def>.
         column_def: ColumnDef,
     },
+    /// DISABLE TRIGGER [ trigger_name | ALL | USER ]
+    /// DISABLE RULE rewrite_rule_name
+    /// DISABLE ROW LEVEL SECURITY
+    ///
+    /// Note: this is a PostgreSQL-specific operation.
+    DisableTrigger {
+        name: Ident,
+    },
+    DisableRule {
+        name: Ident,
+    },
+    DisableRowLevelSecurity,
     /// `DROP CONSTRAINT [ IF EXISTS ] <name>`
     DropConstraint {
         if_exists: bool,
@@ -143,6 +155,15 @@ impl fmt::Display for AlterTableOperation {
             AlterTableOperation::AlterColumn { column_name, op } => {
                 write!(f, "ALTER COLUMN {column_name} {op}")
             }
+            AlterTableOperation::DisableTrigger { name } => {
+                write!(f, "DISABLE TRIGGER {name}")
+            }
+            AlterTableOperation::DisableRule { name } => {
+                write!(f, "DISABLE RULE {name}")
+            }
+            AlterTableOperation::DisableRowLevelSecurity => {
+                write!(f, "DISABLE ROW LEVEL SECURITY")
+            }
             AlterTableOperation::DropPartitions {
                 partitions,
                 if_exists,

src/keywords.rs

@@ -222,6 +222,7 @@ define_keywords!(
     DETAIL,
     DETERMINISTIC,
     DIRECTORY,
+    DISABLE,
     DISCARD,
     DISCONNECT,
     DISTINCT,
@@ -557,6 +558,7 @@ define_keywords!(
     ROWID,
     ROWS,
     ROW_NUMBER,
+    RULE,
     RUN,
     SAFE_CAST,
     SAVEPOINT,

src/parser/mod.rs

@@ -4636,6 +4636,26 @@ impl<'a> Parser<'a> {
                     new_column_name,
                 }
             }
+        } else if self.parse_keyword(Keyword::DISABLE) {
+            if self.parse_keyword(Keyword::TRIGGER) {
+                // let all = self.parse_keyword(Keyword::ALL);
+                // let user = self.parse_keyword(Keyword::ALL);
+                let name = self.parse_identifier()?;
+
+                // println!("parse all: {}", all);
+
+                AlterTableOperation::DisableTrigger { name }
+            } else if self.parse_keyword(Keyword::RULE) {
+                let name = self.parse_identifier()?;
+                AlterTableOperation::DisableRule { name }
+            } else if self.parse_keyword(Keyword::ROW) {
+                AlterTableOperation::DisableRowLevelSecurity {}
+            } else {
+                return self.expected(
+                    "TRIGGER, RULE or ROW LEVEL SECURITY after DISABLE",
+                    self.peek_token(),
+                );
+            }
         } else if self.parse_keyword(Keyword::DROP) {
             if self.parse_keywords(&[Keyword::IF, Keyword::EXISTS, Keyword::PARTITION]) {
                 self.expect_token(&Token::LParen)?;

tests/sqlparser_postgres.rs

@@ -563,6 +563,13 @@ fn parse_alter_table_constraints_rename() {
     }
 }
 
+#[test]
+fn parse_alter_table_disable_trigger() {
+    pg_and_generic().verified_stmt("ALTER TABLE tab DISABLE TRIGGER ALL");
+    pg_and_generic().verified_stmt("ALTER TABLE tab DISABLE TRIGGER USER");
+    pg_and_generic().verified_stmt("ALTER TABLE tab DISABLE TRIGGER trigger_name");
+}
+
 #[test]
 fn parse_alter_table_alter_column() {
     pg().one_statement_parses_to(