switch interner to use arena

This commit is contained in:
Aleksey Kladov 2019-01-08 15:53:32 +03:00
parent 5603237c06
commit d4c8310d05
6 changed files with 23 additions and 40 deletions

1
Cargo.lock generated
View file

@ -678,6 +678,7 @@ name = "ra_db"
version = "0.1.0"
dependencies = [
"parking_lot 0.7.1 (registry+https://github.com/rust-lang/crates.io-index)",
"ra_arena 0.1.0",
"ra_editor 0.1.0",
"ra_syntax 0.1.0",
"relative-path 0.4.0 (registry+https://github.com/rust-lang/crates.io-index)",

View file

@ -61,6 +61,9 @@ pub trait ArenaId {
}
impl<ID: ArenaId, T> Arena<ID, T> {
pub fn len(&self) -> usize {
self.data.len()
}
pub fn alloc(&mut self, value: T) -> ID {
let id = RawId(self.data.len() as u32);
self.data.push(value);

View file

@ -9,6 +9,7 @@ relative-path = "0.4.0"
salsa = "0.9.1"
rustc-hash = "1.0"
parking_lot = "0.7.0"
ra_arena = { path = "../ra_arena" }
ra_syntax = { path = "../ra_syntax" }
ra_editor = { path = "../ra_editor" }
test_utils = { path = "../test_utils" }

View file

@ -18,23 +18,9 @@ pub use crate::{
FileTextQuery, FileSourceRootQuery, SourceRootQuery, LocalRootsQuery, LibraryRootsQuery, CrateGraphQuery,
FileRelativePathQuery
},
loc2id::{LocationIntener, NumericId},
loc2id::LocationIntener,
};
#[macro_export]
macro_rules! impl_numeric_id {
($id:ident) => {
impl $crate::NumericId for $id {
fn from_u32(id: u32) -> Self {
$id(id)
}
fn to_u32(self) -> u32 {
self.0
}
}
};
}
pub trait BaseDatabase: salsa::Database {
fn check_canceled(&self) -> Cancelable<()> {
if self.salsa_runtime().is_current_revision_canceled() {

View file

@ -1,8 +1,8 @@
use parking_lot::Mutex;
use std::hash::Hash;
use parking_lot::Mutex;
use rustc_hash::FxHashMap;
use ra_arena::{Arena, ArenaId};
/// There are two principle ways to refer to things:
/// - by their locatinon (module in foo/bar/baz.rs at line 42)
@ -17,33 +17,33 @@ use rustc_hash::FxHashMap;
#[derive(Debug)]
struct Loc2IdMap<LOC, ID>
where
ID: NumericId,
ID: ArenaId + Clone,
LOC: Clone + Eq + Hash,
{
id2loc: Arena<ID, LOC>,
loc2id: FxHashMap<LOC, ID>,
id2loc: FxHashMap<ID, LOC>,
}
impl<LOC, ID> Default for Loc2IdMap<LOC, ID>
where
ID: NumericId,
ID: ArenaId + Clone,
LOC: Clone + Eq + Hash,
{
fn default() -> Self {
Loc2IdMap {
id2loc: Arena::default(),
loc2id: FxHashMap::default(),
id2loc: FxHashMap::default(),
}
}
}
impl<LOC, ID> Loc2IdMap<LOC, ID>
where
ID: NumericId,
ID: ArenaId + Clone,
LOC: Clone + Eq + Hash,
{
pub fn len(&self) -> usize {
self.loc2id.len()
self.id2loc.len()
}
pub fn loc2id(&mut self, loc: &LOC) -> ID {
@ -51,28 +51,20 @@ where
Some(id) => return id.clone(),
None => (),
}
let id = self.loc2id.len();
assert!(id < u32::max_value() as usize);
let id = ID::from_u32(id as u32);
let id = self.id2loc.alloc(loc.clone());
self.loc2id.insert(loc.clone(), id.clone());
self.id2loc.insert(id.clone(), loc.clone());
id
}
pub fn id2loc(&self, id: ID) -> LOC {
self.id2loc[&id].clone()
self.id2loc[id].clone()
}
}
pub trait NumericId: Clone + Eq + Hash {
fn from_u32(id: u32) -> Self;
fn to_u32(self) -> u32;
}
#[derive(Debug)]
pub struct LocationIntener<LOC, ID>
where
ID: NumericId,
ID: ArenaId + Clone,
LOC: Clone + Eq + Hash,
{
map: Mutex<Loc2IdMap<LOC, ID>>,
@ -80,7 +72,7 @@ where
impl<LOC, ID> Default for LocationIntener<LOC, ID>
where
ID: NumericId,
ID: ArenaId + Clone,
LOC: Clone + Eq + Hash,
{
fn default() -> Self {
@ -92,7 +84,7 @@ where
impl<LOC, ID> LocationIntener<LOC, ID>
where
ID: NumericId,
ID: ArenaId + Clone,
LOC: Clone + Eq + Hash,
{
pub fn len(&self) -> usize {

View file

@ -93,8 +93,8 @@ impl From<MacroCallId> for HirFileId {
/// `MacroCallId` identifies a particular macro invocation, like
/// `println!("Hello, {}", world)`.
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
pub struct MacroCallId(u32);
ra_db::impl_numeric_id!(MacroCallId);
pub struct MacroCallId(RawId);
impl_arena_id!(MacroCallId);
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
pub struct MacroCallLoc {
@ -125,8 +125,8 @@ impl MacroCallLoc {
/// Def's are a core concept of hir. A `Def` is an Item (function, module, etc)
/// in a specific module.
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
pub struct DefId(u32);
ra_db::impl_numeric_id!(DefId);
pub struct DefId(RawId);
impl_arena_id!(DefId);
#[derive(Clone, Debug, PartialEq, Eq, Hash)]
pub struct DefLoc {