Function compileSortDescriptionIfNeeded

src/Core/SortDescription.cpp:188–252

}

void compileSortDescriptionIfNeeded(SortDescription & description, const DataTypes & sort_description_types, bool increase_compile_attempts)
{
    static std::unordered_map<UInt128, UInt64, UInt128Hash> counter;
    static std::mutex mutex;

    if (!description.compile_sort_description || sort_description_types.empty())
        return;

    for (size_t i = 0; i < description.size(); ++i)
    {
        auto nested_type = removeNullable(sort_description_types[i]);
        if (!sort_description_types[i]->createColumn()->isComparatorCompilable() ||
            (!canBeNativeType(*sort_description_types[i]) && !WhichDataType(nested_type).isString() && !WhichDataType(nested_type).isFixedString()))
            return;

        /// JIT comparator does not support collation-aware comparison
        if (description[i].collator)
            return;
    }


    auto description_dump = getSortDescriptionDump(description, sort_description_types);

    SipHash sort_description_dump_hash;
    sort_description_dump_hash.update(description_dump);

    const auto sort_description_hash_key = sort_description_dump_hash.get128();

    {
        std::lock_guard lock(mutex);
        UInt64 & current_counter = counter[sort_description_hash_key];
        if (current_counter < description.min_count_to_compile_sort_description)
        {
            current_counter += static_cast<UInt64>(increase_compile_attempts);
            return;
        }
    }

    std::shared_ptr<CompiledSortDescriptionFunctionHolder> compiled_sort_description_holder;

    if (auto * compilation_cache = CompiledExpressionCacheFactory::instance().tryGetCache())
    {
        auto [compiled_function_cache_entry, _] = compilation_cache->getOrSet(sort_description_hash_key, [&] ()
        {
            LOG_TRACE(getLogger(), "Compile sort description {}", description_dump);

            auto jit_owner = getJITInstancePtr();
            auto compiled_sort_description = compileSortDescription(*jit_owner, description, sort_description_types, description_dump);
            return std::make_shared<CompiledSortDescriptionFunctionHolder>(std::move(compiled_sort_description), std::move(jit_owner));
        });

        compiled_sort_description_holder = std::static_pointer_cast<CompiledSortDescriptionFunctionHolder>(compiled_function_cache_entry);
    }
    else
    {
        LOG_TRACE(getLogger(), "Compile sort description {}", description_dump);
        auto jit_owner = getJITInstancePtr();
        auto compiled_sort_description = compileSortDescription(*jit_owner, description, sort_description_types, description_dump);