mirror of
https://github.com/salsa-rs/salsa.git
synced 2025-01-23 13:10:19 +00:00
93c30a953d
Switch to a procedural implementation of the `query_group!` macro,
residing in the `components/salsa_macros` subcrate.
Allow the user to override the invoked function via `salsa::invoke(...)`
and the name of the generated query type via `salsa::query_type(...)`.
In all tests, replace the `salsa::query_group! { ... }` invocations with
the new attribute-style `#[salsa::query_group]` macro, and change them
to the new naming scheme for query types (`...Query`).
Update README, examples, and documentation.
86 lines
2 KiB
Rust
86 lines
2 KiB
Rust
use crate::db;
|
|
use crate::group::{FibonacciQuery, GcDatabase};
|
|
use salsa::debug::DebugQueryTable;
|
|
use salsa::{Database, SweepStrategy};
|
|
|
|
// For constant values (like `fibonacci`), we only keep the values
|
|
// that were used in the latest revision, not the sub-values that
|
|
// they required to be computed.
|
|
|
|
#[test]
|
|
fn one_rev() {
|
|
let db = db::DatabaseImpl::default();
|
|
|
|
db.fibonacci(5);
|
|
|
|
let k: Vec<_> = db.query(FibonacciQuery).keys();
|
|
assert_eq!(k.len(), 6);
|
|
|
|
// Everything was used in this revision, so
|
|
// nothing gets collected.
|
|
db.sweep_all(SweepStrategy::default());
|
|
assert_eq!(k.len(), 6);
|
|
}
|
|
|
|
#[test]
|
|
fn two_rev_nothing() {
|
|
let db = db::DatabaseImpl::default();
|
|
|
|
db.fibonacci(5);
|
|
|
|
let k: Vec<_> = db.query(FibonacciQuery).keys();
|
|
assert_eq!(k.len(), 6);
|
|
|
|
db.salsa_runtime().next_revision();
|
|
|
|
// Nothing was used in this revision, so
|
|
// everything gets collected.
|
|
db.sweep_all(SweepStrategy::default());
|
|
|
|
let k: Vec<_> = db.query(FibonacciQuery).keys();
|
|
assert_eq!(k.len(), 0);
|
|
}
|
|
|
|
#[test]
|
|
fn two_rev_one_use() {
|
|
let db = db::DatabaseImpl::default();
|
|
|
|
db.fibonacci(5);
|
|
|
|
let k: Vec<_> = db.query(FibonacciQuery).keys();
|
|
assert_eq!(k.len(), 6);
|
|
|
|
db.salsa_runtime().next_revision();
|
|
|
|
db.fibonacci(5);
|
|
|
|
// fibonacci is a constant, so it will not be invalidated,
|
|
// hence we keep `fibonacci(5)` but remove 0..=4.
|
|
db.sweep_all(SweepStrategy::default());
|
|
|
|
let k: Vec<_> = db.query(FibonacciQuery).keys();
|
|
assert_eq!(k, vec![5]);
|
|
}
|
|
|
|
#[test]
|
|
fn two_rev_two_uses() {
|
|
let db = db::DatabaseImpl::default();
|
|
|
|
db.fibonacci(5);
|
|
|
|
let k: Vec<_> = db.query(FibonacciQuery).keys();
|
|
assert_eq!(k.len(), 6);
|
|
|
|
db.salsa_runtime().next_revision();
|
|
|
|
db.fibonacci(5);
|
|
db.fibonacci(3);
|
|
|
|
// fibonacci is a constant, so it will not be invalidated,
|
|
// hence we keep 3 and 5 but remove the rest.
|
|
db.sweep_all(SweepStrategy::default());
|
|
|
|
let mut k: Vec<_> = db.query(FibonacciQuery).keys();
|
|
k.sort();
|
|
assert_eq!(k, vec![3, 5]);
|
|
}
|