fix: Support SET ROLE with TO/=

Author: MazterQyou

src/parser/mod.rs

@@ -15140,6 +15140,7 @@ impl<'a> Parser<'a> {
         modifier: Option<ContextModifier>,
     ) -> Result<Statement, ParserError> {
         self.expect_keyword_is(Keyword::ROLE)?;
+        let _ = self.consume_token(&Token::Eq) || self.parse_keyword(Keyword::TO);
 
         let role_name = if self.parse_keyword(Keyword::NONE) {
             None

tests/sqlparser_common.rs

@@ -9316,22 +9316,16 @@ fn parse_set_variable() {
 
 #[test]
 fn parse_set_role_as_variable() {
-    match verified_stmt("SET role = 'foobar'") {
-        Statement::Set(Set::SingleAssignment {
-            scope,
-            hivevar,
-            variable,
-            values,
+    // In this fork, `SET ROLE` (including the `=`/`TO` forms) is always parsed as a
+    // dedicated SET ROLE statement rather than a generic variable assignment, so
+    // `SET role = 'foobar'` becomes a `Set::SetRole` and round-trips to `SET ROLE 'foobar'`.
+    match one_statement_parses_to("SET role = 'foobar'", "SET ROLE 'foobar'") {
+        Statement::Set(Set::SetRole {
+            context_modifier,
+            role_name,
         }) => {
-            assert_eq!(scope, None);
-            assert!(!hivevar);
-            assert_eq!(variable, ObjectName::from(vec!["role".into()]));
-            assert_eq!(
-                values,
-                vec![Expr::Value(
-                    (Value::SingleQuotedString("foobar".into())).with_empty_span()
-                )]
-            );
+            assert_eq!(context_modifier, None);
+            assert_eq!(role_name, Some(Ident::with_quote('\'', "foobar")));
         }
         _ => unreachable!(),
     }

tests/sqlparser_postgres.rs

@@ -1878,6 +1878,10 @@ fn parse_set_role() {
 
     let query = "SET LOCAL ROLE \"rolename\"";
     let stmt = pg_and_generic().verified_stmt(query);
+    pg_and_generic().one_statement_parses_to(
+        "SET LOCAL ROLE = \"rolename\"",
+        "SET LOCAL ROLE \"rolename\"",
+    );
     assert_eq!(
         stmt,
         Statement::Set(Set::SetRole {
@@ -1893,6 +1897,7 @@ fn parse_set_role() {
 
     let query = "SET ROLE 'rolename'";
     let stmt = pg_and_generic().verified_stmt(query);
+    pg_and_generic().one_statement_parses_to("SET ROLE TO 'rolename'", "SET ROLE 'rolename'");
     assert_eq!(
         stmt,
         Statement::Set(Set::SetRole {