diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-01-31 05:44:41 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-01-31 05:44:41 +0000 |
commit | 376de8b6892deca7dc5d83035c047f1e13eb67ea (patch) | |
tree | 334a1753cd914294aa99128fac3fb59bf14dc10f /sqlglot/optimizer/unnest_subqueries.py | |
parent | Releasing debian version 20.9.0-1. (diff) | |
download | sqlglot-376de8b6892deca7dc5d83035c047f1e13eb67ea.tar.xz sqlglot-376de8b6892deca7dc5d83035c047f1e13eb67ea.zip |
Merging upstream version 20.11.0.
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'sqlglot/optimizer/unnest_subqueries.py')
-rw-r--r-- | sqlglot/optimizer/unnest_subqueries.py | 15 |
1 files changed, 11 insertions, 4 deletions
diff --git a/sqlglot/optimizer/unnest_subqueries.py b/sqlglot/optimizer/unnest_subqueries.py index 4d35175..26f4159 100644 --- a/sqlglot/optimizer/unnest_subqueries.py +++ b/sqlglot/optimizer/unnest_subqueries.py @@ -50,11 +50,12 @@ def unnest(select, parent_select, next_alias_name): ): return + clause = predicate.find_ancestor(exp.Having, exp.Where, exp.Join) + # This subquery returns a scalar and can just be converted to a cross join if not isinstance(predicate, (exp.In, exp.Any)): column = exp.column(select.selects[0].alias_or_name, alias) - clause = predicate.find_ancestor(exp.Having, exp.Where, exp.Join) clause_parent_select = clause.parent_select if clause else None if (isinstance(clause, exp.Having) and clause_parent_select is parent_select) or ( @@ -84,12 +85,18 @@ def unnest(select, parent_select, next_alias_name): column = _other_operand(predicate) value = select.selects[0] - on = exp.condition(f'{column} = "{alias}"."{value.alias}"') - _replace(predicate, f"NOT {on.right} IS NULL") + join_key = exp.column(value.alias, alias) + join_key_not_null = join_key.is_(exp.null()).not_() + + if isinstance(clause, exp.Join): + _replace(predicate, exp.true()) + parent_select.where(join_key_not_null, copy=False) + else: + _replace(predicate, join_key_not_null) parent_select.join( select.group_by(value.this, copy=False), - on=on, + on=column.eq(join_key), join_type="LEFT", join_alias=alias, copy=False, |