rust/crates/vfs/src/path_interner.rs

46 lines
1.3 KiB
Rust
Raw Normal View History

2020-06-15 06:29:07 -05:00
//! Maps paths to compact integer ids. We don't care about clearings paths which
//! no longer exist -- the assumption is total size of paths we ever look at is
//! not too big.
use rustc_hash::FxHashMap;
use crate::{FileId, VfsPath};
2021-01-12 10:41:45 -06:00
/// Structure to map between [`VfsPath`] and [`FileId`].
2020-06-15 06:29:07 -05:00
#[derive(Default)]
pub(crate) struct PathInterner {
map: FxHashMap<VfsPath, FileId>,
vec: Vec<VfsPath>,
}
impl PathInterner {
2021-01-12 10:41:45 -06:00
/// Get the id corresponding to `path`.
///
/// If `path` does not exists in `self`, returns [`None`].
2020-06-15 06:29:07 -05:00
pub(crate) fn get(&self, path: &VfsPath) -> Option<FileId> {
self.map.get(path).copied()
}
2021-01-12 10:41:45 -06:00
/// Insert `path` in `self`.
///
/// - If `path` already exists in `self`, returns its associated id;
/// - Else, returns a newly allocated id.
2020-06-15 06:29:07 -05:00
pub(crate) fn intern(&mut self, path: VfsPath) -> FileId {
if let Some(id) = self.get(&path) {
return id;
}
let id = FileId(self.vec.len() as u32);
self.map.insert(path.clone(), id);
self.vec.push(path);
id
}
2021-01-12 10:41:45 -06:00
/// Returns the path corresponding to `id`.
///
/// # Panics
///
/// Panics if `id` does not exists in `self`.
2020-06-15 06:29:07 -05:00
pub(crate) fn lookup(&self, id: FileId) -> &VfsPath {
&self.vec[id.0 as usize]
}
}