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

Support multiple-table DELETE syntax (#855)

Browse source
This commit is contained in:
AviRaboah 2023-04-27 18:41:20 +03:00 • committed by GitHub
parent 5ecf633e31
commit f72e2ec382
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 131 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

19
src/ast/mod.rs
View file

@ -1229,10 +1229,12 @@ pub enum Statement {
},
/// DELETE
Delete {
/// Multi tables delete are supported in mysql
tables: Vec<ObjectName>,
/// FROM
table_name: TableFactor,
/// USING (Snowflake, Postgres)
using: Option<TableFactor>,
from: Vec<TableWithJoins>,
/// USING (Snowflake, Postgres, MySQL)
using: Option<Vec<TableWithJoins>>,
/// WHERE
selection: Option<Expr>,
/// RETURNING
@ -1982,14 +1984,19 @@ impl fmt::Display for Statement {
Ok(())
}
Statement::Delete {
table_name,
tables,
from,
using,
selection,
returning,
} => {
write!(f, "DELETE FROM {table_name}")?;
write!(f, "DELETE ")?;
if !tables.is_empty() {
write!(f, "{} ", display_comma_separated(tables))?;
}
write!(f, "FROM {}", display_comma_separated(from))?;
if let Some(using) = using {
write!(f, " USING {using}")?;
write!(f, " USING {}", display_comma_separated(using))?;
}
if let Some(selection) = selection {
write!(f, " WHERE {selection}")?;

16
src/parser.rs
View file

@ -4813,10 +4813,17 @@ impl<'a> Parser<'a> {
}
pub fn parse_delete(&mut self) -> Result<Statement, ParserError> {
self.expect_keyword(Keyword::FROM)?;
let table_name = self.parse_table_factor()?;
let tables = if !self.parse_keyword(Keyword::FROM) {
let tables = self.parse_comma_separated(Parser::parse_object_name)?;
self.expect_keyword(Keyword::FROM)?;
tables
} else {
vec![]
};
let from = self.parse_comma_separated(Parser::parse_table_and_joins)?;
let using = if self.parse_keyword(Keyword::USING) {
Some(self.parse_table_factor()?)
Some(self.parse_comma_separated(Parser::parse_table_and_joins)?)
} else {
None
};
@ -4833,7 +4840,8 @@ impl<'a> Parser<'a> {
};
Ok(Statement::Delete {
table_name,
tables,
from,
using,
selection,
returning,