use crate::{GlobalTableId, fs::FsFile};
use quick_cache::{UnitWeighter, sync::Cache as QuickCache};
use std::sync::Arc;
const TAG_BLOCK: u8 = 0;
const TAG_BLOB: u8 = 1;
type Item = Arc<dyn FsFile>;
#[derive(Eq, std::hash::Hash, PartialEq)]
struct CacheKey(u8, u64, u64);
pub struct DescriptorTable {
inner: QuickCache<CacheKey, Item, UnitWeighter, rustc_hash::FxBuildHasher>,
}
impl DescriptorTable {
#[must_use]
pub fn new(capacity: usize) -> Self {
use quick_cache::sync::DefaultLifecycle;
let quick_cache = QuickCache::with(
1_000,
capacity as u64,
UnitWeighter,
rustc_hash::FxBuildHasher,
DefaultLifecycle::default(),
);
Self { inner: quick_cache }
}
pub(crate) fn len(&self) -> usize {
self.inner.len()
}
#[must_use]
pub fn access_for_table(&self, id: &GlobalTableId) -> Option<Arc<dyn FsFile>> {
let key = CacheKey(TAG_BLOCK, id.tree_id(), id.table_id());
self.inner.get(&key)
}
pub fn insert_for_table(&self, id: GlobalTableId, item: Item) {
let key = CacheKey(TAG_BLOCK, id.tree_id(), id.table_id());
self.inner.insert(key, item);
}
#[must_use]
pub fn access_for_blob_file(&self, id: &GlobalTableId) -> Option<Arc<dyn FsFile>> {
let key = CacheKey(TAG_BLOB, id.tree_id(), id.table_id());
self.inner.get(&key)
}
pub fn insert_for_blob_file(&self, id: GlobalTableId, item: Item) {
let key = CacheKey(TAG_BLOB, id.tree_id(), id.table_id());
self.inner.insert(key, item);
}
pub fn remove_for_table(&self, id: &GlobalTableId) {
let key = CacheKey(TAG_BLOCK, id.tree_id(), id.table_id());
self.inner.remove(&key);
}
pub fn remove_for_blob_file(&self, id: &GlobalTableId) {
let key = CacheKey(TAG_BLOB, id.tree_id(), id.table_id());
self.inner.remove(&key);
}
}