mirror of
https://github.com/apache/datafusion-sqlparser-rs.git
synced 2025-08-04 14:28:22 +00:00
BigQuery: Add support for select expr star (#1680)
This commit is contained in:
parent
f7b0812b01
commit
8de3a62948
10 changed files with 183 additions and 30 deletions
|
@ -1540,9 +1540,6 @@ fn parse_hyphenated_table_identifiers() {
|
|||
]))
|
||||
})
|
||||
);
|
||||
|
||||
let error_sql = "select foo-bar.* from foo-bar";
|
||||
assert!(bigquery().parse_sql_statements(error_sql).is_err());
|
||||
}
|
||||
|
||||
#[test]
|
||||
|
@ -2204,6 +2201,14 @@ fn parse_extract_weekday() {
|
|||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn bigquery_select_expr_star() {
|
||||
bigquery()
|
||||
.verified_only_select("SELECT STRUCT<STRING>((SELECT foo FROM T WHERE true)).* FROM T");
|
||||
bigquery().verified_only_select("SELECT [STRUCT<STRING>('foo')][0].* EXCEPT (foo) FROM T");
|
||||
bigquery().verified_only_select("SELECT myfunc()[0].* FROM T");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_select_as_struct() {
|
||||
bigquery().verified_only_select("SELECT * FROM (SELECT AS VALUE STRUCT(123 AS a, false AS b))");
|
||||
|
|
|
@ -1002,7 +1002,7 @@ fn parse_select_wildcard() {
|
|||
let select = verified_only_select(sql);
|
||||
assert_eq!(
|
||||
&SelectItem::QualifiedWildcard(
|
||||
ObjectName::from(vec![Ident::new("foo")]),
|
||||
SelectItemQualifiedWildcardKind::ObjectName(ObjectName::from(vec![Ident::new("foo")])),
|
||||
WildcardAdditionalOptions::default()
|
||||
),
|
||||
only(&select.projection)
|
||||
|
@ -1012,7 +1012,10 @@ fn parse_select_wildcard() {
|
|||
let select = verified_only_select(sql);
|
||||
assert_eq!(
|
||||
&SelectItem::QualifiedWildcard(
|
||||
ObjectName::from(vec![Ident::new("myschema"), Ident::new("mytable"),]),
|
||||
SelectItemQualifiedWildcardKind::ObjectName(ObjectName::from(vec![
|
||||
Ident::new("myschema"),
|
||||
Ident::new("mytable"),
|
||||
])),
|
||||
WildcardAdditionalOptions::default(),
|
||||
),
|
||||
only(&select.projection)
|
||||
|
@ -1057,6 +1060,84 @@ fn parse_column_aliases() {
|
|||
one_statement_parses_to("SELECT a.col + 1 newname FROM foo AS a", sql);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn parse_select_expr_star() {
|
||||
let dialects = all_dialects_where(|d| d.supports_select_expr_star());
|
||||
|
||||
// Identifier wildcard expansion.
|
||||
let select = dialects.verified_only_select("SELECT foo.bar.* FROM T");
|
||||
let SelectItem::QualifiedWildcard(SelectItemQualifiedWildcardKind::ObjectName(object_name), _) =
|
||||
only(&select.projection)
|
||||
else {
|
||||
unreachable!(
|
||||
"expected wildcard select item: got {:?}",
|
||||
&select.projection[0]
|
||||
)
|
||||
};
|
||||
assert_eq!(
|
||||
&ObjectName::from(
|
||||
["foo", "bar"]
|
||||
.into_iter()
|
||||
.map(Ident::new)
|
||||
.collect::<Vec<_>>()
|
||||
),
|
||||
object_name
|
||||
);
|
||||
|
||||
// Arbitrary compound expression with wildcard expansion.
|
||||
let select = dialects.verified_only_select("SELECT foo - bar.* FROM T");
|
||||
let SelectItem::QualifiedWildcard(
|
||||
SelectItemQualifiedWildcardKind::Expr(Expr::BinaryOp { left, op, right }),
|
||||
_,
|
||||
) = only(&select.projection)
|
||||
else {
|
||||
unreachable!(
|
||||
"expected wildcard select item: got {:?}",
|
||||
&select.projection[0]
|
||||
)
|
||||
};
|
||||
let (Expr::Identifier(left), BinaryOperator::Minus, Expr::Identifier(right)) =
|
||||
(left.as_ref(), op, right.as_ref())
|
||||
else {
|
||||
unreachable!("expected binary op expr: got {:?}", &select.projection[0])
|
||||
};
|
||||
assert_eq!(&Ident::new("foo"), left);
|
||||
assert_eq!(&Ident::new("bar"), right);
|
||||
|
||||
// Arbitrary expression wildcard expansion.
|
||||
let select = dialects.verified_only_select("SELECT myfunc().foo.* FROM T");
|
||||
let SelectItem::QualifiedWildcard(
|
||||
SelectItemQualifiedWildcardKind::Expr(Expr::CompoundFieldAccess { root, access_chain }),
|
||||
_,
|
||||
) = only(&select.projection)
|
||||
else {
|
||||
unreachable!("expected wildcard expr: got {:?}", &select.projection[0])
|
||||
};
|
||||
assert!(matches!(root.as_ref(), Expr::Function(_)));
|
||||
assert_eq!(1, access_chain.len());
|
||||
assert!(matches!(
|
||||
&access_chain[0],
|
||||
AccessExpr::Dot(Expr::Identifier(_))
|
||||
));
|
||||
|
||||
dialects.one_statement_parses_to(
|
||||
"SELECT 2. * 3 FROM T",
|
||||
#[cfg(feature = "bigdecimal")]
|
||||
"SELECT 2 * 3 FROM T",
|
||||
#[cfg(not(feature = "bigdecimal"))]
|
||||
"SELECT 2. * 3 FROM T",
|
||||
);
|
||||
dialects.verified_only_select("SELECT myfunc().* FROM T");
|
||||
dialects.verified_only_select("SELECT myfunc().* EXCEPT (foo) FROM T");
|
||||
|
||||
// Invalid
|
||||
let res = dialects.parse_sql_statements("SELECT foo.*.* FROM T");
|
||||
assert_eq!(
|
||||
ParserError::ParserError("Expected: end of statement, found: .".to_string()),
|
||||
res.unwrap_err()
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_eof_after_as() {
|
||||
let res = parse_sql_statements("SELECT foo AS");
|
||||
|
|
|
@ -160,7 +160,7 @@ fn test_select_wildcard_with_exclude() {
|
|||
let select =
|
||||
duckdb().verified_only_select("SELECT name.* EXCLUDE department_id FROM employee_table");
|
||||
let expected = SelectItem::QualifiedWildcard(
|
||||
ObjectName::from(vec![Ident::new("name")]),
|
||||
SelectItemQualifiedWildcardKind::ObjectName(ObjectName::from(vec![Ident::new("name")])),
|
||||
WildcardAdditionalOptions {
|
||||
opt_exclude: Some(ExcludeSelectItem::Single(Ident::new("department_id"))),
|
||||
..Default::default()
|
||||
|
|
|
@ -1368,7 +1368,7 @@ fn test_select_wildcard_with_exclude() {
|
|||
let select = snowflake_and_generic()
|
||||
.verified_only_select("SELECT name.* EXCLUDE department_id FROM employee_table");
|
||||
let expected = SelectItem::QualifiedWildcard(
|
||||
ObjectName::from(vec![Ident::new("name")]),
|
||||
SelectItemQualifiedWildcardKind::ObjectName(ObjectName::from(vec![Ident::new("name")])),
|
||||
WildcardAdditionalOptions {
|
||||
opt_exclude: Some(ExcludeSelectItem::Single(Ident::new("department_id"))),
|
||||
..Default::default()
|
||||
|
@ -1405,7 +1405,7 @@ fn test_select_wildcard_with_rename() {
|
|||
"SELECT name.* RENAME (department_id AS new_dep, employee_id AS new_emp) FROM employee_table",
|
||||
);
|
||||
let expected = SelectItem::QualifiedWildcard(
|
||||
ObjectName::from(vec![Ident::new("name")]),
|
||||
SelectItemQualifiedWildcardKind::ObjectName(ObjectName::from(vec![Ident::new("name")])),
|
||||
WildcardAdditionalOptions {
|
||||
opt_rename: Some(RenameSelectItem::Multiple(vec![
|
||||
IdentWithAlias {
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue