//! FIXME: write short doc here use std::sync::Arc; use ra_db::{ salsa::{self, Database, Durability}, Canceled, CheckCanceled, CrateId, FileId, FileLoader, FileLoaderDelegate, SourceDatabase, SourceDatabaseExt, SourceRootId, }; use relative_path::RelativePath; use rustc_hash::FxHashMap; use crate::{ symbol_index::{self, SymbolsDatabase}, FeatureFlags, LineIndex, }; #[salsa::database( ra_db::SourceDatabaseStorage, ra_db::SourceDatabaseExtStorage, LineIndexDatabaseStorage, symbol_index::SymbolsDatabaseStorage, hir::db::InternDatabaseStorage, hir::db::AstDatabaseStorage, hir::db::DefDatabaseStorage, hir::db::HirDatabaseStorage )] #[derive(Debug)] pub(crate) struct RootDatabase { runtime: salsa::Runtime, pub(crate) feature_flags: Arc, pub(crate) debug_data: Arc, pub(crate) last_gc: crate::wasm_shims::Instant, pub(crate) last_gc_check: crate::wasm_shims::Instant, } impl FileLoader for RootDatabase { fn file_text(&self, file_id: FileId) -> Arc { FileLoaderDelegate(self).file_text(file_id) } fn resolve_relative_path( &self, anchor: FileId, relative_path: &RelativePath, ) -> Option { FileLoaderDelegate(self).resolve_relative_path(anchor, relative_path) } fn relevant_crates(&self, file_id: FileId) -> Arc> { FileLoaderDelegate(self).relevant_crates(file_id) } } impl hir::debug::HirDebugHelper for RootDatabase { fn crate_name(&self, krate: CrateId) -> Option { self.debug_data.crate_names.get(&krate).cloned() } fn file_path(&self, file_id: FileId) -> Option { let source_root_id = self.file_source_root(file_id); let source_root_path = self.debug_data.root_paths.get(&source_root_id)?; let file_path = self.file_relative_path(file_id); Some(format!("{}/{}", source_root_path, file_path.display())) } } impl salsa::Database for RootDatabase { fn salsa_runtime(&self) -> &salsa::Runtime { &self.runtime } fn on_propagated_panic(&self) -> ! { Canceled::throw() } fn salsa_event(&self, event: impl Fn() -> salsa::Event) { match event().kind { salsa::EventKind::DidValidateMemoizedValue { .. } | salsa::EventKind::WillExecute { .. } => { self.check_canceled(); } _ => (), } } } impl Default for RootDatabase { fn default() -> RootDatabase { RootDatabase::new(None, FeatureFlags::default()) } } impl RootDatabase { pub fn new(lru_capacity: Option, feature_flags: FeatureFlags) -> RootDatabase { let mut db = RootDatabase { runtime: salsa::Runtime::default(), last_gc: crate::wasm_shims::Instant::now(), last_gc_check: crate::wasm_shims::Instant::now(), feature_flags: Arc::new(feature_flags), debug_data: Default::default(), }; db.set_crate_graph_with_durability(Default::default(), Durability::HIGH); db.set_local_roots_with_durability(Default::default(), Durability::HIGH); db.set_library_roots_with_durability(Default::default(), Durability::HIGH); let lru_capacity = lru_capacity.unwrap_or(ra_db::DEFAULT_LRU_CAP); db.query_mut(ra_db::ParseQuery).set_lru_capacity(lru_capacity); db.query_mut(hir::db::ParseMacroQuery).set_lru_capacity(lru_capacity); db.query_mut(hir::db::MacroExpandQuery).set_lru_capacity(lru_capacity); db } } impl salsa::ParallelDatabase for RootDatabase { fn snapshot(&self) -> salsa::Snapshot { salsa::Snapshot::new(RootDatabase { runtime: self.runtime.snapshot(self), last_gc: self.last_gc, last_gc_check: self.last_gc_check, feature_flags: Arc::clone(&self.feature_flags), debug_data: Arc::clone(&self.debug_data), }) } } #[salsa::query_group(LineIndexDatabaseStorage)] pub(crate) trait LineIndexDatabase: ra_db::SourceDatabase + CheckCanceled { fn line_index(&self, file_id: FileId) -> Arc; } fn line_index(db: &impl ra_db::SourceDatabase, file_id: FileId) -> Arc { let text = db.file_text(file_id); Arc::new(LineIndex::new(&*text)) } #[derive(Debug, Default, Clone)] pub(crate) struct DebugData { pub(crate) root_paths: FxHashMap, pub(crate) crate_names: FxHashMap, } impl DebugData { pub(crate) fn merge(&mut self, other: DebugData) { self.root_paths.extend(other.root_paths.into_iter()); self.crate_names.extend(other.crate_names.into_iter()); } }