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_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 {