Skip to content
6 changes: 4 additions & 2 deletions crates/ourios-bench/benches/b1.rs
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,7 @@ use std::path::Path;

use criterion::{BenchmarkId, Criterion, criterion_group, criterion_main};

use ourios_bench::{ReferenceCorpus, build_b1_store};
use ourios_bench::{ReferenceCorpus, TxtSeverity, build_b1_store};
use ourios_core::audit::ParamType;
use ourios_core::record::{BodyKind, MinedRecord, Param};
use ourios_core::tenant::TenantId;
Expand Down Expand Up @@ -245,7 +245,9 @@ fn real_corpus(c: &mut Criterion) {
continue;
}
let bucket = tempfile::TempDir::new().expect("temp bucket");
let built = build_b1_store(path, bucket.path(), ZSTD_LEVEL).expect("build b1 store");
let severity = TxtSeverity::from_env().expect("OURIOS_CORPUS_SEVERITY");
let built =
build_b1_store(path, bucket.path(), ZSTD_LEVEL, severity).expect("build b1 store");
Comment thread
jensholdgaard marked this conversation as resolved.
Resolved
let Some((severity, query)) = severity_query(&built, dir) else {
continue;
};
Expand Down
5 changes: 3 additions & 2 deletions crates/ourios-bench/benches/b2.rs
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@ use std::path::Path;

use criterion::{BenchmarkId, Criterion, criterion_group, criterion_main};

use ourios_bench::build_query_store;
use ourios_bench::{TxtSeverity, build_query_store};
use ourios_core::audit::ParamType;
use ourios_core::record::{BodyKind, MinedRecord, Param};
use ourios_core::tenant::TenantId;
Expand Down Expand Up @@ -192,7 +192,8 @@ fn real_corpus(c: &mut Criterion) {
continue;
}
let bucket = tempfile::TempDir::new().expect("temp bucket");
let built = build_query_store(path, bucket.path()).expect("build query store");
let severity = TxtSeverity::from_env().expect("OURIOS_CORPUS_SEVERITY");
let built = build_query_store(path, bucket.path(), severity).expect("build query store");
Comment thread
jensholdgaard marked this conversation as resolved.
Resolved
if built.busiest_template_rows == 0 {
eprintln!("b2/real-corpus: {dir} produced no rows — skipping");
continue;
Expand Down
Loading