cargo fmt, fix compiler warnings

This commit is contained in:
Andy Grove 2018-12-16 13:57:01 -07:00
parent 24af049930
commit e863bc041c
8 changed files with 93 additions and 77 deletions

View file

@ -3,7 +3,6 @@ mod generic_sql;
pub mod keywords; pub mod keywords;
mod postgresql; mod postgresql;
pub use self::ansi_sql::AnsiSqlDialect; pub use self::ansi_sql::AnsiSqlDialect;
pub use self::generic_sql::GenericSqlDialect; pub use self::generic_sql::GenericSqlDialect;
pub use self::postgresql::PostgreSqlDialect; pub use self::postgresql::PostgreSqlDialect;

View file

@ -13,8 +13,8 @@ impl Dialect for PostgreSqlDialect {
CHAR, CHARACTER, VARYING, LARGE, VARCHAR, CLOB, BINARY, VARBINARY, BLOB, FLOAT, REAL, CHAR, CHARACTER, VARYING, LARGE, VARCHAR, CLOB, BINARY, VARBINARY, BLOB, FLOAT, REAL,
DOUBLE, PRECISION, INT, INTEGER, SMALLINT, BIGINT, NUMERIC, DECIMAL, DEC, BOOLEAN, DOUBLE, PRECISION, INT, INTEGER, SMALLINT, BIGINT, NUMERIC, DECIMAL, DEC, BOOLEAN,
DATE, TIME, TIMESTAMP, VALUES, DEFAULT, ZONE, REGCLASS, TEXT, BYTEA, TRUE, FALSE, COPY, DATE, TIME, TIMESTAMP, VALUES, DEFAULT, ZONE, REGCLASS, TEXT, BYTEA, TRUE, FALSE, COPY,
STDIN, PRIMARY, KEY, UNIQUE, UUID, ADD, CONSTRAINT, FOREIGN, REFERENCES, STDIN, PRIMARY, KEY, UNIQUE, UUID, ADD, CONSTRAINT, FOREIGN, REFERENCES, CASE, WHEN,
CASE, WHEN, THEN, ELSE, END, LIKE, THEN, ELSE, END, LIKE,
]; ];
} }

View file

