rust/greptimedb
1
0
Fork 0
mirror of https://github.com/GreptimeTeam/greptimedb.git synced 2026-05-21 07:20:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove keys

Browse Source 
Signed-off-by: Ruihang Xia <waynestxia@gmail.com>
This commit is contained in:
Ruihang Xia
2026-05-18 18:46:25 +08:00
parent 8accd23416
commit bebcfc69b5
1 changed files with 28 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
src/mito2/src/cache.rs
View File

@@ -1246,21 +1246,30 @@ pub struct PageRangeCache {
impl PageRangeCache {
fn new(capacity: u64) -> Arc<PageRangeCache> {
let cache = Cache::builder()
.max_capacity(capacity)
.weigher(page_cache_weight)
.eviction_listener(|k, v, cause| {
let size = page_cache_weight(&k, &v);
CACHE_BYTES.with_label_values(&[PAGE_TYPE]).sub(size.into());
CACHE_EVICTION
.with_label_values(&[PAGE_TYPE, removal_cause_to_str(cause)])
.inc();
})
.build();
Arc::new_cyclic(|weak_cache: &std::sync::Weak<PageRangeCache>| {
let cache = Cache::builder()
.max_capacity(capacity)
.weigher(page_cache_weight)
.eviction_listener({
let weak_cache = weak_cache.clone();
move |k, v, cause| {
let size = page_cache_weight(&k, &v);
CACHE_BYTES.with_label_values(&[PAGE_TYPE]).sub(size.into());
CACHE_EVICTION
.with_label_values(&[PAGE_TYPE, removal_cause_to_str(cause)])
.inc();
Arc::new(PageRangeCache {
cache,
index: new_page_fragment_index_shards(),
if let Some(cache) = weak_cache.upgrade() {
cache.remove_stale_index_entries(vec![*k]);
}
}
})
.build();
PageRangeCache {
cache,
index: new_page_fragment_index_shards(),
}
})
}
@@ -1747,6 +1756,10 @@ mod tests {
&[Bytes::from(vec![2; 10])],
);
cache.cache.run_pending_tasks();
assert_eq!(
vec![PageFragmentKey::new(file_id, 0, &range)],
cache.find_index_candidates(file_id, 0, &range)
);
let lookup = cache.lookup(file_id, 0, std::slice::from_ref(&range));
assert!(lookup.is_fully_cached());
@@ -1802,6 +1815,7 @@ mod tests {
cache.cache.invalidate(&key);
cache.cache.run_pending_tasks();
assert!(cache.find_index_candidates(file_id, 0, &range).is_empty());
let lookup = cache.lookup(file_id, 0, std::slice::from_ref(&range));
assert!(!lookup.is_fully_cached());