From f0b610ff07fb2da6b491ebbf5388d0dd948624db Mon Sep 17 00:00:00 2001
From: Geoffrey Yu <geoffxy@mit.edu>
Date: Wed, 1 May 2024 18:49:54 -0400
Subject: [PATCH] Add zipfian argument to the calibration scripts

---
 tools/calibration/transactions/chbenchmark/COND            | 3 +++
 tools/calibration/transactions/chbenchmark/run_instance.sh | 7 ++++++-
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/tools/calibration/transactions/chbenchmark/COND b/tools/calibration/transactions/chbenchmark/COND
index f59e559a..f8901f59 100644
--- a/tools/calibration/transactions/chbenchmark/COND
+++ b/tools/calibration/transactions/chbenchmark/COND
@@ -13,6 +13,8 @@ COND_INSTANCES = {
   instance: instance.replace(".", "_").replace("db.", "") for instance in INSTANCES
 }
 
+ZIPFIAN_ALPHA = 5.0
+
 combine(
   name="all",
   deps=[
@@ -36,6 +38,7 @@ for instance in INSTANCES:
           "txn-warehouses": 1740,
           "txn-config-file": "aurora.config",
           "schema-name": "chbenchmark",
+          "txn-zipfian-alpha": ZIPFIAN_ALPHA,
           "instance": instance,
         },
       )
diff --git a/tools/calibration/transactions/chbenchmark/run_instance.sh b/tools/calibration/transactions/chbenchmark/run_instance.sh
index dfe0c6b5..3890358c 100755
--- a/tools/calibration/transactions/chbenchmark/run_instance.sh
+++ b/tools/calibration/transactions/chbenchmark/run_instance.sh
@@ -38,6 +38,10 @@ function extract_named_arguments() {
     if [[ $phys_arg =~ --instance=.+ ]]; then
       instance=${phys_arg:11}
     fi
+
+    if [[ $phys_arg =~ --txn-zipfian-alpha=.+ ]]; then
+      txn_zipfian_alpha=${phys_arg:20}
+    fi
   done
 }
 
@@ -74,7 +78,8 @@ RECORD_DETAILED_STATS=1 python3 -m pytpcc.tpcc aurora \
   --duration $run_for_s \
   --clients $t_clients \
   --scalefactor 1 \
-  --lat-sample-prob 0.25
+  --lat-sample-prob 0.25 \
+  --txn-zipfian-alpha $txn_zipfian_alpha
 popd
 
 >&2 echo "Waiting 10 seconds before retrieving metrics..."