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

Merge pull request #77 from benesch/count-distinct

Browse source 
Support COUNT(DISTINCT x) and similar
This commit is contained in:
Nickolay Ponomarev 2019-05-30 02:35:49 +03:00 • committed by GitHub
commit 646479e56c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 65 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

15
src/sqlparser.rs
View file

@ -270,6 +270,14 @@ impl Parser {
pub fn parse_function(&mut self, name: SQLObjectName) -> Result<ASTNode, ParserError> {
self.expect_token(&Token::LParen)?;
let all = self.parse_keyword("ALL");
let distinct = self.parse_keyword("DISTINCT");
if all && distinct {
return parser_err!(format!(
"Cannot specify both ALL and DISTINCT in function: {}",
name.to_string(),
));
}
let args = self.parse_optional_args()?;
let over = if self.parse_keyword("OVER") {
// TBD: support window names (`OVER mywin`) in place of inline specification
@ -296,7 +304,12 @@ impl Parser {
None
};
Ok(ASTNode::SQLFunction { name, args, over })
Ok(ASTNode::SQLFunction {
name,
args,
over,
distinct,
})
}
pub fn parse_window_frame(&mut self) -> Result<Option<SQLWindowFrame>, ParserError> {