@ -167,10 +167,16 @@ impl ToString for ASTNode {
.collect::<Vec<String>>() .collect::<Vec<String>>()
.join(", ") .join(", ")
), ),
ASTNode::SQLCase { conditions, results, else_result } => { ASTNode::SQLCase {
conditions,
results,
else_result,
} => {
let mut s = format!( let mut s = format!(
"CASE {}", "CASE {}",
conditions.iter().zip(results) conditions
.iter()
.zip(results)
.map(|(c, r)| format!("WHEN {} THEN {}", c.to_string(), r.to_string())) .map(|(c, r)| format!("WHEN {} THEN {}", c.to_string(), r.to_string()))
.collect::<Vec<String>>() .collect::<Vec<String>>()
.join(" ") .join(" ")
@ -179,7 +185,7 @@ impl ToString for ASTNode {
s += &format!(" ELSE {}", else_result.to_string()) s += &format!(" ELSE {}", else_result.to_string())
} }
s + " END" s + " END"
}, }
ASTNode::SQLSelect { ASTNode::SQLSelect {
projection, projection,
relation, relation,

View file

@ -1,7 +1,4 @@
use chrono::{ use chrono::{offset::FixedOffset, DateTime, NaiveDate, NaiveDateTime, NaiveTime};
offset::{FixedOffset},
DateTime, NaiveDate, NaiveDateTime, NaiveTime,
};
use uuid::Uuid; use uuid::Uuid;

View file

@ -17,10 +17,7 @@
use super::dialect::Dialect; use super::dialect::Dialect;
use super::sqlast::*; use super::sqlast::*;
use super::sqltokenizer::*; use super::sqltokenizer::*;
use chrono::{ use chrono::{offset::FixedOffset, DateTime, NaiveDate, NaiveDateTime, NaiveTime};
offset::FixedOffset,
DateTime, NaiveDate, NaiveDateTime, NaiveTime,
};
#[derive(Debug, Clone)] #[derive(Debug, Clone)]
pub enum ParserError { pub enum ParserError {
@ -109,10 +106,8 @@ impl Parser {
"NULL" => { "NULL" => {
self.prev_token(); self.prev_token();
self.parse_sql_value() self.parse_sql_value()
},
"CASE" => {
self.parse_case_expression()
} }
"CASE" => self.parse_case_expression(),
_ => return parser_err!(format!("No prefix parser for keyword {}", k)), _ => return parser_err!(format!("No prefix parser for keyword {}", k)),
}, },
Token::Mult => Ok(ASTNode::SQLWildcard), Token::Mult => Ok(ASTNode::SQLWildcard),
@ -156,14 +151,14 @@ impl Parser {
Token::DoubleQuotedString(_) => { Token::DoubleQuotedString(_) => {
self.prev_token(); self.prev_token();
self.parse_sql_value() self.parse_sql_value()
}, }
Token::LParen => { Token::LParen => {
let expr = self.parse(); let expr = self.parse();
if !self.consume_token(&Token::RParen)? { if !self.consume_token(&Token::RParen)? {
return parser_err!(format!("expected token RParen")); return parser_err!(format!("expected token RParen"));
} }
expr expr
}, }
_ => parser_err!(format!( _ => parser_err!(format!(
"Prefix parser expected a keyword but found {:?}", "Prefix parser expected a keyword but found {:?}",
t t
@ -211,20 +206,20 @@ impl Parser {
if self.parse_keywords(vec!["ELSE"]) { if self.parse_keywords(vec!["ELSE"]) {
else_result = Some(Box::new(self.parse_expr(0)?)); else_result = Some(Box::new(self.parse_expr(0)?));
if self.parse_keywords(vec!["END"]) { if self.parse_keywords(vec!["END"]) {
break break;
} else { } else {
return parser_err!("Expecting END after a CASE..ELSE"); return parser_err!("Expecting END after a CASE..ELSE");
} }
} }
if self.parse_keywords(vec!["END"]) { if self.parse_keywords(vec!["END"]) {
break break;
} }
self.consume_token(&Token::Keyword("WHEN".to_string()))?; self.consume_token(&Token::Keyword("WHEN".to_string()))?;
} }
Ok(ASTNode::SQLCase { Ok(ASTNode::SQLCase {
conditions, conditions,
results, results,
else_result else_result,
}) })
} else { } else {
// TODO: implement "simple" case // TODO: implement "simple" case
@ -356,7 +351,7 @@ impl Parser {
&Token::Keyword(ref k) if k == "OR" => Ok(5), &Token::Keyword(ref k) if k == "OR" => Ok(5),
&Token::Keyword(ref k) if k == "AND" => Ok(10), &Token::Keyword(ref k) if k == "AND" => Ok(10),
&Token::Keyword(ref k) if k == "IS" => Ok(15), &Token::Keyword(ref k) if k == "IS" => Ok(15),
&Token::Keyword(ref k )if k == "LIKE" => Ok(20), &Token::Keyword(ref k) if k == "LIKE" => Ok(20),
&Token::Eq | &Token::Lt | &Token::LtEq | &Token::Neq | &Token::Gt | &Token::GtEq => { &Token::Eq | &Token::Lt | &Token::LtEq | &Token::Neq | &Token::Gt | &Token::GtEq => {
Ok(20) Ok(20)
} }
@ -753,12 +748,12 @@ impl Parser {
}, },
Token::Number(ref n) => match n.parse::<i64>() { Token::Number(ref n) => match n.parse::<i64>() {
Ok(n) => { Ok(n) => {
// if let Some(Token::Minus) = self.peek_token() { // if let Some(Token::Minus) = self.peek_token() {
// self.prev_token(); // self.prev_token();
// self.parse_timestamp_value() // self.parse_timestamp_value()
// } else { // } else {
Ok(Value::Long(n)) Ok(Value::Long(n))
// } // }
} }
Err(e) => parser_err!(format!("Could not parse '{}' as i64: {}", n, e)), Err(e) => parser_err!(format!("Could not parse '{}' as i64: {}", n, e)),
}, },
@ -1217,19 +1212,17 @@ impl Parser {
// look for optional ASC / DESC specifier // look for optional ASC / DESC specifier
let asc = match self.peek_token() { let asc = match self.peek_token() {
Some(Token::Keyword(k)) => { Some(Token::Keyword(k)) => match k.to_uppercase().as_ref() {
match k.to_uppercase().as_ref() { "ASC" => {
"ASC" => { self.next_token();
self.next_token(); true
true
},
"DESC" => {
self.next_token();
false
},
_ => true
} }
} "DESC" => {
self.next_token();
false
}
_ => true,
},
Some(Token::Comma) => true, Some(Token::Comma) => true,
_ => true, _ => true,
}; };

View file

@ -11,16 +11,13 @@ fn parse_simple_select() {
let sql = String::from("SELECT id, fname, lname FROM customer WHERE id = 1"); let sql = String::from("SELECT id, fname, lname FROM customer WHERE id = 1");
let ast = parse_sql(&sql); let ast = parse_sql(&sql);
match ast { match ast {
ASTNode::SQLSelect { ASTNode::SQLSelect { projection, .. } => {
projection, ..
} => {
assert_eq!(3, projection.len()); assert_eq!(3, projection.len());
} }
_ => assert!(false), _ => assert!(false),
} }
} }
fn parse_sql(sql: &str) -> ASTNode { fn parse_sql(sql: &str) -> ASTNode {
let dialect = AnsiSqlDialect {}; let dialect = AnsiSqlDialect {};
let mut tokenizer = Tokenizer::new(&dialect, &sql); let mut tokenizer = Tokenizer::new(&dialect, &sql);
@ -29,4 +26,3 @@ fn parse_sql(sql: &str) -> ASTNode {
let ast = parser.parse().unwrap(); let ast = parser.parse().unwrap();
ast ast
} }

View file

@ -13,7 +13,9 @@ fn parse_delete_statement() {
match parse_sql(&sql) { match parse_sql(&sql) {
ASTNode::SQLDelete { relation, .. } => { ASTNode::SQLDelete { relation, .. } => {
assert_eq!( assert_eq!(
Some(Box::new(ASTNode::SQLValue(Value::SingleQuotedString("table".to_string())))), Some(Box::new(ASTNode::SQLValue(Value::SingleQuotedString(
"table".to_string()
)))),
relation relation
); );
} }
@ -36,7 +38,9 @@ fn parse_where_delete_statement() {
.. ..
} => { } => {
assert_eq!( assert_eq!(
Some(Box::new(ASTNode::SQLValue(Value::SingleQuotedString("table".to_string())))), Some(Box::new(ASTNode::SQLValue(Value::SingleQuotedString(
"table".to_string()
)))),
relation relation
); );
@ -207,7 +211,9 @@ fn parse_select_order_by_limit() {
); );
let ast = parse_sql(&sql); let ast = parse_sql(&sql);
match ast { match ast {
ASTNode::SQLSelect { order_by, limit, .. } => { ASTNode::SQLSelect {
order_by, limit, ..
} => {
assert_eq!( assert_eq!(
Some(vec![ Some(vec![
SQLOrderByExpr { SQLOrderByExpr {
@ -341,7 +347,10 @@ fn parse_literal_string() {
let sql = "SELECT 'one'"; let sql = "SELECT 'one'";
match parse_sql(&sql) { match parse_sql(&sql) {
ASTNode::SQLSelect { ref projection, .. } => { ASTNode::SQLSelect { ref projection, .. } => {
assert_eq!(projection[0], ASTNode::SQLValue(Value::SingleQuotedString("one".to_string()))); assert_eq!(
projection[0],
ASTNode::SQLValue(Value::SingleQuotedString("one".to_string()))
);
} }
_ => panic!(), _ => panic!(),
} }
@ -392,8 +401,9 @@ fn parse_parens() {
op: Plus, op: Plus,
right: Box::new(SQLIdentifier("d".to_string())) right: Box::new(SQLIdentifier("d".to_string()))
}) })
} },
, ast); ast
);
} }
#[test] #[test]
@ -410,17 +420,28 @@ fn parse_case_expression() {
SQLCase { SQLCase {
conditions: vec![ conditions: vec![
SQLIsNull(Box::new(SQLIdentifier("bar".to_string()))), SQLIsNull(Box::new(SQLIdentifier("bar".to_string()))),
SQLBinaryExpr { left: Box::new(SQLIdentifier("bar".to_string())), SQLBinaryExpr {
op: Eq, right: Box::new(SQLValue(Value::Long(0))) }, left: Box::new(SQLIdentifier("bar".to_string())),
SQLBinaryExpr { left: Box::new(SQLIdentifier("bar".to_string())), op: Eq,
op: GtEq, right: Box::new(SQLValue(Value::Long(0))) } right: Box::new(SQLValue(Value::Long(0)))
},
SQLBinaryExpr {
left: Box::new(SQLIdentifier("bar".to_string())),
op: GtEq,
right: Box::new(SQLValue(Value::Long(0)))
}
], ],
results: vec![SQLValue(Value::SingleQuotedString("null".to_string())), results: vec![
SQLValue(Value::SingleQuotedString("=0".to_string())), SQLValue(Value::SingleQuotedString("null".to_string())),
SQLValue(Value::SingleQuotedString(">=0".to_string()))], SQLValue(Value::SingleQuotedString("=0".to_string())),
else_result: Some(Box::new(SQLValue(Value::SingleQuotedString("<0".to_string())))) SQLValue(Value::SingleQuotedString(">=0".to_string()))
],
else_result: Some(Box::new(SQLValue(Value::SingleQuotedString(
"<0".to_string()
))))
}, },
projection[0]); projection[0]
);
} }
_ => assert!(false), _ => assert!(false),
} }
@ -445,7 +466,9 @@ fn parse_delete_with_semi_colon() {
match parse_sql(&sql) { match parse_sql(&sql) {
ASTNode::SQLDelete { relation, .. } => { ASTNode::SQLDelete { relation, .. } => {
assert_eq!( assert_eq!(
Some(Box::new(ASTNode::SQLValue(Value::SingleQuotedString("table".to_string())))), Some(Box::new(ASTNode::SQLValue(Value::SingleQuotedString(
"table".to_string()
)))),
relation relation
); );
} }
@ -455,10 +478,9 @@ fn parse_delete_with_semi_colon() {
fn parse_sql(sql: &str) -> ASTNode { fn parse_sql(sql: &str) -> ASTNode {
let dialect = GenericSqlDialect {}; let dialect = GenericSqlDialect {};
let mut tokenizer = Tokenizer::new(&dialect,&sql, ); let mut tokenizer = Tokenizer::new(&dialect, &sql);
let tokens = tokenizer.tokenize().unwrap(); let tokens = tokenizer.tokenize().unwrap();
let mut parser = Parser::new(tokens); let mut parser = Parser::new(tokens);
let ast = parser.parse().unwrap(); let ast = parser.parse().unwrap();
ast ast
} }

View file

@ -632,17 +632,18 @@ PHP ₱ USD $
//assert_eq!(sql, ast.to_string()); //assert_eq!(sql, ast.to_string());
} }
#[ignore] #[test]
fn parse_timestamps_example() { fn parse_timestamps_example() {
let sql = "2016-02-15 09:43:33"; let sql = "2016-02-15 09:43:33";
let ast = parse_sql(sql); let _ = parse_sql(sql);
assert_eq!(sql, ast.to_string()); //TODO add assertion
} }
#[test] #[test]
fn parse_timestamps_with_millis_example() { fn parse_timestamps_with_millis_example() {
let sql = "2017-11-02 19:15:42.308637"; let sql = "2017-11-02 19:15:42.308637";
let _ = parse_sql(sql); let _ = parse_sql(sql);
//TODO add assertion
} }
#[test] #[test]
@ -742,7 +743,9 @@ fn parse_like() {
ASTNode::SQLBinaryExpr { ASTNode::SQLBinaryExpr {
left: Box::new(ASTNode::SQLIdentifier("name".to_string())), left: Box::new(ASTNode::SQLIdentifier("name".to_string())),
op: SQLOperator::Like, op: SQLOperator::Like,
right: Box::new(ASTNode::SQLValue(Value::SingleQuotedString("%a".to_string()))), right: Box::new(ASTNode::SQLValue(Value::SingleQuotedString(
"%a".to_string()
))),
}, },
*selection.unwrap() *selection.unwrap()
); );