52 lines
1.3 KiB
C++
52 lines
1.3 KiB
C++
|
|
#pragma once
|
|
|
|
#include "json_benchmark/file_runner.h"
|
|
#include <vector>
|
|
|
|
namespace distinct_user_id {
|
|
|
|
template<typename I>
|
|
struct runner : public json_benchmark::file_runner<I> {
|
|
std::vector<uint64_t> result{};
|
|
|
|
bool setup(benchmark::State &state) {
|
|
return this->load_json(state, json_benchmark::TWITTER_JSON);
|
|
}
|
|
|
|
bool before_run(benchmark::State &state) {
|
|
if (!json_benchmark::file_runner<I>::before_run(state)) { return false; }
|
|
result.clear();
|
|
return true;
|
|
}
|
|
|
|
bool run(benchmark::State &) {
|
|
return this->implementation.run(this->json, result);
|
|
}
|
|
|
|
bool after_run(benchmark::State &state) {
|
|
if (!json_benchmark::file_runner<I>::after_run(state)) { return false; }
|
|
std::sort(result.begin(), result.end());
|
|
auto last = std::unique(result.begin(), result.end());
|
|
result.erase(last, result.end());
|
|
return true;
|
|
}
|
|
|
|
template<typename R>
|
|
bool diff(benchmark::State &state, runner<R> &reference) {
|
|
return diff_results(state, result, reference.result);
|
|
}
|
|
|
|
size_t items_per_iteration() {
|
|
return result.size();
|
|
}
|
|
};
|
|
|
|
struct simdjson_dom;
|
|
|
|
template<typename I> simdjson_really_inline static void distinct_user_id(benchmark::State &state) {
|
|
json_benchmark::run_json_benchmark<runner<I>, runner<simdjson_dom>>(state);
|
|
}
|
|
|
|
} // namespace distinct_user_id
|