//! This crate defines the core datastructure representing IDE state -- `RootDatabase`. //! //! It is mainly a `HirDatabase` for semantic analysis, plus a `SymbolsDatabase`, for fuzzy search. mod apply_change; pub mod label; pub mod line_index; pub mod symbol_index; pub mod defs; pub mod search; pub mod imports_locator; pub mod source_change; pub mod ty_filter; pub mod traits; pub mod call_info; pub mod helpers; use std::{fmt, sync::Arc}; use base_db::{ salsa::{self, Durability}, AnchoredPath, Canceled, CheckCanceled, CrateId, FileId, FileLoader, FileLoaderDelegate, SourceDatabase, Upcast, }; use hir::db::{AstDatabase, DefDatabase, HirDatabase}; use rustc_hash::FxHashSet; use crate::{line_index::LineIndex, symbol_index::SymbolsDatabase}; /// `base_db` is normally also needed in places where `ide_db` is used, so this re-export is for convenience. pub use base_db; #[salsa::database( base_db::SourceDatabaseStorage, base_db::SourceDatabaseExtStorage, LineIndexDatabaseStorage, symbol_index::SymbolsDatabaseStorage, hir::db::InternDatabaseStorage, hir::db::AstDatabaseStorage, hir::db::DefDatabaseStorage, hir::db::HirDatabaseStorage )] pub struct RootDatabase { storage: salsa::Storage<RootDatabase>, } impl fmt::Debug for RootDatabase { fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { f.debug_struct("RootDatabase").finish() } } impl Upcast<dyn AstDatabase> for RootDatabase { fn upcast(&self) -> &(dyn AstDatabase + 'static) { &*self } } impl Upcast<dyn DefDatabase> for RootDatabase { fn upcast(&self) -> &(dyn DefDatabase + 'static) { &*self } } impl Upcast<dyn HirDatabase> for RootDatabase { fn upcast(&self) -> &(dyn HirDatabase + 'static) { &*self } } impl FileLoader for RootDatabase { fn file_text(&self, file_id: FileId) -> Arc<String> { FileLoaderDelegate(self).file_text(file_id) } fn resolve_path(&self, path: AnchoredPath) -> Option<FileId> { FileLoaderDelegate(self).resolve_path(path) } fn relevant_crates(&self, file_id: FileId) -> Arc<FxHashSet<CrateId>> { FileLoaderDelegate(self).relevant_crates(file_id) } } impl salsa::Database for RootDatabase { fn on_propagated_panic(&self) -> ! { Canceled::throw() } fn salsa_event(&self, event: salsa::Event) { match event.kind { salsa::EventKind::DidValidateMemoizedValue { .. } | salsa::EventKind::WillExecute { .. } => { self.check_canceled(); } _ => (), } } } impl Default for RootDatabase { fn default() -> RootDatabase { RootDatabase::new(None) } } impl RootDatabase { pub fn new(lru_capacity: Option<usize>) -> RootDatabase { let mut db = RootDatabase { storage: salsa::Storage::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); db.update_lru_capacity(lru_capacity); db } pub fn update_lru_capacity(&mut self, lru_capacity: Option<usize>) { let lru_capacity = lru_capacity.unwrap_or(base_db::DEFAULT_LRU_CAP); base_db::ParseQuery.in_db_mut(self).set_lru_capacity(lru_capacity); hir::db::ParseMacroExpansionQuery.in_db_mut(self).set_lru_capacity(lru_capacity); hir::db::MacroExpandQuery.in_db_mut(self).set_lru_capacity(lru_capacity); } } impl salsa::ParallelDatabase for RootDatabase { fn snapshot(&self) -> salsa::Snapshot<RootDatabase> { salsa::Snapshot::new(RootDatabase { storage: self.storage.snapshot() }) } } #[salsa::query_group(LineIndexDatabaseStorage)] pub trait LineIndexDatabase: base_db::SourceDatabase + CheckCanceled { fn line_index(&self, file_id: FileId) -> Arc<LineIndex>; } fn line_index(db: &dyn LineIndexDatabase, file_id: FileId) -> Arc<LineIndex> { let text = db.file_text(file_id); Arc::new(LineIndex::new(&*text)) }