From 9864552a3f2b533c47076a7f6efe246e77deb440 Mon Sep 17 00:00:00 2001
From: Douglas Rumbaugh <dbr4@psu.edu>
Date: Mon, 26 Jun 2023 10:09:46 -0400
Subject: Added static benchmarking

---
 benchmarks/include/bench.h | 41 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 41 insertions(+)

(limited to 'benchmarks/include')

diff --git a/benchmarks/include/bench.h b/benchmarks/include/bench.h
index 6a9c263..db8fc77 100644
--- a/benchmarks/include/bench.h
+++ b/benchmarks/include/bench.h
@@ -108,3 +108,44 @@ static bool query_latency_bench(DE &de_index, std::vector<QP> queries, size_t tr
 
     return true;
 }
+
+
+template <typename Shard, de::RecordInterface R, typename QP, QueryInterface Q, bool PROGRESS=true>
+static bool static_latency_bench(Shard *shard, std::vector<QP> queries, size_t trial_cnt=100) {
+    char progbuf[25];
+    if constexpr (PROGRESS) {
+        sprintf(progbuf, "querying:");
+    }
+
+    size_t total_time = 0;
+    size_t total_results = 0;
+
+    for (size_t i=0; i<trial_cnt; i++) {
+        if constexpr (PROGRESS) {
+            progress_update((double) (i) / (double) trial_cnt, progbuf);
+        }
+
+        std::vector<void *> states(1);
+
+        auto start = std::chrono::high_resolution_clock::now();
+        for (size_t j=0; j<queries.size(); j++) {
+            states[0] = Q::get_query_state(shard, &queries[i]);
+            Q::process_query_states(shard, states, nullptr);
+            auto res = Q::query(shard, states[0], &queries[i]);
+            total_results += res.size();
+            Q::delete_query_state(states[0]);
+        }
+        auto stop = std::chrono::high_resolution_clock::now();
+
+        total_time += std::chrono::duration_cast<std::chrono::nanoseconds>(stop - start).count();
+    }
+
+    progress_update(1.0, progbuf);
+
+    size_t query_latency = total_time / (trial_cnt * queries.size());
+
+    fprintf(stdout, "%ld\t", query_latency);
+    fflush(stdout);
+
+    return true;
+}
-- 
cgit v1.2.3