zed-industries/zed
1
1
Fork 0
mirror of https://github.com/zed-industries/zed.git synced 2024-12-29 06:24:14 +03:00
Code Issues Packages Projects Releases Wiki Activity

Increment job counter on JobClient::new

Browse Source 
Co-authored-by: Kyle <kyle@zed.dev>
This commit is contained in:
Piotr Osiewicz 2023-08-21 13:00:56 +02:00
parent def215af9f
commit 1a88444f2f
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
crates/semantic_index/src/semantic_index.rs
View File

@ -103,6 +103,7 @@ struct JobHandle {
impl JobHandle { impl JobHandle {
fn new(tx: &Arc<Mutex<watch::Sender<usize>>>) -> Self { fn new(tx: &Arc<Mutex<watch::Sender<usize>>>) -> Self {
*tx.lock().borrow_mut() += 1;
Self { Self {
tx: Arc::new(Arc::downgrade(&tx)), tx: Arc::new(Arc::downgrade(&tx)),
} }
@ -656,10 +657,8 @@ impl SemanticIndex {
if !already_stored { if !already_stored {
count += 1; count += 1;
*job_count_tx.lock().borrow_mut() += 1;
let job_handle = JobHandle { let job_handle = JobHandle::new(&job_count_tx);
tx: Arc::new(Arc::downgrade(&job_count_tx)),
};
parsing_files_tx parsing_files_tx
.try_send(PendingFile { .try_send(PendingFile {
worktree_db_id: db_ids_by_worktree_id[&worktree.id()], worktree_db_id: db_ids_by_worktree_id[&worktree.id()],
@ -865,8 +864,14 @@ mod tests {
fn test_job_handle() { fn test_job_handle() {
let (job_count_tx, job_count_rx) = watch::channel_with(0); let (job_count_tx, job_count_rx) = watch::channel_with(0);
let tx = Arc::new(Mutex::new(job_count_tx)); let tx = Arc::new(Mutex::new(job_count_tx));
let job_handle = JobHandle::new(tx); let job_handle = JobHandle::new(&tx);
assert_eq!(1, *job_count_rx.borrow_mut()); assert_eq!(1, *job_count_rx.borrow());
let new_job_handle = job_handle.clone();
assert_eq!(1, *job_count_rx.borrow());
drop(job_handle);
assert_eq!(1, *job_count_rx.borrow());
drop(new_job_handle);
assert_eq!(0, *job_count_rx.borrow());
} }
} }