mirrors/datafusion-sqlparse
1
0
Fork 0
mirror of https://github.com/apache/datafusion-sqlparser-rs.git synced 2025-10-27 05:36:23 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Fix #1469 (SET ROLE regression) (#1474)

Browse source
This commit is contained in:
Ophir LOJKINE 2024-10-14 19:14:40 +02:00 • committed by GitHub
parent 749b061fbf
commit 7c20d4ae1f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 48 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

40
src/parser/mod.rs
View file

@ -9416,27 +9416,35 @@ impl<'a> Parser<'a> {
}
}
/// Parse a `SET ROLE` statement. Expects SET to be consumed already.
fn parse_set_role(&mut self, modifier: Option<Keyword>) -> Result<Statement, ParserError> {
self.expect_keyword(Keyword::ROLE)?;
let context_modifier = match modifier {
Some(Keyword::LOCAL) => ContextModifier::Local,
Some(Keyword::SESSION) => ContextModifier::Session,
_ => ContextModifier::None,
};
let role_name = if self.parse_keyword(Keyword::NONE) {
None
} else {
Some(self.parse_identifier(false)?)
};
Ok(Statement::SetRole {
context_modifier,
role_name,
})
}
pub fn parse_set(&mut self) -> Result<Statement, ParserError> {
let modifier =
self.parse_one_of_keywords(&[Keyword::SESSION, Keyword::LOCAL, Keyword::HIVEVAR]);
if let Some(Keyword::HIVEVAR) = modifier {
self.expect_token(&Token::Colon)?;
} else if self.parse_keyword(Keyword::ROLE) {
let context_modifier = match modifier {
Some(Keyword::LOCAL) => ContextModifier::Local,
Some(Keyword::SESSION) => ContextModifier::Session,
_ => ContextModifier::None,
};
let role_name = if self.parse_keyword(Keyword::NONE) {
None
} else {
Some(self.parse_identifier(false)?)
};
return Ok(Statement::SetRole {
context_modifier,
role_name,
});
} else if let Some(set_role_stmt) =
self.maybe_parse(|parser| parser.parse_set_role(modifier))
{
return Ok(set_role_stmt);
}
let variables = if self.parse_keywords(&[Keyword::TIME, Keyword::ZONE]) {