#include <algorithm>
#include <cstddef>
#include <string>
#include <vector>
#include "absl/container/linked_hash_map.h"
#include "absl/functional/function_ref.h"
#include "absl/strings/str_format.h"
#include "absl/strings/string_view.h"
#include "benchmark/benchmark.h"
namespace {
struct BenchmarkData {
std::vector<std::string> sample;
size_t str_bytes = 0;
explicit BenchmarkData(int size,
absl::FunctionRef<std::string(int)> factory) {
sample.reserve(size);
for (int i = 0; i < size; ++i) {
sample.push_back(factory(i));
str_bytes += sample.back().size();
}
}
};
void BenchmarkTryEmplaceStrings(benchmark::State& state,
absl::FunctionRef<std::string(int)> factory) {
BenchmarkData data(state.range(0), factory);
const size_t batch_size =
std::max(size_t{1}, size_t{1000000} / data.str_bytes);
std::vector<absl::linked_hash_map<std::string, int>> sets(batch_size);
while (state.KeepRunningBatch(batch_size)) {
state.PauseTiming();
for (auto& set : sets) set.clear();
state.ResumeTiming();
for (auto& set : sets) {
for (const auto& str : data.sample) {
benchmark::DoNotOptimize(set.try_emplace(str));
}
}
}
state.SetItemsProcessed(state.iterations() * state.range(0));
state.SetBytesProcessed(state.iterations() * data.str_bytes);
}
void BenchmarkInsertOrAssignStrings(
benchmark::State& state, absl::FunctionRef<std::string(int)> factory) {
BenchmarkData data(state.range(0), factory);
const size_t batch_size = std::max(size_t{1}, 1000000 / data.str_bytes);
std::vector<absl::linked_hash_map<std::string, int>> sets(batch_size);
while (state.KeepRunningBatch(batch_size)) {
state.PauseTiming();
for (auto& set : sets) set.clear();
state.ResumeTiming();
for (auto& set : sets) {
for (const auto& str : data.sample) {
benchmark::DoNotOptimize(set.insert_or_assign(str, 0));
}
}
}
state.SetItemsProcessed(state.iterations() * state.range(0));
state.SetBytesProcessed(state.iterations() * data.str_bytes);
}
constexpr absl::string_view kFormatShort = "%10d";
constexpr absl::string_view kFormatLong =
"a longer string that exceeds the SSO %10d";
void BM_TryEmplaceShortStrings_Hit(benchmark::State& state) {
BenchmarkTryEmplaceStrings(
state, [](int i) { return absl::StrFormat(kFormatShort, i); });
}
BENCHMARK(BM_TryEmplaceShortStrings_Hit)->Range(1, 1 << 16);
void BM_TryEmplaceLongStrings_Hit(benchmark::State& state) {
BenchmarkTryEmplaceStrings(
state, [](int i) { return absl::StrFormat(kFormatLong, i); });
}
BENCHMARK(BM_TryEmplaceLongStrings_Hit)->Range(1, 1 << 16);
void BM_TryEmplaceShortStrings_Miss(benchmark::State& state) {
BenchmarkTryEmplaceStrings(
state, [](int i) { return absl::StrFormat(kFormatShort, i % 20); });
}
BENCHMARK(BM_TryEmplaceShortStrings_Miss)->Range(1, 1 << 16);
void BM_TryEmplaceLongStrings_Miss(benchmark::State& state) {
BenchmarkTryEmplaceStrings(
state, [](int i) { return absl::StrFormat(kFormatLong, i % 20); });
}
BENCHMARK(BM_TryEmplaceLongStrings_Miss)->Range(1, 1 << 16);
void BM_InsertOrAssignShortStrings_Hit(benchmark::State& state) {
BenchmarkInsertOrAssignStrings(
state, [](int i) { return absl::StrFormat(kFormatShort, i); });
}
BENCHMARK(BM_InsertOrAssignShortStrings_Hit)->Range(1, 1 << 16);
void BM_InsertOrAssignLongStrings_Hit(benchmark::State& state) {
BenchmarkInsertOrAssignStrings(
state, [](int i) { return absl::StrFormat(kFormatLong, i); });
}
BENCHMARK(BM_InsertOrAssignLongStrings_Hit)->Range(1, 1 << 16);
void BM_InsertOrAssignShortStrings_Miss(benchmark::State& state) {
BenchmarkInsertOrAssignStrings(
state, [](int i) { return absl::StrFormat(kFormatShort, i % 20); });
}
BENCHMARK(BM_InsertOrAssignShortStrings_Miss)->Range(1, 1 << 16);
void BM_InsertOrAssignLongStrings_Miss(benchmark::State& state) {
BenchmarkInsertOrAssignStrings(
state, [](int i) { return absl::StrFormat(kFormatLong, i % 20); });
}
BENCHMARK(BM_InsertOrAssignLongStrings_Miss)->Range(1, 1 << 16);
}