Function addQueryTreePasses

src/Analyzer/QueryTreePassManager.cpp:268–353

}

void addQueryTreePasses(QueryTreePassManager & manager, bool only_analyze)
{
    manager.addPass(std::make_unique<QueryAnalysisPass>(only_analyze));
    manager.addPass(std::make_unique<GroupingFunctionsResolvePass>());
    manager.addPass(std::make_unique<AutoFinalOnQueryPass>());
    /// This pass should be run for the secondary queries
    /// to ensure that the only required columns are read from VIEWs on the shards.
    manager.addPass(std::make_unique<RemoveUnusedProjectionColumnsPass>());
    manager.addPass(std::make_unique<PruneArrayJoinColumnsPass>());

    manager.addPass(std::make_unique<DictGetTupleElementPass>());
    manager.addPass(std::make_unique<ConvertEmptyStringComparisonToFunctionPass>());
    manager.addPass(std::make_unique<FunctionToSubcolumnsPass>());

    manager.addPass(std::make_unique<ConvertLogicalExpressionToCNFPass>());
    manager.addPass(std::make_unique<RegexpFunctionRewritePass>());

    manager.addPass(std::make_unique<RewriteSumFunctionWithSumAndCountPass>());
    manager.addPass(std::make_unique<CountDistinctPass>());
    manager.addPass(std::make_unique<UniqToCountPass>());
    manager.addPass(std::make_unique<RewriteArrayExistsToHasPass>());
    manager.addPass(std::make_unique<RewriteHasToInPass>());
    manager.addPass(std::make_unique<NormalizeCountVariantsPass>());

    manager.addPass(std::make_unique<DistanceTransposedPartialReadsPass>());

    /// should before AggregateFunctionsArithmeticOperationsPass
    manager.addPass(std::make_unique<AggregateFunctionOfGroupByKeysPass>());

    manager.addPass(std::make_unique<AggregateFunctionsArithmeticOperationsPass>());
    manager.addPass(std::make_unique<UniqInjectiveFunctionsEliminationPass>());

    // Should run before optimization of GROUP BY keys to allow the removal of
    // toString function.
    manager.addPass(std::make_unique<IfTransformStringsToEnumPass>());

    /// These passes can rewrite a predicate so that it depends on base columns instead of
    /// the derived expression that appears in `GROUP BY`. They must run before
    /// `OptimizeGroupByFunctionKeysPass` and `OptimizeGroupByInjectiveFunctionsPass`,
    /// because those passes remove grouping keys based on the current expression tree.
    /// For example, `SELECT toYear(d), toYear(d) = 2024, count() FROM ... GROUP BY ALL`:
    /// `OptimizeGroupByFunctionKeysPass` can keep only `toYear(d)` as the grouping key,
    /// and if `OptimizeDateOrDateTimeConverterWithPreimagePass` runs later it rewrites
    /// `toYear(d) = 2024` to a range on raw `d`. After aggregation only `toYear(d)` and
    /// `count()` remain, so the final projection would need `d` that is no longer present.
    manager.addPass(std::make_unique<InverseDictionaryLookupPass>());
    manager.addPass(std::make_unique<OptimizeDateOrDateTimeConverterWithPreimagePass>());
    manager.addPass(std::make_unique<ComparisonTupleEliminationPass>());

    manager.addPass(std::make_unique<OptimizeGroupByFunctionKeysPass>());
    manager.addPass(std::make_unique<OptimizeGroupByInjectiveFunctionsPass>());

    /// The order here is important as we want to keep collapsing in order
    manager.addPass(std::make_unique<MultiIfToIfPass>());
    manager.addPass(std::make_unique<IfConstantConditionPass>());
    manager.addPass(std::make_unique<IfChainToMultiIfPass>());
    manager.addPass(std::make_unique<RewriteAggregateFunctionWithIfPass>());
    manager.addPass(std::make_unique<SumIfToCountIfPass>());