set time zone shorthand

Author: MohamedAbdeen21

src/parser/mod.rs

@@ -11042,13 +11042,15 @@ impl<'a> Parser<'a> {
                     variables: OneOrManyWithParens::One(ObjectName::from(vec!["TIMEZONE".into()])),
                     value: self.parse_set_values(false)?,
                 });
+            } else if self.dialect.is::<PostgreSqlDialect>() {
+                // Special case for Postgres
+                return Ok(Statement::SetTimeZone {
+                    local: modifier == Some(Keyword::LOCAL),
+                    value: self.parse_expr()?,
+                });
+            } else {
+                return self.expected("assignment operator", self.peek_token());
             }
-
-            // Special case for Postgres
-            return Ok(Statement::SetTimeZone {
-                local: modifier == Some(Keyword::LOCAL),
-                value: self.parse_expr()?,
-            });
         } else if self.dialect.supports_set_names() && self.parse_keyword(Keyword::NAMES) {
             if self.parse_keyword(Keyword::DEFAULT) {
                 return Ok(Statement::SetNamesDefault {});

tests/sqlparser_common.rs

@@ -14656,3 +14656,11 @@ fn parse_multiple_set_statements() -> Result<(), ParserError> {
 
     Ok(())
 }
+
+#[test]
+fn parse_set_time_zone_alias() {
+    // not sure what other dialects support this
+    all_dialects_but_pg()
+        .parse_sql_statements("SET TIME ZONE 'UTC'")
+        .unwrap_err();
+}