Method runBenchmarkWithGenerator

programs/keeper-bench/Runner.cpp:1214–1318

}

void Runner::runBenchmarkWithGenerator()
{
    pool.emplace(CurrentMetrics::LocalThread, CurrentMetrics::LocalThreadActive, CurrentMetrics::LocalThreadScheduled, concurrency);
    createConnections();

    std::cerr << "Preparing to run\n";
    benchmark_context.startup(*connections[0]);
    std::cerr << "Prepared\n";

    warmup_complete = warmup_seconds <= 0;

    int64_t start_timestamp_ms = 0;
    threads = std::vector<ThreadState>(concurrency);

    /// All threads must finish generator initialization (which resolves
    /// `children_of` paths) before any thread starts executing requests.
    generator_init_barrier = std::make_unique<std::barrier<>>(concurrency);

    try
    {
        for (size_t i = 0; i < concurrency; ++i)
        {
            threads[i].thread_idx = i;
            auto thread_connections = connections;
            pool->scheduleOrThrowOnError([this, i, my_connections = std::move(thread_connections)]() mutable { thread(my_connections, threads.at(i)); });
        }
    }
    catch (...)
    {
        shutdown = true;
        pool->wait();
        throw;
    }

    DB::InterruptListener interrupt_listener;
    /// Reset regardless of warmup so setup time is excluded from throughput and time limit
    info->elapsed.restart();
    total_watch.restart();
    start_timestamp_ms = Poco::Timestamp().epochMicroseconds() / 1000;
    Stopwatch warmup_watch;
    delay_watch.restart();

    /// Accumulates stats across all periods for the final report.
    auto cumulative_info = std::make_shared<Stats>();
    cumulative_info->elapsed.restart();

    while (!shutdown)
    {
        if (max_time > 0 && total_watch.elapsedSeconds() >= max_time)
        {
            std::cerr << "Stopping launch of queries. Requested time limit is exhausted.\n";
            shutdown = true;
            break;
        }

        if (interrupt_listener.check())
        {
            std::cerr << "Stopping launch of queries. SIGINT received." << std::endl;