fix: Support SET ROLE with TO/=

Author: MazterQyou

src/parser.rs

@@ -3629,6 +3629,7 @@ impl<'a> Parser<'a> {
         ]);
 
         if modifier != Some(Keyword::HIVEVAR) && self.parse_keyword(Keyword::ROLE) {
+            let _ = self.consume_token(&Token::Eq) || self.parse_keyword(Keyword::TO);
             let role_name = if self.parse_keyword(Keyword::NONE) {
                 None
             } else {

tests/sqlparser_postgres.rs

@@ -960,7 +960,10 @@ fn parse_set_role() {
         }
     );
 
-    let stmt = pg_and_generic().verified_stmt("SET LOCAL ROLE \"rolename\"");
+    let stmt = pg_and_generic().one_statement_parses_to(
+        "SET LOCAL ROLE = \"rolename\"",
+        "SET LOCAL ROLE \"rolename\"",
+    );
     assert_eq!(
         stmt,
         Statement::SetRole {
@@ -973,7 +976,8 @@ fn parse_set_role() {
         }
     );
 
-    let stmt = pg_and_generic().verified_stmt("SET ROLE 'rolename'");
+    let stmt =
+        pg_and_generic().one_statement_parses_to("SET ROLE TO 'rolename'", "SET ROLE 'rolename'");
     assert_eq!(
         stmt,
         Statement::SetRole {