mirrors/datafusion-sqlparse
1
0
Fork 0
mirror of https://github.com/apache/datafusion-sqlparser-rs.git synced 2025-07-18 22:04:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

Merge pull request #96 from benesch/extract

Browse source 
Support EXTRACT function-like operator
This commit is contained in:
Nikhil Benesch 2019-06-04 00:07:23 -04:00 • committed by GitHub
commit 11fc833433
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 85 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

29
tests/sqlparser_common.rs
View file

@ -797,6 +797,35 @@ fn parse_cast() {
);
}
#[test]
fn parse_extract() {
let sql = "SELECT EXTRACT(YEAR FROM d)";
let select = verified_only_select(sql);
assert_eq!(
&ASTNode::SQLExtract {
field: SQLDateTimeField::Year,
expr: Box::new(ASTNode::SQLIdentifier("d".to_string())),
},
expr_from_projection(only(&select.projection)),
);
one_statement_parses_to("SELECT EXTRACT(year from d)", "SELECT EXTRACT(YEAR FROM d)");
verified_stmt("SELECT EXTRACT(MONTH FROM d)");
verified_stmt("SELECT EXTRACT(DAY FROM d)");
verified_stmt("SELECT EXTRACT(HOUR FROM d)");
verified_stmt("SELECT EXTRACT(MINUTE FROM d)");
verified_stmt("SELECT EXTRACT(SECOND FROM d)");
let res = parse_sql_statements("SELECT EXTRACT(MILLISECOND FROM d)");
assert_eq!(
ParserError::ParserError(
"Expected Date/time field inside of EXTRACT function, found: MILLISECOND".to_string()
),
res.unwrap_err()
);
}
#[test]
fn parse_create_table() {
let sql = "CREATE TABLE uk_cities (\