mirror of
https://github.com/salsa-rs/salsa.git
synced 2024-11-24 12:16:25 +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.
45 lines
971 B
Rust
45 lines
971 B
Rust
mod common;
|
|
|
|
use expect_test::expect;
|
|
use salsa::{Accumulator, Database};
|
|
use test_log::test;
|
|
|
|
#[salsa::input]
|
|
struct MyInput {
|
|
count: u32,
|
|
}
|
|
|
|
#[salsa::accumulator(no_debug)]
|
|
struct Log(String);
|
|
|
|
impl std::fmt::Debug for Log {
|
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
|
f.debug_tuple("CustomLog").field(&self.0).finish()
|
|
}
|
|
}
|
|
|
|
#[salsa::tracked]
|
|
fn push_logs(db: &dyn salsa::Database, input: MyInput) {
|
|
for i in 0..input.count(db) {
|
|
Log(format!("#{i}")).accumulate(db);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn accumulate_custom_debug() {
|
|
salsa::DatabaseImpl::new().attach(|db| {
|
|
let input = MyInput::new(db, 2);
|
|
let logs = push_logs::accumulated::<Log>(db, input);
|
|
expect![[r##"
|
|
[
|
|
CustomLog(
|
|
"#0",
|
|
),
|
|
CustomLog(
|
|
"#1",
|
|
),
|
|
]
|
|
"##]]
|
|
.assert_debug_eq(&logs);
|
|
})
|
|
}
|