From b1f966353695a0e06948df5332acccb84bbbcda0 Mon Sep 17 00:00:00 2001
From: Douglas Rumbaugh <dbr4@psu.edu>
Date: Wed, 31 Jan 2024 14:26:34 -0500
Subject: Query/Insert intermix benchmarks

---
 benchmarks/insert_query_tput.cpp | 87 ++++++++++++++++++++++------------------
 1 file changed, 49 insertions(+), 38 deletions(-)

(limited to 'benchmarks/insert_query_tput.cpp')

diff --git a/benchmarks/insert_query_tput.cpp b/benchmarks/insert_query_tput.cpp
index 865e82c..8844d04 100644
--- a/benchmarks/insert_query_tput.cpp
+++ b/benchmarks/insert_query_tput.cpp
@@ -23,71 +23,82 @@ typedef de::DynamicExtension<Rec, ISAM, Q> Ext;
 
 std::atomic<bool> inserts_done = false;
 
-void insert_thread(Ext *extension, size_t n, size_t k) {
+
+void query_thread(Ext *extension, double selectivity, size_t k, gsl_rng *rng) {
     TIMER_INIT();
-    for (int64_t i=0; i<n; i+=k) {
-        TIMER_START();
-        for (int64_t j=0; j<k; j++) {
-            Rec r = {i+j, i+j};
-            while (!extension->insert(r)) {
-                _mm_pause();
-            }
-        }
-        TIMER_STOP();
-        auto insert_lat = TIMER_RESULT();
 
-        fprintf(stdout, "I\t%ld\t%ld\t%ld\n", extension->get_record_count(), insert_lat, k);
-    }
+    size_t reccnt = extension->get_record_count();
 
-    inserts_done.store(true);
-}
+    size_t range = reccnt * selectivity;
 
-void query_thread(Ext *extension, double selectivity, size_t k, gsl_rng *rng) {
-    TIMER_INIT();
+    auto q = new de::rc::Parms<Rec>();
 
-    while (!inserts_done.load()) {
-        size_t reccnt = extension->get_record_count();
-        if (reccnt == 0) {
-            continue; // don't start querying until there is data
-        }
+    TIMER_START();
+    for (int64_t i=0; i<k; i++) {
+        size_t start = gsl_rng_uniform_int(rng, reccnt - range);
 
-        size_t range = reccnt * selectivity;
+        q->lower_bound = start;
+        q->upper_bound = start + range;
+        auto res = extension->query(q);
+        auto r = res.get();
+    }
+    TIMER_STOP();
+    auto query_lat = TIMER_RESULT();
+    fprintf(stdout, "Q\t%ld\t%ld\t%ld\n", reccnt, query_lat, k);
+    delete q;
+}
+
+void insert_thread(Ext *extension, size_t n, size_t k, gsl_rng *rng) {
+    TIMER_INIT();
 
-        auto q = new de::rc::Parms<Rec>();
+    size_t reccnt = 0;
+    Rec r;
+    while (reccnt < n) {
+        auto old_reccnt = reccnt;
 
         TIMER_START();
-        for (int64_t i=0; i<k; i++) {
-            size_t start = gsl_rng_uniform_int(rng, reccnt - range);
+        for (size_t i=0; i<k; i++) {
+            r.key = reccnt;
+            r.value = reccnt;
 
-            q->lower_bound = start;
-            q->upper_bound = start + range;
-            auto res = extension->query(q);
-            auto r = res.get();
+            if (extension->insert(r)) {
+                reccnt++;
+            }
         }
         TIMER_STOP();
-        auto query_lat = TIMER_RESULT();
-        fprintf(stdout, "Q\t%ld\t%ld\t%ld\n", reccnt, query_lat, k);
-        delete q;
+        auto insert_lat = TIMER_RESULT();
+
+        fprintf(stdout, "I\t%ld\t%ld\t%ld\n", reccnt, insert_lat, reccnt - old_reccnt);
+
+        if (reccnt % 100000 == 0 && reccnt != n)  {
+            auto a = std::thread(query_thread, extension, .01, 20, rng);
+            a.detach();
+        }
     }
 }
 
 int main(int argc, char **argv) {
 
     /* the closeout routine takes _forever_ ... so we'll just leak the memory */
-    auto extension = new Ext(1000, 10000, 2);
+    auto extension = new Ext(1000, 12000, 8);
     size_t n = 10000000;
     size_t per_trial = 1000;
     double selectivity = .001;
 
-    gsl_rng * rng = gsl_rng_alloc(gsl_rng_mt19937);
+    TIMER_INIT();
 
-    std::thread i_thrd(insert_thread, extension, n, per_trial);
-    std::thread q_thrd(query_thread, extension, selectivity, 1, rng);
+    gsl_rng * rng = gsl_rng_alloc(gsl_rng_mt19937);
 
-    q_thrd.join();
+    TIMER_START();
+    std::thread i_thrd(insert_thread, extension, n, per_trial, rng);
     i_thrd.join();
+    TIMER_STOP();
+
+    auto total_latency = TIMER_RESULT();
+    fprintf(stdout, "T\t%ld\n", total_latency);
 
     gsl_rng_free(rng);
+    delete extension;
     fflush(stderr);
 }
 
-- 
cgit v1.2.3