From d6679a01790537b7020e49ab9813e26925401cf3 Mon Sep 17 00:00:00 2001
From: levlam <levlam@telegram.org>
Date: Sun, 22 Aug 2021 10:59:50 +0300
Subject: [PATCH] Add even more DuplicateChecker benchmarks.

---
 benchmark/bench_misc.cpp | 38 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 38 insertions(+)

diff --git a/benchmark/bench_misc.cpp b/benchmark/bench_misc.cpp
index 4d7627f24..2edc3659f 100644
--- a/benchmark/bench_misc.cpp
+++ b/benchmark/bench_misc.cpp
@@ -601,9 +601,47 @@ class DuplicateCheckerBenchRepeat final : public td::Benchmark {
   }
 };
 
+template <class T>
+class DuplicateCheckerBenchRepeatOnly final : public td::Benchmark {
+  td::string get_description() const final {
+    return PSTRING() << "DuplicateCheckerBenchRepeatOnly" << T::get_description();
+  }
+  void run(int n) final {
+    T checker_;
+    for (int i = 0; i < n; i++) {
+      auto result = checker_.check(i & 255);
+      CHECK(result.is_error() == i >= 256);
+    }
+  }
+};
+
+template <class T>
+class DuplicateCheckerBenchReverse final : public td::Benchmark {
+  td::string get_description() const final {
+    return PSTRING() << "DuplicateCheckerBenchReverseAdd" << T::get_description();
+  }
+  void run(int n) final {
+    T checker_;
+    for (int i = 0; i < n; i++) {
+      auto pos = i & 255;
+      checker_.check(i - pos + (255 - pos)).ensure();
+    }
+  }
+};
+
 int main() {
   SET_VERBOSITY_LEVEL(VERBOSITY_NAME(DEBUG));
 
+  td::bench(DuplicateCheckerBenchReverse<MessageIdDuplicateCheckerNew<1000>>());
+  td::bench(DuplicateCheckerBenchReverse<MessageIdDuplicateCheckerNew<300>>());
+  td::bench(DuplicateCheckerBenchReverse<MessageIdDuplicateCheckerArray<1000>>());
+  td::bench(DuplicateCheckerBenchReverse<MessageIdDuplicateCheckerArray<300>>());
+
+  td::bench(DuplicateCheckerBenchRepeatOnly<MessageIdDuplicateCheckerNew<1000>>());
+  td::bench(DuplicateCheckerBenchRepeatOnly<MessageIdDuplicateCheckerNew<300>>());
+  td::bench(DuplicateCheckerBenchRepeatOnly<MessageIdDuplicateCheckerArray<1000>>());
+  td::bench(DuplicateCheckerBenchRepeatOnly<MessageIdDuplicateCheckerArray<300>>());
+
   td::bench(DuplicateCheckerBenchRepeat<MessageIdDuplicateCheckerOld>());
   td::bench(DuplicateCheckerBenchRepeat<MessageIdDuplicateCheckerNew<1000>>());
   td::bench(DuplicateCheckerBenchRepeat<MessageIdDuplicateCheckerNewOther>());