From cd961cc231ca3073f03d82b99153a6d0f5a4c7f1 Mon Sep 17 00:00:00 2001
From: Douglas Rumbaugh <dbr4@psu.edu>
Date: Sun, 25 May 2025 14:15:36 -0400
Subject: Added selectivity sweep benchmark

---
 benchmarks/include/file_util.h | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

(limited to 'benchmarks/include')

diff --git a/benchmarks/include/file_util.h b/benchmarks/include/file_util.h
index 39b297b..2d340e6 100644
--- a/benchmarks/include/file_util.h
+++ b/benchmarks/include/file_util.h
@@ -86,6 +86,26 @@ static std::vector<QP> read_range_queries(std::string &fname,
   return queries;
 }
 
+template <typename QP>
+static std::vector<QP> generate_uniform_range_queries(size_t n, size_t key_limit, double selectivity) {
+  std::vector<QP> queries;
+
+  gsl_rng *rng = gsl_rng_alloc(gsl_rng_mt19937);
+  size_t query_size = key_limit * selectivity;
+
+  for (size_t i=0; i<n; i++) {
+    QP q;
+    q.lower_bound = gsl_rng_uniform_int(rng, key_limit - query_size);
+    q.upper_bound = q.lower_bound + query_size;
+
+    queries.push_back(q);
+  }
+
+  gsl_rng_free(rng);
+
+  return queries;
+}
+
 template <typename QP>
 static std::vector<QP> read_sosd_point_lookups(std::string &fname, size_t n) {
   std::vector<QP> queries;
-- 
cgit v1.2.3