Function TEST_P

library/cpp/yt/memory/unittests/free_list_ut.cpp:94–145

TEST_P(TFreeListStressTest, Stress)
{
    TTestItemSet::Reset();
    SetRandomSeed(0x424242);
    auto params = GetParam();

    TFreeList<TTestItem> list;

    std::latch start(params.Threads);

    std::atomic<bool> running{true};
    std::atomic<ui64> put{0};
    std::atomic<ui64> extracted{0};

    std::vector<std::thread> workers;
    for (ui64 i = 0; i < params.Threads; ++i) {
        auto itemSet = TTestItemSet::Allocate(params.MaxBatchSize);
        workers.emplace_back([&, params, itemSet = std::move(itemSet)]() mutable {
            start.arrive_and_wait();

            while (running.load(std::memory_order::relaxed)) {
                // Push batch of items.
                ui64 batchSize = 1 + RandomNumber<ui64>(params.MaxBatchSize);
                for (ui64 i = 0; i < batchSize; ++i) {
                    auto* item = itemSet.Release();
                    item->Value = 1 + RandomNumber<ui64>(1e9);
                    put.fetch_add(item->Value, std::memory_order::relaxed);
                    list.Put(item);
                }

                // Pop batch of items.
                for (ui64 i = 0; i < batchSize; ++i) {
                    auto* item = list.Extract();
                    ASSERT_NE(item, nullptr);
                    extracted.fetch_add(item->Value, std::memory_order::relaxed);
                    itemSet.Acquire(item);
                }
            }
        });
    }

    Sleep(params.TimeLimit);
    running.store(false);

    for (auto& worker : workers) {
        worker.join();
    }

    Cerr << "Put: " << put.load() << Endl;
    Cerr << "Extracted: " << extracted.load() << Endl;
    EXPECT_EQ(put.load(), extracted.load());
}

INSTANTIATE_TEST_SUITE_P(
    TFreeListTest,