salsa/tests/storage_varieties/queries.rs

pub(crate) trait Counter: salsa::Database {
    fn increment(&self) -> usize;
}

salsa::query_group! {
    pub(crate) trait Database: Counter {
        fn memoized(key: ()) -> usize {
            type Memoized;
        }
        fn volatile(key: ()) -> usize {
            type Volatile;
            storage volatile;
        }
    }
}

/// Because this query is memoized, we only increment the counter
/// the first time it is invoked.
fn memoized(db: &impl Database, (): ()) -> usize {
    db.volatile(())
}

/// Because this query is volatile, each time it is invoked,
/// we will increment the counter.
fn volatile(db: &impl Database, (): ()) -> usize {
    db.increment()
}