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

Support VALUES and parenthesized SELECTs as top-level statements

Browse source
This commit is contained in:
Nikhil Benesch 2019-06-10 10:58:10 -04:00
parent b841dccc2c
commit ce171c2a3d
No known key found for this signature in database
GPG key ID: FCF98542083C5A69
2 changed files with 13 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/sqlparser.rs
View file

@ -109,7 +109,7 @@ impl Parser {
match self.next_token() { match self.next_token() {
Some(t) => match t { Some(t) => match t {
Token::SQLWord(ref w) if w.keyword != "" => match w.keyword.as_ref() { Token::SQLWord(ref w) if w.keyword != "" => match w.keyword.as_ref() {
"SELECT" | "WITH" => { "SELECT" | "WITH" | "VALUES" => {
self.prev_token(); self.prev_token();
Ok(SQLStatement::SQLQuery(Box::new(self.parse_query()?))) Ok(SQLStatement::SQLQuery(Box::new(self.parse_query()?)))
} }
@ -133,6 +133,10 @@ impl Parser {
w.to_string() w.to_string()
)), )),
}, },
Token::LParen => {
self.prev_token();
Ok(SQLStatement::SQLQuery(Box::new(self.parse_query()?)))
}
unexpected => self.expected( unexpected => self.expected(
"a keyword at the beginning of a statement", "a keyword at the beginning of a statement",
Some(unexpected), Some(unexpected),

8
tests/sqlparser_common.rs
View file

@ -181,6 +181,14 @@ fn parse_where_delete_statement() {
} }
} }
#[test]
fn parse_top_level() {
verified_stmt("SELECT 1");
verified_stmt("(SELECT 1)");
verified_stmt("((SELECT 1))");
verified_stmt("VALUES (1)");
}
#[test] #[test]
fn parse_simple_select() { fn parse_simple_select() {
let sql = "SELECT id, fname, lname FROM customer WHERE id = 1 LIMIT 5"; let sql = "SELECT id, fname, lname FROM customer WHERE id = 1 LIMIT 5";