Method addPlanToCache

src/QueryPlan/PlanCache.cpp:157–196

}

bool PlanCacheManager::addPlanToCache(UInt128 query_hash, QueryPlanPtr & plan, AnalysisPtr analysis, ContextMutablePtr & context)
{
    if (!context->getPlanCacheManager())
        throw Exception("plan cache has to be initialized", ErrorCodes::LOGICAL_ERROR);

    auto & cache = context->getPlanCacheManager()->instance();
    auto root = plan->getPlanNode();
    UInt32 size = QueryPlan::getPlanNodeCount(root);

    if (size > context->getSettingsRef().max_plannode_count)
        return false;

    PlanNodeId max_id;
    PlanCacheManager::PlanObjectValue plan_object{};
    plan_object.plan_root = PlanCacheManager::getNewPlanNode(root, context, true, max_id);

    for (const auto & cte : plan->getCTEInfo().getCTEs())
        plan_object.cte_map.emplace(cte.first, PlanCacheManager::getNewPlanNode(cte.second, context, true, max_id));

    plan_object.query_info = std::make_shared<PlanCacheManager::QueryInfo>();
    const auto & used_columns_map = analysis->getUsedColumns();
    for (const auto & [table_ast, storage_analysis] : analysis->getStorages())
    {
        if (!storage_analysis.storage)
            continue;
        auto storage_id = storage_analysis.storage->getStorageID();
        if (auto it = used_columns_map.find(storage_analysis.storage->getStorageID()); it != used_columns_map.end())
        {
            for (const auto & column : it->second)
                plan_object.query_info->query_access_info[backQuoteIfNeed(storage_id.getDatabaseName())][storage_id.getFullTableName()].emplace_back(column);
        }

        Statistics::StatsTableIdentifier table_identifier{storage_id};
        auto version_value = Statistics::getVersion(context, table_identifier);
        Int64 version = version_value.has_value() ? version_value.value().convertTo<Int64>() : 0;
        plan_object.query_info->stats_version[storage_id] = version;
    }
    cache.add(query_hash, plan_object);
    return true;
}

bool PlanCacheManager::enableCachePlan(const ASTPtr & query_ast, ContextPtr context)
{