fix 02911_support_alias_column_in_indices

vdimir · vdimir · commit 973c2536ff0d · 2024-02-28T10:44:09.000Z
diff --git a/src/Processors/QueryPlan/ReadFromMergeTree.cpp b/src/Processors/QueryPlan/ReadFromMergeTree.cpp
@@ -1434,8 +1434,13 @@ void ReadFromMergeTree::applyFilters(ActionDAGNodes added_filter_nodes)
         if (query_info.planner_context)
         {
             const auto & table_expression_data = query_info.planner_context->getTableExpressionDataOrThrow(query_info.table_expression);
+            const auto & alias_column_expressions = table_expression_data.getAliasColumnExpressions();
             for (const auto & [column_identifier, column_name] : table_expression_data.getColumnIdentifierToColumnName())
             {
+                /// ALIAS columns cannot be used in the filter expression without being calculated in ActionsDAG,
+                /// so they should not be added to the input nodes.
+                if (alias_column_expressions.contains(column_name))
+                    continue;
                 const auto & column = table_expression_data.getColumnOrThrow(column_name);
                 node_name_to_input_node_column.emplace(column_identifier, ColumnWithTypeAndName(column.type, column_name));
             }

Original file line number	Diff line number	Diff line change
`@@ -1434,8 +1434,13 @@ void ReadFromMergeTree::applyFilters(ActionDAGNodes added_filter_nodes)`
`1434`	`1434`	`if (query_info.planner_context)`
`1435`	`1435`	`{`
`1436`	`1436`	`const auto & table_expression_data = query_info.planner_context->getTableExpressionDataOrThrow(query_info.table_expression);`
	`1437`	`+ const auto & alias_column_expressions = table_expression_data.getAliasColumnExpressions();`
`1437`	`1438`	`for (const auto & [column_identifier, column_name] : table_expression_data.getColumnIdentifierToColumnName())`
`1438`	`1439`	`{`
	`1440`	`+ /// ALIAS columns cannot be used in the filter expression without being calculated in ActionsDAG,`
	`1441`	`+ /// so they should not be added to the input nodes.`
	`1442`	`+ if (alias_column_expressions.contains(column_name))`
	`1443`	`+ continue;`
`1439`	`1444`	`const auto & column = table_expression_data.getColumnOrThrow(column_name);`
`1440`	`1445`	`node_name_to_input_node_column.emplace(column_identifier, ColumnWithTypeAndName(column.type, column_name));`
`1441`	`1446`	`}`