mirror of
https://github.com/salsa-rs/salsa.git
synced 2024-10-23 12:49:57 +00:00
daaa78056a
Under this design, *all* databases are a `DatabaseImpl<U>`, where the `U` implements `UserData` (you can use `()` if there is none). Code would default to `&dyn salsa::Database` but if you want to give access to the userdata, you can define a custom database trait `MyDatabase: salsa::Databse` so long as you * annotate `MyDatabase` trait definition of impls of `MyDatabase` with `#[salsa::db]` * implement `MyDatabase` for `DatabaseImpl<U>` where `U` is your userdata (this could be a blanket impl, if you don't know the precise userdata type). The `tests/common/mod.rs` shows the pattern.
55 lines
1.4 KiB
Rust
55 lines
1.4 KiB
Rust
use codspeed_criterion_compat::{criterion_group, criterion_main, BatchSize, Criterion};
|
|
use salsa::Setter;
|
|
|
|
#[salsa::input]
|
|
struct Input {
|
|
field: usize,
|
|
}
|
|
|
|
#[salsa::tracked]
|
|
struct Tracked<'db> {
|
|
number: usize,
|
|
}
|
|
|
|
#[salsa::tracked(return_ref)]
|
|
fn index<'db>(db: &'db dyn salsa::Database, input: Input) -> Vec<Tracked<'db>> {
|
|
(0..input.field(db)).map(|i| Tracked::new(db, i)).collect()
|
|
}
|
|
|
|
#[salsa::tracked]
|
|
fn root(db: &dyn salsa::Database, input: Input) -> usize {
|
|
let index = index(db, input);
|
|
index.len()
|
|
}
|
|
|
|
fn many_tracked_structs(criterion: &mut Criterion) {
|
|
criterion.bench_function("many_tracked_structs", |b| {
|
|
b.iter_batched_ref(
|
|
|| {
|
|
let db = salsa::DatabaseImpl::new();
|
|
|
|
let input = Input::new(&db, 1_000);
|
|
let input2 = Input::new(&db, 1);
|
|
|
|
// prewarm cache
|
|
let _ = root(&db, input);
|
|
let _ = root(&db, input2);
|
|
|
|
(db, input, input2)
|
|
},
|
|
|(db, input, input2)| {
|
|
// Make a change, but fetch the result for the other input
|
|
input2.set_field(db).to(2);
|
|
|
|
let result = root(db, *input);
|
|
|
|
assert_eq!(result, 1_000);
|
|
},
|
|
BatchSize::LargeInput,
|
|
);
|
|
});
|
|
}
|
|
|
|
criterion_group!(benches, many_tracked_structs);
|
|
criterion_main!(benches);
|