wip

cswinter · Apr 1, 2024 · 9947869 · 9947869
1 parent 62d0806
commit 9947869
Show file tree

Hide file tree

Showing 11 changed files with 300 additions and 562 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/locustdb-compression-utils/Cargo.lock b/locustdb-compression-utils/Cargo.lock
diff --git a/locustdb-compression-utils/Cargo.toml b/locustdb-compression-utils/Cargo.toml
@@ -11,6 +11,8 @@ license-file = "../LICENSE"
 serde = { version = "1.0", features = ["derive"] }
 serde_json = "1.0"
 log = "0.4"
+bitter = "0.6"
+bitbuffer = "0.10"
 
 [dev-dependencies]
 clap = { version = "4", features = ["derive"] }

diff --git a/locustdb-compression-utils/examples/gorilla_time.rs b/locustdb-compression-utils/examples/gorilla_time.rs
@@ -50,7 +50,6 @@ fn main() {
     let opt = Opt::parse();
 
     if let Some(mibibytes) = opt.benchmark {
-        assert!(!opt.single, "Benchmarking single precision is not supported");
         assert!(opt.max_regret.len() == 1, "Benchmarking multiple max-regret values is not supported");
         // create 1GiB of random floats
         let len = (1 << 20) * mibibytes / 8;
@@ -59,7 +58,11 @@ fn main() {
         let start_time = std::time::Instant::now();
         let mut fast_rng = rand::rngs::SmallRng::seed_from_u64(42);
         for _ in 0..len {
-            data.push(fast_rng.gen::<f64>());
+            if opt.single {
+                data.push(fast_rng.gen::<f32>() as f64);
+            } else {
+                data.push(fast_rng.gen::<f64>());
+            }
         }
         println!(
             "Generated {mibibytes} MiB of random data in {:?}",