| 1 | #include "duckdb/planner/expression_binder/qualify_binder.hpp" |
| 2 | |
| 3 | #include "duckdb/parser/expression/columnref_expression.hpp" |
| 4 | #include "duckdb/planner/binder.hpp" |
| 5 | #include "duckdb/planner/expression_binder/aggregate_binder.hpp" |
| 6 | #include "duckdb/common/string_util.hpp" |
| 7 | #include "duckdb/planner/query_node/bound_select_node.hpp" |
| 8 | #include "duckdb/parser/expression/window_expression.hpp" |
| 9 | |
| 10 | namespace duckdb { |
| 11 | |
| 12 | QualifyBinder::QualifyBinder(Binder &binder, ClientContext &context, BoundSelectNode &node, BoundGroupInformation &info, |
| 13 | case_insensitive_map_t<idx_t> &alias_map) |
| 14 | : BaseSelectBinder(binder, context, node, info), column_alias_binder(node, alias_map) { |
| 15 | target_type = LogicalType(LogicalTypeId::BOOLEAN); |
| 16 | } |
| 17 | |
| 18 | BindResult QualifyBinder::BindColumnRef(unique_ptr<ParsedExpression> &expr_ptr, idx_t depth, bool root_expression) { |
| 19 | auto &expr = expr_ptr->Cast<ColumnRefExpression>(); |
| 20 | auto result = duckdb::BaseSelectBinder::BindExpression(expr_ptr, depth); |
| 21 | if (!result.HasError()) { |
| 22 | return result; |
| 23 | } |
| 24 | |
| 25 | auto alias_result = column_alias_binder.BindAlias(enclosing_binder&: *this, expr, depth, root_expression); |
| 26 | if (!alias_result.HasError()) { |
| 27 | return alias_result; |
| 28 | } |
| 29 | |
| 30 | return BindResult(StringUtil::Format(fmt_str: "Referenced column %s not found in FROM clause and can't find in alias map." , |
| 31 | params: expr.ToString())); |
| 32 | } |
| 33 | |
| 34 | BindResult QualifyBinder::BindExpression(unique_ptr<ParsedExpression> &expr_ptr, idx_t depth, bool root_expression) { |
| 35 | auto &expr = *expr_ptr; |
| 36 | // check if the expression binds to one of the groups |
| 37 | auto group_index = TryBindGroup(expr, depth); |
| 38 | if (group_index != DConstants::INVALID_INDEX) { |
| 39 | return BindGroup(expr, depth, group_index); |
| 40 | } |
| 41 | switch (expr.expression_class) { |
| 42 | case ExpressionClass::WINDOW: |
| 43 | return BindWindow(expr&: expr.Cast<WindowExpression>(), depth); |
| 44 | case ExpressionClass::COLUMN_REF: |
| 45 | return BindColumnRef(expr_ptr, depth, root_expression); |
| 46 | default: |
| 47 | return duckdb::BaseSelectBinder::BindExpression(expr_ptr, depth); |
| 48 | } |
| 49 | } |
| 50 | |
| 51 | } // namespace duckdb |
| 52 | |