Fix bench

whamcloud · Aug 8, 2024 · ce96431 · ce96431
1 parent c6df20b
commit ce96431
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 8 deletions.
diff --git a/lustre-collector/Cargo.toml b/lustre-collector/Cargo.toml
@@ -19,3 +19,6 @@ tracing.workspace = true
 [dev-dependencies]
 include_dir.workspace = true
 insta.workspace = true
+
+[lib]
+bench = false
diff --git a/lustrefs-exporter/benches/jobstats.rs b/lustrefs-exporter/benches/jobstats.rs
@@ -37,28 +37,24 @@ job_stats:{}"#,
     str_repeat!(JOBSTAT_JOB, 1000)
 );
 
-async fn parse_synthetic_yaml(input: &'static str, len: usize) {
+async fn parse_synthetic_yaml(input: &'static str) {
     let f = BufReader::with_capacity(128 * 1_024, input.as_bytes());
 
     let (fut, mut rx) = jobstats_stream(f);
 
-    let mut cnt = 0;
-
-    while rx.recv().await.is_some() {
-        cnt += 1;
-    }
+    while rx.recv().await.is_some() {}
 
     fut.await.unwrap();
 }
 
 fn criterion_benchmark_fast(c: &mut Criterion) {
     c.bench_function("jobstats 100", |b| {
         b.to_async(tokio::runtime::Builder::new_multi_thread().build().unwrap())
-            .iter(|| black_box(parse_synthetic_yaml(INPUT_100_JOBS, 100)))
+            .iter(|| black_box(parse_synthetic_yaml(INPUT_100_JOBS)))
     });
     c.bench_function("jobstats 1000", |b| {
         b.to_async(tokio::runtime::Builder::new_multi_thread().build().unwrap())
-            .iter(|| black_box(parse_synthetic_yaml(INPUT_1000_JOBS, 1000)))
+            .iter(|| black_box(parse_synthetic_yaml(INPUT_1000_JOBS)))
     });
 }
 criterion_group! {