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

Add support for parenthesized subquery as IN predicate (#1793)

Browse source
This commit is contained in:
Adam Johnson 2025-04-15 18:01:18 +01:00 • committed by GitHub
parent 514d2ecdaf
commit 3ad13af563
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 22 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/ast/mod.rs
View file

@ -728,7 +728,7 @@ pub enum Expr {
/// `[ NOT ] IN (SELECT ...)` /// `[ NOT ] IN (SELECT ...)`
InSubquery { InSubquery {
expr: Box<Expr>, expr: Box<Expr>,
subquery: Box<Query>, subquery: Box<SetExpr>,
negated: bool, negated: bool,
}, },
/// `[ NOT ] IN UNNEST(array_expression)` /// `[ NOT ] IN UNNEST(array_expression)`

14
src/parser/mod.rs
View file

@ -3756,15 +3756,13 @@ impl<'a> Parser<'a> {
}); });
} }
self.expect_token(&Token::LParen)?; self.expect_token(&Token::LParen)?;
let in_op = if self.parse_keyword(Keyword::SELECT) || self.parse_keyword(Keyword::WITH) { let in_op = match self.maybe_parse(|p| p.parse_query_body(p.dialect.prec_unknown()))? {
self.prev_token(); Some(subquery) => Expr::InSubquery {
Expr::InSubquery {
expr: Box::new(expr), expr: Box::new(expr),
subquery: self.parse_query()?, subquery,
negated, negated,
} },
} else { None => Expr::InList {
Expr::InList {
expr: Box::new(expr), expr: Box::new(expr),
list: if self.dialect.supports_in_empty_list() { list: if self.dialect.supports_in_empty_list() {
self.parse_comma_separated0(Parser::parse_expr, Token::RParen)? self.parse_comma_separated0(Parser::parse_expr, Token::RParen)?
@ -3772,7 +3770,7 @@ impl<'a> Parser<'a> {
self.parse_comma_separated(Parser::parse_expr)? self.parse_comma_separated(Parser::parse_expr)?
}, },
negated, negated,
} },
}; };
self.expect_token(&Token::RParen)?; self.expect_token(&Token::RParen)?;
Ok(in_op) Ok(in_op)

16
tests/sqlparser_common.rs
View file

@ -2224,7 +2224,21 @@ fn parse_in_subquery() {
assert_eq!( assert_eq!(
Expr::InSubquery { Expr::InSubquery {
expr: Box::new(Expr::Identifier(Ident::new("segment"))), expr: Box::new(Expr::Identifier(Ident::new("segment"))),
subquery: Box::new(verified_query("SELECT segm FROM bar")), subquery: verified_query("SELECT segm FROM bar").body,
negated: false,
},
select.selection.unwrap()
);
}
#[test]
fn parse_in_union() {
let sql = "SELECT * FROM customers WHERE segment IN ((SELECT segm FROM bar) UNION (SELECT segm FROM bar2))";
let select = verified_only_select(sql);
assert_eq!(
Expr::InSubquery {
expr: Box::new(Expr::Identifier(Ident::new("segment"))),
subquery: verified_query("(SELECT segm FROM bar) UNION (SELECT segm FROM bar2)").body,
negated: false, negated: false,
}, },
select.selection.unwrap() select.selection.unwrap()