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

Merge 60acdc5bb6 into 93450cc250

Browse source
This commit is contained in:
Sergey Olontsev 2025-07-07 08:24:34 -07:00 • committed by GitHub
commit c8acafd68c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 51 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

10
src/parser/mod.rs
View file

@ -3478,10 +3478,18 @@ impl<'a> Parser<'a> {
| BinaryOperator::LtEq | BinaryOperator::LtEq
| BinaryOperator::Eq | BinaryOperator::Eq
| BinaryOperator::NotEq | BinaryOperator::NotEq
| BinaryOperator::PGRegexMatch
| BinaryOperator::PGRegexIMatch
| BinaryOperator::PGRegexNotMatch
| BinaryOperator::PGRegexNotIMatch
| BinaryOperator::PGLikeMatch
| BinaryOperator::PGILikeMatch
| BinaryOperator::PGNotLikeMatch
| BinaryOperator::PGNotILikeMatch
) { ) {
return parser_err!( return parser_err!(
format!( format!(
"Expected one of [=, >, <, =>, =<, !=] as comparison operator, found: {op}" "Expected one of [=, >, <, =>, =<, !=, ~, ~*, !~, !~*, ~~, ~~*, !~~, !~~*] as comparison operator, found: {op}"
), ),
span.start span.start
); );

56
tests/sqlparser_postgres.rs
View file

@ -2181,19 +2181,35 @@ fn parse_pg_regex_match_ops() {
]; ];
for (str_op, op) in pg_regex_match_ops { for (str_op, op) in pg_regex_match_ops {
let select = pg().verified_only_select(&format!("SELECT 'abc' {} '^a'", &str_op)); // Basic binary operator usage
let select = pg().verified_only_select(&format!("SELECT 'abc' {str_op} '^a'"));
assert_eq!( assert_eq!(
SelectItem::UnnamedExpr(Expr::BinaryOp { SelectItem::UnnamedExpr(Expr::BinaryOp {
left: Box::new(Expr::Value( left: Box::new(Expr::Value(single_quoted_string("abc").with_empty_span(),)),
(Value::SingleQuotedString("abc".into())).with_empty_span()
)),
op: op.clone(), op: op.clone(),
right: Box::new(Expr::Value( right: Box::new(Expr::Value(single_quoted_string("^a").with_empty_span(),)),
(Value::SingleQuotedString("^a".into())).with_empty_span()
)),
}), }),
select.projection[0] select.projection[0]
); );
// Binary operator with ANY operator
let select =
pg().verified_only_select(&format!("SELECT 'abc' {str_op} ANY(ARRAY['^a', 'x'])"));
assert_eq!(
SelectItem::UnnamedExpr(Expr::AnyOp {
left: Box::new(Expr::Value(single_quoted_string("abc").with_empty_span(),)),
compare_op: op.clone(),
right: Box::new(Expr::Array(Array {
elem: vec![
Expr::Value(single_quoted_string("^a").with_empty_span()),
Expr::Value(single_quoted_string("x").with_empty_span()),
],
named: true,
})),
is_some: false,
}),
select.projection[0]
)
} }
} }
@ -2207,19 +2223,31 @@ fn parse_pg_like_match_ops() {
]; ];
for (str_op, op) in pg_like_match_ops { for (str_op, op) in pg_like_match_ops {
let select = pg().verified_only_select(&format!("SELECT 'abc' {} 'a_c%'", &str_op)); // Basic binary operator usage
let select = pg().verified_only_select(&format!("SELECT 'abc' {str_op} 'a_c%'"));
assert_eq!( assert_eq!(
SelectItem::UnnamedExpr(Expr::BinaryOp { SelectItem::UnnamedExpr(Expr::BinaryOp {
left: Box::new(Expr::Value( left: Box::new(Expr::Value(single_quoted_string("abc").with_empty_span(),)),
(Value::SingleQuotedString("abc".into())).with_empty_span()
)),
op: op.clone(), op: op.clone(),
right: Box::new(Expr::Value( right: Box::new(Expr::Value(single_quoted_string("a_c%").with_empty_span(),)),
(Value::SingleQuotedString("a_c%".into())).with_empty_span()
)),
}), }),
select.projection[0] select.projection[0]
); );
// Binary operator with ALL operator
let select =
pg().verified_only_select(&format!("SELECT 'abc' {str_op} ALL(ARRAY['a_c%'])"));
assert_eq!(
SelectItem::UnnamedExpr(Expr::AllOp {
left: Box::new(Expr::Value(single_quoted_string("abc").with_empty_span(),)),
compare_op: op.clone(),
right: Box::new(Expr::Array(Array {
elem: vec![Expr::Value(single_quoted_string("a_c%").with_empty_span())],
named: true,
})),
}),
select.projection[0]
)
} }
} }