rust-analyzer/crates/ra-salsa/tests/storage_varieties/queries.rs
2024-10-14 10:09:22 -04:00

22 lines
598 B
Rust

pub(crate) trait Counter: ra_salsa::Database {
fn increment(&self) -> usize;
}
#[ra_salsa::query_group(GroupStruct)]
pub(crate) trait Database: Counter {
fn memoized(&self) -> usize;
fn volatile(&self) -> usize;
}
/// Because this query is memoized, we only increment the counter
/// the first time it is invoked.
fn memoized(db: &dyn Database) -> usize {
db.volatile()
}
/// Because this query is volatile, each time it is invoked,
/// we will increment the counter.
fn volatile(db: &dyn Database) -> usize {
db.salsa_runtime().report_untracked_read();
db.increment()
}