mirror of
https://github.com/apache/datafusion-sqlparser-rs.git
synced 2025-08-17 20:50:16 +00:00
4907 lines
190 KiB
Rust
4907 lines
190 KiB
Rust
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
//! SQL Parser
|
|
|
|
#[cfg(not(feature = "std"))]
|
|
use alloc::{
|
|
boxed::Box,
|
|
format,
|
|
string::{String, ToString},
|
|
vec,
|
|
vec::Vec,
|
|
};
|
|
use core::fmt;
|
|
|
|
use log::debug;
|
|
|
|
use crate::ast::*;
|
|
use crate::dialect::*;
|
|
use crate::keywords::{self, Keyword};
|
|
use crate::tokenizer::*;
|
|
|
|
#[derive(Debug, Clone, PartialEq, Eq)]
|
|
pub enum ParserError {
|
|
TokenizerError(String),
|
|
ParserError(String),
|
|
}
|
|
|
|
// Use `Parser::expected` instead, if possible
|
|
macro_rules! parser_err {
|
|
($MSG:expr) => {
|
|
Err(ParserError::ParserError($MSG.to_string()))
|
|
};
|
|
}
|
|
|
|
// Returns a successful result if the optional expression is some
|
|
macro_rules! return_ok_if_some {
|
|
($e:expr) => {{
|
|
if let Some(v) = $e {
|
|
return Ok(v);
|
|
}
|
|
}};
|
|
}
|
|
|
|
#[derive(PartialEq, Eq)]
|
|
pub enum IsOptional {
|
|
Optional,
|
|
Mandatory,
|
|
}
|
|
|
|
use IsOptional::*;
|
|
|
|
pub enum IsLateral {
|
|
Lateral,
|
|
NotLateral,
|
|
}
|
|
|
|
use IsLateral::*;
|
|
|
|
pub enum WildcardExpr {
|
|
Expr(Expr),
|
|
QualifiedWildcard(ObjectName),
|
|
Wildcard,
|
|
}
|
|
|
|
impl From<WildcardExpr> for FunctionArgExpr {
|
|
fn from(wildcard_expr: WildcardExpr) -> Self {
|
|
match wildcard_expr {
|
|
WildcardExpr::Expr(expr) => Self::Expr(expr),
|
|
WildcardExpr::QualifiedWildcard(prefix) => Self::QualifiedWildcard(prefix),
|
|
WildcardExpr::Wildcard => Self::Wildcard,
|
|
}
|
|
}
|
|
}
|
|
|
|
impl From<TokenizerError> for ParserError {
|
|
fn from(e: TokenizerError) -> Self {
|
|
ParserError::TokenizerError(e.to_string())
|
|
}
|
|
}
|
|
|
|
impl fmt::Display for ParserError {
|
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
|
write!(
|
|
f,
|
|
"sql parser error: {}",
|
|
match self {
|
|
ParserError::TokenizerError(s) => s,
|
|
ParserError::ParserError(s) => s,
|
|
}
|
|
)
|
|
}
|
|
}
|
|
|
|
#[cfg(feature = "std")]
|
|
impl std::error::Error for ParserError {}
|
|
|
|
pub struct Parser<'a> {
|
|
tokens: Vec<Token>,
|
|
/// The index of the first unprocessed token in `self.tokens`
|
|
index: usize,
|
|
dialect: &'a dyn Dialect,
|
|
}
|
|
|
|
impl<'a> Parser<'a> {
|
|
/// Parse the specified tokens
|
|
pub fn new(tokens: Vec<Token>, dialect: &'a dyn Dialect) -> Self {
|
|
Parser {
|
|
tokens,
|
|
index: 0,
|
|
dialect,
|
|
}
|
|
}
|
|
|
|
/// Parse a SQL statement and produce an Abstract Syntax Tree (AST)
|
|
pub fn parse_sql(dialect: &dyn Dialect, sql: &str) -> Result<Vec<Statement>, ParserError> {
|
|
let mut tokenizer = Tokenizer::new(dialect, sql);
|
|
let tokens = tokenizer.tokenize()?;
|
|
let mut parser = Parser::new(tokens, dialect);
|
|
let mut stmts = Vec::new();
|
|
let mut expecting_statement_delimiter = false;
|
|
debug!("Parsing sql '{}'...", sql);
|
|
loop {
|
|
// ignore empty statements (between successive statement delimiters)
|
|
while parser.consume_token(&Token::SemiColon) {
|
|
expecting_statement_delimiter = false;
|
|
}
|
|
|
|
if parser.peek_token() == Token::EOF {
|
|
break;
|
|
}
|
|
if expecting_statement_delimiter {
|
|
return parser.expected("end of statement", parser.peek_token());
|
|
}
|
|
|
|
let statement = parser.parse_statement()?;
|
|
stmts.push(statement);
|
|
expecting_statement_delimiter = true;
|
|
}
|
|
Ok(stmts)
|
|
}
|
|
|
|
/// Parse a single top-level statement (such as SELECT, INSERT, CREATE, etc.),
|
|
/// stopping before the statement separator, if any.
|
|
pub fn parse_statement(&mut self) -> Result<Statement, ParserError> {
|
|
match self.next_token() {
|
|
Token::Word(w) => match w.keyword {
|
|
Keyword::KILL => Ok(self.parse_kill()?),
|
|
Keyword::DESCRIBE => Ok(self.parse_explain(true)?),
|
|
Keyword::EXPLAIN => Ok(self.parse_explain(false)?),
|
|
Keyword::ANALYZE => Ok(self.parse_analyze()?),
|
|
Keyword::SELECT | Keyword::WITH | Keyword::VALUES => {
|
|
self.prev_token();
|
|
Ok(Statement::Query(Box::new(self.parse_query()?)))
|
|
}
|
|
Keyword::TRUNCATE => Ok(self.parse_truncate()?),
|
|
Keyword::MSCK => Ok(self.parse_msck()?),
|
|
Keyword::CREATE => Ok(self.parse_create()?),
|
|
Keyword::DROP => Ok(self.parse_drop()?),
|
|
Keyword::DISCARD => Ok(self.parse_discard()?),
|
|
Keyword::DECLARE => Ok(self.parse_declare()?),
|
|
Keyword::FETCH => Ok(self.parse_fetch_statement()?),
|
|
Keyword::DELETE => Ok(self.parse_delete()?),
|
|
Keyword::INSERT => Ok(self.parse_insert()?),
|
|
Keyword::UPDATE => Ok(self.parse_update()?),
|
|
Keyword::ALTER => Ok(self.parse_alter()?),
|
|
Keyword::COPY => Ok(self.parse_copy()?),
|
|
Keyword::CLOSE => Ok(self.parse_close()?),
|
|
Keyword::SET => Ok(self.parse_set()?),
|
|
Keyword::SHOW => Ok(self.parse_show()?),
|
|
Keyword::USE => Ok(self.parse_use()?),
|
|
Keyword::GRANT => Ok(self.parse_grant()?),
|
|
Keyword::REVOKE => Ok(self.parse_revoke()?),
|
|
Keyword::START => Ok(self.parse_start_transaction()?),
|
|
// `BEGIN` is a nonstandard but common alias for the
|
|
// standard `START TRANSACTION` statement. It is supported
|
|
// by at least PostgreSQL and MySQL.
|
|
Keyword::BEGIN => Ok(self.parse_begin()?),
|
|
Keyword::SAVEPOINT => Ok(self.parse_savepoint()?),
|
|
Keyword::COMMIT => Ok(self.parse_commit()?),
|
|
Keyword::ROLLBACK => Ok(self.parse_rollback()?),
|
|
Keyword::ASSERT => Ok(self.parse_assert()?),
|
|
// `PREPARE`, `EXECUTE` and `DEALLOCATE` are Postgres-specific
|
|
// syntaxes. They are used for Postgres prepared statement.
|
|
Keyword::DEALLOCATE => Ok(self.parse_deallocate()?),
|
|
Keyword::EXECUTE => Ok(self.parse_execute()?),
|
|
Keyword::PREPARE => Ok(self.parse_prepare()?),
|
|
Keyword::MERGE => Ok(self.parse_merge()?),
|
|
Keyword::REPLACE if dialect_of!(self is SQLiteDialect ) => {
|
|
self.prev_token();
|
|
Ok(self.parse_insert()?)
|
|
}
|
|
Keyword::COMMENT if dialect_of!(self is PostgreSqlDialect) => {
|
|
Ok(self.parse_comment()?)
|
|
}
|
|
_ => self.expected("an SQL statement", Token::Word(w)),
|
|
},
|
|
Token::LParen => {
|
|
self.prev_token();
|
|
Ok(Statement::Query(Box::new(self.parse_query()?)))
|
|
}
|
|
unexpected => self.expected("an SQL statement", unexpected),
|
|
}
|
|
}
|
|
|
|
pub fn parse_msck(&mut self) -> Result<Statement, ParserError> {
|
|
let repair = self.parse_keyword(Keyword::REPAIR);
|
|
self.expect_keyword(Keyword::TABLE)?;
|
|
let table_name = self.parse_object_name()?;
|
|
let partition_action = self
|
|
.maybe_parse(|parser| {
|
|
let pa = match parser.parse_one_of_keywords(&[
|
|
Keyword::ADD,
|
|
Keyword::DROP,
|
|
Keyword::SYNC,
|
|
]) {
|
|
Some(Keyword::ADD) => Some(AddDropSync::ADD),
|
|
Some(Keyword::DROP) => Some(AddDropSync::DROP),
|
|
Some(Keyword::SYNC) => Some(AddDropSync::SYNC),
|
|
_ => None,
|
|
};
|
|
parser.expect_keyword(Keyword::PARTITIONS)?;
|
|
Ok(pa)
|
|
})
|
|
.unwrap_or_default();
|
|
Ok(Statement::Msck {
|
|
repair,
|
|
table_name,
|
|
partition_action,
|
|
})
|
|
}
|
|
|
|
pub fn parse_truncate(&mut self) -> Result<Statement, ParserError> {
|
|
self.expect_keyword(Keyword::TABLE)?;
|
|
let table_name = self.parse_object_name()?;
|
|
let mut partitions = None;
|
|
if self.parse_keyword(Keyword::PARTITION) {
|
|
self.expect_token(&Token::LParen)?;
|
|
partitions = Some(self.parse_comma_separated(Parser::parse_expr)?);
|
|
self.expect_token(&Token::RParen)?;
|
|
}
|
|
Ok(Statement::Truncate {
|
|
table_name,
|
|
partitions,
|
|
})
|
|
}
|
|
|
|
pub fn parse_analyze(&mut self) -> Result<Statement, ParserError> {
|
|
self.expect_keyword(Keyword::TABLE)?;
|
|
let table_name = self.parse_object_name()?;
|
|
let mut for_columns = false;
|
|
let mut cache_metadata = false;
|
|
let mut noscan = false;
|
|
let mut partitions = None;
|
|
let mut compute_statistics = false;
|
|
let mut columns = vec![];
|
|
loop {
|
|
match self.parse_one_of_keywords(&[
|
|
Keyword::PARTITION,
|
|
Keyword::FOR,
|
|
Keyword::CACHE,
|
|
Keyword::NOSCAN,
|
|
Keyword::COMPUTE,
|
|
]) {
|
|
Some(Keyword::PARTITION) => {
|
|
self.expect_token(&Token::LParen)?;
|
|
partitions = Some(self.parse_comma_separated(Parser::parse_expr)?);
|
|
self.expect_token(&Token::RParen)?;
|
|
}
|
|
Some(Keyword::NOSCAN) => noscan = true,
|
|
Some(Keyword::FOR) => {
|
|
self.expect_keyword(Keyword::COLUMNS)?;
|
|
|
|
columns = self
|
|
.maybe_parse(|parser| {
|
|
parser.parse_comma_separated(Parser::parse_identifier)
|
|
})
|
|
.unwrap_or_default();
|
|
for_columns = true
|
|
}
|
|
Some(Keyword::CACHE) => {
|
|
self.expect_keyword(Keyword::METADATA)?;
|
|
cache_metadata = true
|
|
}
|
|
Some(Keyword::COMPUTE) => {
|
|
self.expect_keyword(Keyword::STATISTICS)?;
|
|
compute_statistics = true
|
|
}
|
|
_ => break,
|
|
}
|
|
}
|
|
|
|
Ok(Statement::Analyze {
|
|
table_name,
|
|
for_columns,
|
|
columns,
|
|
partitions,
|
|
cache_metadata,
|
|
noscan,
|
|
compute_statistics,
|
|
})
|
|
}
|
|
|
|
/// Parse a new expression including wildcard & qualified wildcard
|
|
pub fn parse_wildcard_expr(&mut self) -> Result<WildcardExpr, ParserError> {
|
|
let index = self.index;
|
|
|
|
match self.next_token() {
|
|
Token::Word(w) if self.peek_token() == Token::Period => {
|
|
let mut id_parts: Vec<Ident> = vec![w.to_ident()];
|
|
|
|
while self.consume_token(&Token::Period) {
|
|
match self.next_token() {
|
|
Token::Word(w) => id_parts.push(w.to_ident()),
|
|
Token::Mul => {
|
|
return Ok(WildcardExpr::QualifiedWildcard(ObjectName(id_parts)));
|
|
}
|
|
unexpected => {
|
|
return self.expected("an identifier or a '*' after '.'", unexpected);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
Token::Mul => {
|
|
return Ok(WildcardExpr::Wildcard);
|
|
}
|
|
_ => (),
|
|
};
|
|
|
|
self.index = index;
|
|
self.parse_expr().map(WildcardExpr::Expr)
|
|
}
|
|
|
|
/// Parse a new expression
|
|
pub fn parse_expr(&mut self) -> Result<Expr, ParserError> {
|
|
self.parse_subexpr(0)
|
|
}
|
|
|
|
/// Parse tokens until the precedence changes
|
|
pub fn parse_subexpr(&mut self, precedence: u8) -> Result<Expr, ParserError> {
|
|
debug!("parsing expr");
|
|
let mut expr = self.parse_prefix()?;
|
|
debug!("prefix: {:?}", expr);
|
|
loop {
|
|
let next_precedence = self.get_next_precedence()?;
|
|
debug!("next precedence: {:?}", next_precedence);
|
|
|
|
if precedence >= next_precedence {
|
|
break;
|
|
}
|
|
|
|
expr = self.parse_infix(expr, next_precedence)?;
|
|
}
|
|
Ok(expr)
|
|
}
|
|
|
|
pub fn parse_assert(&mut self) -> Result<Statement, ParserError> {
|
|
let condition = self.parse_expr()?;
|
|
let message = if self.parse_keyword(Keyword::AS) {
|
|
Some(self.parse_expr()?)
|
|
} else {
|
|
None
|
|
};
|
|
|
|
Ok(Statement::Assert { condition, message })
|
|
}
|
|
|
|
pub fn parse_savepoint(&mut self) -> Result<Statement, ParserError> {
|
|
let name = self.parse_identifier()?;
|
|
Ok(Statement::Savepoint { name })
|
|
}
|
|
|
|
/// Parse an expression prefix
|
|
pub fn parse_prefix(&mut self) -> Result<Expr, ParserError> {
|
|
// PostgreSQL allows any string literal to be preceded by a type name, indicating that the
|
|
// string literal represents a literal of that type. Some examples:
|
|
//
|
|
// DATE '2020-05-20'
|
|
// TIMESTAMP WITH TIME ZONE '2020-05-20 7:43:54'
|
|
// BOOL 'true'
|
|
//
|
|
// The first two are standard SQL, while the latter is a PostgreSQL extension. Complicating
|
|
// matters is the fact that INTERVAL string literals may optionally be followed by special
|
|
// keywords, e.g.:
|
|
//
|
|
// INTERVAL '7' DAY
|
|
//
|
|
// Note also that naively `SELECT date` looks like a syntax error because the `date` type
|
|
// name is not followed by a string literal, but in fact in PostgreSQL it is a valid
|
|
// expression that should parse as the column name "date".
|
|
return_ok_if_some!(self.maybe_parse(|parser| {
|
|
match parser.parse_data_type()? {
|
|
DataType::Interval => parser.parse_literal_interval(),
|
|
// PostgreSQL allows almost any identifier to be used as custom data type name,
|
|
// and we support that in `parse_data_type()`. But unlike Postgres we don't
|
|
// have a list of globally reserved keywords (since they vary across dialects),
|
|
// so given `NOT 'a' LIKE 'b'`, we'd accept `NOT` as a possible custom data type
|
|
// name, resulting in `NOT 'a'` being recognized as a `TypedString` instead of
|
|
// an unary negation `NOT ('a' LIKE 'b')`. To solve this, we don't accept the
|
|
// `type 'string'` syntax for the custom data types at all.
|
|
DataType::Custom(..) => parser_err!("dummy"),
|
|
data_type => Ok(Expr::TypedString {
|
|
data_type,
|
|
value: parser.parse_literal_string()?,
|
|
}),
|
|
}
|
|
}));
|
|
|
|
let expr = match self.next_token() {
|
|
Token::Word(w) => match w.keyword {
|
|
Keyword::TRUE | Keyword::FALSE | Keyword::NULL => {
|
|
self.prev_token();
|
|
Ok(Expr::Value(self.parse_value()?))
|
|
}
|
|
Keyword::CURRENT_CATALOG
|
|
| Keyword::CURRENT_USER
|
|
| Keyword::SESSION_USER
|
|
| Keyword::USER
|
|
if dialect_of!(self is PostgreSqlDialect | GenericDialect) =>
|
|
{
|
|
Ok(Expr::Function(Function {
|
|
name: ObjectName(vec![w.to_ident()]),
|
|
args: vec![],
|
|
over: None,
|
|
distinct: false,
|
|
special: true,
|
|
}))
|
|
}
|
|
Keyword::CURRENT_TIMESTAMP | Keyword::CURRENT_TIME | Keyword::CURRENT_DATE => {
|
|
self.parse_time_functions(ObjectName(vec![w.to_ident()]))
|
|
}
|
|
Keyword::CASE => self.parse_case_expr(),
|
|
Keyword::CAST => self.parse_cast_expr(),
|
|
Keyword::TRY_CAST => self.parse_try_cast_expr(),
|
|
Keyword::SAFE_CAST => self.parse_safe_cast_expr(),
|
|
Keyword::EXISTS => self.parse_exists_expr(false),
|
|
Keyword::EXTRACT => self.parse_extract_expr(),
|
|
Keyword::POSITION => self.parse_position_expr(),
|
|
Keyword::SUBSTRING => self.parse_substring_expr(),
|
|
Keyword::TRIM => self.parse_trim_expr(),
|
|
Keyword::INTERVAL => self.parse_literal_interval(),
|
|
Keyword::LISTAGG => self.parse_listagg_expr(),
|
|
// Treat ARRAY[1,2,3] as an array [1,2,3], otherwise try as function call
|
|
Keyword::ARRAY if self.peek_token() == Token::LBracket => {
|
|
self.expect_token(&Token::LBracket)?;
|
|
self.parse_array_expr(true)
|
|
}
|
|
Keyword::NOT => self.parse_not(),
|
|
// Here `w` is a word, check if it's a part of a multi-part
|
|
// identifier, a function call, or a simple identifier:
|
|
_ => match self.peek_token() {
|
|
Token::LParen | Token::Period => {
|
|
let mut id_parts: Vec<Ident> = vec![w.to_ident()];
|
|
while self.consume_token(&Token::Period) {
|
|
match self.next_token() {
|
|
Token::Word(w) => id_parts.push(w.to_ident()),
|
|
unexpected => {
|
|
return self
|
|
.expected("an identifier or a '*' after '.'", unexpected);
|
|
}
|
|
}
|
|
}
|
|
|
|
if self.consume_token(&Token::LParen) {
|
|
self.prev_token();
|
|
self.parse_function(ObjectName(id_parts))
|
|
} else {
|
|
Ok(Expr::CompoundIdentifier(id_parts))
|
|
}
|
|
}
|
|
_ => Ok(Expr::Identifier(w.to_ident())),
|
|
},
|
|
}, // End of Token::Word
|
|
// array `[1, 2, 3]`
|
|
Token::LBracket => self.parse_array_expr(false),
|
|
tok @ Token::Minus | tok @ Token::Plus => {
|
|
let op = if tok == Token::Plus {
|
|
UnaryOperator::Plus
|
|
} else {
|
|
UnaryOperator::Minus
|
|
};
|
|
Ok(Expr::UnaryOp {
|
|
op,
|
|
expr: Box::new(self.parse_subexpr(Self::PLUS_MINUS_PREC)?),
|
|
})
|
|
}
|
|
tok @ Token::DoubleExclamationMark
|
|
| tok @ Token::PGSquareRoot
|
|
| tok @ Token::PGCubeRoot
|
|
| tok @ Token::AtSign
|
|
| tok @ Token::Tilde
|
|
if dialect_of!(self is PostgreSqlDialect) =>
|
|
{
|
|
let op = match tok {
|
|
Token::DoubleExclamationMark => UnaryOperator::PGPrefixFactorial,
|
|
Token::PGSquareRoot => UnaryOperator::PGSquareRoot,
|
|
Token::PGCubeRoot => UnaryOperator::PGCubeRoot,
|
|
Token::AtSign => UnaryOperator::PGAbs,
|
|
Token::Tilde => UnaryOperator::PGBitwiseNot,
|
|
_ => unreachable!(),
|
|
};
|
|
Ok(Expr::UnaryOp {
|
|
op,
|
|
expr: Box::new(self.parse_subexpr(Self::PLUS_MINUS_PREC)?),
|
|
})
|
|
}
|
|
Token::EscapedStringLiteral(_) if dialect_of!(self is PostgreSqlDialect | GenericDialect) =>
|
|
{
|
|
self.prev_token();
|
|
Ok(Expr::Value(self.parse_value()?))
|
|
}
|
|
Token::Number(_, _)
|
|
| Token::SingleQuotedString(_)
|
|
| Token::DoubleQuotedString(_)
|
|
| Token::NationalStringLiteral(_)
|
|
| Token::HexStringLiteral(_) => {
|
|
self.prev_token();
|
|
Ok(Expr::Value(self.parse_value()?))
|
|
}
|
|
|
|
Token::LParen => {
|
|
let expr =
|
|
if self.parse_keyword(Keyword::SELECT) || self.parse_keyword(Keyword::WITH) {
|
|
self.prev_token();
|
|
Expr::Subquery(Box::new(self.parse_query()?))
|
|
} else {
|
|
let exprs = self.parse_comma_separated(Parser::parse_expr)?;
|
|
match exprs.len() {
|
|
0 => unreachable!(), // parse_comma_separated ensures 1 or more
|
|
1 => Expr::Nested(Box::new(exprs.into_iter().next().unwrap())),
|
|
_ => Expr::Tuple(exprs),
|
|
}
|
|
};
|
|
self.expect_token(&Token::RParen)?;
|
|
if !self.consume_token(&Token::Period) {
|
|
Ok(expr)
|
|
} else {
|
|
let tok = self.next_token();
|
|
let key = match tok {
|
|
Token::Word(word) => word.to_ident(),
|
|
_ => return parser_err!(format!("Expected identifier, found: {}", tok)),
|
|
};
|
|
Ok(Expr::CompositeAccess {
|
|
expr: Box::new(expr),
|
|
key,
|
|
})
|
|
}
|
|
}
|
|
Token::Placeholder(_) => {
|
|
self.prev_token();
|
|
Ok(Expr::Value(self.parse_value()?))
|
|
}
|
|
unexpected => self.expected("an expression:", unexpected),
|
|
}?;
|
|
|
|
if self.parse_keyword(Keyword::COLLATE) {
|
|
Ok(Expr::Collate {
|
|
expr: Box::new(expr),
|
|
collation: self.parse_object_name()?,
|
|
})
|
|
} else {
|
|
Ok(expr)
|
|
}
|
|
}
|
|
|
|
pub fn parse_function(&mut self, name: ObjectName) -> Result<Expr, ParserError> {
|
|
self.expect_token(&Token::LParen)?;
|
|
let distinct = self.parse_all_or_distinct()?;
|
|
let args = self.parse_optional_args()?;
|
|
let over = if self.parse_keyword(Keyword::OVER) {
|
|
// TBD: support window names (`OVER mywin`) in place of inline specification
|
|
self.expect_token(&Token::LParen)?;
|
|
let partition_by = if self.parse_keywords(&[Keyword::PARTITION, Keyword::BY]) {
|
|
// a list of possibly-qualified column names
|
|
self.parse_comma_separated(Parser::parse_expr)?
|
|
} else {
|
|
vec![]
|
|
};
|
|
let order_by = if self.parse_keywords(&[Keyword::ORDER, Keyword::BY]) {
|
|
self.parse_comma_separated(Parser::parse_order_by_expr)?
|
|
} else {
|
|
vec![]
|
|
};
|
|
let window_frame = if !self.consume_token(&Token::RParen) {
|
|
let window_frame = self.parse_window_frame()?;
|
|
self.expect_token(&Token::RParen)?;
|
|
Some(window_frame)
|
|
} else {
|
|
None
|
|
};
|
|
|
|
Some(WindowSpec {
|
|
partition_by,
|
|
order_by,
|
|
window_frame,
|
|
})
|
|
} else {
|
|
None
|
|
};
|
|
|
|
Ok(Expr::Function(Function {
|
|
name,
|
|
args,
|
|
over,
|
|
distinct,
|
|
special: false,
|
|
}))
|
|
}
|
|
|
|
pub fn parse_time_functions(&mut self, name: ObjectName) -> Result<Expr, ParserError> {
|
|
let args = if self.consume_token(&Token::LParen) {
|
|
self.parse_optional_args()?
|
|
} else {
|
|
vec![]
|
|
};
|
|
Ok(Expr::Function(Function {
|
|
name,
|
|
args,
|
|
over: None,
|
|
distinct: false,
|
|
special: false,
|
|
}))
|
|
}
|
|
|
|
pub fn parse_window_frame_units(&mut self) -> Result<WindowFrameUnits, ParserError> {
|
|
match self.next_token() {
|
|
Token::Word(w) => match w.keyword {
|
|
Keyword::ROWS => Ok(WindowFrameUnits::Rows),
|
|
Keyword::RANGE => Ok(WindowFrameUnits::Range),
|
|
Keyword::GROUPS => Ok(WindowFrameUnits::Groups),
|
|
_ => self.expected("ROWS, RANGE, GROUPS", Token::Word(w))?,
|
|
},
|
|
unexpected => self.expected("ROWS, RANGE, GROUPS", unexpected),
|
|
}
|
|
}
|
|
|
|
pub fn parse_window_frame(&mut self) -> Result<WindowFrame, ParserError> {
|
|
let units = self.parse_window_frame_units()?;
|
|
let (start_bound, end_bound) = if self.parse_keyword(Keyword::BETWEEN) {
|
|
let start_bound = self.parse_window_frame_bound()?;
|
|
self.expect_keyword(Keyword::AND)?;
|
|
let end_bound = Some(self.parse_window_frame_bound()?);
|
|
(start_bound, end_bound)
|
|
} else {
|
|
(self.parse_window_frame_bound()?, None)
|
|
};
|
|
Ok(WindowFrame {
|
|
units,
|
|
start_bound,
|
|
end_bound,
|
|
})
|
|
}
|
|
|
|
/// Parse `CURRENT ROW` or `{ <positive number> | UNBOUNDED } { PRECEDING | FOLLOWING }`
|
|
pub fn parse_window_frame_bound(&mut self) -> Result<WindowFrameBound, ParserError> {
|
|
if self.parse_keywords(&[Keyword::CURRENT, Keyword::ROW]) {
|
|
Ok(WindowFrameBound::CurrentRow)
|
|
} else {
|
|
let rows = if self.parse_keyword(Keyword::UNBOUNDED) {
|
|
None
|
|
} else {
|
|
Some(self.parse_literal_uint()?)
|
|
};
|
|
if self.parse_keyword(Keyword::PRECEDING) {
|
|
Ok(WindowFrameBound::Preceding(rows))
|
|
} else if self.parse_keyword(Keyword::FOLLOWING) {
|
|
Ok(WindowFrameBound::Following(rows))
|
|
} else {
|
|
self.expected("PRECEDING or FOLLOWING", self.peek_token())
|
|
}
|
|
}
|
|
}
|
|
|
|
/// parse a group by expr. a group by expr can be one of group sets, roll up, cube, or simple
|
|
/// expr.
|
|
fn parse_group_by_expr(&mut self) -> Result<Expr, ParserError> {
|
|
if dialect_of!(self is PostgreSqlDialect) {
|
|
if self.parse_keywords(&[Keyword::GROUPING, Keyword::SETS]) {
|
|
self.expect_token(&Token::LParen)?;
|
|
let result = self.parse_comma_separated(|p| p.parse_tuple(false, true))?;
|
|
self.expect_token(&Token::RParen)?;
|
|
Ok(Expr::GroupingSets(result))
|
|
} else if self.parse_keyword(Keyword::CUBE) {
|
|
self.expect_token(&Token::LParen)?;
|
|
let result = self.parse_comma_separated(|p| p.parse_tuple(true, true))?;
|
|
self.expect_token(&Token::RParen)?;
|
|
Ok(Expr::Cube(result))
|
|
} else if self.parse_keyword(Keyword::ROLLUP) {
|
|
self.expect_token(&Token::LParen)?;
|
|
let result = self.parse_comma_separated(|p| p.parse_tuple(true, true))?;
|
|
self.expect_token(&Token::RParen)?;
|
|
Ok(Expr::Rollup(result))
|
|
} else {
|
|
self.parse_expr()
|
|
}
|
|
} else {
|
|
// TODO parse rollup for other dialects
|
|
self.parse_expr()
|
|
}
|
|
}
|
|
|
|
/// parse a tuple with `(` and `)`.
|
|
/// If `lift_singleton` is true, then a singleton tuple is lifted to a tuple of length 1, otherwise it will fail.
|
|
/// If `allow_empty` is true, then an empty tuple is allowed.
|
|
fn parse_tuple(
|
|
&mut self,
|
|
lift_singleton: bool,
|
|
allow_empty: bool,
|
|
) -> Result<Vec<Expr>, ParserError> {
|
|
if lift_singleton {
|
|
if self.consume_token(&Token::LParen) {
|
|
let result = if allow_empty && self.consume_token(&Token::RParen) {
|
|
vec![]
|
|
} else {
|
|
let result = self.parse_comma_separated(Parser::parse_expr)?;
|
|
self.expect_token(&Token::RParen)?;
|
|
result
|
|
};
|
|
Ok(result)
|
|
} else {
|
|
Ok(vec![self.parse_expr()?])
|
|
}
|
|
} else {
|
|
self.expect_token(&Token::LParen)?;
|
|
let result = if allow_empty && self.consume_token(&Token::RParen) {
|
|
vec![]
|
|
} else {
|
|
let result = self.parse_comma_separated(Parser::parse_expr)?;
|
|
self.expect_token(&Token::RParen)?;
|
|
result
|
|
};
|
|
Ok(result)
|
|
}
|
|
}
|
|
|
|
pub fn parse_case_expr(&mut self) -> Result<Expr, ParserError> {
|
|
let mut operand = None;
|
|
if !self.parse_keyword(Keyword::WHEN) {
|
|
operand = Some(Box::new(self.parse_expr()?));
|
|
self.expect_keyword(Keyword::WHEN)?;
|
|
}
|
|
let mut conditions = vec![];
|
|
let mut results = vec![];
|
|
loop {
|
|
conditions.push(self.parse_expr()?);
|
|
self.expect_keyword(Keyword::THEN)?;
|
|
results.push(self.parse_expr()?);
|
|
if !self.parse_keyword(Keyword::WHEN) {
|
|
break;
|
|
}
|
|
}
|
|
let else_result = if self.parse_keyword(Keyword::ELSE) {
|
|
Some(Box::new(self.parse_expr()?))
|
|
} else {
|
|
None
|
|
};
|
|
self.expect_keyword(Keyword::END)?;
|
|
Ok(Expr::Case {
|
|
operand,
|
|
conditions,
|
|
results,
|
|
else_result,
|
|
})
|
|
}
|
|
|
|
/// Parse a SQL CAST function e.g. `CAST(expr AS FLOAT)`
|
|
pub fn parse_cast_expr(&mut self) -> Result<Expr, ParserError> {
|
|
self.expect_token(&Token::LParen)?;
|
|
let expr = self.parse_expr()?;
|
|
self.expect_keyword(Keyword::AS)?;
|
|
let data_type = self.parse_data_type()?;
|
|
self.expect_token(&Token::RParen)?;
|
|
Ok(Expr::Cast {
|
|
expr: Box::new(expr),
|
|
data_type,
|
|
})
|
|
}
|
|
|
|
/// Parse a SQL TRY_CAST function e.g. `TRY_CAST(expr AS FLOAT)`
|
|
pub fn parse_try_cast_expr(&mut self) -> Result<Expr, ParserError> {
|
|
self.expect_token(&Token::LParen)?;
|
|
let expr = self.parse_expr()?;
|
|
self.expect_keyword(Keyword::AS)?;
|
|
let data_type = self.parse_data_type()?;
|
|
self.expect_token(&Token::RParen)?;
|
|
Ok(Expr::TryCast {
|
|
expr: Box::new(expr),
|
|
data_type,
|
|
})
|
|
}
|
|
|
|
/// Parse a BigQuery SAFE_CAST function e.g. `SAFE_CAST(expr AS FLOAT64)`
|
|
pub fn parse_safe_cast_expr(&mut self) -> Result<Expr, ParserError> {
|
|
self.expect_token(&Token::LParen)?;
|
|
let expr = self.parse_expr()?;
|
|
self.expect_keyword(Keyword::AS)?;
|
|
let data_type = self.parse_data_type()?;
|
|
self.expect_token(&Token::RParen)?;
|
|
Ok(Expr::SafeCast {
|
|
expr: Box::new(expr),
|
|
data_type,
|
|
})
|
|
}
|
|
|
|
/// Parse a SQL EXISTS expression e.g. `WHERE EXISTS(SELECT ...)`.
|
|
pub fn parse_exists_expr(&mut self, negated: bool) -> Result<Expr, ParserError> {
|
|
self.expect_token(&Token::LParen)?;
|
|
let exists_node = Expr::Exists {
|
|
negated,
|
|
subquery: Box::new(self.parse_query()?),
|
|
};
|
|
self.expect_token(&Token::RParen)?;
|
|
Ok(exists_node)
|
|
}
|
|
|
|
pub fn parse_extract_expr(&mut self) -> Result<Expr, ParserError> {
|
|
self.expect_token(&Token::LParen)?;
|
|
let field = self.parse_date_time_field()?;
|
|
self.expect_keyword(Keyword::FROM)?;
|
|
let expr = self.parse_expr()?;
|
|
self.expect_token(&Token::RParen)?;
|
|
Ok(Expr::Extract {
|
|
field,
|
|
expr: Box::new(expr),
|
|
})
|
|
}
|
|
|
|
pub fn parse_position_expr(&mut self) -> Result<Expr, ParserError> {
|
|
// PARSE SELECT POSITION('@' in field)
|
|
self.expect_token(&Token::LParen)?;
|
|
|
|
// Parse the subexpr till the IN keyword
|
|
let expr = self.parse_subexpr(Self::BETWEEN_PREC)?;
|
|
if self.parse_keyword(Keyword::IN) {
|
|
let from = self.parse_expr()?;
|
|
self.expect_token(&Token::RParen)?;
|
|
Ok(Expr::Position {
|
|
expr: Box::new(expr),
|
|
r#in: Box::new(from),
|
|
})
|
|
} else {
|
|
parser_err!("Position function must include IN keyword".to_string())
|
|
}
|
|
}
|
|
|
|
pub fn parse_substring_expr(&mut self) -> Result<Expr, ParserError> {
|
|
// PARSE SUBSTRING (EXPR [FROM 1] [FOR 3])
|
|
self.expect_token(&Token::LParen)?;
|
|
let expr = self.parse_expr()?;
|
|
let mut from_expr = None;
|
|
if self.parse_keyword(Keyword::FROM) || self.consume_token(&Token::Comma) {
|
|
from_expr = Some(self.parse_expr()?);
|
|
}
|
|
|
|
let mut to_expr = None;
|
|
if self.parse_keyword(Keyword::FOR) || self.consume_token(&Token::Comma) {
|
|
to_expr = Some(self.parse_expr()?);
|
|
}
|
|
self.expect_token(&Token::RParen)?;
|
|
|
|
Ok(Expr::Substring {
|
|
expr: Box::new(expr),
|
|
substring_from: from_expr.map(Box::new),
|
|
substring_for: to_expr.map(Box::new),
|
|
})
|
|
}
|
|
|
|
/// TRIM (WHERE 'text' FROM 'text')\
|
|
/// TRIM ('text')
|
|
pub fn parse_trim_expr(&mut self) -> Result<Expr, ParserError> {
|
|
self.expect_token(&Token::LParen)?;
|
|
let mut where_expr = None;
|
|
if let Token::Word(word) = self.peek_token() {
|
|
if [Keyword::BOTH, Keyword::LEADING, Keyword::TRAILING]
|
|
.iter()
|
|
.any(|d| word.keyword == *d)
|
|
{
|
|
let trim_where = self.parse_trim_where()?;
|
|
let sub_expr = self.parse_expr()?;
|
|
self.expect_keyword(Keyword::FROM)?;
|
|
where_expr = Some((trim_where, Box::new(sub_expr)));
|
|
}
|
|
}
|
|
let expr = self.parse_expr()?;
|
|
self.expect_token(&Token::RParen)?;
|
|
|
|
Ok(Expr::Trim {
|
|
expr: Box::new(expr),
|
|
trim_where: where_expr,
|
|
})
|
|
}
|
|
|
|
pub fn parse_trim_where(&mut self) -> Result<TrimWhereField, ParserError> {
|
|
match self.next_token() {
|
|
Token::Word(w) => match w.keyword {
|
|
Keyword::BOTH => Ok(TrimWhereField::Both),
|
|
Keyword::LEADING => Ok(TrimWhereField::Leading),
|
|
Keyword::TRAILING => Ok(TrimWhereField::Trailing),
|
|
_ => self.expected("trim_where field", Token::Word(w))?,
|
|
},
|
|
unexpected => self.expected("trim_where field", unexpected),
|
|
}
|
|
}
|
|
|
|
/// Parses an array expression `[ex1, ex2, ..]`
|
|
/// if `named` is `true`, came from an expression like `ARRAY[ex1, ex2]`
|
|
pub fn parse_array_expr(&mut self, named: bool) -> Result<Expr, ParserError> {
|
|
if self.peek_token() == Token::RBracket {
|
|
let _ = self.next_token();
|
|
Ok(Expr::Array(Array {
|
|
elem: vec![],
|
|
named,
|
|
}))
|
|
} else {
|
|
let exprs = self.parse_comma_separated(Parser::parse_expr)?;
|
|
self.expect_token(&Token::RBracket)?;
|
|
Ok(Expr::Array(Array { elem: exprs, named }))
|
|
}
|
|
}
|
|
|
|
/// Parse a SQL LISTAGG expression, e.g. `LISTAGG(...) WITHIN GROUP (ORDER BY ...)`.
|
|
pub fn parse_listagg_expr(&mut self) -> Result<Expr, ParserError> {
|
|
self.expect_token(&Token::LParen)?;
|
|
let distinct = self.parse_all_or_distinct()?;
|
|
let expr = Box::new(self.parse_expr()?);
|
|
// While ANSI SQL would would require the separator, Redshift makes this optional. Here we
|
|
// choose to make the separator optional as this provides the more general implementation.
|
|
let separator = if self.consume_token(&Token::Comma) {
|
|
Some(Box::new(self.parse_expr()?))
|
|
} else {
|
|
None
|
|
};
|
|
let on_overflow = if self.parse_keywords(&[Keyword::ON, Keyword::OVERFLOW]) {
|
|
if self.parse_keyword(Keyword::ERROR) {
|
|
Some(ListAggOnOverflow::Error)
|
|
} else {
|
|
self.expect_keyword(Keyword::TRUNCATE)?;
|
|
let filler = match self.peek_token() {
|
|
Token::Word(w)
|
|
if w.keyword == Keyword::WITH || w.keyword == Keyword::WITHOUT =>
|
|
{
|
|
None
|
|
}
|
|
Token::SingleQuotedString(_)
|
|
| Token::EscapedStringLiteral(_)
|
|
| Token::NationalStringLiteral(_)
|
|
| Token::HexStringLiteral(_) => Some(Box::new(self.parse_expr()?)),
|
|
unexpected => {
|
|
self.expected("either filler, WITH, or WITHOUT in LISTAGG", unexpected)?
|
|
}
|
|
};
|
|
let with_count = self.parse_keyword(Keyword::WITH);
|
|
if !with_count && !self.parse_keyword(Keyword::WITHOUT) {
|
|
self.expected("either WITH or WITHOUT in LISTAGG", self.peek_token())?;
|
|
}
|
|
self.expect_keyword(Keyword::COUNT)?;
|
|
Some(ListAggOnOverflow::Truncate { filler, with_count })
|
|
}
|
|
} else {
|
|
None
|
|
};
|
|
self.expect_token(&Token::RParen)?;
|
|
// Once again ANSI SQL requires WITHIN GROUP, but Redshift does not. Again we choose the
|
|
// more general implementation.
|
|
let within_group = if self.parse_keywords(&[Keyword::WITHIN, Keyword::GROUP]) {
|
|
self.expect_token(&Token::LParen)?;
|
|
self.expect_keywords(&[Keyword::ORDER, Keyword::BY])?;
|
|
let order_by_expr = self.parse_comma_separated(Parser::parse_order_by_expr)?;
|
|
self.expect_token(&Token::RParen)?;
|
|
order_by_expr
|
|
} else {
|
|
vec![]
|
|
};
|
|
Ok(Expr::ListAgg(ListAgg {
|
|
distinct,
|
|
expr,
|
|
separator,
|
|
on_overflow,
|
|
within_group,
|
|
}))
|
|
}
|
|
|
|
// This function parses date/time fields for both the EXTRACT function-like
|
|
// operator and interval qualifiers. EXTRACT supports a wider set of
|
|
// date/time fields than interval qualifiers, so this function may need to
|
|
// be split in two.
|
|
pub fn parse_date_time_field(&mut self) -> Result<DateTimeField, ParserError> {
|
|
match self.next_token() {
|
|
Token::Word(w) => match w.keyword {
|
|
Keyword::YEAR => Ok(DateTimeField::Year),
|
|
Keyword::MONTH => Ok(DateTimeField::Month),
|
|
Keyword::WEEK => Ok(DateTimeField::Week),
|
|
Keyword::DAY => Ok(DateTimeField::Day),
|
|
Keyword::HOUR => Ok(DateTimeField::Hour),
|
|
Keyword::MINUTE => Ok(DateTimeField::Minute),
|
|
Keyword::SECOND => Ok(DateTimeField::Second),
|
|
Keyword::CENTURY => Ok(DateTimeField::Century),
|
|
Keyword::DECADE => Ok(DateTimeField::Decade),
|
|
Keyword::DOY => Ok(DateTimeField::Doy),
|
|
Keyword::DOW => Ok(DateTimeField::Dow),
|
|
Keyword::EPOCH => Ok(DateTimeField::Epoch),
|
|
Keyword::ISODOW => Ok(DateTimeField::Isodow),
|
|
Keyword::ISOYEAR => Ok(DateTimeField::Isoyear),
|
|
Keyword::JULIAN => Ok(DateTimeField::Julian),
|
|
Keyword::MICROSECONDS => Ok(DateTimeField::Microseconds),
|
|
Keyword::MILLENIUM => Ok(DateTimeField::Millenium),
|
|
Keyword::MILLISECONDS => Ok(DateTimeField::Milliseconds),
|
|
Keyword::QUARTER => Ok(DateTimeField::Quarter),
|
|
Keyword::TIMEZONE => Ok(DateTimeField::Timezone),
|
|
Keyword::TIMEZONE_HOUR => Ok(DateTimeField::TimezoneHour),
|
|
Keyword::TIMEZONE_MINUTE => Ok(DateTimeField::TimezoneMinute),
|
|
_ => self.expected("date/time field", Token::Word(w))?,
|
|
},
|
|
unexpected => self.expected("date/time field", unexpected),
|
|
}
|
|
}
|
|
|
|
pub fn parse_not(&mut self) -> Result<Expr, ParserError> {
|
|
match self.peek_token() {
|
|
Token::Word(w) => match w.keyword {
|
|
Keyword::EXISTS => {
|
|
let negated = true;
|
|
let _ = self.parse_keyword(Keyword::EXISTS);
|
|
self.parse_exists_expr(negated)
|
|
}
|
|
_ => Ok(Expr::UnaryOp {
|
|
op: UnaryOperator::Not,
|
|
expr: Box::new(self.parse_subexpr(Self::UNARY_NOT_PREC)?),
|
|
}),
|
|
},
|
|
_ => Ok(Expr::UnaryOp {
|
|
op: UnaryOperator::Not,
|
|
expr: Box::new(self.parse_subexpr(Self::UNARY_NOT_PREC)?),
|
|
}),
|
|
}
|
|
}
|
|
|
|
/// Parse an INTERVAL literal.
|
|
///
|
|
/// Some syntactically valid intervals:
|
|
///
|
|
/// 1. `INTERVAL '1' DAY`
|
|
/// 2. `INTERVAL '1-1' YEAR TO MONTH`
|
|
/// 3. `INTERVAL '1' SECOND`
|
|
/// 4. `INTERVAL '1:1:1.1' HOUR (5) TO SECOND (5)`
|
|
/// 5. `INTERVAL '1.1' SECOND (2, 2)`
|
|
/// 6. `INTERVAL '1:1' HOUR (5) TO MINUTE (5)`
|
|
/// 7. (MySql and BigQuey only):`INTERVAL 1 DAY`
|
|
///
|
|
/// Note that we do not currently attempt to parse the quoted value.
|
|
pub fn parse_literal_interval(&mut self) -> Result<Expr, ParserError> {
|
|
// The SQL standard allows an optional sign before the value string, but
|
|
// it is not clear if any implementations support that syntax, so we
|
|
// don't currently try to parse it. (The sign can instead be included
|
|
// inside the value string.)
|
|
|
|
// The first token in an interval is a string literal which specifies
|
|
// the duration of the interval.
|
|
let value = self.parse_expr()?;
|
|
|
|
// Following the string literal is a qualifier which indicates the units
|
|
// of the duration specified in the string literal.
|
|
//
|
|
// Note that PostgreSQL allows omitting the qualifier, so we provide
|
|
// this more general implementation.
|
|
let leading_field = match self.peek_token() {
|
|
Token::Word(kw)
|
|
if [
|
|
Keyword::YEAR,
|
|
Keyword::MONTH,
|
|
Keyword::WEEK,
|
|
Keyword::DAY,
|
|
Keyword::HOUR,
|
|
Keyword::MINUTE,
|
|
Keyword::SECOND,
|
|
Keyword::CENTURY,
|
|
Keyword::DECADE,
|
|
Keyword::DOW,
|
|
Keyword::DOY,
|
|
Keyword::EPOCH,
|
|
Keyword::ISODOW,
|
|
Keyword::ISOYEAR,
|
|
Keyword::JULIAN,
|
|
Keyword::MICROSECONDS,
|
|
Keyword::MILLENIUM,
|
|
Keyword::MILLISECONDS,
|
|
Keyword::QUARTER,
|
|
Keyword::TIMEZONE,
|
|
Keyword::TIMEZONE_HOUR,
|
|
Keyword::TIMEZONE_MINUTE,
|
|
]
|
|
.iter()
|
|
.any(|d| kw.keyword == *d) =>
|
|
{
|
|
Some(self.parse_date_time_field()?)
|
|
}
|
|
_ => None,
|
|
};
|
|
|
|
let (leading_precision, last_field, fsec_precision) =
|
|
if leading_field == Some(DateTimeField::Second) {
|
|
// SQL mandates special syntax for `SECOND TO SECOND` literals.
|
|
// Instead of
|
|
// `SECOND [(<leading precision>)] TO SECOND[(<fractional seconds precision>)]`
|
|
// one must use the special format:
|
|
// `SECOND [( <leading precision> [ , <fractional seconds precision>] )]`
|
|
let last_field = None;
|
|
let (leading_precision, fsec_precision) = self.parse_optional_precision_scale()?;
|
|
(leading_precision, last_field, fsec_precision)
|
|
} else {
|
|
let leading_precision = self.parse_optional_precision()?;
|
|
if self.parse_keyword(Keyword::TO) {
|
|
let last_field = Some(self.parse_date_time_field()?);
|
|
let fsec_precision = if last_field == Some(DateTimeField::Second) {
|
|
self.parse_optional_precision()?
|
|
} else {
|
|
None
|
|
};
|
|
(leading_precision, last_field, fsec_precision)
|
|
} else {
|
|
(leading_precision, None, None)
|
|
}
|
|
};
|
|
|
|
Ok(Expr::Value(Value::Interval {
|
|
value: Box::new(value),
|
|
leading_field,
|
|
leading_precision,
|
|
last_field,
|
|
fractional_seconds_precision: fsec_precision,
|
|
}))
|
|
}
|
|
|
|
/// Parse an operator following an expression
|
|
pub fn parse_infix(&mut self, expr: Expr, precedence: u8) -> Result<Expr, ParserError> {
|
|
let tok = self.next_token();
|
|
|
|
let regular_binary_operator = match &tok {
|
|
Token::Spaceship => Some(BinaryOperator::Spaceship),
|
|
Token::DoubleEq => Some(BinaryOperator::Eq),
|
|
Token::Eq => Some(BinaryOperator::Eq),
|
|
Token::Neq => Some(BinaryOperator::NotEq),
|
|
Token::Gt => Some(BinaryOperator::Gt),
|
|
Token::GtEq => Some(BinaryOperator::GtEq),
|
|
Token::Lt => Some(BinaryOperator::Lt),
|
|
Token::LtEq => Some(BinaryOperator::LtEq),
|
|
Token::Plus => Some(BinaryOperator::Plus),
|
|
Token::Minus => Some(BinaryOperator::Minus),
|
|
Token::Mul => Some(BinaryOperator::Multiply),
|
|
Token::Mod => Some(BinaryOperator::Modulo),
|
|
Token::StringConcat => Some(BinaryOperator::StringConcat),
|
|
Token::Pipe => Some(BinaryOperator::BitwiseOr),
|
|
Token::Caret => Some(BinaryOperator::BitwiseXor),
|
|
Token::Ampersand => Some(BinaryOperator::BitwiseAnd),
|
|
Token::Div => Some(BinaryOperator::Divide),
|
|
Token::ShiftLeft if dialect_of!(self is PostgreSqlDialect | GenericDialect) => {
|
|
Some(BinaryOperator::PGBitwiseShiftLeft)
|
|
}
|
|
Token::ShiftRight if dialect_of!(self is PostgreSqlDialect | GenericDialect) => {
|
|
Some(BinaryOperator::PGBitwiseShiftRight)
|
|
}
|
|
Token::Sharp if dialect_of!(self is PostgreSqlDialect) => {
|
|
Some(BinaryOperator::PGBitwiseXor)
|
|
}
|
|
Token::Tilde => Some(BinaryOperator::PGRegexMatch),
|
|
Token::TildeAsterisk => Some(BinaryOperator::PGRegexIMatch),
|
|
Token::ExclamationMarkTilde => Some(BinaryOperator::PGRegexNotMatch),
|
|
Token::ExclamationMarkTildeAsterisk => Some(BinaryOperator::PGRegexNotIMatch),
|
|
Token::Word(w) => match w.keyword {
|
|
Keyword::AND => Some(BinaryOperator::And),
|
|
Keyword::OR => Some(BinaryOperator::Or),
|
|
Keyword::XOR => Some(BinaryOperator::Xor),
|
|
Keyword::OPERATOR if dialect_of!(self is PostgreSqlDialect | GenericDialect) => {
|
|
self.expect_token(&Token::LParen)?;
|
|
// there are special rules for operator names in
|
|
// postgres so we can not use 'parse_object'
|
|
// or similar.
|
|
// See https://www.postgresql.org/docs/current/sql-createoperator.html
|
|
let mut idents = vec![];
|
|
loop {
|
|
idents.push(self.next_token().to_string());
|
|
if !self.consume_token(&Token::Period) {
|
|
break;
|
|
}
|
|
}
|
|
self.expect_token(&Token::RParen)?;
|
|
Some(BinaryOperator::PGCustomBinaryOperator(idents))
|
|
}
|
|
_ => None,
|
|
},
|
|
_ => None,
|
|
};
|
|
|
|
if let Some(op) = regular_binary_operator {
|
|
if let Some(keyword) = self.parse_one_of_keywords(&[Keyword::ANY, Keyword::ALL]) {
|
|
self.expect_token(&Token::LParen)?;
|
|
let right = self.parse_subexpr(precedence)?;
|
|
self.expect_token(&Token::RParen)?;
|
|
|
|
let right = match keyword {
|
|
Keyword::ALL => Box::new(Expr::AllOp(Box::new(right))),
|
|
Keyword::ANY => Box::new(Expr::AnyOp(Box::new(right))),
|
|
_ => unreachable!(),
|
|
};
|
|
|
|
Ok(Expr::BinaryOp {
|
|
left: Box::new(expr),
|
|
op,
|
|
right,
|
|
})
|
|
} else {
|
|
Ok(Expr::BinaryOp {
|
|
left: Box::new(expr),
|
|
op,
|
|
right: Box::new(self.parse_subexpr(precedence)?),
|
|
})
|
|
}
|
|
} else if let Token::Word(w) = &tok {
|
|
match w.keyword {
|
|
Keyword::IS => {
|
|
if self.parse_keyword(Keyword::NULL) {
|
|
Ok(Expr::IsNull(Box::new(expr)))
|
|
} else if self.parse_keywords(&[Keyword::NOT, Keyword::NULL]) {
|
|
Ok(Expr::IsNotNull(Box::new(expr)))
|
|
} else if self.parse_keywords(&[Keyword::TRUE]) {
|
|
Ok(Expr::IsTrue(Box::new(expr)))
|
|
} else if self.parse_keywords(&[Keyword::FALSE]) {
|
|
Ok(Expr::IsFalse(Box::new(expr)))
|
|
} else if self.parse_keywords(&[Keyword::DISTINCT, Keyword::FROM]) {
|
|
let expr2 = self.parse_expr()?;
|
|
Ok(Expr::IsDistinctFrom(Box::new(expr), Box::new(expr2)))
|
|
} else if self.parse_keywords(&[Keyword::NOT, Keyword::DISTINCT, Keyword::FROM])
|
|
{
|
|
let expr2 = self.parse_expr()?;
|
|
Ok(Expr::IsNotDistinctFrom(Box::new(expr), Box::new(expr2)))
|
|
} else {
|
|
self.expected(
|
|
"[NOT] NULL or TRUE|FALSE or [NOT] DISTINCT FROM after IS",
|
|
self.peek_token(),
|
|
)
|
|
}
|
|
}
|
|
Keyword::AT => {
|
|
// if self.parse_keyword(Keyword::TIME) {
|
|
// self.expect_keyword(Keyword::ZONE)?;
|
|
if self.parse_keywords(&[Keyword::TIME, Keyword::ZONE]) {
|
|
let time_zone = self.next_token();
|
|
match time_zone {
|
|
Token::SingleQuotedString(time_zone) => {
|
|
log::trace!("Peek token: {:?}", self.peek_token());
|
|
Ok(Expr::AtTimeZone {
|
|
timestamp: Box::new(expr),
|
|
time_zone,
|
|
})
|
|
}
|
|
tok => self.expected(
|
|
"Expected Token::SingleQuotedString after AT TIME ZONE",
|
|
tok,
|
|
),
|
|
}
|
|
} else {
|
|
self.expected("Expected Token::Word after AT", tok)
|
|
}
|
|
}
|
|
Keyword::NOT
|
|
| Keyword::IN
|
|
| Keyword::BETWEEN
|
|
| Keyword::LIKE
|
|
| Keyword::ILIKE
|
|
| Keyword::SIMILAR => {
|
|
self.prev_token();
|
|
let negated = self.parse_keyword(Keyword::NOT);
|
|
if self.parse_keyword(Keyword::IN) {
|
|
self.parse_in(expr, negated)
|
|
} else if self.parse_keyword(Keyword::BETWEEN) {
|
|
self.parse_between(expr, negated)
|
|
} else if self.parse_keyword(Keyword::LIKE) {
|
|
Ok(Expr::Like {
|
|
negated,
|
|
expr: Box::new(expr),
|
|
pattern: Box::new(self.parse_value()?),
|
|
escape_char: self.parse_escape_char()?,
|
|
})
|
|
} else if self.parse_keyword(Keyword::ILIKE) {
|
|
Ok(Expr::ILike {
|
|
negated,
|
|
expr: Box::new(expr),
|
|
pattern: Box::new(self.parse_value()?),
|
|
escape_char: self.parse_escape_char()?,
|
|
})
|
|
} else if self.parse_keywords(&[Keyword::SIMILAR, Keyword::TO]) {
|
|
Ok(Expr::SimilarTo {
|
|
negated,
|
|
expr: Box::new(expr),
|
|
pattern: Box::new(self.parse_value()?),
|
|
escape_char: self.parse_escape_char()?,
|
|
})
|
|
} else {
|
|
self.expected("IN or BETWEEN after NOT", self.peek_token())
|
|
}
|
|
}
|
|
// Can only happen if `get_next_precedence` got out of sync with this function
|
|
_ => parser_err!(format!("No infix parser for token {:?}", tok)),
|
|
}
|
|
} else if Token::DoubleColon == tok {
|
|
self.parse_pg_cast(expr)
|
|
} else if Token::ExclamationMark == tok {
|
|
// PostgreSQL factorial operation
|
|
Ok(Expr::UnaryOp {
|
|
op: UnaryOperator::PGPostfixFactorial,
|
|
expr: Box::new(expr),
|
|
})
|
|
} else if Token::LBracket == tok {
|
|
if dialect_of!(self is PostgreSqlDialect | GenericDialect) {
|
|
// parse index
|
|
return self.parse_array_index(expr);
|
|
}
|
|
self.parse_map_access(expr)
|
|
} else if Token::Arrow == tok
|
|
|| Token::LongArrow == tok
|
|
|| Token::HashArrow == tok
|
|
|| Token::HashLongArrow == tok
|
|
{
|
|
let operator = match tok {
|
|
Token::Arrow => JsonOperator::Arrow,
|
|
Token::LongArrow => JsonOperator::LongArrow,
|
|
Token::HashArrow => JsonOperator::HashArrow,
|
|
Token::HashLongArrow => JsonOperator::HashLongArrow,
|
|
_ => unreachable!(),
|
|
};
|
|
Ok(Expr::JsonAccess {
|
|
left: Box::new(expr),
|
|
operator,
|
|
right: Box::new(self.parse_expr()?),
|
|
})
|
|
} else {
|
|
// Can only happen if `get_next_precedence` got out of sync with this function
|
|
parser_err!(format!("No infix parser for token {:?}", tok))
|
|
}
|
|
}
|
|
|
|
/// parse the ESCAPE CHAR portion of LIKE, ILIKE, and SIMILAR TO
|
|
pub fn parse_escape_char(&mut self) -> Result<Option<char>, ParserError> {
|
|
if self.parse_keyword(Keyword::ESCAPE) {
|
|
Ok(Some(self.parse_literal_char()?))
|
|
} else {
|
|
Ok(None)
|
|
}
|
|
}
|
|
|
|
pub fn parse_array_index(&mut self, expr: Expr) -> Result<Expr, ParserError> {
|
|
let index = self.parse_expr()?;
|
|
self.expect_token(&Token::RBracket)?;
|
|
let mut indexes: Vec<Expr> = vec![index];
|
|
while self.consume_token(&Token::LBracket) {
|
|
let index = self.parse_expr()?;
|
|
self.expect_token(&Token::RBracket)?;
|
|
indexes.push(index);
|
|
}
|
|
Ok(Expr::ArrayIndex {
|
|
obj: Box::new(expr),
|
|
indexes,
|
|
})
|
|
}
|
|
|
|
pub fn parse_map_access(&mut self, expr: Expr) -> Result<Expr, ParserError> {
|
|
let key = self.parse_map_key()?;
|
|
let tok = self.consume_token(&Token::RBracket);
|
|
debug!("Tok: {}", tok);
|
|
let mut key_parts: Vec<Expr> = vec![key];
|
|
while self.consume_token(&Token::LBracket) {
|
|
let key = self.parse_map_key()?;
|
|
let tok = self.consume_token(&Token::RBracket);
|
|
debug!("Tok: {}", tok);
|
|
key_parts.push(key);
|
|
}
|
|
match expr {
|
|
e @ Expr::Identifier(_) | e @ Expr::CompoundIdentifier(_) => Ok(Expr::MapAccess {
|
|
column: Box::new(e),
|
|
keys: key_parts,
|
|
}),
|
|
_ => Ok(expr),
|
|
}
|
|
}
|
|
|
|
/// Parses the parens following the `[ NOT ] IN` operator
|
|
pub fn parse_in(&mut self, expr: Expr, negated: bool) -> Result<Expr, ParserError> {
|
|
// BigQuery allows `IN UNNEST(array_expression)`
|
|
// https://cloud.google.com/bigquery/docs/reference/standard-sql/operators#in_operators
|
|
if self.parse_keyword(Keyword::UNNEST) {
|
|
self.expect_token(&Token::LParen)?;
|
|
let array_expr = self.parse_expr()?;
|
|
self.expect_token(&Token::RParen)?;
|
|
return Ok(Expr::InUnnest {
|
|
expr: Box::new(expr),
|
|
array_expr: Box::new(array_expr),
|
|
negated,
|
|
});
|
|
}
|
|
self.expect_token(&Token::LParen)?;
|
|
let in_op = if self.parse_keyword(Keyword::SELECT) || self.parse_keyword(Keyword::WITH) {
|
|
self.prev_token();
|
|
Expr::InSubquery {
|
|
expr: Box::new(expr),
|
|
subquery: Box::new(self.parse_query()?),
|
|
negated,
|
|
}
|
|
} else {
|
|
Expr::InList {
|
|
expr: Box::new(expr),
|
|
list: self.parse_comma_separated(Parser::parse_expr)?,
|
|
negated,
|
|
}
|
|
};
|
|
self.expect_token(&Token::RParen)?;
|
|
Ok(in_op)
|
|
}
|
|
|
|
/// Parses `BETWEEN <low> AND <high>`, assuming the `BETWEEN` keyword was already consumed
|
|
pub fn parse_between(&mut self, expr: Expr, negated: bool) -> Result<Expr, ParserError> {
|
|
// Stop parsing subexpressions for <low> and <high> on tokens with
|
|
// precedence lower than that of `BETWEEN`, such as `AND`, `IS`, etc.
|
|
let low = self.parse_subexpr(Self::BETWEEN_PREC)?;
|
|
self.expect_keyword(Keyword::AND)?;
|
|
let high = self.parse_subexpr(Self::BETWEEN_PREC)?;
|
|
Ok(Expr::Between {
|
|
expr: Box::new(expr),
|
|
negated,
|
|
low: Box::new(low),
|
|
high: Box::new(high),
|
|
})
|
|
}
|
|
|
|
/// Parse a postgresql casting style which is in the form of `expr::datatype`
|
|
pub fn parse_pg_cast(&mut self, expr: Expr) -> Result<Expr, ParserError> {
|
|
Ok(Expr::Cast {
|
|
expr: Box::new(expr),
|
|
data_type: self.parse_data_type()?,
|
|
})
|
|
}
|
|
|
|
const UNARY_NOT_PREC: u8 = 15;
|
|
const BETWEEN_PREC: u8 = 20;
|
|
const PLUS_MINUS_PREC: u8 = 30;
|
|
const TIME_ZONE_PREC: u8 = 20;
|
|
|
|
/// Get the precedence of the next token
|
|
pub fn get_next_precedence(&self) -> Result<u8, ParserError> {
|
|
let token = self.peek_token();
|
|
debug!("get_next_precedence() {:?}", token);
|
|
let token_0 = self.peek_nth_token(0);
|
|
let token_1 = self.peek_nth_token(1);
|
|
let token_2 = self.peek_nth_token(2);
|
|
debug!("0: {token_0} 1: {token_1} 2: {token_2}");
|
|
match token {
|
|
Token::Word(w) if w.keyword == Keyword::OR => Ok(5),
|
|
Token::Word(w) if w.keyword == Keyword::AND => Ok(10),
|
|
Token::Word(w) if w.keyword == Keyword::XOR => Ok(24),
|
|
|
|
Token::Word(w) if w.keyword == Keyword::AT => {
|
|
match (self.peek_nth_token(1), self.peek_nth_token(2)) {
|
|
(Token::Word(w), Token::Word(w2))
|
|
if w.keyword == Keyword::TIME && w2.keyword == Keyword::ZONE =>
|
|
{
|
|
Ok(Self::TIME_ZONE_PREC)
|
|
}
|
|
_ => Ok(0),
|
|
}
|
|
}
|
|
|
|
Token::Word(w) if w.keyword == Keyword::NOT => match self.peek_nth_token(1) {
|
|
// The precedence of NOT varies depending on keyword that
|
|
// follows it. If it is followed by IN, BETWEEN, or LIKE,
|
|
// it takes on the precedence of those tokens. Otherwise it
|
|
// is not an infix operator, and therefore has zero
|
|
// precedence.
|
|
Token::Word(w) if w.keyword == Keyword::IN => Ok(Self::BETWEEN_PREC),
|
|
Token::Word(w) if w.keyword == Keyword::BETWEEN => Ok(Self::BETWEEN_PREC),
|
|
Token::Word(w) if w.keyword == Keyword::LIKE => Ok(Self::BETWEEN_PREC),
|
|
Token::Word(w) if w.keyword == Keyword::ILIKE => Ok(Self::BETWEEN_PREC),
|
|
Token::Word(w) if w.keyword == Keyword::SIMILAR => Ok(Self::BETWEEN_PREC),
|
|
_ => Ok(0),
|
|
},
|
|
Token::Word(w) if w.keyword == Keyword::IS => Ok(17),
|
|
Token::Word(w) if w.keyword == Keyword::IN => Ok(Self::BETWEEN_PREC),
|
|
Token::Word(w) if w.keyword == Keyword::BETWEEN => Ok(Self::BETWEEN_PREC),
|
|
Token::Word(w) if w.keyword == Keyword::LIKE => Ok(Self::BETWEEN_PREC),
|
|
Token::Word(w) if w.keyword == Keyword::ILIKE => Ok(Self::BETWEEN_PREC),
|
|
Token::Word(w) if w.keyword == Keyword::OPERATOR => Ok(Self::BETWEEN_PREC),
|
|
Token::Word(w) if w.keyword == Keyword::SIMILAR => Ok(Self::BETWEEN_PREC),
|
|
Token::Eq
|
|
| Token::Lt
|
|
| Token::LtEq
|
|
| Token::Neq
|
|
| Token::Gt
|
|
| Token::GtEq
|
|
| Token::DoubleEq
|
|
| Token::Tilde
|
|
| Token::TildeAsterisk
|
|
| Token::ExclamationMarkTilde
|
|
| Token::ExclamationMarkTildeAsterisk
|
|
| Token::Spaceship => Ok(20),
|
|
Token::Pipe => Ok(21),
|
|
Token::Caret | Token::Sharp | Token::ShiftRight | Token::ShiftLeft => Ok(22),
|
|
Token::Ampersand => Ok(23),
|
|
Token::Plus | Token::Minus => Ok(Self::PLUS_MINUS_PREC),
|
|
Token::Mul | Token::Div | Token::Mod | Token::StringConcat => Ok(40),
|
|
Token::DoubleColon => Ok(50),
|
|
Token::ExclamationMark => Ok(50),
|
|
Token::LBracket
|
|
| Token::LongArrow
|
|
| Token::Arrow
|
|
| Token::HashArrow
|
|
| Token::HashLongArrow => Ok(50),
|
|
_ => Ok(0),
|
|
}
|
|
}
|
|
|
|
/// Return the first non-whitespace token that has not yet been processed
|
|
/// (or None if reached end-of-file)
|
|
pub fn peek_token(&self) -> Token {
|
|
self.peek_nth_token(0)
|
|
}
|
|
|
|
/// Return nth non-whitespace token that has not yet been processed
|
|
pub fn peek_nth_token(&self, mut n: usize) -> Token {
|
|
let mut index = self.index;
|
|
loop {
|
|
index += 1;
|
|
match self.tokens.get(index - 1) {
|
|
Some(Token::Whitespace(_)) => continue,
|
|
non_whitespace => {
|
|
if n == 0 {
|
|
return non_whitespace.cloned().unwrap_or(Token::EOF);
|
|
}
|
|
n -= 1;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Return the first non-whitespace token that has not yet been processed
|
|
/// (or None if reached end-of-file) and mark it as processed. OK to call
|
|
/// repeatedly after reaching EOF.
|
|
pub fn next_token(&mut self) -> Token {
|
|
loop {
|
|
self.index += 1;
|
|
match self.tokens.get(self.index - 1) {
|
|
Some(Token::Whitespace(_)) => continue,
|
|
token => return token.cloned().unwrap_or(Token::EOF),
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Return the first unprocessed token, possibly whitespace.
|
|
pub fn next_token_no_skip(&mut self) -> Option<&Token> {
|
|
self.index += 1;
|
|
self.tokens.get(self.index - 1)
|
|
}
|
|
|
|
/// Push back the last one non-whitespace token. Must be called after
|
|
/// `next_token()`, otherwise might panic. OK to call after
|
|
/// `next_token()` indicates an EOF.
|
|
pub fn prev_token(&mut self) {
|
|
loop {
|
|
assert!(self.index > 0);
|
|
self.index -= 1;
|
|
if let Some(Token::Whitespace(_)) = self.tokens.get(self.index) {
|
|
continue;
|
|
}
|
|
return;
|
|
}
|
|
}
|
|
|
|
/// Report unexpected token
|
|
fn expected<T>(&self, expected: &str, found: Token) -> Result<T, ParserError> {
|
|
parser_err!(format!("Expected {}, found: {}", expected, found))
|
|
}
|
|
|
|
/// Look for an expected keyword and consume it if it exists
|
|
#[must_use]
|
|
pub fn parse_keyword(&mut self, expected: Keyword) -> bool {
|
|
match self.peek_token() {
|
|
Token::Word(w) if expected == w.keyword => {
|
|
self.next_token();
|
|
true
|
|
}
|
|
_ => false,
|
|
}
|
|
}
|
|
|
|
/// Look for an expected sequence of keywords and consume them if they exist
|
|
#[must_use]
|
|
pub fn parse_keywords(&mut self, keywords: &[Keyword]) -> bool {
|
|
let index = self.index;
|
|
for &keyword in keywords {
|
|
if !self.parse_keyword(keyword) {
|
|
// println!("parse_keywords aborting .. did not find {:?}", keyword);
|
|
// reset index and return immediately
|
|
self.index = index;
|
|
return false;
|
|
}
|
|
}
|
|
true
|
|
}
|
|
|
|
/// Look for one of the given keywords and return the one that matches.
|
|
#[must_use]
|
|
pub fn parse_one_of_keywords(&mut self, keywords: &[Keyword]) -> Option<Keyword> {
|
|
match self.peek_token() {
|
|
Token::Word(w) => {
|
|
keywords
|
|
.iter()
|
|
.find(|keyword| **keyword == w.keyword)
|
|
.map(|keyword| {
|
|
self.next_token();
|
|
*keyword
|
|
})
|
|
}
|
|
_ => None,
|
|
}
|
|
}
|
|
|
|
/// Bail out if the current token is not one of the expected keywords, or consume it if it is
|
|
pub fn expect_one_of_keywords(&mut self, keywords: &[Keyword]) -> Result<Keyword, ParserError> {
|
|
if let Some(keyword) = self.parse_one_of_keywords(keywords) {
|
|
Ok(keyword)
|
|
} else {
|
|
let keywords: Vec<String> = keywords.iter().map(|x| format!("{:?}", x)).collect();
|
|
self.expected(
|
|
&format!("one of {}", keywords.join(" or ")),
|
|
self.peek_token(),
|
|
)
|
|
}
|
|
}
|
|
|
|
/// Bail out if the current token is not an expected keyword, or consume it if it is
|
|
pub fn expect_keyword(&mut self, expected: Keyword) -> Result<(), ParserError> {
|
|
if self.parse_keyword(expected) {
|
|
Ok(())
|
|
} else {
|
|
self.expected(format!("{:?}", &expected).as_str(), self.peek_token())
|
|
}
|
|
}
|
|
|
|
/// Bail out if the following tokens are not the expected sequence of
|
|
/// keywords, or consume them if they are.
|
|
pub fn expect_keywords(&mut self, expected: &[Keyword]) -> Result<(), ParserError> {
|
|
for &kw in expected {
|
|
self.expect_keyword(kw)?;
|
|
}
|
|
Ok(())
|
|
}
|
|
|
|
/// Consume the next token if it matches the expected token, otherwise return false
|
|
#[must_use]
|
|
pub fn consume_token(&mut self, expected: &Token) -> bool {
|
|
if self.peek_token() == *expected {
|
|
self.next_token();
|
|
true
|
|
} else {
|
|
false
|
|
}
|
|
}
|
|
|
|
/// Bail out if the current token is not an expected keyword, or consume it if it is
|
|
pub fn expect_token(&mut self, expected: &Token) -> Result<(), ParserError> {
|
|
if self.consume_token(expected) {
|
|
Ok(())
|
|
} else {
|
|
self.expected(&expected.to_string(), self.peek_token())
|
|
}
|
|
}
|
|
|
|
/// Parse a comma-separated list of 1+ SelectItem
|
|
pub fn parse_projection(&mut self) -> Result<Vec<SelectItem>, ParserError> {
|
|
let mut values = vec![];
|
|
loop {
|
|
values.push(self.parse_select_item()?);
|
|
if !self.consume_token(&Token::Comma) {
|
|
break;
|
|
} else if dialect_of!(self is BigQueryDialect) {
|
|
// BigQuery allows trailing commas.
|
|
// e.g. `SELECT 1, 2, FROM t`
|
|
// https://cloud.google.com/bigquery/docs/reference/standard-sql/lexical#trailing_commas
|
|
match self.peek_token() {
|
|
Token::Word(kw)
|
|
if keywords::RESERVED_FOR_COLUMN_ALIAS
|
|
.iter()
|
|
.any(|d| kw.keyword == *d) =>
|
|
{
|
|
break
|
|
}
|
|
Token::RParen | Token::EOF => break,
|
|
_ => continue,
|
|
}
|
|
}
|
|
}
|
|
Ok(values)
|
|
}
|
|
|
|
/// Parse a comma-separated list of 1+ items accepted by `F`
|
|
pub fn parse_comma_separated<T, F>(&mut self, mut f: F) -> Result<Vec<T>, ParserError>
|
|
where
|
|
F: FnMut(&mut Parser<'a>) -> Result<T, ParserError>,
|
|
{
|
|
let mut values = vec![];
|
|
loop {
|
|
values.push(f(self)?);
|
|
if !self.consume_token(&Token::Comma) {
|
|
break;
|
|
}
|
|
}
|
|
Ok(values)
|
|
}
|
|
|
|
/// Run a parser method `f`, reverting back to the current position
|
|
/// if unsuccessful.
|
|
#[must_use]
|
|
fn maybe_parse<T, F>(&mut self, mut f: F) -> Option<T>
|
|
where
|
|
F: FnMut(&mut Parser) -> Result<T, ParserError>,
|
|
{
|
|
let index = self.index;
|
|
if let Ok(t) = f(self) {
|
|
Some(t)
|
|
} else {
|
|
self.index = index;
|
|
None
|
|
}
|
|
}
|
|
|
|
/// Parse either `ALL` or `DISTINCT`. Returns `true` if `DISTINCT` is parsed and results in a
|
|
/// `ParserError` if both `ALL` and `DISTINCT` are fround.
|
|
pub fn parse_all_or_distinct(&mut self) -> Result<bool, ParserError> {
|
|
let all = self.parse_keyword(Keyword::ALL);
|
|
let distinct = self.parse_keyword(Keyword::DISTINCT);
|
|
if all && distinct {
|
|
parser_err!("Cannot specify both ALL and DISTINCT".to_string())
|
|
} else {
|
|
Ok(distinct)
|
|
}
|
|
}
|
|
|
|
/// Parse a SQL CREATE statement
|
|
pub fn parse_create(&mut self) -> Result<Statement, ParserError> {
|
|
let or_replace = self.parse_keywords(&[Keyword::OR, Keyword::REPLACE]);
|
|
let local = self.parse_one_of_keywords(&[Keyword::LOCAL]).is_some();
|
|
let global = self.parse_one_of_keywords(&[Keyword::GLOBAL]).is_some();
|
|
let global: Option<bool> = if global {
|
|
Some(true)
|
|
} else if local {
|
|
Some(false)
|
|
} else {
|
|
None
|
|
};
|
|
let temporary = self
|
|
.parse_one_of_keywords(&[Keyword::TEMP, Keyword::TEMPORARY])
|
|
.is_some();
|
|
if self.parse_keyword(Keyword::TABLE) {
|
|
self.parse_create_table(or_replace, temporary, global)
|
|
} else if self.parse_keyword(Keyword::MATERIALIZED) || self.parse_keyword(Keyword::VIEW) {
|
|
self.prev_token();
|
|
self.parse_create_view(or_replace)
|
|
} else if self.parse_keyword(Keyword::EXTERNAL) {
|
|
self.parse_create_external_table(or_replace)
|
|
} else if or_replace {
|
|
self.expected(
|
|
"[EXTERNAL] TABLE or [MATERIALIZED] VIEW after CREATE OR REPLACE",
|
|
self.peek_token(),
|
|
)
|
|
} else if self.parse_keyword(Keyword::INDEX) {
|
|
self.parse_create_index(false)
|
|
} else if self.parse_keywords(&[Keyword::UNIQUE, Keyword::INDEX]) {
|
|
self.parse_create_index(true)
|
|
} else if self.parse_keyword(Keyword::VIRTUAL) {
|
|
self.parse_create_virtual_table()
|
|
} else if self.parse_keyword(Keyword::SCHEMA) {
|
|
self.parse_create_schema()
|
|
} else if self.parse_keyword(Keyword::DATABASE) {
|
|
self.parse_create_database()
|
|
} else if dialect_of!(self is HiveDialect) && self.parse_keyword(Keyword::FUNCTION) {
|
|
self.parse_create_function(temporary)
|
|
} else {
|
|
self.expected("an object type after CREATE", self.peek_token())
|
|
}
|
|
}
|
|
|
|
/// SQLite-specific `CREATE VIRTUAL TABLE`
|
|
pub fn parse_create_virtual_table(&mut self) -> Result<Statement, ParserError> {
|
|
self.expect_keyword(Keyword::TABLE)?;
|
|
let if_not_exists = self.parse_keywords(&[Keyword::IF, Keyword::NOT, Keyword::EXISTS]);
|
|
let table_name = self.parse_object_name()?;
|
|
self.expect_keyword(Keyword::USING)?;
|
|
let module_name = self.parse_identifier()?;
|
|
// SQLite docs note that module "arguments syntax is sufficiently
|
|
// general that the arguments can be made to appear as column
|
|
// definitions in a traditional CREATE TABLE statement", but
|
|
// we don't implement that.
|
|
let module_args = self.parse_parenthesized_column_list(Optional)?;
|
|
Ok(Statement::CreateVirtualTable {
|
|
name: table_name,
|
|
if_not_exists,
|
|
module_name,
|
|
module_args,
|
|
})
|
|
}
|
|
|
|
pub fn parse_create_schema(&mut self) -> Result<Statement, ParserError> {
|
|
let if_not_exists = self.parse_keywords(&[Keyword::IF, Keyword::NOT, Keyword::EXISTS]);
|
|
let schema_name = self.parse_object_name()?;
|
|
Ok(Statement::CreateSchema {
|
|
schema_name,
|
|
if_not_exists,
|
|
})
|
|
}
|
|
|
|
pub fn parse_create_database(&mut self) -> Result<Statement, ParserError> {
|
|
let ine = self.parse_keywords(&[Keyword::IF, Keyword::NOT, Keyword::EXISTS]);
|
|
let db_name = self.parse_object_name()?;
|
|
let mut location = None;
|
|
let mut managed_location = None;
|
|
loop {
|
|
match self.parse_one_of_keywords(&[Keyword::LOCATION, Keyword::MANAGEDLOCATION]) {
|
|
Some(Keyword::LOCATION) => location = Some(self.parse_literal_string()?),
|
|
Some(Keyword::MANAGEDLOCATION) => {
|
|
managed_location = Some(self.parse_literal_string()?)
|
|
}
|
|
_ => break,
|
|
}
|
|
}
|
|
Ok(Statement::CreateDatabase {
|
|
db_name,
|
|
if_not_exists: ine,
|
|
location,
|
|
managed_location,
|
|
})
|
|
}
|
|
|
|
pub fn parse_optional_create_function_using(
|
|
&mut self,
|
|
) -> Result<Option<CreateFunctionUsing>, ParserError> {
|
|
if !self.parse_keyword(Keyword::USING) {
|
|
return Ok(None);
|
|
};
|
|
let keyword =
|
|
self.expect_one_of_keywords(&[Keyword::JAR, Keyword::FILE, Keyword::ARCHIVE])?;
|
|
|
|
let uri = self.parse_literal_string()?;
|
|
|
|
match keyword {
|
|
Keyword::JAR => Ok(Some(CreateFunctionUsing::Jar(uri))),
|
|
Keyword::FILE => Ok(Some(CreateFunctionUsing::File(uri))),
|
|
Keyword::ARCHIVE => Ok(Some(CreateFunctionUsing::Archive(uri))),
|
|
_ => self.expected(
|
|
"JAR, FILE or ARCHIVE, got {:?}",
|
|
Token::make_keyword(format!("{:?}", keyword).as_str()),
|
|
),
|
|
}
|
|
}
|
|
|
|
pub fn parse_create_function(&mut self, temporary: bool) -> Result<Statement, ParserError> {
|
|
let name = self.parse_object_name()?;
|
|
self.expect_keyword(Keyword::AS)?;
|
|
let class_name = self.parse_literal_string()?;
|
|
let using = self.parse_optional_create_function_using()?;
|
|
|
|
Ok(Statement::CreateFunction {
|
|
temporary,
|
|
name,
|
|
class_name,
|
|
using,
|
|
})
|
|
}
|
|
|
|
pub fn parse_create_external_table(
|
|
&mut self,
|
|
or_replace: bool,
|
|
) -> Result<Statement, ParserError> {
|
|
self.expect_keyword(Keyword::TABLE)?;
|
|
let if_not_exists = self.parse_keywords(&[Keyword::IF, Keyword::NOT, Keyword::EXISTS]);
|
|
let table_name = self.parse_object_name()?;
|
|
let (columns, constraints) = self.parse_columns()?;
|
|
|
|
let hive_distribution = self.parse_hive_distribution()?;
|
|
let hive_formats = self.parse_hive_formats()?;
|
|
|
|
let file_format = if let Some(ff) = &hive_formats.storage {
|
|
match ff {
|
|
HiveIOFormat::FileFormat { format } => Some(format.clone()),
|
|
_ => None,
|
|
}
|
|
} else {
|
|
None
|
|
};
|
|
let location = hive_formats.location.clone();
|
|
let table_properties = self.parse_options(Keyword::TBLPROPERTIES)?;
|
|
Ok(Statement::CreateTable {
|
|
name: table_name,
|
|
columns,
|
|
constraints,
|
|
hive_distribution,
|
|
hive_formats: Some(hive_formats),
|
|
with_options: vec![],
|
|
table_properties,
|
|
or_replace,
|
|
if_not_exists,
|
|
external: true,
|
|
global: None,
|
|
temporary: false,
|
|
file_format,
|
|
location,
|
|
query: None,
|
|
without_rowid: false,
|
|
like: None,
|
|
clone: None,
|
|
default_charset: None,
|
|
engine: None,
|
|
collation: None,
|
|
on_commit: None,
|
|
on_cluster: None,
|
|
})
|
|
}
|
|
|
|
pub fn parse_file_format(&mut self) -> Result<FileFormat, ParserError> {
|
|
match self.next_token() {
|
|
Token::Word(w) => match w.keyword {
|
|
Keyword::AVRO => Ok(FileFormat::AVRO),
|
|
Keyword::JSONFILE => Ok(FileFormat::JSONFILE),
|
|
Keyword::ORC => Ok(FileFormat::ORC),
|
|
Keyword::PARQUET => Ok(FileFormat::PARQUET),
|
|
Keyword::RCFILE => Ok(FileFormat::RCFILE),
|
|
Keyword::SEQUENCEFILE => Ok(FileFormat::SEQUENCEFILE),
|
|
Keyword::TEXTFILE => Ok(FileFormat::TEXTFILE),
|
|
_ => self.expected("fileformat", Token::Word(w)),
|
|
},
|
|
unexpected => self.expected("fileformat", unexpected),
|
|
}
|
|
}
|
|
|
|
pub fn parse_create_view(&mut self, or_replace: bool) -> Result<Statement, ParserError> {
|
|
let materialized = self.parse_keyword(Keyword::MATERIALIZED);
|
|
self.expect_keyword(Keyword::VIEW)?;
|
|
// Many dialects support `OR ALTER` right after `CREATE`, but we don't (yet).
|
|
// ANSI SQL and Postgres support RECURSIVE here, but we don't support it either.
|
|
let name = self.parse_object_name()?;
|
|
let columns = self.parse_parenthesized_column_list(Optional)?;
|
|
let with_options = self.parse_options(Keyword::WITH)?;
|
|
self.expect_keyword(Keyword::AS)?;
|
|
let query = Box::new(self.parse_query()?);
|
|
// Optional `WITH [ CASCADED | LOCAL ] CHECK OPTION` is widely supported here.
|
|
Ok(Statement::CreateView {
|
|
name,
|
|
columns,
|
|
query,
|
|
materialized,
|
|
or_replace,
|
|
with_options,
|
|
})
|
|
}
|
|
|
|
pub fn parse_drop(&mut self) -> Result<Statement, ParserError> {
|
|
let object_type = if self.parse_keyword(Keyword::TABLE) {
|
|
ObjectType::Table
|
|
} else if self.parse_keyword(Keyword::VIEW) {
|
|
ObjectType::View
|
|
} else if self.parse_keyword(Keyword::INDEX) {
|
|
ObjectType::Index
|
|
} else if self.parse_keyword(Keyword::SCHEMA) {
|
|
ObjectType::Schema
|
|
} else {
|
|
return self.expected("TABLE, VIEW, INDEX or SCHEMA after DROP", self.peek_token());
|
|
};
|
|
// Many dialects support the non standard `IF EXISTS` clause and allow
|
|
// specifying multiple objects to delete in a single statement
|
|
let if_exists = self.parse_keywords(&[Keyword::IF, Keyword::EXISTS]);
|
|
let names = self.parse_comma_separated(Parser::parse_object_name)?;
|
|
let cascade = self.parse_keyword(Keyword::CASCADE);
|
|
let restrict = self.parse_keyword(Keyword::RESTRICT);
|
|
let purge = self.parse_keyword(Keyword::PURGE);
|
|
if cascade && restrict {
|
|
return parser_err!("Cannot specify both CASCADE and RESTRICT in DROP");
|
|
}
|
|
Ok(Statement::Drop {
|
|
object_type,
|
|
if_exists,
|
|
names,
|
|
cascade,
|
|
purge,
|
|
})
|
|
}
|
|
|
|
/// DECLARE name [ BINARY ] [ ASENSITIVE | INSENSITIVE ] [ [ NO ] SCROLL ]
|
|
// CURSOR [ { WITH | WITHOUT } HOLD ] FOR query
|
|
pub fn parse_declare(&mut self) -> Result<Statement, ParserError> {
|
|
let name = self.parse_identifier()?;
|
|
|
|
let binary = self.parse_keyword(Keyword::BINARY);
|
|
let sensitive = if self.parse_keyword(Keyword::INSENSITIVE) {
|
|
Some(true)
|
|
} else if self.parse_keyword(Keyword::ASENSITIVE) {
|
|
Some(false)
|
|
} else {
|
|
None
|
|
};
|
|
let scroll = if self.parse_keyword(Keyword::SCROLL) {
|
|
Some(true)
|
|
} else if self.parse_keywords(&[Keyword::NO, Keyword::SCROLL]) {
|
|
Some(false)
|
|
} else {
|
|
None
|
|
};
|
|
|
|
self.expect_keyword(Keyword::CURSOR)?;
|
|
|
|
let hold = match self.parse_one_of_keywords(&[Keyword::WITH, Keyword::WITHOUT]) {
|
|
Some(keyword) => {
|
|
self.expect_keyword(Keyword::HOLD)?;
|
|
|
|
match keyword {
|
|
Keyword::WITH => Some(true),
|
|
Keyword::WITHOUT => Some(false),
|
|
_ => unreachable!(),
|
|
}
|
|
}
|
|
None => None,
|
|
};
|
|
|
|
self.expect_keyword(Keyword::FOR)?;
|
|
|
|
let query = self.parse_query()?;
|
|
|
|
Ok(Statement::Declare {
|
|
name,
|
|
binary,
|
|
sensitive,
|
|
scroll,
|
|
hold,
|
|
query: Box::new(query),
|
|
})
|
|
}
|
|
|
|
// FETCH [ direction { FROM | IN } ] cursor INTO target;
|
|
pub fn parse_fetch_statement(&mut self) -> Result<Statement, ParserError> {
|
|
let direction = if self.parse_keyword(Keyword::NEXT) {
|
|
FetchDirection::Next
|
|
} else if self.parse_keyword(Keyword::PRIOR) {
|
|
FetchDirection::Prior
|
|
} else if self.parse_keyword(Keyword::FIRST) {
|
|
FetchDirection::First
|
|
} else if self.parse_keyword(Keyword::LAST) {
|
|
FetchDirection::Last
|
|
} else if self.parse_keyword(Keyword::ABSOLUTE) {
|
|
FetchDirection::Absolute {
|
|
limit: self.parse_number_value()?,
|
|
}
|
|
} else if self.parse_keyword(Keyword::RELATIVE) {
|
|
FetchDirection::Relative {
|
|
limit: self.parse_number_value()?,
|
|
}
|
|
} else if self.parse_keyword(Keyword::FORWARD) {
|
|
if self.parse_keyword(Keyword::ALL) {
|
|
FetchDirection::ForwardAll
|
|
} else {
|
|
FetchDirection::Forward {
|
|
// TODO: Support optional
|
|
limit: Some(self.parse_number_value()?),
|
|
}
|
|
}
|
|
} else if self.parse_keyword(Keyword::BACKWARD) {
|
|
if self.parse_keyword(Keyword::ALL) {
|
|
FetchDirection::BackwardAll
|
|
} else {
|
|
FetchDirection::Backward {
|
|
// TODO: Support optional
|
|
limit: Some(self.parse_number_value()?),
|
|
}
|
|
}
|
|
} else if self.parse_keyword(Keyword::ALL) {
|
|
FetchDirection::All
|
|
} else {
|
|
FetchDirection::Count {
|
|
limit: self.parse_number_value()?,
|
|
}
|
|
};
|
|
|
|
self.expect_one_of_keywords(&[Keyword::FROM, Keyword::IN])?;
|
|
|
|
let name = self.parse_identifier()?;
|
|
|
|
let into = if self.parse_keyword(Keyword::INTO) {
|
|
Some(self.parse_object_name()?)
|
|
} else {
|
|
None
|
|
};
|
|
|
|
Ok(Statement::Fetch {
|
|
name,
|
|
direction,
|
|
into,
|
|
})
|
|
}
|
|
|
|
pub fn parse_discard(&mut self) -> Result<Statement, ParserError> {
|
|
let object_type = if self.parse_keyword(Keyword::ALL) {
|
|
DiscardObject::ALL
|
|
} else if self.parse_keyword(Keyword::PLANS) {
|
|
DiscardObject::PLANS
|
|
} else if self.parse_keyword(Keyword::SEQUENCES) {
|
|
DiscardObject::SEQUENCES
|
|
} else if self.parse_keyword(Keyword::TEMP) || self.parse_keyword(Keyword::TEMPORARY) {
|
|
DiscardObject::TEMP
|
|
} else {
|
|
return self.expected(
|
|
"ALL, PLANS, SEQUENCES, TEMP or TEMPORARY after DISCARD",
|
|
self.peek_token(),
|
|
);
|
|
};
|
|
Ok(Statement::Discard { object_type })
|
|
}
|
|
|
|
pub fn parse_create_index(&mut self, unique: bool) -> Result<Statement, ParserError> {
|
|
let if_not_exists = self.parse_keywords(&[Keyword::IF, Keyword::NOT, Keyword::EXISTS]);
|
|
let index_name = self.parse_object_name()?;
|
|
self.expect_keyword(Keyword::ON)?;
|
|
let table_name = self.parse_object_name()?;
|
|
self.expect_token(&Token::LParen)?;
|
|
let columns = self.parse_comma_separated(Parser::parse_order_by_expr)?;
|
|
self.expect_token(&Token::RParen)?;
|
|
Ok(Statement::CreateIndex {
|
|
name: index_name,
|
|
table_name,
|
|
columns,
|
|
unique,
|
|
if_not_exists,
|
|
})
|
|
}
|
|
|
|
//TODO: Implement parsing for Skewed and Clustered
|
|
pub fn parse_hive_distribution(&mut self) -> Result<HiveDistributionStyle, ParserError> {
|
|
if self.parse_keywords(&[Keyword::PARTITIONED, Keyword::BY]) {
|
|
self.expect_token(&Token::LParen)?;
|
|
let columns = self.parse_comma_separated(Parser::parse_column_def)?;
|
|
self.expect_token(&Token::RParen)?;
|
|
Ok(HiveDistributionStyle::PARTITIONED { columns })
|
|
} else {
|
|
Ok(HiveDistributionStyle::NONE)
|
|
}
|
|
}
|
|
|
|
pub fn parse_hive_formats(&mut self) -> Result<HiveFormat, ParserError> {
|
|
let mut hive_format = HiveFormat::default();
|
|
loop {
|
|
match self.parse_one_of_keywords(&[Keyword::ROW, Keyword::STORED, Keyword::LOCATION]) {
|
|
Some(Keyword::ROW) => {
|
|
hive_format.row_format = Some(self.parse_row_format()?);
|
|
}
|
|
Some(Keyword::STORED) => {
|
|
self.expect_keyword(Keyword::AS)?;
|
|
if self.parse_keyword(Keyword::INPUTFORMAT) {
|
|
let input_format = self.parse_expr()?;
|
|
self.expect_keyword(Keyword::OUTPUTFORMAT)?;
|
|
let output_format = self.parse_expr()?;
|
|
hive_format.storage = Some(HiveIOFormat::IOF {
|
|
input_format,
|
|
output_format,
|
|
});
|
|
} else {
|
|
let format = self.parse_file_format()?;
|
|
hive_format.storage = Some(HiveIOFormat::FileFormat { format });
|
|
}
|
|
}
|
|
Some(Keyword::LOCATION) => {
|
|
hive_format.location = Some(self.parse_literal_string()?);
|
|
}
|
|
None => break,
|
|
_ => break,
|
|
}
|
|
}
|
|
|
|
Ok(hive_format)
|
|
}
|
|
|
|
pub fn parse_row_format(&mut self) -> Result<HiveRowFormat, ParserError> {
|
|
self.expect_keyword(Keyword::FORMAT)?;
|
|
match self.parse_one_of_keywords(&[Keyword::SERDE, Keyword::DELIMITED]) {
|
|
Some(Keyword::SERDE) => {
|
|
let class = self.parse_literal_string()?;
|
|
Ok(HiveRowFormat::SERDE { class })
|
|
}
|
|
_ => Ok(HiveRowFormat::DELIMITED),
|
|
}
|
|
}
|
|
|
|
pub fn parse_create_table(
|
|
&mut self,
|
|
or_replace: bool,
|
|
temporary: bool,
|
|
global: Option<bool>,
|
|
) -> Result<Statement, ParserError> {
|
|
let if_not_exists = self.parse_keywords(&[Keyword::IF, Keyword::NOT, Keyword::EXISTS]);
|
|
let table_name = self.parse_object_name()?;
|
|
|
|
// Clickhouse has `ON CLUSTER 'cluster'` syntax for DDLs
|
|
let on_cluster = if self.parse_keywords(&[Keyword::ON, Keyword::CLUSTER]) {
|
|
match self.next_token() {
|
|
Token::SingleQuotedString(s) => Some(s),
|
|
Token::Word(s) => Some(s.to_string()),
|
|
unexpected => self.expected("identifier or cluster literal", unexpected)?,
|
|
}
|
|
} else {
|
|
None
|
|
};
|
|
|
|
let like = if self.parse_keyword(Keyword::LIKE) || self.parse_keyword(Keyword::ILIKE) {
|
|
self.parse_object_name().ok()
|
|
} else {
|
|
None
|
|
};
|
|
|
|
let clone = if self.parse_keyword(Keyword::CLONE) {
|
|
self.parse_object_name().ok()
|
|
} else {
|
|
None
|
|
};
|
|
|
|
// parse optional column list (schema)
|
|
let (columns, constraints) = self.parse_columns()?;
|
|
|
|
// SQLite supports `WITHOUT ROWID` at the end of `CREATE TABLE`
|
|
let without_rowid = self.parse_keywords(&[Keyword::WITHOUT, Keyword::ROWID]);
|
|
|
|
let hive_distribution = self.parse_hive_distribution()?;
|
|
let hive_formats = self.parse_hive_formats()?;
|
|
// PostgreSQL supports `WITH ( options )`, before `AS`
|
|
let with_options = self.parse_options(Keyword::WITH)?;
|
|
let table_properties = self.parse_options(Keyword::TBLPROPERTIES)?;
|
|
// Parse optional `AS ( query )`
|
|
let query = if self.parse_keyword(Keyword::AS) {
|
|
Some(Box::new(self.parse_query()?))
|
|
} else {
|
|
None
|
|
};
|
|
|
|
let engine = if self.parse_keyword(Keyword::ENGINE) {
|
|
self.expect_token(&Token::Eq)?;
|
|
match self.next_token() {
|
|
Token::Word(w) => Some(w.value),
|
|
unexpected => self.expected("identifier", unexpected)?,
|
|
}
|
|
} else {
|
|
None
|
|
};
|
|
|
|
let default_charset = if self.parse_keywords(&[Keyword::DEFAULT, Keyword::CHARSET]) {
|
|
self.expect_token(&Token::Eq)?;
|
|
match self.next_token() {
|
|
Token::Word(w) => Some(w.value),
|
|
unexpected => self.expected("identifier", unexpected)?,
|
|
}
|
|
} else {
|
|
None
|
|
};
|
|
|
|
let collation = if self.parse_keywords(&[Keyword::COLLATE]) {
|
|
self.expect_token(&Token::Eq)?;
|
|
match self.next_token() {
|
|
Token::Word(w) => Some(w.value),
|
|
unexpected => self.expected("identifier", unexpected)?,
|
|
}
|
|
} else {
|
|
None
|
|
};
|
|
|
|
let on_commit: Option<OnCommit> =
|
|
if self.parse_keywords(&[Keyword::ON, Keyword::COMMIT, Keyword::DELETE, Keyword::ROWS])
|
|
{
|
|
Some(OnCommit::DeleteRows)
|
|
} else if self.parse_keywords(&[
|
|
Keyword::ON,
|
|
Keyword::COMMIT,
|
|
Keyword::PRESERVE,
|
|
Keyword::ROWS,
|
|
]) {
|
|
Some(OnCommit::PreserveRows)
|
|
} else if self.parse_keywords(&[Keyword::ON, Keyword::COMMIT, Keyword::DROP]) {
|
|
Some(OnCommit::Drop)
|
|
} else {
|
|
None
|
|
};
|
|
|
|
Ok(Statement::CreateTable {
|
|
name: table_name,
|
|
temporary,
|
|
columns,
|
|
constraints,
|
|
with_options,
|
|
table_properties,
|
|
or_replace,
|
|
if_not_exists,
|
|
hive_distribution,
|
|
hive_formats: Some(hive_formats),
|
|
external: false,
|
|
global,
|
|
file_format: None,
|
|
location: None,
|
|
query,
|
|
without_rowid,
|
|
like,
|
|
clone,
|
|
engine,
|
|
default_charset,
|
|
collation,
|
|
on_commit,
|
|
on_cluster,
|
|
})
|
|
}
|
|
|
|
pub fn parse_columns(&mut self) -> Result<(Vec<ColumnDef>, Vec<TableConstraint>), ParserError> {
|
|
let mut columns = vec![];
|
|
let mut constraints = vec![];
|
|
if !self.consume_token(&Token::LParen) || self.consume_token(&Token::RParen) {
|
|
return Ok((columns, constraints));
|
|
}
|
|
|
|
loop {
|
|
if let Some(constraint) = self.parse_optional_table_constraint()? {
|
|
constraints.push(constraint);
|
|
} else if let Token::Word(_) = self.peek_token() {
|
|
columns.push(self.parse_column_def()?);
|
|
} else {
|
|
return self.expected("column name or constraint definition", self.peek_token());
|
|
}
|
|
let comma = self.consume_token(&Token::Comma);
|
|
if self.consume_token(&Token::RParen) {
|
|
// allow a trailing comma, even though it's not in standard
|
|
break;
|
|
} else if !comma {
|
|
return self.expected("',' or ')' after column definition", self.peek_token());
|
|
}
|
|
}
|
|
|
|
Ok((columns, constraints))
|
|
}
|
|
|
|
pub fn parse_column_def(&mut self) -> Result<ColumnDef, ParserError> {
|
|
let name = self.parse_identifier()?;
|
|
let data_type = self.parse_data_type()?;
|
|
let collation = if self.parse_keyword(Keyword::COLLATE) {
|
|
Some(self.parse_object_name()?)
|
|
} else {
|
|
None
|
|
};
|
|
let mut options = vec![];
|
|
loop {
|
|
if self.parse_keyword(Keyword::CONSTRAINT) {
|
|
let name = Some(self.parse_identifier()?);
|
|
if let Some(option) = self.parse_optional_column_option()? {
|
|
options.push(ColumnOptionDef { name, option });
|
|
} else {
|
|
return self.expected(
|
|
"constraint details after CONSTRAINT <name>",
|
|
self.peek_token(),
|
|
);
|
|
}
|
|
} else if let Some(option) = self.parse_optional_column_option()? {
|
|
options.push(ColumnOptionDef { name: None, option });
|
|
} else {
|
|
break;
|
|
};
|
|
}
|
|
Ok(ColumnDef {
|
|
name,
|
|
data_type,
|
|
collation,
|
|
options,
|
|
})
|
|
}
|
|
|
|
pub fn parse_optional_column_option(&mut self) -> Result<Option<ColumnOption>, ParserError> {
|
|
if self.parse_keywords(&[Keyword::CHARACTER, Keyword::SET]) {
|
|
Ok(Some(ColumnOption::CharacterSet(self.parse_object_name()?)))
|
|
} else if self.parse_keywords(&[Keyword::NOT, Keyword::NULL]) {
|
|
Ok(Some(ColumnOption::NotNull))
|
|
} else if self.parse_keywords(&[Keyword::COMMENT]) {
|
|
match self.next_token() {
|
|
Token::SingleQuotedString(value, ..) => Ok(Some(ColumnOption::Comment(value))),
|
|
unexpected => self.expected("string", unexpected),
|
|
}
|
|
} else if self.parse_keyword(Keyword::NULL) {
|
|
Ok(Some(ColumnOption::Null))
|
|
} else if self.parse_keyword(Keyword::DEFAULT) {
|
|
Ok(Some(ColumnOption::Default(self.parse_expr()?)))
|
|
} else if self.parse_keywords(&[Keyword::PRIMARY, Keyword::KEY]) {
|
|
Ok(Some(ColumnOption::Unique { is_primary: true }))
|
|
} else if self.parse_keyword(Keyword::UNIQUE) {
|
|
Ok(Some(ColumnOption::Unique { is_primary: false }))
|
|
} else if self.parse_keyword(Keyword::REFERENCES) {
|
|
let foreign_table = self.parse_object_name()?;
|
|
// PostgreSQL allows omitting the column list and
|
|
// uses the primary key column of the foreign table by default
|
|
let referred_columns = self.parse_parenthesized_column_list(Optional)?;
|
|
let mut on_delete = None;
|
|
let mut on_update = None;
|
|
loop {
|
|
if on_delete.is_none() && self.parse_keywords(&[Keyword::ON, Keyword::DELETE]) {
|
|
on_delete = Some(self.parse_referential_action()?);
|
|
} else if on_update.is_none()
|
|
&& self.parse_keywords(&[Keyword::ON, Keyword::UPDATE])
|
|
{
|
|
on_update = Some(self.parse_referential_action()?);
|
|
} else {
|
|
break;
|
|
}
|
|
}
|
|
Ok(Some(ColumnOption::ForeignKey {
|
|
foreign_table,
|
|
referred_columns,
|
|
on_delete,
|
|
on_update,
|
|
}))
|
|
} else if self.parse_keyword(Keyword::CHECK) {
|
|
self.expect_token(&Token::LParen)?;
|
|
let expr = self.parse_expr()?;
|
|
self.expect_token(&Token::RParen)?;
|
|
Ok(Some(ColumnOption::Check(expr)))
|
|
} else if self.parse_keyword(Keyword::AUTO_INCREMENT)
|
|
&& dialect_of!(self is MySqlDialect | GenericDialect)
|
|
{
|
|
// Support AUTO_INCREMENT for MySQL
|
|
Ok(Some(ColumnOption::DialectSpecific(vec![
|
|
Token::make_keyword("AUTO_INCREMENT"),
|
|
])))
|
|
} else if self.parse_keyword(Keyword::AUTOINCREMENT)
|
|
&& dialect_of!(self is SQLiteDialect | GenericDialect)
|
|
{
|
|
// Support AUTOINCREMENT for SQLite
|
|
Ok(Some(ColumnOption::DialectSpecific(vec![
|
|
Token::make_keyword("AUTOINCREMENT"),
|
|
])))
|
|
} else if self.parse_keywords(&[Keyword::ON, Keyword::UPDATE])
|
|
&& dialect_of!(self is MySqlDialect)
|
|
{
|
|
Ok(Some(ColumnOption::DialectSpecific(vec![
|
|
Token::make_keyword("ON UPDATE"),
|
|
])))
|
|
} else {
|
|
Ok(None)
|
|
}
|
|
}
|
|
|
|
pub fn parse_referential_action(&mut self) -> Result<ReferentialAction, ParserError> {
|
|
if self.parse_keyword(Keyword::RESTRICT) {
|
|
Ok(ReferentialAction::Restrict)
|
|
} else if self.parse_keyword(Keyword::CASCADE) {
|
|
Ok(ReferentialAction::Cascade)
|
|
} else if self.parse_keywords(&[Keyword::SET, Keyword::NULL]) {
|
|
Ok(ReferentialAction::SetNull)
|
|
} else if self.parse_keywords(&[Keyword::NO, Keyword::ACTION]) {
|
|
Ok(ReferentialAction::NoAction)
|
|
} else if self.parse_keywords(&[Keyword::SET, Keyword::DEFAULT]) {
|
|
Ok(ReferentialAction::SetDefault)
|
|
} else {
|
|
self.expected(
|
|
"one of RESTRICT, CASCADE, SET NULL, NO ACTION or SET DEFAULT",
|
|
self.peek_token(),
|
|
)
|
|
}
|
|
}
|
|
|
|
pub fn parse_optional_table_constraint(
|
|
&mut self,
|
|
) -> Result<Option<TableConstraint>, ParserError> {
|
|
let name = if self.parse_keyword(Keyword::CONSTRAINT) {
|
|
Some(self.parse_identifier()?)
|
|
} else {
|
|
None
|
|
};
|
|
match self.next_token() {
|
|
Token::Word(w) if w.keyword == Keyword::PRIMARY || w.keyword == Keyword::UNIQUE => {
|
|
let is_primary = w.keyword == Keyword::PRIMARY;
|
|
if is_primary {
|
|
self.expect_keyword(Keyword::KEY)?;
|
|
}
|
|
let columns = self.parse_parenthesized_column_list(Mandatory)?;
|
|
Ok(Some(TableConstraint::Unique {
|
|
name,
|
|
columns,
|
|
is_primary,
|
|
}))
|
|
}
|
|
Token::Word(w) if w.keyword == Keyword::FOREIGN => {
|
|
self.expect_keyword(Keyword::KEY)?;
|
|
let columns = self.parse_parenthesized_column_list(Mandatory)?;
|
|
self.expect_keyword(Keyword::REFERENCES)?;
|
|
let foreign_table = self.parse_object_name()?;
|
|
let referred_columns = self.parse_parenthesized_column_list(Mandatory)?;
|
|
let mut on_delete = None;
|
|
let mut on_update = None;
|
|
loop {
|
|
if on_delete.is_none() && self.parse_keywords(&[Keyword::ON, Keyword::DELETE]) {
|
|
on_delete = Some(self.parse_referential_action()?);
|
|
} else if on_update.is_none()
|
|
&& self.parse_keywords(&[Keyword::ON, Keyword::UPDATE])
|
|
{
|
|
on_update = Some(self.parse_referential_action()?);
|
|
} else {
|
|
break;
|
|
}
|
|
}
|
|
Ok(Some(TableConstraint::ForeignKey {
|
|
name,
|
|
columns,
|
|
foreign_table,
|
|
referred_columns,
|
|
on_delete,
|
|
on_update,
|
|
}))
|
|
}
|
|
Token::Word(w) if w.keyword == Keyword::CHECK => {
|
|
self.expect_token(&Token::LParen)?;
|
|
let expr = Box::new(self.parse_expr()?);
|
|
self.expect_token(&Token::RParen)?;
|
|
Ok(Some(TableConstraint::Check { name, expr }))
|
|
}
|
|
unexpected => {
|
|
if name.is_some() {
|
|
self.expected("PRIMARY, UNIQUE, FOREIGN, or CHECK", unexpected)
|
|
} else {
|
|
self.prev_token();
|
|
Ok(None)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn parse_options(&mut self, keyword: Keyword) -> Result<Vec<SqlOption>, ParserError> {
|
|
if self.parse_keyword(keyword) {
|
|
self.expect_token(&Token::LParen)?;
|
|
let options = self.parse_comma_separated(Parser::parse_sql_option)?;
|
|
self.expect_token(&Token::RParen)?;
|
|
Ok(options)
|
|
} else {
|
|
Ok(vec![])
|
|
}
|
|
}
|
|
|
|
pub fn parse_sql_option(&mut self) -> Result<SqlOption, ParserError> {
|
|
let name = self.parse_identifier()?;
|
|
self.expect_token(&Token::Eq)?;
|
|
let value = self.parse_value()?;
|
|
Ok(SqlOption { name, value })
|
|
}
|
|
|
|
pub fn parse_alter(&mut self) -> Result<Statement, ParserError> {
|
|
self.expect_keyword(Keyword::TABLE)?;
|
|
let _ = self.parse_keyword(Keyword::ONLY);
|
|
let table_name = self.parse_object_name()?;
|
|
let operation = if self.parse_keyword(Keyword::ADD) {
|
|
if let Some(constraint) = self.parse_optional_table_constraint()? {
|
|
AlterTableOperation::AddConstraint(constraint)
|
|
} else {
|
|
let if_not_exists =
|
|
self.parse_keywords(&[Keyword::IF, Keyword::NOT, Keyword::EXISTS]);
|
|
if self.parse_keyword(Keyword::PARTITION) {
|
|
self.expect_token(&Token::LParen)?;
|
|
let partitions = self.parse_comma_separated(Parser::parse_expr)?;
|
|
self.expect_token(&Token::RParen)?;
|
|
AlterTableOperation::AddPartitions {
|
|
if_not_exists,
|
|
new_partitions: partitions,
|
|
}
|
|
} else {
|
|
let _ = self.parse_keyword(Keyword::COLUMN);
|
|
let column_def = self.parse_column_def()?;
|
|
AlterTableOperation::AddColumn { column_def }
|
|
}
|
|
}
|
|
} else if self.parse_keyword(Keyword::RENAME) {
|
|
if dialect_of!(self is PostgreSqlDialect) && self.parse_keyword(Keyword::CONSTRAINT) {
|
|
let old_name = self.parse_identifier()?;
|
|
self.expect_keyword(Keyword::TO)?;
|
|
let new_name = self.parse_identifier()?;
|
|
AlterTableOperation::RenameConstraint { old_name, new_name }
|
|
} else if self.parse_keyword(Keyword::TO) {
|
|
let table_name = self.parse_object_name()?;
|
|
AlterTableOperation::RenameTable { table_name }
|
|
} else {
|
|
let _ = self.parse_keyword(Keyword::COLUMN);
|
|
let old_column_name = self.parse_identifier()?;
|
|
self.expect_keyword(Keyword::TO)?;
|
|
let new_column_name = self.parse_identifier()?;
|
|
AlterTableOperation::RenameColumn {
|
|
old_column_name,
|
|
new_column_name,
|
|
}
|
|
}
|
|
} else if self.parse_keyword(Keyword::DROP) {
|
|
if self.parse_keywords(&[Keyword::IF, Keyword::EXISTS, Keyword::PARTITION]) {
|
|
self.expect_token(&Token::LParen)?;
|
|
let partitions = self.parse_comma_separated(Parser::parse_expr)?;
|
|
self.expect_token(&Token::RParen)?;
|
|
AlterTableOperation::DropPartitions {
|
|
partitions,
|
|
if_exists: true,
|
|
}
|
|
} else if self.parse_keyword(Keyword::PARTITION) {
|
|
self.expect_token(&Token::LParen)?;
|
|
let partitions = self.parse_comma_separated(Parser::parse_expr)?;
|
|
self.expect_token(&Token::RParen)?;
|
|
AlterTableOperation::DropPartitions {
|
|
partitions,
|
|
if_exists: false,
|
|
}
|
|
} else if self.parse_keyword(Keyword::CONSTRAINT) {
|
|
let if_exists = self.parse_keywords(&[Keyword::IF, Keyword::EXISTS]);
|
|
let name = self.parse_identifier()?;
|
|
let cascade = self.parse_keyword(Keyword::CASCADE);
|
|
AlterTableOperation::DropConstraint {
|
|
if_exists,
|
|
name,
|
|
cascade,
|
|
}
|
|
} else {
|
|
let _ = self.parse_keyword(Keyword::COLUMN);
|
|
let if_exists = self.parse_keywords(&[Keyword::IF, Keyword::EXISTS]);
|
|
let column_name = self.parse_identifier()?;
|
|
let cascade = self.parse_keyword(Keyword::CASCADE);
|
|
AlterTableOperation::DropColumn {
|
|
column_name,
|
|
if_exists,
|
|
cascade,
|
|
}
|
|
}
|
|
} else if self.parse_keyword(Keyword::PARTITION) {
|
|
self.expect_token(&Token::LParen)?;
|
|
let before = self.parse_comma_separated(Parser::parse_expr)?;
|
|
self.expect_token(&Token::RParen)?;
|
|
self.expect_keyword(Keyword::RENAME)?;
|
|
self.expect_keywords(&[Keyword::TO, Keyword::PARTITION])?;
|
|
self.expect_token(&Token::LParen)?;
|
|
let renames = self.parse_comma_separated(Parser::parse_expr)?;
|
|
self.expect_token(&Token::RParen)?;
|
|
AlterTableOperation::RenamePartitions {
|
|
old_partitions: before,
|
|
new_partitions: renames,
|
|
}
|
|
} else if self.parse_keyword(Keyword::CHANGE) {
|
|
let _ = self.parse_keyword(Keyword::COLUMN);
|
|
let old_name = self.parse_identifier()?;
|
|
let new_name = self.parse_identifier()?;
|
|
let data_type = self.parse_data_type()?;
|
|
let mut options = vec![];
|
|
while let Some(option) = self.parse_optional_column_option()? {
|
|
options.push(option);
|
|
}
|
|
|
|
AlterTableOperation::ChangeColumn {
|
|
old_name,
|
|
new_name,
|
|
data_type,
|
|
options,
|
|
}
|
|
} else if self.parse_keyword(Keyword::ALTER) {
|
|
let _ = self.parse_keyword(Keyword::COLUMN);
|
|
let column_name = self.parse_identifier()?;
|
|
let is_postgresql = dialect_of!(self is PostgreSqlDialect);
|
|
|
|
let op = if self.parse_keywords(&[Keyword::SET, Keyword::NOT, Keyword::NULL]) {
|
|
AlterColumnOperation::SetNotNull {}
|
|
} else if self.parse_keywords(&[Keyword::DROP, Keyword::NOT, Keyword::NULL]) {
|
|
AlterColumnOperation::DropNotNull {}
|
|
} else if self.parse_keywords(&[Keyword::SET, Keyword::DEFAULT]) {
|
|
AlterColumnOperation::SetDefault {
|
|
value: self.parse_expr()?,
|
|
}
|
|
} else if self.parse_keywords(&[Keyword::DROP, Keyword::DEFAULT]) {
|
|
AlterColumnOperation::DropDefault {}
|
|
} else if self.parse_keywords(&[Keyword::SET, Keyword::DATA, Keyword::TYPE])
|
|
|| (is_postgresql && self.parse_keyword(Keyword::TYPE))
|
|
{
|
|
let data_type = self.parse_data_type()?;
|
|
let using = if is_postgresql && self.parse_keyword(Keyword::USING) {
|
|
Some(self.parse_expr()?)
|
|
} else {
|
|
None
|
|
};
|
|
AlterColumnOperation::SetDataType { data_type, using }
|
|
} else {
|
|
return self.expected(
|
|
"SET/DROP NOT NULL, SET DEFAULT, SET DATA TYPE after ALTER COLUMN",
|
|
self.peek_token(),
|
|
);
|
|
};
|
|
AlterTableOperation::AlterColumn { column_name, op }
|
|
} else {
|
|
return self.expected(
|
|
"ADD, RENAME, PARTITION or DROP after ALTER TABLE",
|
|
self.peek_token(),
|
|
);
|
|
};
|
|
Ok(Statement::AlterTable {
|
|
name: table_name,
|
|
operation,
|
|
})
|
|
}
|
|
|
|
/// Parse a copy statement
|
|
pub fn parse_copy(&mut self) -> Result<Statement, ParserError> {
|
|
let table_name = self.parse_object_name()?;
|
|
let columns = self.parse_parenthesized_column_list(Optional)?;
|
|
let to = match self.parse_one_of_keywords(&[Keyword::FROM, Keyword::TO]) {
|
|
Some(Keyword::FROM) => false,
|
|
Some(Keyword::TO) => true,
|
|
_ => self.expected("FROM or TO", self.peek_token())?,
|
|
};
|
|
let target = if self.parse_keyword(Keyword::STDIN) {
|
|
CopyTarget::Stdin
|
|
} else if self.parse_keyword(Keyword::STDOUT) {
|
|
CopyTarget::Stdout
|
|
} else if self.parse_keyword(Keyword::PROGRAM) {
|
|
CopyTarget::Program {
|
|
command: self.parse_literal_string()?,
|
|
}
|
|
} else {
|
|
CopyTarget::File {
|
|
filename: self.parse_literal_string()?,
|
|
}
|
|
};
|
|
let _ = self.parse_keyword(Keyword::WITH);
|
|
let mut options = vec![];
|
|
if self.consume_token(&Token::LParen) {
|
|
options = self.parse_comma_separated(Parser::parse_copy_option)?;
|
|
self.expect_token(&Token::RParen)?;
|
|
}
|
|
let mut legacy_options = vec![];
|
|
while let Some(opt) = self.maybe_parse(|parser| parser.parse_copy_legacy_option()) {
|
|
legacy_options.push(opt);
|
|
}
|
|
let values = if let CopyTarget::Stdin = target {
|
|
self.expect_token(&Token::SemiColon)?;
|
|
self.parse_tsv()
|
|
} else {
|
|
vec![]
|
|
};
|
|
Ok(Statement::Copy {
|
|
table_name,
|
|
columns,
|
|
to,
|
|
target,
|
|
options,
|
|
legacy_options,
|
|
values,
|
|
})
|
|
}
|
|
|
|
pub fn parse_close(&mut self) -> Result<Statement, ParserError> {
|
|
let cursor = if self.parse_keyword(Keyword::ALL) {
|
|
CloseCursor::All
|
|
} else {
|
|
let name = self.parse_identifier()?;
|
|
|
|
CloseCursor::Specific { name }
|
|
};
|
|
|
|
Ok(Statement::Close { cursor })
|
|
}
|
|
|
|
fn parse_copy_option(&mut self) -> Result<CopyOption, ParserError> {
|
|
let ret = match self.parse_one_of_keywords(&[
|
|
Keyword::FORMAT,
|
|
Keyword::FREEZE,
|
|
Keyword::DELIMITER,
|
|
Keyword::NULL,
|
|
Keyword::HEADER,
|
|
Keyword::QUOTE,
|
|
Keyword::ESCAPE,
|
|
Keyword::FORCE_QUOTE,
|
|
Keyword::FORCE_NOT_NULL,
|
|
Keyword::FORCE_NULL,
|
|
Keyword::ENCODING,
|
|
]) {
|
|
Some(Keyword::FORMAT) => CopyOption::Format(self.parse_identifier()?),
|
|
Some(Keyword::FREEZE) => CopyOption::Freeze(!matches!(
|
|
self.parse_one_of_keywords(&[Keyword::TRUE, Keyword::FALSE]),
|
|
Some(Keyword::FALSE)
|
|
)),
|
|
Some(Keyword::DELIMITER) => CopyOption::Delimiter(self.parse_literal_char()?),
|
|
Some(Keyword::NULL) => CopyOption::Null(self.parse_literal_string()?),
|
|
Some(Keyword::HEADER) => CopyOption::Header(!matches!(
|
|
self.parse_one_of_keywords(&[Keyword::TRUE, Keyword::FALSE]),
|
|
Some(Keyword::FALSE)
|
|
)),
|
|
Some(Keyword::QUOTE) => CopyOption::Quote(self.parse_literal_char()?),
|
|
Some(Keyword::ESCAPE) => CopyOption::Escape(self.parse_literal_char()?),
|
|
Some(Keyword::FORCE_QUOTE) => {
|
|
CopyOption::ForceQuote(self.parse_parenthesized_column_list(Mandatory)?)
|
|
}
|
|
Some(Keyword::FORCE_NOT_NULL) => {
|
|
CopyOption::ForceNotNull(self.parse_parenthesized_column_list(Mandatory)?)
|
|
}
|
|
Some(Keyword::FORCE_NULL) => {
|
|
CopyOption::ForceNull(self.parse_parenthesized_column_list(Mandatory)?)
|
|
}
|
|
Some(Keyword::ENCODING) => CopyOption::Encoding(self.parse_literal_string()?),
|
|
_ => self.expected("option", self.peek_token())?,
|
|
};
|
|
Ok(ret)
|
|
}
|
|
|
|
fn parse_copy_legacy_option(&mut self) -> Result<CopyLegacyOption, ParserError> {
|
|
let ret = match self.parse_one_of_keywords(&[
|
|
Keyword::BINARY,
|
|
Keyword::DELIMITER,
|
|
Keyword::NULL,
|
|
Keyword::CSV,
|
|
]) {
|
|
Some(Keyword::BINARY) => CopyLegacyOption::Binary,
|
|
Some(Keyword::DELIMITER) => {
|
|
let _ = self.parse_keyword(Keyword::AS); // [ AS ]
|
|
CopyLegacyOption::Delimiter(self.parse_literal_char()?)
|
|
}
|
|
Some(Keyword::NULL) => {
|
|
let _ = self.parse_keyword(Keyword::AS); // [ AS ]
|
|
CopyLegacyOption::Null(self.parse_literal_string()?)
|
|
}
|
|
Some(Keyword::CSV) => CopyLegacyOption::Csv({
|
|
let mut opts = vec![];
|
|
while let Some(opt) =
|
|
self.maybe_parse(|parser| parser.parse_copy_legacy_csv_option())
|
|
{
|
|
opts.push(opt);
|
|
}
|
|
opts
|
|
}),
|
|
_ => self.expected("option", self.peek_token())?,
|
|
};
|
|
Ok(ret)
|
|
}
|
|
|
|
fn parse_copy_legacy_csv_option(&mut self) -> Result<CopyLegacyCsvOption, ParserError> {
|
|
let ret = match self.parse_one_of_keywords(&[
|
|
Keyword::HEADER,
|
|
Keyword::QUOTE,
|
|
Keyword::ESCAPE,
|
|
Keyword::FORCE,
|
|
]) {
|
|
Some(Keyword::HEADER) => CopyLegacyCsvOption::Header,
|
|
Some(Keyword::QUOTE) => {
|
|
let _ = self.parse_keyword(Keyword::AS); // [ AS ]
|
|
CopyLegacyCsvOption::Quote(self.parse_literal_char()?)
|
|
}
|
|
Some(Keyword::ESCAPE) => {
|
|
let _ = self.parse_keyword(Keyword::AS); // [ AS ]
|
|
CopyLegacyCsvOption::Escape(self.parse_literal_char()?)
|
|
}
|
|
Some(Keyword::FORCE) if self.parse_keywords(&[Keyword::NOT, Keyword::NULL]) => {
|
|
CopyLegacyCsvOption::ForceNotNull(
|
|
self.parse_comma_separated(Parser::parse_identifier)?,
|
|
)
|
|
}
|
|
Some(Keyword::FORCE) if self.parse_keywords(&[Keyword::QUOTE]) => {
|
|
CopyLegacyCsvOption::ForceQuote(
|
|
self.parse_comma_separated(Parser::parse_identifier)?,
|
|
)
|
|
}
|
|
_ => self.expected("csv option", self.peek_token())?,
|
|
};
|
|
Ok(ret)
|
|
}
|
|
|
|
fn parse_literal_char(&mut self) -> Result<char, ParserError> {
|
|
let s = self.parse_literal_string()?;
|
|
if s.len() != 1 {
|
|
return parser_err!(format!("Expect a char, found {:?}", s));
|
|
}
|
|
Ok(s.chars().next().unwrap())
|
|
}
|
|
|
|
/// Parse a tab separated values in
|
|
/// COPY payload
|
|
pub fn parse_tsv(&mut self) -> Vec<Option<String>> {
|
|
self.parse_tab_value()
|
|
}
|
|
|
|
pub fn parse_tab_value(&mut self) -> Vec<Option<String>> {
|
|
let mut values = vec![];
|
|
let mut content = String::from("");
|
|
while let Some(t) = self.next_token_no_skip() {
|
|
match t {
|
|
Token::Whitespace(Whitespace::Tab) => {
|
|
values.push(Some(content.to_string()));
|
|
content.clear();
|
|
}
|
|
Token::Whitespace(Whitespace::Newline) => {
|
|
values.push(Some(content.to_string()));
|
|
content.clear();
|
|
}
|
|
Token::Backslash => {
|
|
if self.consume_token(&Token::Period) {
|
|
return values;
|
|
}
|
|
if let Token::Word(w) = self.next_token() {
|
|
if w.value == "N" {
|
|
values.push(None);
|
|
}
|
|
}
|
|
}
|
|
_ => {
|
|
content.push_str(&t.to_string());
|
|
}
|
|
}
|
|
}
|
|
values
|
|
}
|
|
|
|
/// Parse a literal value (numbers, strings, date/time, booleans)
|
|
pub fn parse_value(&mut self) -> Result<Value, ParserError> {
|
|
match self.next_token() {
|
|
Token::Word(w) => match w.keyword {
|
|
Keyword::TRUE => Ok(Value::Boolean(true)),
|
|
Keyword::FALSE => Ok(Value::Boolean(false)),
|
|
Keyword::NULL => Ok(Value::Null),
|
|
Keyword::NoKeyword if w.quote_style.is_some() => match w.quote_style {
|
|
Some('"') => Ok(Value::DoubleQuotedString(w.value)),
|
|
Some('\'') => Ok(Value::SingleQuotedString(w.value)),
|
|
_ => self.expected("A value?", Token::Word(w))?,
|
|
},
|
|
_ => self.expected("a concrete value", Token::Word(w)),
|
|
},
|
|
// The call to n.parse() returns a bigdecimal when the
|
|
// bigdecimal feature is enabled, and is otherwise a no-op
|
|
// (i.e., it returns the input string).
|
|
Token::Number(ref n, l) => match n.parse() {
|
|
Ok(n) => Ok(Value::Number(n, l)),
|
|
Err(e) => parser_err!(format!("Could not parse '{}' as number: {}", n, e)),
|
|
},
|
|
Token::SingleQuotedString(ref s) => Ok(Value::SingleQuotedString(s.to_string())),
|
|
Token::DoubleQuotedString(ref s) => Ok(Value::DoubleQuotedString(s.to_string())),
|
|
Token::NationalStringLiteral(ref s) => Ok(Value::NationalStringLiteral(s.to_string())),
|
|
Token::EscapedStringLiteral(ref s) => Ok(Value::EscapedStringLiteral(s.to_string())),
|
|
Token::HexStringLiteral(ref s) => Ok(Value::HexStringLiteral(s.to_string())),
|
|
Token::Placeholder(ref s) => Ok(Value::Placeholder(s.to_string())),
|
|
unexpected => self.expected("a value", unexpected),
|
|
}
|
|
}
|
|
|
|
pub fn parse_number_value(&mut self) -> Result<Value, ParserError> {
|
|
match self.parse_value()? {
|
|
v @ Value::Number(_, _) => Ok(v),
|
|
v @ Value::Placeholder(_) => Ok(v),
|
|
_ => {
|
|
self.prev_token();
|
|
self.expected("literal number", self.peek_token())
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Parse an unsigned literal integer/long
|
|
pub fn parse_literal_uint(&mut self) -> Result<u64, ParserError> {
|
|
match self.next_token() {
|
|
Token::Number(s, _) => s.parse::<u64>().map_err(|e| {
|
|
ParserError::ParserError(format!("Could not parse '{}' as u64: {}", s, e))
|
|
}),
|
|
unexpected => self.expected("literal int", unexpected),
|
|
}
|
|
}
|
|
|
|
/// Parse a literal string
|
|
pub fn parse_literal_string(&mut self) -> Result<String, ParserError> {
|
|
match self.next_token() {
|
|
Token::Word(Word { value, keyword, .. }) if keyword == Keyword::NoKeyword => Ok(value),
|
|
Token::SingleQuotedString(s) => Ok(s),
|
|
Token::DoubleQuotedString(s) => Ok(s),
|
|
Token::EscapedStringLiteral(s) if dialect_of!(self is PostgreSqlDialect | GenericDialect) => {
|
|
Ok(s)
|
|
}
|
|
unexpected => self.expected("literal string", unexpected),
|
|
}
|
|
}
|
|
|
|
/// Parse a map key string
|
|
pub fn parse_map_key(&mut self) -> Result<Expr, ParserError> {
|
|
match self.next_token() {
|
|
Token::Word(Word { value, keyword, .. }) if keyword == Keyword::NoKeyword => {
|
|
if self.peek_token() == Token::LParen {
|
|
return self.parse_function(ObjectName(vec![Ident::new(value)]));
|
|
}
|
|
Ok(Expr::Value(Value::SingleQuotedString(value)))
|
|
}
|
|
Token::SingleQuotedString(s) => Ok(Expr::Value(Value::SingleQuotedString(s))),
|
|
#[cfg(not(feature = "bigdecimal"))]
|
|
Token::Number(s, _) => Ok(Expr::Value(Value::Number(s, false))),
|
|
#[cfg(feature = "bigdecimal")]
|
|
Token::Number(s, _) => Ok(Expr::Value(Value::Number(s.parse().unwrap(), false))),
|
|
unexpected => self.expected("literal string, number or function", unexpected),
|
|
}
|
|
}
|
|
|
|
/// Parse a SQL datatype (in the context of a CREATE TABLE statement for example)
|
|
pub fn parse_data_type(&mut self) -> Result<DataType, ParserError> {
|
|
let mut data = match self.next_token() {
|
|
Token::Word(w) => match w.keyword {
|
|
Keyword::BOOLEAN => Ok(DataType::Boolean),
|
|
Keyword::FLOAT => Ok(DataType::Float(self.parse_optional_precision()?)),
|
|
Keyword::REAL => Ok(DataType::Real),
|
|
Keyword::DOUBLE => {
|
|
let _ = self.parse_keyword(Keyword::PRECISION);
|
|
Ok(DataType::Double)
|
|
}
|
|
Keyword::TINYINT => {
|
|
let optional_precision = self.parse_optional_precision();
|
|
if self.parse_keyword(Keyword::UNSIGNED) {
|
|
Ok(DataType::UnsignedTinyInt(optional_precision?))
|
|
} else {
|
|
Ok(DataType::TinyInt(optional_precision?))
|
|
}
|
|
}
|
|
Keyword::SMALLINT => {
|
|
let optional_precision = self.parse_optional_precision();
|
|
if self.parse_keyword(Keyword::UNSIGNED) {
|
|
Ok(DataType::UnsignedSmallInt(optional_precision?))
|
|
} else {
|
|
Ok(DataType::SmallInt(optional_precision?))
|
|
}
|
|
}
|
|
Keyword::INT => {
|
|
let optional_precision = self.parse_optional_precision();
|
|
if self.parse_keyword(Keyword::UNSIGNED) {
|
|
Ok(DataType::UnsignedInt(optional_precision?))
|
|
} else {
|
|
Ok(DataType::Int(optional_precision?))
|
|
}
|
|
}
|
|
Keyword::INTEGER => {
|
|
let optional_precision = self.parse_optional_precision();
|
|
if self.parse_keyword(Keyword::UNSIGNED) {
|
|
Ok(DataType::UnsignedInteger(optional_precision?))
|
|
} else {
|
|
Ok(DataType::Integer(optional_precision?))
|
|
}
|
|
}
|
|
Keyword::BIGINT => {
|
|
let optional_precision = self.parse_optional_precision();
|
|
if self.parse_keyword(Keyword::UNSIGNED) {
|
|
Ok(DataType::UnsignedBigInt(optional_precision?))
|
|
} else {
|
|
Ok(DataType::BigInt(optional_precision?))
|
|
}
|
|
}
|
|
Keyword::VARCHAR => Ok(DataType::Varchar(self.parse_optional_precision()?)),
|
|
Keyword::NVARCHAR => Ok(DataType::Nvarchar(self.parse_optional_precision()?)),
|
|
Keyword::CHAR | Keyword::CHARACTER => {
|
|
if self.parse_keyword(Keyword::VARYING) {
|
|
Ok(DataType::Varchar(self.parse_optional_precision()?))
|
|
} else {
|
|
Ok(DataType::Char(self.parse_optional_precision()?))
|
|
}
|
|
}
|
|
Keyword::UUID => Ok(DataType::Uuid),
|
|
Keyword::DATE => Ok(DataType::Date),
|
|
Keyword::DATETIME => Ok(DataType::Datetime),
|
|
Keyword::TIMESTAMP => {
|
|
// TBD: we throw away "with/without timezone" information
|
|
if self.parse_keyword(Keyword::WITH) || self.parse_keyword(Keyword::WITHOUT) {
|
|
self.expect_keywords(&[Keyword::TIME, Keyword::ZONE])?;
|
|
}
|
|
Ok(DataType::Timestamp)
|
|
}
|
|
Keyword::TIME => {
|
|
// TBD: we throw away "with/without timezone" information
|
|
if self.parse_keyword(Keyword::WITH) || self.parse_keyword(Keyword::WITHOUT) {
|
|
self.expect_keywords(&[Keyword::TIME, Keyword::ZONE])?;
|
|
}
|
|
Ok(DataType::Time)
|
|
}
|
|
// Interval types can be followed by a complicated interval
|
|
// qualifier that we don't currently support. See
|
|
// parse_interval_literal for a taste.
|
|
Keyword::INTERVAL => Ok(DataType::Interval),
|
|
Keyword::REGCLASS => Ok(DataType::Regclass),
|
|
Keyword::STRING => Ok(DataType::String),
|
|
Keyword::TEXT => Ok(DataType::Text),
|
|
Keyword::BYTEA => Ok(DataType::Bytea),
|
|
Keyword::NUMERIC | Keyword::DECIMAL | Keyword::DEC => {
|
|
let (precision, scale) = self.parse_optional_precision_scale()?;
|
|
Ok(DataType::Decimal(precision, scale))
|
|
}
|
|
Keyword::ENUM => Ok(DataType::Enum(self.parse_string_values()?)),
|
|
Keyword::SET => Ok(DataType::Set(self.parse_string_values()?)),
|
|
Keyword::ARRAY => {
|
|
// Hive array syntax. Note that nesting arrays - or other Hive syntax
|
|
// that ends with > will fail due to "C++" problem - >> is parsed as
|
|
// Token::ShiftRight
|
|
self.expect_token(&Token::Lt)?;
|
|
let inside_type = self.parse_data_type()?;
|
|
self.expect_token(&Token::Gt)?;
|
|
Ok(DataType::Array(Box::new(inside_type)))
|
|
}
|
|
_ => {
|
|
self.prev_token();
|
|
let type_name = self.parse_object_name()?;
|
|
Ok(DataType::Custom(type_name))
|
|
}
|
|
},
|
|
unexpected => self.expected("a data type name", unexpected),
|
|
}?;
|
|
|
|
// Parse array data types. Note: this is postgresql-specific and different from
|
|
// Keyword::ARRAY syntax from above
|
|
while self.consume_token(&Token::LBracket) {
|
|
self.expect_token(&Token::RBracket)?;
|
|
data = DataType::Array(Box::new(data))
|
|
}
|
|
Ok(data)
|
|
}
|
|
|
|
pub fn parse_string_values(&mut self) -> Result<Vec<String>, ParserError> {
|
|
self.expect_token(&Token::LParen)?;
|
|
let mut values = Vec::new();
|
|
loop {
|
|
match self.next_token() {
|
|
Token::SingleQuotedString(value) => values.push(value),
|
|
unexpected => self.expected("a string", unexpected)?,
|
|
}
|
|
match self.next_token() {
|
|
Token::Comma => (),
|
|
Token::RParen => break,
|
|
unexpected => self.expected(", or }", unexpected)?,
|
|
}
|
|
}
|
|
Ok(values)
|
|
}
|
|
|
|
/// Parse `AS identifier` (or simply `identifier` if it's not a reserved keyword)
|
|
/// Some examples with aliases: `SELECT 1 foo`, `SELECT COUNT(*) AS cnt`,
|
|
/// `SELECT ... FROM t1 foo, t2 bar`, `SELECT ... FROM (...) AS bar`
|
|
pub fn parse_optional_alias(
|
|
&mut self,
|
|
reserved_kwds: &[Keyword],
|
|
) -> Result<Option<Ident>, ParserError> {
|
|
let after_as = self.parse_keyword(Keyword::AS);
|
|
match self.next_token() {
|
|
// Accept any identifier after `AS` (though many dialects have restrictions on
|
|
// keywords that may appear here). If there's no `AS`: don't parse keywords,
|
|
// which may start a construct allowed in this position, to be parsed as aliases.
|
|
// (For example, in `FROM t1 JOIN` the `JOIN` will always be parsed as a keyword,
|
|
// not an alias.)
|
|
Token::Word(w) if after_as || !reserved_kwds.contains(&w.keyword) => {
|
|
Ok(Some(w.to_ident()))
|
|
}
|
|
// MSSQL supports single-quoted strings as aliases for columns
|
|
// We accept them as table aliases too, although MSSQL does not.
|
|
//
|
|
// Note, that this conflicts with an obscure rule from the SQL
|
|
// standard, which we don't implement:
|
|
// https://crate.io/docs/sql-99/en/latest/chapters/07.html#character-string-literal-s
|
|
// "[Obscure Rule] SQL allows you to break a long <character
|
|
// string literal> up into two or more smaller <character string
|
|
// literal>s, split by a <separator> that includes a newline
|
|
// character. When it sees such a <literal>, your DBMS will
|
|
// ignore the <separator> and treat the multiple strings as
|
|
// a single <literal>."
|
|
Token::SingleQuotedString(s) => Ok(Some(Ident::with_quote('\'', s))),
|
|
not_an_ident => {
|
|
if after_as {
|
|
return self.expected("an identifier after AS", not_an_ident);
|
|
}
|
|
self.prev_token();
|
|
Ok(None) // no alias found
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Parse `AS identifier` when the AS is describing a table-valued object,
|
|
/// like in `... FROM generate_series(1, 10) AS t (col)`. In this case
|
|
/// the alias is allowed to optionally name the columns in the table, in
|
|
/// addition to the table itself.
|
|
pub fn parse_optional_table_alias(
|
|
&mut self,
|
|
reserved_kwds: &[Keyword],
|
|
) -> Result<Option<TableAlias>, ParserError> {
|
|
match self.parse_optional_alias(reserved_kwds)? {
|
|
Some(name) => {
|
|
let columns = self.parse_parenthesized_column_list(Optional)?;
|
|
Ok(Some(TableAlias { name, columns }))
|
|
}
|
|
None => Ok(None),
|
|
}
|
|
}
|
|
|
|
/// Parse a possibly qualified, possibly quoted identifier, e.g.
|
|
/// `foo` or `myschema."table"
|
|
pub fn parse_object_name(&mut self) -> Result<ObjectName, ParserError> {
|
|
let mut idents = vec![];
|
|
loop {
|
|
idents.push(self.parse_identifier()?);
|
|
if !self.consume_token(&Token::Period) {
|
|
break;
|
|
}
|
|
}
|
|
Ok(ObjectName(idents))
|
|
}
|
|
|
|
/// Parse identifiers strictly i.e. don't parse keywords
|
|
pub fn parse_identifiers_non_keywords(&mut self) -> Result<Vec<Ident>, ParserError> {
|
|
let mut idents = vec![];
|
|
loop {
|
|
match self.peek_token() {
|
|
Token::Word(w) => {
|
|
if w.keyword != Keyword::NoKeyword {
|
|
break;
|
|
}
|
|
|
|
idents.push(w.to_ident());
|
|
}
|
|
Token::EOF | Token::Eq => break,
|
|
_ => {}
|
|
}
|
|
|
|
self.next_token();
|
|
}
|
|
|
|
Ok(idents)
|
|
}
|
|
|
|
/// Parse identifiers
|
|
pub fn parse_identifiers(&mut self) -> Result<Vec<Ident>, ParserError> {
|
|
let mut idents = vec![];
|
|
loop {
|
|
match self.next_token() {
|
|
Token::Word(w) => {
|
|
idents.push(w.to_ident());
|
|
}
|
|
Token::EOF => break,
|
|
_ => {}
|
|
}
|
|
}
|
|
|
|
Ok(idents)
|
|
}
|
|
|
|
/// Parse a simple one-word identifier (possibly quoted, possibly a keyword)
|
|
pub fn parse_identifier(&mut self) -> Result<Ident, ParserError> {
|
|
match self.next_token() {
|
|
Token::Word(w) => Ok(w.to_ident()),
|
|
Token::SingleQuotedString(s) => Ok(Ident::with_quote('\'', s)),
|
|
unexpected => self.expected("identifier", unexpected),
|
|
}
|
|
}
|
|
|
|
/// Parse a parenthesized comma-separated list of unqualified, possibly quoted identifiers
|
|
pub fn parse_parenthesized_column_list(
|
|
&mut self,
|
|
optional: IsOptional,
|
|
) -> Result<Vec<Ident>, ParserError> {
|
|
if self.consume_token(&Token::LParen) {
|
|
let cols = self.parse_comma_separated(Parser::parse_identifier)?;
|
|
self.expect_token(&Token::RParen)?;
|
|
Ok(cols)
|
|
} else if optional == Optional {
|
|
Ok(vec![])
|
|
} else {
|
|
self.expected("a list of columns in parentheses", self.peek_token())
|
|
}
|
|
}
|
|
|
|
pub fn parse_optional_precision(&mut self) -> Result<Option<u64>, ParserError> {
|
|
if self.consume_token(&Token::LParen) {
|
|
let n = self.parse_literal_uint()?;
|
|
self.expect_token(&Token::RParen)?;
|
|
Ok(Some(n))
|
|
} else {
|
|
Ok(None)
|
|
}
|
|
}
|
|
|
|
pub fn parse_optional_precision_scale(
|
|
&mut self,
|
|
) -> Result<(Option<u64>, Option<u64>), ParserError> {
|
|
if self.consume_token(&Token::LParen) {
|
|
let n = self.parse_literal_uint()?;
|
|
let scale = if self.consume_token(&Token::Comma) {
|
|
Some(self.parse_literal_uint()?)
|
|
} else {
|
|
None
|
|
};
|
|
self.expect_token(&Token::RParen)?;
|
|
Ok((Some(n), scale))
|
|
} else {
|
|
Ok((None, None))
|
|
}
|
|
}
|
|
|
|
pub fn parse_delete(&mut self) -> Result<Statement, ParserError> {
|
|
self.expect_keyword(Keyword::FROM)?;
|
|
let table_name = self.parse_table_factor()?;
|
|
let using = if self.parse_keyword(Keyword::USING) {
|
|
Some(self.parse_table_factor()?)
|
|
} else {
|
|
None
|
|
};
|
|
let selection = if self.parse_keyword(Keyword::WHERE) {
|
|
Some(self.parse_expr()?)
|
|
} else {
|
|
None
|
|
};
|
|
|
|
Ok(Statement::Delete {
|
|
table_name,
|
|
using,
|
|
selection,
|
|
})
|
|
}
|
|
|
|
// KILL [CONNECTION | QUERY | MUTATION] processlist_id
|
|
pub fn parse_kill(&mut self) -> Result<Statement, ParserError> {
|
|
let modifier_keyword =
|
|
self.parse_one_of_keywords(&[Keyword::CONNECTION, Keyword::QUERY, Keyword::MUTATION]);
|
|
|
|
let id = self.parse_literal_uint()?;
|
|
|
|
let modifier = match modifier_keyword {
|
|
Some(Keyword::CONNECTION) => Some(KillType::Connection),
|
|
Some(Keyword::QUERY) => Some(KillType::Query),
|
|
Some(Keyword::MUTATION) => {
|
|
if dialect_of!(self is ClickHouseDialect | GenericDialect) {
|
|
Some(KillType::Mutation)
|
|
} else {
|
|
self.expected(
|
|
"Unsupported type for KILL, allowed: CONNECTION | QUERY",
|
|
self.peek_token(),
|
|
)?
|
|
}
|
|
}
|
|
_ => None,
|
|
};
|
|
|
|
Ok(Statement::Kill { modifier, id })
|
|
}
|
|
|
|
pub fn parse_explain(&mut self, describe_alias: bool) -> Result<Statement, ParserError> {
|
|
let analyze = self.parse_keyword(Keyword::ANALYZE);
|
|
let verbose = self.parse_keyword(Keyword::VERBOSE);
|
|
|
|
if let Some(statement) = self.maybe_parse(|parser| parser.parse_statement()) {
|
|
Ok(Statement::Explain {
|
|
describe_alias,
|
|
analyze,
|
|
verbose,
|
|
statement: Box::new(statement),
|
|
})
|
|
} else {
|
|
let table_name = self.parse_object_name()?;
|
|
|
|
Ok(Statement::ExplainTable {
|
|
describe_alias,
|
|
table_name,
|
|
})
|
|
}
|
|
}
|
|
|
|
/// Parse a query expression, i.e. a `SELECT` statement optionally
|
|
/// preceeded with some `WITH` CTE declarations and optionally followed
|
|
/// by `ORDER BY`. Unlike some other parse_... methods, this one doesn't
|
|
/// expect the initial keyword to be already consumed
|
|
pub fn parse_query(&mut self) -> Result<Query, ParserError> {
|
|
let with = if self.parse_keyword(Keyword::WITH) {
|
|
Some(With {
|
|
recursive: self.parse_keyword(Keyword::RECURSIVE),
|
|
cte_tables: self.parse_comma_separated(Parser::parse_cte)?,
|
|
})
|
|
} else {
|
|
None
|
|
};
|
|
|
|
if !self.parse_keyword(Keyword::INSERT) {
|
|
let body = Box::new(self.parse_query_body(0)?);
|
|
|
|
let order_by = if self.parse_keywords(&[Keyword::ORDER, Keyword::BY]) {
|
|
self.parse_comma_separated(Parser::parse_order_by_expr)?
|
|
} else {
|
|
vec![]
|
|
};
|
|
|
|
let mut limit = None;
|
|
let mut offset = None;
|
|
|
|
for _x in 0..2 {
|
|
if limit.is_none() && self.parse_keyword(Keyword::LIMIT) {
|
|
limit = self.parse_limit()?
|
|
}
|
|
|
|
if offset.is_none() && self.parse_keyword(Keyword::OFFSET) {
|
|
offset = Some(self.parse_offset()?)
|
|
}
|
|
|
|
if offset.is_none() && self.consume_token(&Token::Comma) {
|
|
// mysql style LIMIT 10, offset 5
|
|
offset = Some(self.parse_offset()?)
|
|
}
|
|
}
|
|
|
|
let fetch = if self.parse_keyword(Keyword::FETCH) {
|
|
Some(self.parse_fetch()?)
|
|
} else {
|
|
None
|
|
};
|
|
|
|
let lock = if self.parse_keyword(Keyword::FOR) {
|
|
Some(self.parse_lock()?)
|
|
} else {
|
|
None
|
|
};
|
|
|
|
Ok(Query {
|
|
with,
|
|
body,
|
|
order_by,
|
|
limit,
|
|
offset,
|
|
fetch,
|
|
lock,
|
|
})
|
|
} else {
|
|
let insert = self.parse_insert()?;
|
|
|
|
Ok(Query {
|
|
with,
|
|
body: Box::new(SetExpr::Insert(insert)),
|
|
limit: None,
|
|
order_by: vec![],
|
|
offset: None,
|
|
fetch: None,
|
|
lock: None,
|
|
})
|
|
}
|
|
}
|
|
|
|
/// Parse a CTE (`alias [( col1, col2, ... )] AS (subquery)`)
|
|
pub fn parse_cte(&mut self) -> Result<Cte, ParserError> {
|
|
let name = self.parse_identifier()?;
|
|
|
|
let mut cte = if self.parse_keyword(Keyword::AS) {
|
|
self.expect_token(&Token::LParen)?;
|
|
let query = self.parse_query()?;
|
|
self.expect_token(&Token::RParen)?;
|
|
let alias = TableAlias {
|
|
name,
|
|
columns: vec![],
|
|
};
|
|
Cte {
|
|
alias,
|
|
query,
|
|
from: None,
|
|
}
|
|
} else {
|
|
let columns = self.parse_parenthesized_column_list(Optional)?;
|
|
self.expect_keyword(Keyword::AS)?;
|
|
self.expect_token(&Token::LParen)?;
|
|
let query = self.parse_query()?;
|
|
self.expect_token(&Token::RParen)?;
|
|
let alias = TableAlias { name, columns };
|
|
Cte {
|
|
alias,
|
|
query,
|
|
from: None,
|
|
}
|
|
};
|
|
if self.parse_keyword(Keyword::FROM) {
|
|
cte.from = Some(self.parse_identifier()?);
|
|
}
|
|
Ok(cte)
|
|
}
|
|
|
|
/// Parse a "query body", which is an expression with roughly the
|
|
/// following grammar:
|
|
/// ```text
|
|
/// query_body ::= restricted_select | '(' subquery ')' | set_operation
|
|
/// restricted_select ::= 'SELECT' [expr_list] [ from ] [ where ] [ groupby_having ]
|
|
/// subquery ::= query_body [ order_by_limit ]
|
|
/// set_operation ::= query_body { 'UNION' | 'EXCEPT' | 'INTERSECT' } [ 'ALL' ] query_body
|
|
/// ```
|
|
pub fn parse_query_body(&mut self, precedence: u8) -> Result<SetExpr, ParserError> {
|
|
// We parse the expression using a Pratt parser, as in `parse_expr()`.
|
|
// Start by parsing a restricted SELECT or a `(subquery)`:
|
|
let mut expr = if self.parse_keyword(Keyword::SELECT) {
|
|
SetExpr::Select(Box::new(self.parse_select()?))
|
|
} else if self.consume_token(&Token::LParen) {
|
|
// CTEs are not allowed here, but the parser currently accepts them
|
|
let subquery = self.parse_query()?;
|
|
self.expect_token(&Token::RParen)?;
|
|
SetExpr::Query(Box::new(subquery))
|
|
} else if self.parse_keyword(Keyword::VALUES) {
|
|
SetExpr::Values(self.parse_values()?)
|
|
} else {
|
|
return self.expected(
|
|
"SELECT, VALUES, or a subquery in the query body",
|
|
self.peek_token(),
|
|
);
|
|
};
|
|
|
|
loop {
|
|
// The query can be optionally followed by a set operator:
|
|
let op = self.parse_set_operator(&self.peek_token());
|
|
let next_precedence = match op {
|
|
// UNION and EXCEPT have the same binding power and evaluate left-to-right
|
|
Some(SetOperator::Union) | Some(SetOperator::Except) => 10,
|
|
// INTERSECT has higher precedence than UNION/EXCEPT
|
|
Some(SetOperator::Intersect) => 20,
|
|
// Unexpected token or EOF => stop parsing the query body
|
|
None => break,
|
|
};
|
|
if precedence >= next_precedence {
|
|
break;
|
|
}
|
|
self.next_token(); // skip past the set operator
|
|
expr = SetExpr::SetOperation {
|
|
left: Box::new(expr),
|
|
op: op.unwrap(),
|
|
all: self.parse_keyword(Keyword::ALL),
|
|
right: Box::new(self.parse_query_body(next_precedence)?),
|
|
};
|
|
}
|
|
|
|
Ok(expr)
|
|
}
|
|
|
|
pub fn parse_set_operator(&mut self, token: &Token) -> Option<SetOperator> {
|
|
match token {
|
|
Token::Word(w) if w.keyword == Keyword::UNION => Some(SetOperator::Union),
|
|
Token::Word(w) if w.keyword == Keyword::EXCEPT => Some(SetOperator::Except),
|
|
Token::Word(w) if w.keyword == Keyword::INTERSECT => Some(SetOperator::Intersect),
|
|
_ => None,
|
|
}
|
|
}
|
|
|
|
/// Parse a restricted `SELECT` statement (no CTEs / `UNION` / `ORDER BY`),
|
|
/// assuming the initial `SELECT` was already consumed
|
|
pub fn parse_select(&mut self) -> Result<Select, ParserError> {
|
|
let distinct = self.parse_all_or_distinct()?;
|
|
|
|
let top = if self.parse_keyword(Keyword::TOP) {
|
|
Some(self.parse_top()?)
|
|
} else {
|
|
None
|
|
};
|
|
|
|
let projection = self.parse_projection()?;
|
|
|
|
let into = if self.parse_keyword(Keyword::INTO) {
|
|
let temporary = self
|
|
.parse_one_of_keywords(&[Keyword::TEMP, Keyword::TEMPORARY])
|
|
.is_some();
|
|
let unlogged = self.parse_keyword(Keyword::UNLOGGED);
|
|
let table = self.parse_keyword(Keyword::TABLE);
|
|
let name = self.parse_object_name()?;
|
|
Some(SelectInto {
|
|
temporary,
|
|
unlogged,
|
|
table,
|
|
name,
|
|
})
|
|
} else {
|
|
None
|
|
};
|
|
|
|
// Note that for keywords to be properly handled here, they need to be
|
|
// added to `RESERVED_FOR_COLUMN_ALIAS` / `RESERVED_FOR_TABLE_ALIAS`,
|
|
// otherwise they may be parsed as an alias as part of the `projection`
|
|
// or `from`.
|
|
|
|
let from = if self.parse_keyword(Keyword::FROM) {
|
|
self.parse_comma_separated(Parser::parse_table_and_joins)?
|
|
} else {
|
|
vec![]
|
|
};
|
|
|
|
let mut lateral_views = vec![];
|
|
loop {
|
|
if self.parse_keywords(&[Keyword::LATERAL, Keyword::VIEW]) {
|
|
let outer = self.parse_keyword(Keyword::OUTER);
|
|
let lateral_view = self.parse_expr()?;
|
|
let lateral_view_name = self.parse_object_name()?;
|
|
let lateral_col_alias = self
|
|
.parse_comma_separated(|parser| {
|
|
parser.parse_optional_alias(&[
|
|
Keyword::WHERE,
|
|
Keyword::GROUP,
|
|
Keyword::CLUSTER,
|
|
Keyword::HAVING,
|
|
Keyword::LATERAL,
|
|
]) // This couldn't possibly be a bad idea
|
|
})?
|
|
.into_iter()
|
|
.flatten()
|
|
.collect();
|
|
|
|
lateral_views.push(LateralView {
|
|
lateral_view,
|
|
lateral_view_name,
|
|
lateral_col_alias,
|
|
outer,
|
|
});
|
|
} else {
|
|
break;
|
|
}
|
|
}
|
|
|
|
let selection = if self.parse_keyword(Keyword::WHERE) {
|
|
Some(self.parse_expr()?)
|
|
} else {
|
|
None
|
|
};
|
|
|
|
let group_by = if self.parse_keywords(&[Keyword::GROUP, Keyword::BY]) {
|
|
self.parse_comma_separated(Parser::parse_group_by_expr)?
|
|
} else {
|
|
vec![]
|
|
};
|
|
|
|
let cluster_by = if self.parse_keywords(&[Keyword::CLUSTER, Keyword::BY]) {
|
|
self.parse_comma_separated(Parser::parse_expr)?
|
|
} else {
|
|
vec![]
|
|
};
|
|
|
|
let distribute_by = if self.parse_keywords(&[Keyword::DISTRIBUTE, Keyword::BY]) {
|
|
self.parse_comma_separated(Parser::parse_expr)?
|
|
} else {
|
|
vec![]
|
|
};
|
|
|
|
let sort_by = if self.parse_keywords(&[Keyword::SORT, Keyword::BY]) {
|
|
self.parse_comma_separated(Parser::parse_expr)?
|
|
} else {
|
|
vec![]
|
|
};
|
|
|
|
let having = if self.parse_keyword(Keyword::HAVING) {
|
|
Some(self.parse_expr()?)
|
|
} else {
|
|
None
|
|
};
|
|
|
|
let qualify = if self.parse_keyword(Keyword::QUALIFY) {
|
|
Some(self.parse_expr()?)
|
|
} else {
|
|
None
|
|
};
|
|
|
|
Ok(Select {
|
|
distinct,
|
|
top,
|
|
projection,
|
|
into,
|
|
from,
|
|
lateral_views,
|
|
selection,
|
|
group_by,
|
|
cluster_by,
|
|
distribute_by,
|
|
sort_by,
|
|
having,
|
|
qualify,
|
|
})
|
|
}
|
|
|
|
pub fn parse_set(&mut self) -> Result<Statement, ParserError> {
|
|
let modifier =
|
|
self.parse_one_of_keywords(&[Keyword::SESSION, Keyword::LOCAL, Keyword::HIVEVAR]);
|
|
if let Some(Keyword::HIVEVAR) = modifier {
|
|
self.expect_token(&Token::Colon)?;
|
|
} else if self.parse_keyword(Keyword::ROLE) {
|
|
let role_name = if self.parse_keyword(Keyword::NONE) {
|
|
None
|
|
} else {
|
|
Some(self.parse_identifier()?)
|
|
};
|
|
return Ok(Statement::SetRole {
|
|
local: modifier == Some(Keyword::LOCAL),
|
|
session: modifier == Some(Keyword::SESSION),
|
|
role_name,
|
|
});
|
|
}
|
|
|
|
let variable = self.parse_object_name()?;
|
|
if variable.to_string().eq_ignore_ascii_case("NAMES")
|
|
&& dialect_of!(self is MySqlDialect | GenericDialect)
|
|
{
|
|
if self.parse_keyword(Keyword::DEFAULT) {
|
|
return Ok(Statement::SetNamesDefault {});
|
|
}
|
|
|
|
let charset_name = self.parse_literal_string()?;
|
|
let collation_name = if self.parse_one_of_keywords(&[Keyword::COLLATE]).is_some() {
|
|
Some(self.parse_literal_string()?)
|
|
} else {
|
|
None
|
|
};
|
|
|
|
Ok(Statement::SetNames {
|
|
charset_name,
|
|
collation_name,
|
|
})
|
|
} else if self.consume_token(&Token::Eq) || self.parse_keyword(Keyword::TO) {
|
|
let mut values = vec![];
|
|
loop {
|
|
let token = self.peek_token();
|
|
let value = match (self.parse_value(), token) {
|
|
(Ok(value), _) => SetVariableValue::Literal(value),
|
|
(Err(_), Token::Word(ident)) => SetVariableValue::Ident(ident.to_ident()),
|
|
(Err(_), Token::Minus) => {
|
|
let next_token = self.next_token();
|
|
match next_token {
|
|
Token::Word(ident) => SetVariableValue::Ident(Ident {
|
|
quote_style: ident.quote_style,
|
|
value: format!("-{}", ident.value),
|
|
}),
|
|
_ => self.expected("word", next_token)?,
|
|
}
|
|
}
|
|
(Err(_), unexpected) => self.expected("variable value", unexpected)?,
|
|
};
|
|
values.push(value);
|
|
if self.consume_token(&Token::Comma) {
|
|
continue;
|
|
}
|
|
return Ok(Statement::SetVariable {
|
|
local: modifier == Some(Keyword::LOCAL),
|
|
hivevar: Some(Keyword::HIVEVAR) == modifier,
|
|
variable,
|
|
value: values,
|
|
});
|
|
}
|
|
} else if variable.to_string() == "CHARACTERISTICS" {
|
|
self.expect_keywords(&[Keyword::AS, Keyword::TRANSACTION])?;
|
|
Ok(Statement::SetTransaction {
|
|
modes: self.parse_transaction_modes()?,
|
|
snapshot: None,
|
|
session: true,
|
|
})
|
|
} else if variable.to_string() == "TRANSACTION" && modifier.is_none() {
|
|
if self.parse_keyword(Keyword::SNAPSHOT) {
|
|
let snaphot_id = self.parse_value()?;
|
|
return Ok(Statement::SetTransaction {
|
|
modes: vec![],
|
|
snapshot: Some(snaphot_id),
|
|
session: false,
|
|
});
|
|
}
|
|
Ok(Statement::SetTransaction {
|
|
modes: self.parse_transaction_modes()?,
|
|
snapshot: None,
|
|
session: false,
|
|
})
|
|
} else {
|
|
self.expected("equals sign or TO", self.peek_token())
|
|
}
|
|
}
|
|
|
|
pub fn parse_show(&mut self) -> Result<Statement, ParserError> {
|
|
let extended = self.parse_keyword(Keyword::EXTENDED);
|
|
let full = self.parse_keyword(Keyword::FULL);
|
|
if self
|
|
.parse_one_of_keywords(&[Keyword::COLUMNS, Keyword::FIELDS])
|
|
.is_some()
|
|
{
|
|
Ok(self.parse_show_columns(extended, full)?)
|
|
} else if self.parse_keyword(Keyword::TABLES) {
|
|
Ok(self.parse_show_tables(extended, full)?)
|
|
} else if extended || full {
|
|
Err(ParserError::ParserError(
|
|
"EXTENDED/FULL are not supported with this type of SHOW query".to_string(),
|
|
))
|
|
} else if self.parse_one_of_keywords(&[Keyword::CREATE]).is_some() {
|
|
Ok(self.parse_show_create()?)
|
|
} else if self.parse_keyword(Keyword::COLLATION) {
|
|
Ok(self.parse_show_collation()?)
|
|
} else if self.parse_keyword(Keyword::VARIABLES)
|
|
&& dialect_of!(self is MySqlDialect | GenericDialect)
|
|
{
|
|
// TODO: Support GLOBAL|SESSION
|
|
Ok(Statement::ShowVariables {
|
|
filter: self.parse_show_statement_filter()?,
|
|
})
|
|
} else {
|
|
Ok(Statement::ShowVariable {
|
|
variable: self.parse_identifiers()?,
|
|
})
|
|
}
|
|
}
|
|
|
|
pub fn parse_show_create(&mut self) -> Result<Statement, ParserError> {
|
|
let obj_type = match self.expect_one_of_keywords(&[
|
|
Keyword::TABLE,
|
|
Keyword::TRIGGER,
|
|
Keyword::FUNCTION,
|
|
Keyword::PROCEDURE,
|
|
Keyword::EVENT,
|
|
Keyword::VIEW,
|
|
])? {
|
|
Keyword::TABLE => Ok(ShowCreateObject::Table),
|
|
Keyword::TRIGGER => Ok(ShowCreateObject::Trigger),
|
|
Keyword::FUNCTION => Ok(ShowCreateObject::Function),
|
|
Keyword::PROCEDURE => Ok(ShowCreateObject::Procedure),
|
|
Keyword::EVENT => Ok(ShowCreateObject::Event),
|
|
Keyword::VIEW => Ok(ShowCreateObject::View),
|
|
keyword => Err(ParserError::ParserError(format!(
|
|
"Unable to map keyword to ShowCreateObject: {:?}",
|
|
keyword
|
|
))),
|
|
}?;
|
|
|
|
let obj_name = self.parse_object_name()?;
|
|
|
|
Ok(Statement::ShowCreate { obj_type, obj_name })
|
|
}
|
|
|
|
pub fn parse_show_columns(
|
|
&mut self,
|
|
extended: bool,
|
|
full: bool,
|
|
) -> Result<Statement, ParserError> {
|
|
self.expect_one_of_keywords(&[Keyword::FROM, Keyword::IN])?;
|
|
let object_name = self.parse_object_name()?;
|
|
let table_name = match self.parse_one_of_keywords(&[Keyword::FROM, Keyword::IN]) {
|
|
Some(_) => {
|
|
let db_name = vec![self.parse_identifier()?];
|
|
let ObjectName(table_name) = object_name;
|
|
let object_name = db_name.into_iter().chain(table_name.into_iter()).collect();
|
|
ObjectName(object_name)
|
|
}
|
|
None => object_name,
|
|
};
|
|
let filter = self.parse_show_statement_filter()?;
|
|
Ok(Statement::ShowColumns {
|
|
extended,
|
|
full,
|
|
table_name,
|
|
filter,
|
|
})
|
|
}
|
|
|
|
pub fn parse_show_tables(
|
|
&mut self,
|
|
extended: bool,
|
|
full: bool,
|
|
) -> Result<Statement, ParserError> {
|
|
let db_name = match self.parse_one_of_keywords(&[Keyword::FROM, Keyword::IN]) {
|
|
Some(_) => Some(self.parse_identifier()?),
|
|
None => None,
|
|
};
|
|
let filter = self.parse_show_statement_filter()?;
|
|
Ok(Statement::ShowTables {
|
|
extended,
|
|
full,
|
|
db_name,
|
|
filter,
|
|
})
|
|
}
|
|
|
|
pub fn parse_show_collation(&mut self) -> Result<Statement, ParserError> {
|
|
let filter = self.parse_show_statement_filter()?;
|
|
Ok(Statement::ShowCollation { filter })
|
|
}
|
|
|
|
pub fn parse_show_statement_filter(
|
|
&mut self,
|
|
) -> Result<Option<ShowStatementFilter>, ParserError> {
|
|
if self.parse_keyword(Keyword::LIKE) {
|
|
Ok(Some(ShowStatementFilter::Like(
|
|
self.parse_literal_string()?,
|
|
)))
|
|
} else if self.parse_keyword(Keyword::ILIKE) {
|
|
Ok(Some(ShowStatementFilter::ILike(
|
|
self.parse_literal_string()?,
|
|
)))
|
|
} else if self.parse_keyword(Keyword::WHERE) {
|
|
Ok(Some(ShowStatementFilter::Where(self.parse_expr()?)))
|
|
} else {
|
|
Ok(None)
|
|
}
|
|
}
|
|
|
|
pub fn parse_use(&mut self) -> Result<Statement, ParserError> {
|
|
let db_name = self.parse_identifier()?;
|
|
Ok(Statement::Use { db_name })
|
|
}
|
|
|
|
pub fn parse_table_and_joins(&mut self) -> Result<TableWithJoins, ParserError> {
|
|
let relation = self.parse_table_factor()?;
|
|
// Note that for keywords to be properly handled here, they need to be
|
|
// added to `RESERVED_FOR_TABLE_ALIAS`, otherwise they may be parsed as
|
|
// a table alias.
|
|
let mut joins = vec![];
|
|
loop {
|
|
let join = if self.parse_keyword(Keyword::CROSS) {
|
|
let join_operator = if self.parse_keyword(Keyword::JOIN) {
|
|
JoinOperator::CrossJoin
|
|
} else if self.parse_keyword(Keyword::APPLY) {
|
|
// MSSQL extension, similar to CROSS JOIN LATERAL
|
|
JoinOperator::CrossApply
|
|
} else {
|
|
return self.expected("JOIN or APPLY after CROSS", self.peek_token());
|
|
};
|
|
Join {
|
|
relation: self.parse_table_factor()?,
|
|
join_operator,
|
|
}
|
|
} else if self.parse_keyword(Keyword::OUTER) {
|
|
// MSSQL extension, similar to LEFT JOIN LATERAL .. ON 1=1
|
|
self.expect_keyword(Keyword::APPLY)?;
|
|
Join {
|
|
relation: self.parse_table_factor()?,
|
|
join_operator: JoinOperator::OuterApply,
|
|
}
|
|
} else {
|
|
let natural = self.parse_keyword(Keyword::NATURAL);
|
|
let peek_keyword = if let Token::Word(w) = self.peek_token() {
|
|
w.keyword
|
|
} else {
|
|
Keyword::NoKeyword
|
|
};
|
|
|
|
let join_operator_type = match peek_keyword {
|
|
Keyword::INNER | Keyword::JOIN => {
|
|
let _ = self.parse_keyword(Keyword::INNER);
|
|
self.expect_keyword(Keyword::JOIN)?;
|
|
JoinOperator::Inner
|
|
}
|
|
kw @ Keyword::LEFT | kw @ Keyword::RIGHT | kw @ Keyword::FULL => {
|
|
let _ = self.next_token();
|
|
let _ = self.parse_keyword(Keyword::OUTER);
|
|
self.expect_keyword(Keyword::JOIN)?;
|
|
match kw {
|
|
Keyword::LEFT => JoinOperator::LeftOuter,
|
|
Keyword::RIGHT => JoinOperator::RightOuter,
|
|
Keyword::FULL => JoinOperator::FullOuter,
|
|
_ => unreachable!(),
|
|
}
|
|
}
|
|
Keyword::OUTER => {
|
|
return self.expected("LEFT, RIGHT, or FULL", self.peek_token());
|
|
}
|
|
_ if natural => {
|
|
return self.expected("a join type after NATURAL", self.peek_token());
|
|
}
|
|
_ => break,
|
|
};
|
|
let relation = self.parse_table_factor()?;
|
|
let join_constraint = self.parse_join_constraint(natural)?;
|
|
Join {
|
|
relation,
|
|
join_operator: join_operator_type(join_constraint),
|
|
}
|
|
};
|
|
joins.push(join);
|
|
}
|
|
Ok(TableWithJoins { relation, joins })
|
|
}
|
|
|
|
/// A table name or a parenthesized subquery, followed by optional `[AS] alias`
|
|
pub fn parse_table_factor(&mut self) -> Result<TableFactor, ParserError> {
|
|
if self.parse_keyword(Keyword::LATERAL) {
|
|
// LATERAL must always be followed by a subquery.
|
|
if !self.consume_token(&Token::LParen) {
|
|
self.expected("subquery after LATERAL", self.peek_token())?;
|
|
}
|
|
self.parse_derived_table_factor(Lateral)
|
|
} else if self.parse_keyword(Keyword::TABLE) {
|
|
// parse table function (SELECT * FROM TABLE (<expr>) [ AS <alias> ])
|
|
self.expect_token(&Token::LParen)?;
|
|
let expr = self.parse_expr()?;
|
|
self.expect_token(&Token::RParen)?;
|
|
let alias = self.parse_optional_table_alias(keywords::RESERVED_FOR_TABLE_ALIAS)?;
|
|
Ok(TableFactor::TableFunction { expr, alias })
|
|
} else if self.consume_token(&Token::LParen) {
|
|
// A left paren introduces either a derived table (i.e., a subquery)
|
|
// or a nested join. It's nearly impossible to determine ahead of
|
|
// time which it is... so we just try to parse both.
|
|
//
|
|
// Here's an example that demonstrates the complexity:
|
|
// /-------------------------------------------------------\
|
|
// | /-----------------------------------\ |
|
|
// SELECT * FROM ( ( ( (SELECT 1) UNION (SELECT 2) ) AS t1 NATURAL JOIN t2 ) )
|
|
// ^ ^ ^ ^
|
|
// | | | |
|
|
// | | | |
|
|
// | | | (4) belongs to a SetExpr::Query inside the subquery
|
|
// | | (3) starts a derived table (subquery)
|
|
// | (2) starts a nested join
|
|
// (1) an additional set of parens around a nested join
|
|
//
|
|
|
|
// If the recently consumed '(' starts a derived table, the call to
|
|
// `parse_derived_table_factor` below will return success after parsing the
|
|
// subquery, followed by the closing ')', and the alias of the derived table.
|
|
// In the example above this is case (3).
|
|
return_ok_if_some!(
|
|
self.maybe_parse(|parser| parser.parse_derived_table_factor(NotLateral))
|
|
);
|
|
// A parsing error from `parse_derived_table_factor` indicates that the '(' we've
|
|
// recently consumed does not start a derived table (cases 1, 2, or 4).
|
|
// `maybe_parse` will ignore such an error and rewind to be after the opening '('.
|
|
|
|
// Inside the parentheses we expect to find an (A) table factor
|
|
// followed by some joins or (B) another level of nesting.
|
|
let mut table_and_joins = self.parse_table_and_joins()?;
|
|
|
|
#[allow(clippy::if_same_then_else)]
|
|
if !table_and_joins.joins.is_empty() {
|
|
self.expect_token(&Token::RParen)?;
|
|
let alias = self.parse_optional_table_alias(keywords::RESERVED_FOR_TABLE_ALIAS)?;
|
|
Ok(TableFactor::NestedJoin {
|
|
table_with_joins: Box::new(table_and_joins),
|
|
alias,
|
|
}) // (A)
|
|
} else if let TableFactor::NestedJoin {
|
|
table_with_joins: _,
|
|
alias: _,
|
|
} = &table_and_joins.relation
|
|
{
|
|
// (B): `table_and_joins` (what we found inside the parentheses)
|
|
// is a nested join `(foo JOIN bar)`, not followed by other joins.
|
|
self.expect_token(&Token::RParen)?;
|
|
let alias = self.parse_optional_table_alias(keywords::RESERVED_FOR_TABLE_ALIAS)?;
|
|
Ok(TableFactor::NestedJoin {
|
|
table_with_joins: Box::new(table_and_joins),
|
|
alias,
|
|
})
|
|
} else if dialect_of!(self is SnowflakeDialect | GenericDialect) {
|
|
// Dialect-specific behavior: Snowflake diverges from the
|
|
// standard and from most of the other implementations by
|
|
// allowing extra parentheses not only around a join (B), but
|
|
// around lone table names (e.g. `FROM (mytable [AS alias])`)
|
|
// and around derived tables (e.g. `FROM ((SELECT ...)
|
|
// [AS alias])`) as well.
|
|
self.expect_token(&Token::RParen)?;
|
|
|
|
if let Some(outer_alias) =
|
|
self.parse_optional_table_alias(keywords::RESERVED_FOR_TABLE_ALIAS)?
|
|
{
|
|
// Snowflake also allows specifying an alias *after* parens
|
|
// e.g. `FROM (mytable) AS alias`
|
|
match &mut table_and_joins.relation {
|
|
TableFactor::Derived { alias, .. }
|
|
| TableFactor::Table { alias, .. }
|
|
| TableFactor::UNNEST { alias, .. }
|
|
| TableFactor::TableFunction { alias, .. }
|
|
| TableFactor::NestedJoin { alias, .. } => {
|
|
// but not `FROM (mytable AS alias1) AS alias2`.
|
|
if let Some(inner_alias) = alias {
|
|
return Err(ParserError::ParserError(format!(
|
|
"duplicate alias {}",
|
|
inner_alias
|
|
)));
|
|
}
|
|
// Act as if the alias was specified normally next
|
|
// to the table name: `(mytable) AS alias` ->
|
|
// `(mytable AS alias)`
|
|
alias.replace(outer_alias);
|
|
}
|
|
};
|
|
}
|
|
// Do not store the extra set of parens in the AST
|
|
Ok(table_and_joins.relation)
|
|
} else {
|
|
// The SQL spec prohibits derived tables and bare tables from
|
|
// appearing alone in parentheses (e.g. `FROM (mytable)`)
|
|
self.expected("joined table", self.peek_token())
|
|
}
|
|
} else if dialect_of!(self is BigQueryDialect | GenericDialect)
|
|
&& self.parse_keyword(Keyword::UNNEST)
|
|
{
|
|
self.expect_token(&Token::LParen)?;
|
|
let expr = self.parse_expr()?;
|
|
self.expect_token(&Token::RParen)?;
|
|
|
|
let alias = match self.parse_optional_table_alias(keywords::RESERVED_FOR_TABLE_ALIAS) {
|
|
Ok(Some(alias)) => Some(alias),
|
|
Ok(None) => None,
|
|
Err(e) => return Err(e),
|
|
};
|
|
|
|
let with_offset = match self.expect_keywords(&[Keyword::WITH, Keyword::OFFSET]) {
|
|
Ok(()) => true,
|
|
Err(_) => false,
|
|
};
|
|
|
|
let with_offset_alias =
|
|
match self.parse_optional_alias(keywords::RESERVED_FOR_COLUMN_ALIAS) {
|
|
Ok(Some(alias)) => Some(alias),
|
|
Ok(None) => None,
|
|
Err(e) => return Err(e),
|
|
};
|
|
|
|
Ok(TableFactor::UNNEST {
|
|
alias,
|
|
array_expr: Box::new(expr),
|
|
with_offset,
|
|
with_offset_alias,
|
|
})
|
|
} else {
|
|
let name = self.parse_object_name()?;
|
|
// Postgres, MSSQL: table-valued functions:
|
|
let args = if self.consume_token(&Token::LParen) {
|
|
Some(self.parse_optional_args()?)
|
|
} else {
|
|
None
|
|
};
|
|
let alias = self.parse_optional_table_alias(keywords::RESERVED_FOR_TABLE_ALIAS)?;
|
|
// MSSQL-specific table hints:
|
|
let mut with_hints = vec![];
|
|
if self.parse_keyword(Keyword::WITH) {
|
|
if self.consume_token(&Token::LParen) {
|
|
with_hints = self.parse_comma_separated(Parser::parse_expr)?;
|
|
self.expect_token(&Token::RParen)?;
|
|
} else {
|
|
// rewind, as WITH may belong to the next statement's CTE
|
|
self.prev_token();
|
|
}
|
|
};
|
|
Ok(TableFactor::Table {
|
|
name,
|
|
alias,
|
|
args,
|
|
with_hints,
|
|
})
|
|
}
|
|
}
|
|
|
|
pub fn parse_derived_table_factor(
|
|
&mut self,
|
|
lateral: IsLateral,
|
|
) -> Result<TableFactor, ParserError> {
|
|
let subquery = Box::new(self.parse_query()?);
|
|
self.expect_token(&Token::RParen)?;
|
|
let alias = self.parse_optional_table_alias(keywords::RESERVED_FOR_TABLE_ALIAS)?;
|
|
Ok(TableFactor::Derived {
|
|
lateral: match lateral {
|
|
Lateral => true,
|
|
NotLateral => false,
|
|
},
|
|
subquery,
|
|
alias,
|
|
})
|
|
}
|
|
|
|
pub fn parse_join_constraint(&mut self, natural: bool) -> Result<JoinConstraint, ParserError> {
|
|
if natural {
|
|
Ok(JoinConstraint::Natural)
|
|
} else if self.parse_keyword(Keyword::ON) {
|
|
let constraint = self.parse_expr()?;
|
|
Ok(JoinConstraint::On(constraint))
|
|
} else if self.parse_keyword(Keyword::USING) {
|
|
let columns = self.parse_parenthesized_column_list(Mandatory)?;
|
|
Ok(JoinConstraint::Using(columns))
|
|
} else {
|
|
Ok(JoinConstraint::None)
|
|
//self.expected("ON, or USING after JOIN", self.peek_token())
|
|
}
|
|
}
|
|
|
|
/// Parse a GRANT statement.
|
|
pub fn parse_grant(&mut self) -> Result<Statement, ParserError> {
|
|
let (privileges, objects) = self.parse_grant_revoke_privileges_objects()?;
|
|
|
|
self.expect_keyword(Keyword::TO)?;
|
|
let grantees = self.parse_comma_separated(Parser::parse_identifier)?;
|
|
|
|
let with_grant_option =
|
|
self.parse_keywords(&[Keyword::WITH, Keyword::GRANT, Keyword::OPTION]);
|
|
|
|
let granted_by = self
|
|
.parse_keywords(&[Keyword::GRANTED, Keyword::BY])
|
|
.then(|| self.parse_identifier().unwrap());
|
|
|
|
Ok(Statement::Grant {
|
|
privileges,
|
|
objects,
|
|
grantees,
|
|
with_grant_option,
|
|
granted_by,
|
|
})
|
|
}
|
|
|
|
pub fn parse_grant_revoke_privileges_objects(
|
|
&mut self,
|
|
) -> Result<(Privileges, GrantObjects), ParserError> {
|
|
let privileges = if self.parse_keyword(Keyword::ALL) {
|
|
Privileges::All {
|
|
with_privileges_keyword: self.parse_keyword(Keyword::PRIVILEGES),
|
|
}
|
|
} else {
|
|
let (actions, err): (Vec<_>, Vec<_>) = self
|
|
.parse_comma_separated(Parser::parse_grant_permission)?
|
|
.into_iter()
|
|
.map(|(kw, columns)| match kw {
|
|
Keyword::DELETE => Ok(Action::Delete),
|
|
Keyword::INSERT => Ok(Action::Insert { columns }),
|
|
Keyword::REFERENCES => Ok(Action::References { columns }),
|
|
Keyword::SELECT => Ok(Action::Select { columns }),
|
|
Keyword::TRIGGER => Ok(Action::Trigger),
|
|
Keyword::TRUNCATE => Ok(Action::Truncate),
|
|
Keyword::UPDATE => Ok(Action::Update { columns }),
|
|
Keyword::USAGE => Ok(Action::Usage),
|
|
Keyword::CONNECT => Ok(Action::Connect),
|
|
Keyword::CREATE => Ok(Action::Create),
|
|
Keyword::EXECUTE => Ok(Action::Execute),
|
|
Keyword::TEMPORARY => Ok(Action::Temporary),
|
|
// This will cover all future added keywords to
|
|
// parse_grant_permission and unhandled in this
|
|
// match
|
|
_ => Err(kw),
|
|
})
|
|
.partition(Result::is_ok);
|
|
|
|
if !err.is_empty() {
|
|
let errors: Vec<Keyword> = err.into_iter().filter_map(|x| x.err()).collect();
|
|
return Err(ParserError::ParserError(format!(
|
|
"INTERNAL ERROR: GRANT/REVOKE unexpected keyword(s) - {:?}",
|
|
errors
|
|
)));
|
|
}
|
|
let act = actions.into_iter().filter_map(|x| x.ok()).collect();
|
|
Privileges::Actions(act)
|
|
};
|
|
|
|
self.expect_keyword(Keyword::ON)?;
|
|
|
|
let objects = if self.parse_keywords(&[
|
|
Keyword::ALL,
|
|
Keyword::TABLES,
|
|
Keyword::IN,
|
|
Keyword::SCHEMA,
|
|
]) {
|
|
GrantObjects::AllTablesInSchema {
|
|
schemas: self.parse_comma_separated(Parser::parse_object_name)?,
|
|
}
|
|
} else if self.parse_keywords(&[
|
|
Keyword::ALL,
|
|
Keyword::SEQUENCES,
|
|
Keyword::IN,
|
|
Keyword::SCHEMA,
|
|
]) {
|
|
GrantObjects::AllSequencesInSchema {
|
|
schemas: self.parse_comma_separated(Parser::parse_object_name)?,
|
|
}
|
|
} else {
|
|
let object_type =
|
|
self.parse_one_of_keywords(&[Keyword::SEQUENCE, Keyword::SCHEMA, Keyword::TABLE]);
|
|
let objects = self.parse_comma_separated(Parser::parse_object_name);
|
|
match object_type {
|
|
Some(Keyword::SCHEMA) => GrantObjects::Schemas(objects?),
|
|
Some(Keyword::SEQUENCE) => GrantObjects::Sequences(objects?),
|
|
Some(Keyword::TABLE) | None => GrantObjects::Tables(objects?),
|
|
_ => unreachable!(),
|
|
}
|
|
};
|
|
|
|
Ok((privileges, objects))
|
|
}
|
|
|
|
pub fn parse_grant_permission(&mut self) -> Result<(Keyword, Option<Vec<Ident>>), ParserError> {
|
|
if let Some(kw) = self.parse_one_of_keywords(&[
|
|
Keyword::CONNECT,
|
|
Keyword::CREATE,
|
|
Keyword::DELETE,
|
|
Keyword::EXECUTE,
|
|
Keyword::INSERT,
|
|
Keyword::REFERENCES,
|
|
Keyword::SELECT,
|
|
Keyword::TEMPORARY,
|
|
Keyword::TRIGGER,
|
|
Keyword::TRUNCATE,
|
|
Keyword::UPDATE,
|
|
Keyword::USAGE,
|
|
]) {
|
|
let columns = match kw {
|
|
Keyword::INSERT | Keyword::REFERENCES | Keyword::SELECT | Keyword::UPDATE => {
|
|
let columns = self.parse_parenthesized_column_list(Optional)?;
|
|
if columns.is_empty() {
|
|
None
|
|
} else {
|
|
Some(columns)
|
|
}
|
|
}
|
|
_ => None,
|
|
};
|
|
Ok((kw, columns))
|
|
} else {
|
|
self.expected("a privilege keyword", self.peek_token())?
|
|
}
|
|
}
|
|
|
|
/// Parse a REVOKE statement
|
|
pub fn parse_revoke(&mut self) -> Result<Statement, ParserError> {
|
|
let (privileges, objects) = self.parse_grant_revoke_privileges_objects()?;
|
|
|
|
self.expect_keyword(Keyword::FROM)?;
|
|
let grantees = self.parse_comma_separated(Parser::parse_identifier)?;
|
|
|
|
let granted_by = self
|
|
.parse_keywords(&[Keyword::GRANTED, Keyword::BY])
|
|
.then(|| self.parse_identifier().unwrap());
|
|
|
|
let cascade = self.parse_keyword(Keyword::CASCADE);
|
|
let restrict = self.parse_keyword(Keyword::RESTRICT);
|
|
if cascade && restrict {
|
|
return parser_err!("Cannot specify both CASCADE and RESTRICT in REVOKE");
|
|
}
|
|
|
|
Ok(Statement::Revoke {
|
|
privileges,
|
|
objects,
|
|
grantees,
|
|
granted_by,
|
|
cascade,
|
|
})
|
|
}
|
|
|
|
/// Parse an INSERT statement
|
|
pub fn parse_insert(&mut self) -> Result<Statement, ParserError> {
|
|
let or = if !dialect_of!(self is SQLiteDialect) {
|
|
None
|
|
} else if self.parse_keywords(&[Keyword::OR, Keyword::REPLACE]) {
|
|
Some(SqliteOnConflict::Replace)
|
|
} else if self.parse_keywords(&[Keyword::OR, Keyword::ROLLBACK]) {
|
|
Some(SqliteOnConflict::Rollback)
|
|
} else if self.parse_keywords(&[Keyword::OR, Keyword::ABORT]) {
|
|
Some(SqliteOnConflict::Abort)
|
|
} else if self.parse_keywords(&[Keyword::OR, Keyword::FAIL]) {
|
|
Some(SqliteOnConflict::Fail)
|
|
} else if self.parse_keywords(&[Keyword::OR, Keyword::IGNORE]) {
|
|
Some(SqliteOnConflict::Ignore)
|
|
} else if self.parse_keyword(Keyword::REPLACE) {
|
|
Some(SqliteOnConflict::Replace)
|
|
} else {
|
|
None
|
|
};
|
|
|
|
let action = self.parse_one_of_keywords(&[Keyword::INTO, Keyword::OVERWRITE]);
|
|
let into = action == Some(Keyword::INTO);
|
|
let overwrite = action == Some(Keyword::OVERWRITE);
|
|
|
|
let local = self.parse_keyword(Keyword::LOCAL);
|
|
|
|
if self.parse_keyword(Keyword::DIRECTORY) {
|
|
let path = self.parse_literal_string()?;
|
|
let file_format = if self.parse_keywords(&[Keyword::STORED, Keyword::AS]) {
|
|
Some(self.parse_file_format()?)
|
|
} else {
|
|
None
|
|
};
|
|
let source = Box::new(self.parse_query()?);
|
|
Ok(Statement::Directory {
|
|
local,
|
|
path,
|
|
overwrite,
|
|
file_format,
|
|
source,
|
|
})
|
|
} else {
|
|
// Hive lets you put table here regardless
|
|
let table = self.parse_keyword(Keyword::TABLE);
|
|
let table_name = self.parse_object_name()?;
|
|
let columns = self.parse_parenthesized_column_list(Optional)?;
|
|
|
|
let partitioned = if self.parse_keyword(Keyword::PARTITION) {
|
|
self.expect_token(&Token::LParen)?;
|
|
let r = Some(self.parse_comma_separated(Parser::parse_expr)?);
|
|
self.expect_token(&Token::RParen)?;
|
|
r
|
|
} else {
|
|
None
|
|
};
|
|
|
|
// Hive allows you to specify columns after partitions as well if you want.
|
|
let after_columns = self.parse_parenthesized_column_list(Optional)?;
|
|
|
|
let source = Box::new(self.parse_query()?);
|
|
let on = if self.parse_keyword(Keyword::ON) {
|
|
self.expect_keyword(Keyword::DUPLICATE)?;
|
|
self.expect_keyword(Keyword::KEY)?;
|
|
self.expect_keyword(Keyword::UPDATE)?;
|
|
let l = self.parse_comma_separated(Parser::parse_assignment)?;
|
|
|
|
Some(OnInsert::DuplicateKeyUpdate(l))
|
|
} else {
|
|
None
|
|
};
|
|
|
|
Ok(Statement::Insert {
|
|
or,
|
|
table_name,
|
|
into,
|
|
overwrite,
|
|
partitioned,
|
|
columns,
|
|
after_columns,
|
|
source,
|
|
table,
|
|
on,
|
|
})
|
|
}
|
|
}
|
|
|
|
pub fn parse_update(&mut self) -> Result<Statement, ParserError> {
|
|
let table = self.parse_table_and_joins()?;
|
|
self.expect_keyword(Keyword::SET)?;
|
|
let assignments = self.parse_comma_separated(Parser::parse_assignment)?;
|
|
let from = if self.parse_keyword(Keyword::FROM) && dialect_of!(self is PostgreSqlDialect) {
|
|
Some(self.parse_table_and_joins()?)
|
|
} else {
|
|
None
|
|
};
|
|
let selection = if self.parse_keyword(Keyword::WHERE) {
|
|
Some(self.parse_expr()?)
|
|
} else {
|
|
None
|
|
};
|
|
Ok(Statement::Update {
|
|
table,
|
|
assignments,
|
|
from,
|
|
selection,
|
|
})
|
|
}
|
|
|
|
/// Parse a `var = expr` assignment, used in an UPDATE statement
|
|
pub fn parse_assignment(&mut self) -> Result<Assignment, ParserError> {
|
|
let id = self.parse_identifiers_non_keywords()?;
|
|
self.expect_token(&Token::Eq)?;
|
|
let value = self.parse_expr()?;
|
|
Ok(Assignment { id, value })
|
|
}
|
|
|
|
pub fn parse_function_args(&mut self) -> Result<FunctionArg, ParserError> {
|
|
if self.peek_nth_token(1) == Token::RArrow {
|
|
let name = self.parse_identifier()?;
|
|
|
|
self.expect_token(&Token::RArrow)?;
|
|
let arg = self.parse_wildcard_expr()?.into();
|
|
|
|
Ok(FunctionArg::Named { name, arg })
|
|
} else {
|
|
Ok(FunctionArg::Unnamed(self.parse_wildcard_expr()?.into()))
|
|
}
|
|
}
|
|
|
|
pub fn parse_optional_args(&mut self) -> Result<Vec<FunctionArg>, ParserError> {
|
|
if self.consume_token(&Token::RParen) {
|
|
Ok(vec![])
|
|
} else {
|
|
let args = self.parse_comma_separated(Parser::parse_function_args)?;
|
|
self.expect_token(&Token::RParen)?;
|
|
Ok(args)
|
|
}
|
|
}
|
|
|
|
/// Parse a comma-delimited list of projections after SELECT
|
|
pub fn parse_select_item(&mut self) -> Result<SelectItem, ParserError> {
|
|
match self.parse_wildcard_expr()? {
|
|
WildcardExpr::Expr(expr) => self
|
|
.parse_optional_alias(keywords::RESERVED_FOR_COLUMN_ALIAS)
|
|
.map(|alias| match alias {
|
|
Some(alias) => SelectItem::ExprWithAlias { expr, alias },
|
|
None => SelectItem::UnnamedExpr(expr),
|
|
}),
|
|
WildcardExpr::QualifiedWildcard(prefix) => Ok(SelectItem::QualifiedWildcard(prefix)),
|
|
WildcardExpr::Wildcard => Ok(SelectItem::Wildcard),
|
|
}
|
|
}
|
|
|
|
/// Parse an expression, optionally followed by ASC or DESC (used in ORDER BY)
|
|
pub fn parse_order_by_expr(&mut self) -> Result<OrderByExpr, ParserError> {
|
|
let expr = self.parse_expr()?;
|
|
|
|
let asc = if self.parse_keyword(Keyword::ASC) {
|
|
Some(true)
|
|
} else if self.parse_keyword(Keyword::DESC) {
|
|
Some(false)
|
|
} else {
|
|
None
|
|
};
|
|
|
|
let nulls_first = if self.parse_keywords(&[Keyword::NULLS, Keyword::FIRST]) {
|
|
Some(true)
|
|
} else if self.parse_keywords(&[Keyword::NULLS, Keyword::LAST]) {
|
|
Some(false)
|
|
} else {
|
|
None
|
|
};
|
|
|
|
Ok(OrderByExpr {
|
|
expr,
|
|
asc,
|
|
nulls_first,
|
|
})
|
|
}
|
|
|
|
/// Parse a TOP clause, MSSQL equivalent of LIMIT,
|
|
/// that follows after SELECT [DISTINCT].
|
|
pub fn parse_top(&mut self) -> Result<Top, ParserError> {
|
|
let quantity = if self.consume_token(&Token::LParen) {
|
|
let quantity = self.parse_expr()?;
|
|
self.expect_token(&Token::RParen)?;
|
|
Some(quantity)
|
|
} else {
|
|
Some(Expr::Value(self.parse_number_value()?))
|
|
};
|
|
|
|
let percent = self.parse_keyword(Keyword::PERCENT);
|
|
|
|
let with_ties = self.parse_keywords(&[Keyword::WITH, Keyword::TIES]);
|
|
|
|
Ok(Top {
|
|
with_ties,
|
|
percent,
|
|
quantity,
|
|
})
|
|
}
|
|
|
|
/// Parse a LIMIT clause
|
|
pub fn parse_limit(&mut self) -> Result<Option<Expr>, ParserError> {
|
|
if self.parse_keyword(Keyword::ALL) {
|
|
Ok(None)
|
|
} else {
|
|
Ok(Some(self.parse_expr()?))
|
|
}
|
|
}
|
|
|
|
/// Parse an OFFSET clause
|
|
pub fn parse_offset(&mut self) -> Result<Offset, ParserError> {
|
|
let value = self.parse_expr()?;
|
|
let rows = if self.parse_keyword(Keyword::ROW) {
|
|
OffsetRows::Row
|
|
} else if self.parse_keyword(Keyword::ROWS) {
|
|
OffsetRows::Rows
|
|
} else {
|
|
OffsetRows::None
|
|
};
|
|
Ok(Offset { value, rows })
|
|
}
|
|
|
|
/// Parse a FETCH clause
|
|
pub fn parse_fetch(&mut self) -> Result<Fetch, ParserError> {
|
|
self.expect_one_of_keywords(&[Keyword::FIRST, Keyword::NEXT])?;
|
|
let (quantity, percent) = if self
|
|
.parse_one_of_keywords(&[Keyword::ROW, Keyword::ROWS])
|
|
.is_some()
|
|
{
|
|
(None, false)
|
|
} else {
|
|
let quantity = Expr::Value(self.parse_value()?);
|
|
let percent = self.parse_keyword(Keyword::PERCENT);
|
|
self.expect_one_of_keywords(&[Keyword::ROW, Keyword::ROWS])?;
|
|
(Some(quantity), percent)
|
|
};
|
|
let with_ties = if self.parse_keyword(Keyword::ONLY) {
|
|
false
|
|
} else if self.parse_keywords(&[Keyword::WITH, Keyword::TIES]) {
|
|
true
|
|
} else {
|
|
return self.expected("one of ONLY or WITH TIES", self.peek_token());
|
|
};
|
|
Ok(Fetch {
|
|
with_ties,
|
|
percent,
|
|
quantity,
|
|
})
|
|
}
|
|
|
|
/// Parse a FOR UPDATE/FOR SHARE clause
|
|
pub fn parse_lock(&mut self) -> Result<LockType, ParserError> {
|
|
match self.expect_one_of_keywords(&[Keyword::UPDATE, Keyword::SHARE])? {
|
|
Keyword::UPDATE => Ok(LockType::Update),
|
|
Keyword::SHARE => Ok(LockType::Share),
|
|
_ => unreachable!(),
|
|
}
|
|
}
|
|
|
|
pub fn parse_values(&mut self) -> Result<Values, ParserError> {
|
|
let values = self.parse_comma_separated(|parser| {
|
|
parser.expect_token(&Token::LParen)?;
|
|
let exprs = parser.parse_comma_separated(Parser::parse_expr)?;
|
|
parser.expect_token(&Token::RParen)?;
|
|
Ok(exprs)
|
|
})?;
|
|
Ok(Values(values))
|
|
}
|
|
|
|
pub fn parse_start_transaction(&mut self) -> Result<Statement, ParserError> {
|
|
self.expect_keyword(Keyword::TRANSACTION)?;
|
|
Ok(Statement::StartTransaction {
|
|
modes: self.parse_transaction_modes()?,
|
|
})
|
|
}
|
|
|
|
pub fn parse_begin(&mut self) -> Result<Statement, ParserError> {
|
|
let _ = self.parse_one_of_keywords(&[Keyword::TRANSACTION, Keyword::WORK]);
|
|
Ok(Statement::StartTransaction {
|
|
modes: self.parse_transaction_modes()?,
|
|
})
|
|
}
|
|
|
|
pub fn parse_transaction_modes(&mut self) -> Result<Vec<TransactionMode>, ParserError> {
|
|
let mut modes = vec![];
|
|
let mut required = false;
|
|
loop {
|
|
let mode = if self.parse_keywords(&[Keyword::ISOLATION, Keyword::LEVEL]) {
|
|
let iso_level = if self.parse_keywords(&[Keyword::READ, Keyword::UNCOMMITTED]) {
|
|
TransactionIsolationLevel::ReadUncommitted
|
|
} else if self.parse_keywords(&[Keyword::READ, Keyword::COMMITTED]) {
|
|
TransactionIsolationLevel::ReadCommitted
|
|
} else if self.parse_keywords(&[Keyword::REPEATABLE, Keyword::READ]) {
|
|
TransactionIsolationLevel::RepeatableRead
|
|
} else if self.parse_keyword(Keyword::SERIALIZABLE) {
|
|
TransactionIsolationLevel::Serializable
|
|
} else {
|
|
self.expected("isolation level", self.peek_token())?
|
|
};
|
|
TransactionMode::IsolationLevel(iso_level)
|
|
} else if self.parse_keywords(&[Keyword::READ, Keyword::ONLY]) {
|
|
TransactionMode::AccessMode(TransactionAccessMode::ReadOnly)
|
|
} else if self.parse_keywords(&[Keyword::READ, Keyword::WRITE]) {
|
|
TransactionMode::AccessMode(TransactionAccessMode::ReadWrite)
|
|
} else if required {
|
|
self.expected("transaction mode", self.peek_token())?
|
|
} else {
|
|
break;
|
|
};
|
|
modes.push(mode);
|
|
// ANSI requires a comma after each transaction mode, but
|
|
// PostgreSQL, for historical reasons, does not. We follow
|
|
// PostgreSQL in making the comma optional, since that is strictly
|
|
// more general.
|
|
required = self.consume_token(&Token::Comma);
|
|
}
|
|
Ok(modes)
|
|
}
|
|
|
|
pub fn parse_commit(&mut self) -> Result<Statement, ParserError> {
|
|
Ok(Statement::Commit {
|
|
chain: self.parse_commit_rollback_chain()?,
|
|
})
|
|
}
|
|
|
|
pub fn parse_rollback(&mut self) -> Result<Statement, ParserError> {
|
|
Ok(Statement::Rollback {
|
|
chain: self.parse_commit_rollback_chain()?,
|
|
})
|
|
}
|
|
|
|
pub fn parse_commit_rollback_chain(&mut self) -> Result<bool, ParserError> {
|
|
let _ = self.parse_one_of_keywords(&[Keyword::TRANSACTION, Keyword::WORK]);
|
|
if self.parse_keyword(Keyword::AND) {
|
|
let chain = !self.parse_keyword(Keyword::NO);
|
|
self.expect_keyword(Keyword::CHAIN)?;
|
|
Ok(chain)
|
|
} else {
|
|
Ok(false)
|
|
}
|
|
}
|
|
|
|
pub fn parse_deallocate(&mut self) -> Result<Statement, ParserError> {
|
|
let prepare = self.parse_keyword(Keyword::PREPARE);
|
|
let name = self.parse_identifier()?;
|
|
Ok(Statement::Deallocate { name, prepare })
|
|
}
|
|
|
|
pub fn parse_execute(&mut self) -> Result<Statement, ParserError> {
|
|
let name = self.parse_identifier()?;
|
|
|
|
let mut parameters = vec![];
|
|
if self.consume_token(&Token::LParen) {
|
|
parameters = self.parse_comma_separated(Parser::parse_expr)?;
|
|
self.expect_token(&Token::RParen)?;
|
|
}
|
|
|
|
Ok(Statement::Execute { name, parameters })
|
|
}
|
|
|
|
pub fn parse_prepare(&mut self) -> Result<Statement, ParserError> {
|
|
let name = self.parse_identifier()?;
|
|
|
|
let mut data_types = vec![];
|
|
if self.consume_token(&Token::LParen) {
|
|
data_types = self.parse_comma_separated(Parser::parse_data_type)?;
|
|
self.expect_token(&Token::RParen)?;
|
|
}
|
|
|
|
self.expect_keyword(Keyword::AS)?;
|
|
let statement = Box::new(self.parse_statement()?);
|
|
Ok(Statement::Prepare {
|
|
name,
|
|
data_types,
|
|
statement,
|
|
})
|
|
}
|
|
|
|
pub fn parse_comment(&mut self) -> Result<Statement, ParserError> {
|
|
self.expect_keyword(Keyword::ON)?;
|
|
let token = self.next_token();
|
|
|
|
let (object_type, object_name) = match token {
|
|
Token::Word(w) if w.keyword == Keyword::COLUMN => {
|
|
let object_name = self.parse_object_name()?;
|
|
(CommentObject::Column, object_name)
|
|
}
|
|
Token::Word(w) if w.keyword == Keyword::TABLE => {
|
|
let object_name = self.parse_object_name()?;
|
|
(CommentObject::Table, object_name)
|
|
}
|
|
_ => self.expected("comment object_type", token)?,
|
|
};
|
|
|
|
self.expect_keyword(Keyword::IS)?;
|
|
let comment = if self.parse_keyword(Keyword::NULL) {
|
|
None
|
|
} else {
|
|
Some(self.parse_literal_string()?)
|
|
};
|
|
Ok(Statement::Comment {
|
|
object_type,
|
|
object_name,
|
|
comment,
|
|
})
|
|
}
|
|
|
|
pub fn parse_merge_clauses(&mut self) -> Result<Vec<MergeClause>, ParserError> {
|
|
let mut clauses: Vec<MergeClause> = vec![];
|
|
loop {
|
|
if self.peek_token() == Token::EOF || self.peek_token() == Token::SemiColon {
|
|
break;
|
|
}
|
|
self.expect_keyword(Keyword::WHEN)?;
|
|
|
|
let is_not_matched = self.parse_keyword(Keyword::NOT);
|
|
self.expect_keyword(Keyword::MATCHED)?;
|
|
|
|
let predicate = if self.parse_keyword(Keyword::AND) {
|
|
Some(self.parse_expr()?)
|
|
} else {
|
|
None
|
|
};
|
|
|
|
self.expect_keyword(Keyword::THEN)?;
|
|
|
|
clauses.push(
|
|
match self.parse_one_of_keywords(&[
|
|
Keyword::UPDATE,
|
|
Keyword::INSERT,
|
|
Keyword::DELETE,
|
|
]) {
|
|
Some(Keyword::UPDATE) => {
|
|
if is_not_matched {
|
|
return Err(ParserError::ParserError(
|
|
"UPDATE in NOT MATCHED merge clause".to_string(),
|
|
));
|
|
}
|
|
self.expect_keyword(Keyword::SET)?;
|
|
let assignments = self.parse_comma_separated(Parser::parse_assignment)?;
|
|
MergeClause::MatchedUpdate {
|
|
predicate,
|
|
assignments,
|
|
}
|
|
}
|
|
Some(Keyword::DELETE) => {
|
|
if is_not_matched {
|
|
return Err(ParserError::ParserError(
|
|
"DELETE in NOT MATCHED merge clause".to_string(),
|
|
));
|
|
}
|
|
MergeClause::MatchedDelete(predicate)
|
|
}
|
|
Some(Keyword::INSERT) => {
|
|
if !is_not_matched {
|
|
return Err(ParserError::ParserError(
|
|
"INSERT in MATCHED merge clause".to_string(),
|
|
));
|
|
}
|
|
let columns = self.parse_parenthesized_column_list(Optional)?;
|
|
self.expect_keyword(Keyword::VALUES)?;
|
|
let values = self.parse_values()?;
|
|
MergeClause::NotMatched {
|
|
predicate,
|
|
columns,
|
|
values,
|
|
}
|
|
}
|
|
Some(_) => {
|
|
return Err(ParserError::ParserError(
|
|
"expected UPDATE, DELETE or INSERT in merge clause".to_string(),
|
|
))
|
|
}
|
|
None => {
|
|
return Err(ParserError::ParserError(
|
|
"expected UPDATE, DELETE or INSERT in merge clause".to_string(),
|
|
))
|
|
}
|
|
},
|
|
);
|
|
}
|
|
Ok(clauses)
|
|
}
|
|
|
|
pub fn parse_merge(&mut self) -> Result<Statement, ParserError> {
|
|
let into = self.parse_keyword(Keyword::INTO);
|
|
|
|
let table = self.parse_table_factor()?;
|
|
|
|
self.expect_keyword(Keyword::USING)?;
|
|
let source = self.parse_table_factor()?;
|
|
self.expect_keyword(Keyword::ON)?;
|
|
let on = self.parse_expr()?;
|
|
let clauses = self.parse_merge_clauses()?;
|
|
|
|
Ok(Statement::Merge {
|
|
into,
|
|
table,
|
|
source,
|
|
on: Box::new(on),
|
|
clauses,
|
|
})
|
|
}
|
|
}
|
|
|
|
impl Word {
|
|
pub fn to_ident(&self) -> Ident {
|
|
Ident {
|
|
value: self.value.clone(),
|
|
quote_style: self.quote_style,
|
|
}
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
use crate::test_utils::{all_dialects, TestedDialects};
|
|
|
|
#[test]
|
|
fn test_prev_index() {
|
|
let sql = "SELECT version";
|
|
all_dialects().run_parser_method(sql, |parser| {
|
|
assert_eq!(parser.peek_token(), Token::make_keyword("SELECT"));
|
|
assert_eq!(parser.next_token(), Token::make_keyword("SELECT"));
|
|
parser.prev_token();
|
|
assert_eq!(parser.next_token(), Token::make_keyword("SELECT"));
|
|
assert_eq!(parser.next_token(), Token::make_word("version", None));
|
|
parser.prev_token();
|
|
assert_eq!(parser.peek_token(), Token::make_word("version", None));
|
|
assert_eq!(parser.next_token(), Token::make_word("version", None));
|
|
assert_eq!(parser.peek_token(), Token::EOF);
|
|
parser.prev_token();
|
|
assert_eq!(parser.next_token(), Token::make_word("version", None));
|
|
assert_eq!(parser.next_token(), Token::EOF);
|
|
assert_eq!(parser.next_token(), Token::EOF);
|
|
parser.prev_token();
|
|
});
|
|
}
|
|
|
|
#[test]
|
|
fn test_parse_limit() {
|
|
let sql = "SELECT * FROM user LIMIT 1";
|
|
all_dialects().run_parser_method(sql, |parser| {
|
|
let ast = parser.parse_query().unwrap();
|
|
assert_eq!(ast.to_string(), sql.to_string());
|
|
});
|
|
|
|
let sql = "SELECT * FROM user LIMIT $1 OFFSET $2";
|
|
let dialects = TestedDialects {
|
|
dialects: vec![
|
|
Box::new(PostgreSqlDialect {}),
|
|
Box::new(ClickHouseDialect {}),
|
|
Box::new(GenericDialect {}),
|
|
Box::new(MsSqlDialect {}),
|
|
Box::new(SnowflakeDialect {}),
|
|
],
|
|
};
|
|
|
|
dialects.run_parser_method(sql, |parser| {
|
|
let ast = parser.parse_query().unwrap();
|
|
assert_eq!(ast.to_string(), sql.to_string());
|
|
});
|
|
|
|
let sql = "SELECT * FROM user LIMIT ? OFFSET ?";
|
|
let dialects = TestedDialects {
|
|
dialects: vec![Box::new(MySqlDialect {})],
|
|
};
|
|
dialects.run_parser_method(sql, |parser| {
|
|
let ast = parser.parse_query().unwrap();
|
|
assert_eq!(ast.to_string(), sql.to_string());
|
|
});
|
|
}
|
|
}
|