mirrors/datafusion-sqlparse
1
0
Fork 0
mirror of https://github.com/apache/datafusion-sqlparser-rs.git synced 2025-09-06 22:20:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Simplify CASE parsing

Browse source
This commit is contained in:
Nickolay Ponomarev 2019-04-27 18:15:36 +03:00
parent e5c0c426b7
commit 2aa4c267e7
1 changed files with 8 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

20
src/sqlparser.rs
View file

@ -336,27 +336,23 @@ impl Parser {
} }
pub fn parse_case_expression(&mut self) -> Result<ASTNode, ParserError> { pub fn parse_case_expression(&mut self) -> Result<ASTNode, ParserError> {
if self.parse_keywords(vec!["WHEN"]) { if self.parse_keyword("WHEN") {
let mut conditions = vec![]; let mut conditions = vec![];
let mut results = vec![]; let mut results = vec![];
let mut else_result = None;
loop { loop {
conditions.push(self.parse_expr()?); conditions.push(self.parse_expr()?);
self.expect_keyword("THEN")?; self.expect_keyword("THEN")?;
results.push(self.parse_expr()?); results.push(self.parse_expr()?);
if self.parse_keywords(vec!["ELSE"]) { if !self.parse_keyword("WHEN") {
else_result = Some(Box::new(self.parse_expr()?));
if self.parse_keywords(vec!["END"]) {
break;
} else {
return parser_err!("Expecting END after a CASE..ELSE");
}
}
if self.parse_keywords(vec!["END"]) {
break; break;
} }
self.expect_keyword("WHEN")?;
} }
let else_result = if self.parse_keyword("ELSE") {
Some(Box::new(self.parse_expr()?))
} else {
None
};
self.expect_keyword("END")?;
Ok(ASTNode::SQLCase { Ok(ASTNode::SQLCase {
conditions, conditions,
results, results,