Piotr OsiewiczandSmit Barmase
e85c060625
Follow-up after #40417, which should've fixed hangs. smol::fs uses a separate threadpool, which is a bit yuck. This PR also added a benchmark you can use to run a full worktree scan (initial one, that is) for arbitrary worktree.. and refactored worktree scanner to use async locks, as otherwise tests were deadlocking. :) I've benchmarked it against Zed, Linux and Chromium and saw a ~60% drop in initial worktree scan times across the board. Release Notes: - Significantly (3.3x speedup over the old implementation) improved speed of Zed's worktree scanner, that's responsible for synchronizing the state of your project with the state of files on hard drive. --------- Co-authored-by: Smit Barmase <heysmitbarmase@gmail.com>
55 lines
1.6 KiB
Rust
55 lines
1.6 KiB
Rust
use std::{
|
|
path::Path,
|
|
sync::{Arc, atomic::AtomicUsize},
|
|
};
|
|
|
|
use fs::RealFs;
|
|
use gpui::Application;
|
|
use settings::Settings;
|
|
use worktree::{Worktree, WorktreeSettings};
|
|
|
|
fn main() {
|
|
let Some(worktree_root_path) = std::env::args().nth(1) else {
|
|
println!(
|
|
"Missing path to worktree root\nUsage: bench_background_scan PATH_TO_WORKTREE_ROOT"
|
|
);
|
|
return;
|
|
};
|
|
let app = Application::headless();
|
|
|
|
app.run(|cx| {
|
|
settings::init(cx);
|
|
WorktreeSettings::register(cx);
|
|
let fs = Arc::new(RealFs::new(None, cx.background_executor().clone()));
|
|
|
|
cx.spawn(async move |cx| {
|
|
let worktree = Worktree::local(
|
|
Path::new(&worktree_root_path),
|
|
true,
|
|
fs,
|
|
Arc::new(AtomicUsize::new(0)),
|
|
cx,
|
|
)
|
|
.await
|
|
.expect("Worktree initialization to succeed");
|
|
let did_finish_scan = worktree
|
|
.update(cx, |this, _| this.as_local().unwrap().scan_complete())
|
|
.unwrap();
|
|
let start = std::time::Instant::now();
|
|
did_finish_scan.await;
|
|
let elapsed = start.elapsed();
|
|
let (files, directories) = worktree
|
|
.read_with(cx, |this, _| (this.file_count(), this.dir_count()))
|
|
.unwrap();
|
|
println!(
|
|
"{:?} for {directories} directories and {files} files",
|
|
elapsed
|
|
);
|
|
cx.update(|cx| {
|
|
cx.quit();
|
|
})
|
|
})
|
|
.detach();
|
|
})
|
|
}
|