68 lines
2.2 KiB
C++
68 lines
2.2 KiB
C++
// Copyright 2022 Memgraph Ltd.
|
|
//
|
|
// Use of this software is governed by the Business Source License
|
|
// included in the file licenses/BSL.txt; by using this file, you agree to be bound by the terms of the Business Source
|
|
// License, and you may not use this file except in compliance with the Business Source License.
|
|
//
|
|
// As of the Change Date specified in that file, in accordance with
|
|
// the Business Source License, use of this software will be governed
|
|
// by the Apache License, Version 2.0, included in the file
|
|
// licenses/APL.txt.
|
|
|
|
#include "skip_list_common.hpp"
|
|
|
|
#include "utils/skip_list.hpp"
|
|
|
|
DEFINE_int32(max_element, 20000, "Maximum element in the intial list");
|
|
DEFINE_int32(max_range, 2000000, "Maximum range used for the test");
|
|
|
|
int main(int argc, char **argv) {
|
|
gflags::ParseCommandLineFlags(&argc, &argv, true);
|
|
|
|
memgraph::utils::SkipList<uint64_t> list;
|
|
|
|
{
|
|
auto acc = list.access();
|
|
for (uint64_t i = 0; i <= FLAGS_max_element; ++i) {
|
|
MG_ASSERT(acc.insert(i).second);
|
|
}
|
|
uint64_t val = 0;
|
|
for (auto item : acc) {
|
|
MG_ASSERT(item == val);
|
|
++val;
|
|
}
|
|
MG_ASSERT(val == FLAGS_max_element + 1);
|
|
}
|
|
|
|
RunConcurrentTest([&list](auto *run, auto *stats) {
|
|
std::mt19937 generator(std::random_device{}());
|
|
std::uniform_int_distribution<uint64_t> distribution(0, 3);
|
|
std::mt19937 i_generator(std::random_device{}());
|
|
std::uniform_int_distribution<uint64_t> i_distribution(0, FLAGS_max_range);
|
|
while (run->load(std::memory_order_relaxed)) {
|
|
auto value = distribution(generator);
|
|
auto accessor = list.access();
|
|
auto item = i_distribution(i_generator);
|
|
switch (value) {
|
|
case 0:
|
|
stats->succ[OP_INSERT] += static_cast<uint64_t>(accessor.insert(item).second);
|
|
break;
|
|
case 1:
|
|
stats->succ[OP_CONTAINS] += static_cast<uint64_t>(accessor.contains(item));
|
|
break;
|
|
case 2:
|
|
stats->succ[OP_REMOVE] += static_cast<uint64_t>(accessor.remove(item));
|
|
break;
|
|
case 3:
|
|
stats->succ[OP_FIND] += static_cast<uint64_t>(accessor.find(item) != accessor.end());
|
|
break;
|
|
default:
|
|
std::terminate();
|
|
}
|
|
++stats->total;
|
|
}
|
|
});
|
|
|
|
return 0;
|
|
}
|