mirrors/datafusion-sqlparse
1
0
Fork 0
mirror of https://github.com/apache/datafusion-sqlparser-rs.git synced 2025-12-23 11:12:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

add support for with clauses (CTEs) in delete statements (#1764)

Browse source
This commit is contained in:
Ophir LOJKINE 2025-03-12 11:42:51 +01:00 • committed by GitHub
parent 1e54a34acd
commit 3392623b00
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 52 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

22
src/parser/mod.rs
View file

@ -10050,6 +10050,13 @@ impl<'a> Parser<'a> {
Ok(parent_type(inside_type.into()))
}
/// Parse a DELETE statement, returning a `Box`ed SetExpr
///
/// This is used to reduce the size of the stack frames in debug builds
fn parse_delete_setexpr_boxed(&mut self) -> Result<Box<SetExpr>, ParserError> {
Ok(Box::new(SetExpr::Delete(self.parse_delete()?)))
}
pub fn parse_delete(&mut self) -> Result<Statement, ParserError> {
let (tables, with_from_keyword) = if !self.parse_keyword(Keyword::FROM) {
// `FROM` keyword is optional in BigQuery SQL.
@ -10249,6 +10256,21 @@ impl<'a> Parser<'a> {
format_clause: None,
}
.into())
} else if self.parse_keyword(Keyword::DELETE) {
Ok(Query {
with,
body: self.parse_delete_setexpr_boxed()?,
limit: None,
limit_by: vec![],
order_by: None,
offset: None,
fetch: None,
locks: vec![],
for_clause: None,
settings: None,
format_clause: None,
}
.into())
} else {
let body = self.parse_query_body(self.dialect.prec_unknown())?;