feat: Support multiple values in SET

Author: MazterQyou

src/ast/mod.rs

@@ -1858,12 +1858,13 @@ impl fmt::Display for Statement {
                     .iter()
                     .map(|key_value| {
                         format!(
-                            "{}{}",
+                            "{}{} = {}",
                             if key_value.local { "LOCAL " } else { "" },
+                            key_value.key,
                             key_value
                                 .value
                                 .iter()
-                                .map(|value| format!("{} = {}", key_value.key, value))
+                                .map(|value| format!("{}", value))
                                 .collect::<Vec<String>>()
                                 .join(", ")
                         )

src/parser.rs

@@ -3746,11 +3746,16 @@ impl<'a> Parser<'a> {
         }
 
         let mut key_values: Vec<SetVariableKeyValue> = vec![];
-        loop {
+
+        if dialect_of!(self is PostgreSqlDialect | RedshiftSqlDialect) {
             let variable = self.parse_object_name()?;
             let mut values = vec![];
 
-            if self.consume_token(&Token::Eq) || self.parse_keyword(Keyword::TO) {
+            if !self.consume_token(&Token::Eq) && !self.parse_keyword(Keyword::TO) {
+                return self.expected("equals sign or TO", self.peek_token());
+            }
+
+            loop {
                 let value = if let Ok(expr) = self.parse_expr() {
                     expr
                 } else {
@@ -3759,22 +3764,50 @@ impl<'a> Parser<'a> {
 
                 values.push(value);
 
-                key_values.push(SetVariableKeyValue {
-                    key: variable,
-                    value: values,
-                    local: modifier == Some(Keyword::LOCAL),
-                    hivevar: false,
-                });
-
-                if self.consume_token(&Token::Comma) {
-                    continue;
+                if !self.consume_token(&Token::Comma) {
+                    break;
                 }
+            }
 
-                return Ok(Statement::SetVariable { key_values });
-            } else {
+            key_values.push(SetVariableKeyValue {
+                key: variable,
+                value: values,
+                local: modifier == Some(Keyword::LOCAL),
+                hivevar: false,
+            });
+
+            return Ok(Statement::SetVariable { key_values });
+        }
+
+        loop {
+            let variable = self.parse_object_name()?;
+            let mut values = vec![];
+
+            if !self.consume_token(&Token::Eq) && !self.parse_keyword(Keyword::TO) {
                 return self.expected("equals sign or TO", self.peek_token());
             }
+
+            let value = if let Ok(expr) = self.parse_expr() {
+                expr
+            } else {
+                self.expected("variable value", self.peek_token())?
+            };
+
+            values.push(value);
+
+            key_values.push(SetVariableKeyValue {
+                key: variable,
+                value: values,
+                local: modifier == Some(Keyword::LOCAL),
+                hivevar: false,
+            });
+
+            if !self.consume_token(&Token::Comma) {
+                break;
+            }
         }
+
+        Ok(Statement::SetVariable { key_values })
     }
 
     pub fn parse_show(&mut self) -> Result<Statement, ParserError> {

tests/sqlparser_postgres.rs

@@ -792,6 +792,29 @@ fn parse_set() {
         }
     );
 
+    let stmt = pg().verified_stmt("SET a = b, c");
+    assert_eq!(
+        stmt,
+        Statement::SetVariable {
+            key_values: [SetVariableKeyValue {
+                key: ObjectName(vec![Ident::new("a")]),
+                value: vec![
+                    Expr::Identifier(Ident {
+                        value: "b".into(),
+                        quote_style: None
+                    }),
+                    Expr::Identifier(Ident {
+                        value: "c".into(),
+                        quote_style: None
+                    }),
+                ],
+                local: false,
+                hivevar: false,
+            }]
+            .to_vec()
+        }
+    );
+
     let stmt = pg_and_generic().verified_stmt("SET a = 'b'");
     assert_eq!(
         stmt,