From 35568cf057e116ef3471b3c5e50beeb2f6bc7e70 Mon Sep 17 00:00:00 2001 From: Aleksey Kladov Date: Mon, 29 Oct 2018 13:32:45 +0300 Subject: Index files in parallel --- crates/ra_analysis/src/imp.rs | 22 +++++++++++++--------- 1 file changed, 13 insertions(+), 9 deletions(-) diff --git a/crates/ra_analysis/src/imp.rs b/crates/ra_analysis/src/imp.rs index 5a6e2450d..90446c838 100644 --- a/crates/ra_analysis/src/imp.rs +++ b/crates/ra_analysis/src/imp.rs @@ -10,6 +10,7 @@ use ra_syntax::{ SyntaxKind::*, SyntaxNodeRef, TextRange, TextUnit, }; +use rayon::prelude::*; use relative_path::RelativePath; use rustc_hash::FxHashSet; use salsa::{ParallelDatabase, Database}; @@ -23,6 +24,7 @@ use crate::{ input::{SourceRootId, FilesDatabase, SourceRoot, WORKSPACE}, descriptors::module::{ModulesDatabase, ModuleTree, Problem}, descriptors::{FnDescriptor}, + symbol_index::SymbolIndex, CrateGraph, CrateId, Diagnostic, FileId, FileResolver, FileSystemEdit, Position, Query, SourceChange, SourceFileEdit, Cancelable, }; @@ -180,16 +182,18 @@ impl AnalysisImpl { self.db.file_lines(file_id) } pub fn world_symbols(&self, query: Query) -> Cancelable> { - let mut buf = Vec::new(); - if query.libs { - for &lib_id in self.db.libraries().iter() { - buf.push(self.db.library_symbols(lib_id)); - } + let buf: Vec> = if query.libs { + self.db.libraries().iter() + .map(|&lib_id| self.db.library_symbols(lib_id)) + .collect() } else { - for &file_id in self.db.source_root(WORKSPACE).files.iter() { - buf.push(self.db.file_symbols(file_id)?); - } - } + let files = &self.db.source_root(WORKSPACE).files; + let db = self.db.clone(); + files.par_iter() + .map_with(db, |db, &file_id| db.file_symbols(file_id)) + .filter_map(|it| it.ok()) + .collect() + }; Ok(query.search(&buf)) } fn module_tree(&self, file_id: FileId) -> Cancelable> { -- cgit v1.2.3