thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-12-02 12:32:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

Optimize keys of GROUPING SETS

Browse Source
This commit is contained in:
Dmitry Novik 2022-11-15 16:56:45 +00:00
parent b2eca5c703
commit a7ea30179e
1 changed files with 21 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
src/Analyzer/Passes/OptimizeGroupByFunctionKeysPass.cpp
View File

@ -65,6 +65,19 @@ class OptimizeGroupByFunctionKeysVisitor : public InDepthQueryTreeVisitor<Optimi
return true;
}
static void optimizeGroupingSet(QueryTreeNodes & grouping_set)
{
QueryTreeNodes new_group_by_keys;
new_group_by_keys.reserve(grouping_set.size());
for (auto & group_by_elem : grouping_set)
{
if (!canBeEliminated(group_by_elem, grouping_set))
new_group_by_keys.push_back(group_by_elem);
}
grouping_set = std::move(new_group_by_keys);
}
public:
static bool needChildVisit(QueryTreeNodePtr & /*parent*/, QueryTreeNodePtr & child)
{
@ -81,16 +94,16 @@ public:
return;
auto & group_by = query->getGroupBy().getNodes();
QueryTreeNodes new_group_by_keys;
new_group_by_keys.reserve(group_by.size());
for (auto & group_by_elem : group_by)
if (query->isGroupByWithGroupingSets())
{
if (!canBeEliminated(group_by_elem, group_by))
new_group_by_keys.push_back(group_by_elem);
for (auto & set : group_by)
{
auto & grouping_set = set->as<ListNode>()->getNodes();
optimizeGroupingSet(grouping_set);
}
}
group_by = std::move(new_group_by_keys);
else
optimizeGroupingSet(group_by);
}
};