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

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> {
if self.parse_keywords(vec!["WHEN"]) {
if self.parse_keyword("WHEN") {
let mut conditions = vec![];
let mut results = vec![];
let mut else_result = None;
loop {
conditions.push(self.parse_expr()?);
self.expect_keyword("THEN")?;
results.push(self.parse_expr()?);
if self.parse_keywords(vec!["ELSE"]) {
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"]) {
if !self.parse_keyword("WHEN") {
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 {
conditions,
results,