From d233b9bc1a3445e0c4caafd8d26398edeac495a0 Mon Sep 17 00:00:00 2001
From: Ruslan Nigmatullin <elessar@dropbox.com>
Date: Wed, 7 Jun 2023 00:07:01 -0700
Subject: [PATCH] [SDK] Add benchmark for base2 exponential histogram indexer
 (#2181)

---
 sdk/test/metrics/BUILD                        | 15 +++++
 sdk/test/metrics/CMakeLists.txt               |  6 ++
 ...exponential_histogram_indexer_benchmark.cc | 65 +++++++++++++++++++
 3 files changed, 86 insertions(+)
 create mode 100644 sdk/test/metrics/base2_exponential_histogram_indexer_benchmark.cc

diff --git a/sdk/test/metrics/BUILD b/sdk/test/metrics/BUILD
index 24b426fdf5..3cf379494d 100644
--- a/sdk/test/metrics/BUILD
+++ b/sdk/test/metrics/BUILD
@@ -360,6 +360,21 @@ otel_cc_benchmark(
     ],
 )
 
+otel_cc_benchmark(
+    name = "base2_exponential_histogram_indexer_benchmark",
+    srcs = [
+        "base2_exponential_histogram_indexer_benchmark.cc",
+    ],
+    tags = [
+        "benchmark",
+        "metrics",
+        "test",
+    ],
+    deps = [
+        "metrics_common_test_utils",
+    ],
+)
+
 otel_cc_benchmark(
     name = "histogram_aggregation_benchmark",
     srcs = [
diff --git a/sdk/test/metrics/CMakeLists.txt b/sdk/test/metrics/CMakeLists.txt
index 8e60e3b342..b30d71ae22 100644
--- a/sdk/test/metrics/CMakeLists.txt
+++ b/sdk/test/metrics/CMakeLists.txt
@@ -49,6 +49,12 @@ if(WITH_BENCHMARK)
   target_link_libraries(attributes_hashmap_benchmark benchmark::benchmark
                         ${CMAKE_THREAD_LIBS_INIT} opentelemetry_common)
 
+  add_executable(base2_exponential_histogram_indexer_benchmark
+                 base2_exponential_histogram_indexer_benchmark.cc)
+  target_link_libraries(
+    base2_exponential_histogram_indexer_benchmark benchmark::benchmark
+    opentelemetry_metrics ${CMAKE_THREAD_LIBS_INIT} opentelemetry_common)
+
   add_executable(histogram_aggregation_benchmark
                  histogram_aggregation_benchmark.cc)
   target_link_libraries(
diff --git a/sdk/test/metrics/base2_exponential_histogram_indexer_benchmark.cc b/sdk/test/metrics/base2_exponential_histogram_indexer_benchmark.cc
new file mode 100644
index 0000000000..9f1d0d864d
--- /dev/null
+++ b/sdk/test/metrics/base2_exponential_histogram_indexer_benchmark.cc
@@ -0,0 +1,65 @@
+// Copyright The OpenTelemetry Authors
+// SPDX-License-Identifier: Apache-2.0
+
+#include "opentelemetry/sdk/metrics/aggregation/base2_exponential_histogram_indexer.h"
+
+#include <benchmark/benchmark.h>
+#include <array>
+#include <random>
+
+using namespace opentelemetry::sdk::metrics;
+namespace
+{
+
+void BM_NewIndexer(benchmark::State &state)
+{
+  std::array<int, 1000> batch;
+  std::default_random_engine generator;
+  std::uniform_int_distribution<int> distribution(1, 32);
+
+  while (state.KeepRunningBatch(static_cast<benchmark::IterationCount>(batch.size())))
+  {
+    state.PauseTiming();
+    for (auto &value : batch)
+    {
+      value = distribution(generator);
+    }
+    state.ResumeTiming();
+
+    for (const auto value : batch)
+    {
+      benchmark::DoNotOptimize(Base2ExponentialHistogramIndexer(value));
+    }
+  }
+}
+
+BENCHMARK(BM_NewIndexer);
+
+void BM_ComputeIndex(benchmark::State &state)
+{
+  std::array<double, 1000> batch;
+  std::default_random_engine generator;
+  std::uniform_real_distribution<double> distribution(0, 1000);
+  Base2ExponentialHistogramIndexer indexer(static_cast<int32_t>(state.range(0)));
+
+  while (state.KeepRunningBatch(static_cast<benchmark::IterationCount>(batch.size())))
+  {
+    state.PauseTiming();
+    for (auto &value : batch)
+    {
+      value = distribution(generator);
+    }
+    state.ResumeTiming();
+
+    for (const auto value : batch)
+    {
+      benchmark::DoNotOptimize(indexer.ComputeIndex(value));
+    }
+  }
+}
+
+BENCHMARK(BM_ComputeIndex)->Arg(-1)->Arg(0)->Arg(1)->Arg(20);
+
+}  // namespace
+
+BENCHMARK_MAIN();