Method getPlanSegment

src/Interpreters/InterpreterSelectQueryUseOptimizer.cpp:255–305

}

std::pair<PlanSegmentTreePtr, std::set<StorageID>> InterpreterSelectQueryUseOptimizer::getPlanSegment()
{
    Stopwatch stage_watch, total_watch;
    total_watch.start();
    setUnsupportedSettings(context);
    QueryPlanPtr query_plan = getQueryPlan();

    query_plan->setResetStepId(false);
    stage_watch.start();
    QueryPlan plan = PlanNodeToNodeVisitor::convert(*query_plan);

    LOG_DEBUG(log, "optimizer stage run time: plan normalize, {} ms", stage_watch.elapsedMillisecondsAsDouble());
    stage_watch.restart();

    PlanSegmentTreePtr plan_segment_tree = std::make_unique<PlanSegmentTree>();
    ClusterInfoContext cluster_info_context{.query_plan = *query_plan, .context = context, .plan_segment_tree = plan_segment_tree};
    PlanSegmentContext plan_segment_context = ClusterInfoFinder::find(*query_plan, cluster_info_context);

    stage_watch.restart();
    std::set<StorageID> used_storage_ids = plan.allocateLocalTable(context);

    blockQueryJSONUseOptimizer(used_storage_ids, context);
    // select health worker before split
    if (context->getSettingsRef().enable_adaptive_scheduler && context->tryGetCurrentWorkerGroup())
    {
        context->selectWorkerNodesWithMetrics();
        auto wg_health = context->getWorkerGroupStatusPtr()->getWorkerGroupHealth();
        if (wg_health == WorkerGroupHealthStatus::Critical)
            throw Exception("no worker available", ErrorCodes::LOGICAL_ERROR);
    }

    PlanSegmentSplitter::split(plan, plan_segment_context);
    context->logOptimizerProfile(
        log, "Optimizer total run time: ", "PlanSegment build", std::to_string(stage_watch.elapsedMillisecondsAsDouble()) + "ms");
    ProfileEvents::increment(ProfileEvents::PlanSegmentSplitterTime, stage_watch.elapsedMilliseconds());

    resetFinalSampleSize(plan_segment_tree);
    setPlanSegmentInfoForExplainAnalyze(plan_segment_tree);
    GraphvizPrinter::printPlanSegment(plan_segment_tree, context);
    context->logOptimizerProfile(
        log, "Optimizer total run time: ", "Optimizer Total", std::to_string(total_watch.elapsedMillisecondsAsDouble()) + "ms");

    if (context->getSettingsRef().log_segment_profiles)
    {
        segment_profiles = std::make_shared<std::vector<String>>();
        for (auto & node : plan_segment_tree->getNodes())
            segment_profiles->emplace_back(PlanSegmentDescription::getPlanSegmentDescription(node.plan_segment, true)->jsonPlanSegmentDescriptionAsString({}));
    }

    return std::make_pair(std::move(plan_segment_tree), std::move(used_storage_ids));
}

QueryPipeline executeTEALimit(QueryPipeline & pipeline, ContextMutablePtr context, ASTPtr query_ptr, Poco::Logger * log)
{