Apply fix

Co-authored-by: Ifeanyi Ubah <ifeanyi@validio.io>
Signed-off-by: Guan-Ming (Wesley) Chiu <105915352+guan404ming@users.noreply.github.com>

Author: guan404ming

src/ast/dml.rs

@@ -80,6 +80,7 @@ pub struct Insert {
     /// RETURNING
     pub returning: Option<Vec<SelectItem>>,
     /// OUTPUT (MSSQL)
+    /// See <https://learn.microsoft.com/en-us/sql/t-sql/queries/output-clause-transact-sql>
     pub output: Option<OutputClause>,
     /// Only for mysql
     pub replace_into: bool,
@@ -297,6 +298,7 @@ pub struct Delete {
     /// RETURNING
     pub returning: Option<Vec<SelectItem>>,
     /// OUTPUT (MSSQL)
+    /// See <https://learn.microsoft.com/en-us/sql/t-sql/queries/output-clause-transact-sql>
     pub output: Option<OutputClause>,
     /// ORDER BY (MySQL)
     pub order_by: Vec<OrderByExpr>,
@@ -381,6 +383,7 @@ pub struct Update {
     /// RETURNING
     pub returning: Option<Vec<SelectItem>>,
     /// OUTPUT (MSSQL)
+    /// See <https://learn.microsoft.com/en-us/sql/t-sql/queries/output-clause-transact-sql>
     pub output: Option<OutputClause>,
     /// SQLite-specific conflict resolution clause
     pub or: Option<SqliteOnConflict>,

src/parser/merge.rs

@@ -218,6 +218,20 @@ impl Parser<'_> {
         self.parse_parenthesized_qualified_column_list(IsOptional::Optional, allow_empty)
     }
 
+    /// Parses an `OUTPUT` clause if present (MSSQL).
+    pub(super) fn maybe_parse_output_clause(
+        &mut self,
+    ) -> Result<Option<OutputClause>, ParserError> {
+        if self.parse_keyword(Keyword::OUTPUT) {
+            Ok(Some(self.parse_output(
+                Keyword::OUTPUT,
+                self.get_current_token().clone(),
+            )?))
+        } else {
+            Ok(None)
+        }
+    }
+
     pub(super) fn parse_output(
         &mut self,
         start_keyword: Keyword,

src/parser/mod.rs

@@ -13310,13 +13310,7 @@ impl<'a> Parser<'a> {
 
         let from = self.parse_comma_separated(Parser::parse_table_and_joins)?;
 
-        // MSSQL OUTPUT clause appears after FROM table, before USING/WHERE
-        // https://learn.microsoft.com/en-us/sql/t-sql/queries/output-clause-transact-sql
-        let output = if self.parse_keyword(Keyword::OUTPUT) {
-            Some(self.parse_output(Keyword::OUTPUT, self.get_current_token().clone())?)
-        } else {
-            None
-        };
+        let output = self.maybe_parse_output_clause()?;
 
         let using = if self.parse_keyword(Keyword::USING) {
             Some(self.parse_comma_separated(Parser::parse_table_and_joins)?)
@@ -17305,13 +17299,7 @@ impl<'a> Parser<'a> {
                     Default::default()
                 };
 
-                // MSSQL OUTPUT clause appears between columns and source
-                // https://learn.microsoft.com/en-us/sql/t-sql/queries/output-clause-transact-sql
-                let output = if self.parse_keyword(Keyword::OUTPUT) {
-                    Some(self.parse_output(Keyword::OUTPUT, self.get_current_token().clone())?)
-                } else {
-                    None
-                };
+                let output = self.maybe_parse_output_clause()?;
 
                 let (source, assignments) = if self.peek_keyword(Keyword::FORMAT)
                     || self.peek_keyword(Keyword::SETTINGS)
@@ -17539,13 +17527,7 @@ impl<'a> Parser<'a> {
         self.expect_keyword(Keyword::SET)?;
         let assignments = self.parse_comma_separated(Parser::parse_assignment)?;
 
-        // MSSQL OUTPUT clause appears after SET, before FROM/WHERE
-        // https://learn.microsoft.com/en-us/sql/t-sql/queries/output-clause-transact-sql
-        let output = if self.parse_keyword(Keyword::OUTPUT) {
-            Some(self.parse_output(Keyword::OUTPUT, self.get_current_token().clone())?)
-        } else {
-            None
-        };
+        let output = self.maybe_parse_output_clause()?;
 
         let from = if from_before_set.is_none() && self.parse_keyword(Keyword::FROM) {
             Some(UpdateTableFromKind::AfterSet(