Add support for CREATE TRIGGER for SQL Server

- similar to functions & procedures, this dialect can define triggers with a multi statement block
- there's no `EXECUTE` keyword here, so that means the `exec_body` used by other dialects becomes an `Option`

Author: aharpervc

src/ast/mod.rs

@@ -3623,6 +3623,7 @@ pub enum Statement {
     /// ```
     ///
     /// Postgres: <https://www.postgresql.org/docs/current/sql-createtrigger.html>
+    /// SQL Server: <https://learn.microsoft.com/en-us/sql/t-sql/statements/create-trigger-transact-sql>
     CreateTrigger {
         /// The `OR REPLACE` clause is used to re-create the trigger if it already exists.
         ///
@@ -3684,7 +3685,9 @@ pub enum Statement {
         ///  Triggering conditions
         condition: Option<Expr>,
         /// Execute logic block
-        exec_body: TriggerExecBody,
+        exec_body: Option<TriggerExecBody>,
+        /// For SQL dialect has statements for a body
+        statements: Vec<Statement>,
         /// The characteristic of the trigger, which include whether the trigger is `DEFERRABLE`, `INITIALLY DEFERRED`, or `INITIALLY IMMEDIATE`,
         characteristics: Option<ConstraintCharacteristics>,
     },
@@ -4478,6 +4481,7 @@ impl fmt::Display for Statement {
                 condition,
                 include_each,
                 exec_body,
+                statements,
                 characteristics,
             } => {
                 write!(
@@ -4512,7 +4516,20 @@ impl fmt::Display for Statement {
                 if let Some(condition) = condition {
                     write!(f, " WHEN {condition}")?;
                 }
-                write!(f, " EXECUTE {exec_body}")
+                if let Some(exec_body) = exec_body {
+                    write!(f, " EXECUTE {exec_body}")?;
+                }
+                if !statements.is_empty() {
+                    write!(f, " AS ")?;
+                    if statements.len() > 1 {
+                        write!(f, "BEGIN ")?;
+                    }
+                    write!(f, "{}", display_separated(statements, "; "))?;
+                    if statements.len() > 1 {
+                        write!(f, " END")?;
+                    }
+                }
+                Ok(())
             }
             Statement::DropTrigger {
                 if_exists,

src/ast/trigger.rs

@@ -110,6 +110,7 @@ impl fmt::Display for TriggerEvent {
 #[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
 #[cfg_attr(feature = "visitor", derive(Visit, VisitMut))]
 pub enum TriggerPeriod {
+    For,
     After,
     Before,
     InsteadOf,
@@ -118,6 +119,7 @@ pub enum TriggerPeriod {
 impl fmt::Display for TriggerPeriod {
     fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
         match self {
+            TriggerPeriod::For => write!(f, "FOR"),
             TriggerPeriod::After => write!(f, "AFTER"),
             TriggerPeriod::Before => write!(f, "BEFORE"),
             TriggerPeriod::InsteadOf => write!(f, "INSTEAD OF"),

src/parser/mod.rs

@@ -5208,11 +5208,15 @@ impl<'a> Parser<'a> {
         or_replace: bool,
         is_constraint: bool,
     ) -> Result<Statement, ParserError> {
-        if !dialect_of!(self is PostgreSqlDialect | GenericDialect | MySqlDialect) {
+        if !dialect_of!(self is PostgreSqlDialect | GenericDialect | MySqlDialect | MsSqlDialect) {
             self.prev_token();
             return self.expected("an object type after CREATE", self.peek_token());
         }
 
+        if dialect_of!(self is MsSqlDialect) {
+            return self.parse_mssql_create_trigger(or_replace, is_constraint);
+        }
+
         let name = self.parse_object_name(false)?;
         let period = self.parse_trigger_period()?;
 
@@ -5265,18 +5269,69 @@ impl<'a> Parser<'a> {
             trigger_object,
             include_each,
             condition,
-            exec_body,
+            exec_body: Some(exec_body),
+            statements: vec![],
             characteristics,
         })
     }
 
+    /// Parse `CREATE TRIGGER` for [SQL Server]
+    ///
+    /// [SQL Server]: https://learn.microsoft.com/en-us/sql/t-sql/statements/create-trigger-transact-sql
+    pub fn parse_mssql_create_trigger(
+        &mut self,
+        or_replace: bool,
+        is_constraint: bool,
+    ) -> Result<Statement, ParserError> {
+        let name = self.parse_object_name(false)?;
+        self.expect_keyword_is(Keyword::ON)?;
+        let table_name = self.parse_object_name(false)?;
+        let period = self.parse_trigger_period()?;
+        let events = self.parse_comma_separated(Parser::parse_trigger_event)?;
+
+        self.expect_keyword_is(Keyword::AS)?;
+
+        let statements = if self.peek_keyword(Keyword::BEGIN) {
+            self.next_token();
+            let mut result = self.parse_statements()?;
+            // note: `parse_statements` will consume the `END` token & produce a Commit statement...
+            if let Some(Statement::Commit{ chain, end, modifier }) = result.last() {
+                if *chain == false && *end == true && *modifier == None {
+                    result = result[..result.len() - 1].to_vec();
+                }
+            }
+            result
+        } else {
+            vec![self.parse_statement()?]
+        };
+
+        Ok(Statement::CreateTrigger {
+            or_replace,
+            is_constraint,
+            name,
+            period,
+            events,
+            table_name,
+            referenced_table_name: None,
+            referencing: Vec::new(),
+            trigger_object: TriggerObject::Statement,
+            include_each: false,
+            condition: None,
+            exec_body: None,
+            statements,
+            characteristics: None,
+        })
+    }
+
     pub fn parse_trigger_period(&mut self) -> Result<TriggerPeriod, ParserError> {
         Ok(
             match self.expect_one_of_keywords(&[
+                Keyword::FOR,
                 Keyword::BEFORE,
                 Keyword::AFTER,
                 Keyword::INSTEAD,
             ])? {
+                Keyword::FOR => TriggerPeriod::For,
                 Keyword::BEFORE => TriggerPeriod::Before,
                 Keyword::AFTER => TriggerPeriod::After,
                 Keyword::INSTEAD => self

tests/sqlparser_mssql.rs

@@ -2036,3 +2036,88 @@ fn parse_mssql_merge_with_output() {
         OUTPUT $action, deleted.ProductID INTO dsi.temp_products";
     ms_and_generic().verified_stmt(stmt);
 }
+
+#[test]
+fn parse_create_trigger() {
+    let create_trigger = r#"
+        CREATE TRIGGER reminder1
+        ON Sales.Customer
+        AFTER INSERT, UPDATE
+        AS RAISERROR ('Notify Customer Relations', 16, 10);
+    "#;
+    let create_stmt = ms().one_statement_parses_to(create_trigger, "");
+    assert_eq!(
+        create_stmt,
+        Statement::CreateTrigger {
+            or_replace: false,
+            is_constraint: false,
+            name: ObjectName::from(vec![Ident::new("reminder1")]),
+            period: TriggerPeriod::After,
+            events: vec![
+                TriggerEvent::Insert,
+                TriggerEvent::Update(vec![]),
+            ],
+            table_name: ObjectName::from(vec![Ident::new("Sales"), Ident::new("Customer")]),
+            referenced_table_name: None,
+            referencing: vec![],
+            trigger_object: TriggerObject::Statement,
+            include_each: false,
+            condition: None,
+            exec_body: None,
+            statements: vec![Statement::RaisError {
+                message: Box::new(Expr::Value(
+                    (Value::SingleQuotedString("Notify Customer Relations".to_string())).with_empty_span()
+                )),
+                severity: Box::new(Expr::Value(
+                    (Value::Number("16".parse().unwrap(), false)).with_empty_span()
+                )),
+                state: Box::new(Expr::Value(
+                    (Value::Number("10".parse().unwrap(), false)).with_empty_span()
+                )),
+                arguments: vec![],
+                options: vec![],
+            }],
+            characteristics: None,
+        }
+    );
+
+    let multi_statement_trigger = r#"
+        CREATE TRIGGER some_trigger ON some_table FOR INSERT
+        AS
+        BEGIN
+            RAISERROR('Trigger fired', 10, 1);
+        END
+    "#;
+    let create_stmt = ms().one_statement_parses_to(multi_statement_trigger, "");
+    assert_eq!(
+        create_stmt,
+        Statement::CreateTrigger {
+            or_replace: false,
+            is_constraint: false,
+            name: ObjectName::from(vec![Ident::new("some_trigger")]),
+            period: TriggerPeriod::For,
+            events: vec![TriggerEvent::Insert],
+            table_name: ObjectName::from(vec![Ident::new("some_table")]),
+            referenced_table_name: None,
+            referencing: vec![],
+            trigger_object: TriggerObject::Statement,
+            include_each: false,
+            condition: None,
+            exec_body: None,
+            statements: vec![Statement::RaisError {
+                message: Box::new(Expr::Value(
+                    (Value::SingleQuotedString("Trigger fired".to_string())).with_empty_span()
+                )),
+                severity: Box::new(Expr::Value(
+                    (Value::Number("10".parse().unwrap(), false)).with_empty_span()
+                )),
+                state: Box::new(Expr::Value(
+                    (Value::Number("1".parse().unwrap(), false)).with_empty_span()
+                )),
+                arguments: vec![],
+                options: vec![],
+            }],
+            characteristics: None,
+        }
+    );
+}

tests/sqlparser_mysql.rs

@@ -3410,13 +3410,14 @@ fn parse_create_trigger() {
             trigger_object: TriggerObject::Row,
             include_each: true,
             condition: None,
-            exec_body: TriggerExecBody {
+            exec_body: Some(TriggerExecBody {
                 exec_type: TriggerExecBodyType::Function,
                 func_desc: FunctionDesc {
                     name: ObjectName::from(vec![Ident::new("emp_stamp")]),
                     args: None,
                 }
-            },
+            }),
+            statements: vec![],
             characteristics: None,
         }
     );

tests/sqlparser_postgres.rs

@@ -5059,13 +5059,14 @@ fn parse_create_simple_before_insert_trigger() {
         trigger_object: TriggerObject::Row,
         include_each: true,
         condition: None,
-        exec_body: TriggerExecBody {
+        exec_body: Some(TriggerExecBody {
             exec_type: TriggerExecBodyType::Function,
             func_desc: FunctionDesc {
                 name: ObjectName::from(vec![Ident::new("check_account_insert")]),
                 args: None,
             },
-        },
+        }),
+        statements: vec![],
         characteristics: None,
     };
 
@@ -5094,13 +5095,14 @@ fn parse_create_after_update_trigger_with_condition() {
             op: BinaryOperator::Gt,
             right: Box::new(Expr::value(number("10000"))),
         }))),
-        exec_body: TriggerExecBody {
+        exec_body: Some(TriggerExecBody {
             exec_type: TriggerExecBodyType::Function,
             func_desc: FunctionDesc {
                 name: ObjectName::from(vec![Ident::new("check_account_update")]),
                 args: None,
             },
-        },
+        }),
+        statements: vec![],
         characteristics: None,
     };
 
@@ -5122,13 +5124,14 @@ fn parse_create_instead_of_delete_trigger() {
         trigger_object: TriggerObject::Row,
         include_each: true,
         condition: None,
-        exec_body: TriggerExecBody {
+        exec_body: Some(TriggerExecBody {
             exec_type: TriggerExecBodyType::Function,
             func_desc: FunctionDesc {
                 name: ObjectName::from(vec![Ident::new("check_account_deletes")]),
                 args: None,
             },
-        },
+        }),
+        statements: vec![],
         characteristics: None,
     };
 
@@ -5154,13 +5157,14 @@ fn parse_create_trigger_with_multiple_events_and_deferrable() {
         trigger_object: TriggerObject::Row,
         include_each: true,
         condition: None,
-        exec_body: TriggerExecBody {
+        exec_body: Some(TriggerExecBody {
             exec_type: TriggerExecBodyType::Function,
             func_desc: FunctionDesc {
                 name: ObjectName::from(vec![Ident::new("check_account_changes")]),
                 args: None,
             },
-        },
+        }),
+        statements: vec![],
         characteristics: Some(ConstraintCharacteristics {
             deferrable: Some(true),
             initially: Some(DeferrableInitial::Deferred),
@@ -5197,13 +5201,14 @@ fn parse_create_trigger_with_referencing() {
         trigger_object: TriggerObject::Row,
         include_each: true,
         condition: None,
-        exec_body: TriggerExecBody {
+        exec_body: Some(TriggerExecBody {
             exec_type: TriggerExecBodyType::Function,
             func_desc: FunctionDesc {
                 name: ObjectName::from(vec![Ident::new("check_account_referencing")]),
                 args: None,
             },
-        },
+        }),
+        statements: vec![],
         characteristics: None,
     };
 
@@ -5223,7 +5228,7 @@ fn parse_create_trigger_invalid_cases() {
         ),
         (
             "CREATE TRIGGER check_update TOMORROW UPDATE ON accounts EXECUTE FUNCTION check_account_update",
-            "Expected: one of BEFORE or AFTER or INSTEAD, found: TOMORROW"
+            "Expected: one of FOR or BEFORE or AFTER or INSTEAD, found: TOMORROW"
         ),
         (
             "CREATE TRIGGER check_update BEFORE SAVE ON accounts EXECUTE FUNCTION check_account_update",
@@ -5496,13 +5501,14 @@ fn parse_trigger_related_functions() {
             trigger_object: TriggerObject::Row,
             include_each: true,
             condition: None,
-            exec_body: TriggerExecBody {
+            exec_body: Some(TriggerExecBody {
                 exec_type: TriggerExecBodyType::Function,
                 func_desc: FunctionDesc {
                     name: ObjectName::from(vec![Ident::new("emp_stamp")]),
                     args: None,
                 }
-            },
+            }),
+            statements: vec![],
             characteristics: None
         }
     );