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

Fix limit in subqueries (#1899)
Some checks failed
license / Release Audit Tool (RAT) (push) Has been cancelled
Details
Rust / codestyle (push) Has been cancelled
Details
Rust / lint (push) Has been cancelled
Details
Rust / benchmark-lint (push) Has been cancelled
Details
Rust / compile (push) Has been cancelled
Details
Rust / docs (push) Has been cancelled
Details
Rust / compile-no-std (push) Has been cancelled
Details
Rust / test (beta) (push) Has been cancelled
Details
Rust / test (nightly) (push) Has been cancelled
Details
Rust / test (stable) (push) Has been cancelled
Details

Browse source
This commit is contained in:
Dima 2025-06-22 08:22:45 +01:00 • committed by GitHub
parent 1d0dc7cdd8
commit 7865de015f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 11 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/ast/mod.rs
View file

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

2
src/parser/mod.rs
View file

@ -3817,7 +3817,7 @@ impl<'a> Parser<'a> {
}); });
} }
self.expect_token(&Token::LParen)?; self.expect_token(&Token::LParen)?;
let in_op = match self.maybe_parse(|p| p.parse_query_body(p.dialect.prec_unknown()))? { let in_op = match self.maybe_parse(|p| p.parse_query())? {
Some(subquery) => Expr::InSubquery { Some(subquery) => Expr::InSubquery {
expr: Box::new(expr), expr: Box::new(expr),
subquery, subquery,

11
tests/sqlparser_common.rs
View file

@ -2225,7 +2225,7 @@ 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: verified_query("SELECT segm FROM bar").body, subquery: Box::new(verified_query("SELECT segm FROM bar")),
negated: false, negated: false,
}, },
select.selection.unwrap() select.selection.unwrap()
@ -2239,7 +2239,9 @@ fn parse_in_union() {
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: verified_query("(SELECT segm FROM bar) UNION (SELECT segm FROM bar2)").body, subquery: Box::new(verified_query(
"(SELECT segm FROM bar) UNION (SELECT segm FROM bar2)"
)),
negated: false, negated: false,
}, },
select.selection.unwrap() select.selection.unwrap()
@ -15303,6 +15305,11 @@ fn parse_return() {
let _ = all_dialects().verified_stmt("RETURN 1"); let _ = all_dialects().verified_stmt("RETURN 1");
} }
#[test]
fn parse_subquery_limit() {
let _ = all_dialects().verified_stmt("SELECT t1_id, t1_name FROM t1 WHERE t1_id IN (SELECT t2_id FROM t2 WHERE t1_name = t2_name LIMIT 10)");
}
#[test] #[test]
fn test_open() { fn test_open() {
let open_cursor = "OPEN Employee_Cursor"; let open_cursor = "OPEN Employee_Cursor";