use std::sync::Arc; #[cfg(test)] use parking_lot::Mutex; use ra_editor::LineIndex; use ra_syntax::{SourceFileNode, SyntaxNode}; use salsa::{self, Database}; use crate::{ db, descriptors::{ DescriptorDatabase, FnScopesQuery, FnSyntaxQuery, ModuleTreeQuery, SubmodulesQuery, ItemMapQuery, InputModuleItemsQuery, }, symbol_index::SymbolIndex, syntax_ptr::SyntaxPtr, loc2id::{IdMaps, IdDatabase}, Cancelable, Canceled, FileId, }; #[derive(Debug)] pub(crate) struct RootDatabase { #[cfg(test)] events: Mutex>>>, #[cfg(not(test))] events: (), runtime: salsa::Runtime, id_maps: IdMaps, } impl salsa::Database for RootDatabase { fn salsa_runtime(&self) -> &salsa::Runtime { &self.runtime } #[allow(unused)] fn salsa_event(&self, event: impl Fn() -> salsa::Event) { #[cfg(test)] { let mut events = self.events.lock(); if let Some(events) = &mut *events { events.push(event()); } } } } impl Default for RootDatabase { fn default() -> RootDatabase { let mut db = RootDatabase { events: Default::default(), runtime: salsa::Runtime::default(), id_maps: IdMaps::default(), }; db.query_mut(crate::input::SourceRootQuery) .set(crate::input::WORKSPACE, Default::default()); db.query_mut(crate::input::CrateGraphQuery) .set((), Default::default()); db.query_mut(crate::input::LibrariesQuery) .set((), Default::default()); db } } pub(crate) fn check_canceled(db: &impl salsa::Database) -> Cancelable<()> { if db.salsa_runtime().is_current_revision_canceled() { Err(Canceled) } else { Ok(()) } } impl salsa::ParallelDatabase for RootDatabase { fn snapshot(&self) -> salsa::Snapshot { salsa::Snapshot::new(RootDatabase { events: Default::default(), runtime: self.runtime.snapshot(self), id_maps: self.id_maps.clone(), }) } } impl IdDatabase for RootDatabase { fn id_maps(&self) -> &IdMaps { &self.id_maps } } #[cfg(test)] impl RootDatabase { pub(crate) fn log(&self, f: impl FnOnce()) -> Vec> { *self.events.lock() = Some(Vec::new()); f(); let events = self.events.lock().take().unwrap(); events } pub(crate) fn log_executed(&self, f: impl FnOnce()) -> Vec { let events = self.log(f); events .into_iter() .filter_map(|e| match e.kind { // This pretty horrible, but `Debug` is the only way to inspect // QueryDescriptor at the moment. salsa::EventKind::WillExecute { descriptor } => Some(format!("{:?}", descriptor)), _ => None, }) .collect() } } salsa::database_storage! { pub(crate) struct RootDatabaseStorage for RootDatabase { impl crate::input::FilesDatabase { fn file_text() for crate::input::FileTextQuery; fn file_source_root() for crate::input::FileSourceRootQuery; fn source_root() for crate::input::SourceRootQuery; fn libraries() for crate::input::LibrariesQuery; fn library_symbols() for crate::input::LibrarySymbolsQuery; fn crate_graph() for crate::input::CrateGraphQuery; } impl SyntaxDatabase { fn file_syntax() for FileSyntaxQuery; fn file_lines() for FileLinesQuery; fn file_symbols() for FileSymbolsQuery; fn resolve_syntax_ptr() for ResolveSyntaxPtrQuery; } impl DescriptorDatabase { fn module_tree() for ModuleTreeQuery; fn fn_scopes() for FnScopesQuery; fn _input_module_items() for InputModuleItemsQuery; fn _item_map() for ItemMapQuery; fn _fn_syntax() for FnSyntaxQuery; fn _submodules() for SubmodulesQuery; } } } salsa::query_group! { pub(crate) trait SyntaxDatabase: crate::input::FilesDatabase { fn file_syntax(file_id: FileId) -> SourceFileNode { type FileSyntaxQuery; } fn file_lines(file_id: FileId) -> Arc { type FileLinesQuery; } fn file_symbols(file_id: FileId) -> Cancelable> { type FileSymbolsQuery; } fn resolve_syntax_ptr(ptr: SyntaxPtr) -> SyntaxNode { type ResolveSyntaxPtrQuery; // Don't retain syntax trees in memory storage volatile; use fn crate::syntax_ptr::resolve_syntax_ptr; } } } fn file_syntax(db: &impl SyntaxDatabase, file_id: FileId) -> SourceFileNode { let text = db.file_text(file_id); SourceFileNode::parse(&*text) } fn file_lines(db: &impl SyntaxDatabase, file_id: FileId) -> Arc { let text = db.file_text(file_id); Arc::new(LineIndex::new(&*text)) } fn file_symbols(db: &impl SyntaxDatabase, file_id: FileId) -> Cancelable> { db::check_canceled(db)?; let syntax = db.file_syntax(file_id); Ok(Arc::new(SymbolIndex::for_file(file_id, syntax))) }