Unnamed repository; edit this file 'description' to name the repository.
Diffstat (limited to 'crates/ide/src/prime_caches.rs')
| -rw-r--r-- | crates/ide/src/prime_caches.rs | 157 |
1 files changed, 141 insertions, 16 deletions
diff --git a/crates/ide/src/prime_caches.rs b/crates/ide/src/prime_caches.rs index 5eba1d1e27..b873329e86 100644 --- a/crates/ide/src/prime_caches.rs +++ b/crates/ide/src/prime_caches.rs @@ -2,10 +2,14 @@ //! sometimes is counter productive when, for example, the first goto definition //! request takes longer to compute. This modules implemented prepopulation of //! various caches, it's not really advanced at the moment. +mod topologic_sort; use hir::db::DefDatabase; -use ide_db::base_db::{SourceDatabase, SourceDatabaseExt}; -use rustc_hash::FxHashSet; +use ide_db::base_db::{ + salsa::{Database, ParallelDatabase, Snapshot}, + Cancelled, CrateGraph, CrateId, SourceDatabase, SourceDatabaseExt, +}; +use rustc_hash::{FxHashMap, FxHashSet}; use crate::RootDatabase; @@ -20,11 +24,144 @@ pub struct PrimeCachesProgress { pub(crate) fn prime_caches(db: &RootDatabase, cb: &(dyn Fn(PrimeCachesProgress) + Sync)) { let _p = profile::span("prime_caches"); let graph = db.crate_graph(); + let to_prime = compute_crates_to_prime(db, &graph); + + let n_total = to_prime.len(); + for (n_done, &crate_id) in to_prime.iter().enumerate() { + let crate_name = graph[crate_id].display_name.as_deref().unwrap_or_default().to_string(); + + cb(PrimeCachesProgress { on_crate: crate_name, n_done, n_total }); + // This also computes the DefMap + db.import_map(crate_id); + } +} + +/// We're indexing many crates. +#[derive(Debug)] +pub struct ParallelPrimeCachesProgress { + /// the crates that we are currently priming. + pub crates_currently_indexing: Vec<String>, + /// the total number of crates we want to prime. + pub crates_total: usize, + /// the total number of crates that have finished priming + pub crates_done: usize, +} + +pub(crate) fn parallel_prime_caches<F>(db: &RootDatabase, num_worker_threads: u8, cb: &F) +where + F: Fn(ParallelPrimeCachesProgress) + Sync + std::panic::UnwindSafe, +{ + let _p = profile::span("prime_caches"); + + let graph = db.crate_graph(); + let mut crates_to_prime = { + let crate_ids = compute_crates_to_prime(db, &graph); + + let mut builder = topologic_sort::TopologicalSortIter::builder(); + + for &crate_id in &crate_ids { + let crate_data = &graph[crate_id]; + let dependencies = crate_data + .dependencies + .iter() + .map(|d| d.crate_id) + .filter(|i| crate_ids.contains(i)); + + builder.add(crate_id, dependencies); + } + + builder.build() + }; + + crossbeam_utils::thread::scope(move |s| { + let (work_sender, work_receiver) = crossbeam_channel::unbounded(); + let (progress_sender, progress_receiver) = crossbeam_channel::unbounded(); + + enum ParallelPrimeCacheWorkerProgress { + BeginCrate { crate_id: CrateId, crate_name: String }, + EndCrate { crate_id: CrateId, cancelled: bool }, + } + + let prime_caches_worker = move |db: Snapshot<RootDatabase>| { + while let Ok((crate_id, crate_name)) = work_receiver.recv() { + progress_sender + .send(ParallelPrimeCacheWorkerProgress::BeginCrate { crate_id, crate_name })?; + + let cancelled = Cancelled::catch(|| { + // This also computes the DefMap + db.import_map(crate_id); + }) + .is_err(); + + progress_sender + .send(ParallelPrimeCacheWorkerProgress::EndCrate { crate_id, cancelled })?; + + if cancelled { + break; + } + } + + Ok::<_, crossbeam_channel::SendError<_>>(()) + }; + + for _ in 0..num_worker_threads { + let worker = prime_caches_worker.clone(); + let db = db.snapshot(); + s.spawn(move |_| worker(db)); + } + + let crates_total = crates_to_prime.len(); + let mut crates_done = 0; + + let mut is_cancelled = false; + let mut crates_currently_indexing = + FxHashMap::with_capacity_and_hasher(num_worker_threads as _, Default::default()); + + while !crates_to_prime.is_empty() && !is_cancelled { + for crate_id in &mut crates_to_prime { + work_sender + .send(( + crate_id, + graph[crate_id].display_name.as_deref().unwrap_or_default().to_string(), + )) + .ok(); + } + + let worker_progress = match progress_receiver.recv() { + Ok(p) => p, + Err(_) => break, + }; + match worker_progress { + ParallelPrimeCacheWorkerProgress::BeginCrate { crate_id, crate_name } => { + crates_currently_indexing.insert(crate_id, crate_name); + } + ParallelPrimeCacheWorkerProgress::EndCrate { crate_id, cancelled } => { + crates_currently_indexing.remove(&crate_id); + crates_to_prime.mark_done(crate_id); + crates_done += 1; + is_cancelled = cancelled; + } + }; + + let progress = ParallelPrimeCachesProgress { + crates_currently_indexing: crates_currently_indexing.values().cloned().collect(), + crates_done, + crates_total, + }; + + cb(progress); + db.unwind_if_cancelled(); + } + }) + .unwrap(); +} + +fn compute_crates_to_prime(db: &RootDatabase, graph: &CrateGraph) -> FxHashSet<CrateId> { // We're only interested in the workspace crates and the `ImportMap`s of their direct // dependencies, though in practice the latter also compute the `DefMap`s. // We don't prime transitive dependencies because they're generally not visible in // the current workspace. - let to_prime: FxHashSet<_> = graph + graph .iter() .filter(|&id| { let file_id = graph[id].root_file_id; @@ -32,17 +169,5 @@ pub(crate) fn prime_caches(db: &RootDatabase, cb: &(dyn Fn(PrimeCachesProgress) !db.source_root(root_id).is_library }) .flat_map(|id| graph[id].dependencies.iter().map(|krate| krate.crate_id)) - .collect(); - - // FIXME: This would be easy to parallelize, since it's in the ideal ordering for that. - // Unfortunately rayon prevents panics from propagation out of a `scope`, which breaks - // cancellation, so we cannot use rayon. - let n_total = to_prime.len(); - for (n_done, &crate_id) in to_prime.iter().enumerate() { - let crate_name = graph[crate_id].display_name.as_deref().unwrap_or_default().to_string(); - - cb(PrimeCachesProgress { on_crate: crate_name, n_done, n_total }); - // This also computes the DefMap - db.import_map(crate_id); - } + .collect() } |