Function setCoverageTest

base/base/coverage.cpp:367–417

void setCoverageTest(std::string_view test_name)
{
    /// We collect what we need under the lock, then release it before invoking
    /// the callback (which may do heavy DB work and must not hold the mutex).
    std::string prev_test_name;
    std::vector<CovCounter> name_refs;
    std::vector<IndirectCallEntry> indirect_calls;
    CoverageFlushCallback cb;

    {
        std::lock_guard lock(g_coverage_mutex);

        if (!g_current_test_name.empty() && g_flush_callback)
        {
            /// Collect covered NameRefs and indirect-call observations before reset.
            name_refs      = getCurrentCoveredNameRefs();
            indirect_calls = getCurrentIndirectCalls();
            prev_test_name = g_current_test_name;
            cb = g_flush_callback;
        }

        __llvm_profile_reset_counters(); // NOLINT

#ifdef CLICKHOUSE_XRAY_INSTRUMENT_COVERAGE
        /// Bump XRay generation so every thread resets its depth baseline on next call.
        g_xray_generation.fetch_add(1, std::memory_order_relaxed);
        /// Reset per-function min-depth array.
        for (uint32_t i = 0; i < g_xray_depth_size; ++i)
            g_xray_min_depth[i].store(UINT32_MAX, std::memory_order_relaxed);
        if (!g_current_test_name.empty())
        {
            /// Deactivate XRay after collecting the previous test's data.
            __xray_unpatch();
        }
        if (!test_name.empty())
        {
            /// Build the XRay→profile map lazily on first test start.
            buildXRayProfileMap();
            __xray_set_handler(xrayHandler);
            __xray_patch();
        }
#endif

        g_current_test_name = std::string(test_name);
    }

    /// Invoke the callback outside the lock so the DB layer can look up
    /// source locations and insert into system.coverage_log without risking deadlock.
    if (cb)
        cb(prev_test_name, name_refs, indirect_calls);
}

void resetCoverage()
{