leyoda/cosmic-files
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix more bugs in directory size calculation

Browse source
This commit is contained in:
Jeremy Soller 2024-11-15 15:37:31 -07:00
parent 5f944a3059
commit 0c43b0b342
No known key found for this signature in database
GPG key ID: D02FD439211AF56F
3 changed files with 65 additions and 34 deletions
Download patch file Download diff file Expand all files Collapse all files

79
src/tab.rs
View file

@ -4386,7 +4386,7 @@ impl Tab {
widget::responsive(|size| self.view_responsive(key_binds, size)).into()
}
pub fn subscription(&self) -> Subscription<Message> {
pub fn subscription(&self, preview: bool) -> Subscription<Message> {
let Some(items) = &self.items_opt else {
return Subscription::none();
};
@ -4470,39 +4470,56 @@ impl Tab {
}
}
// Load directory size for selected items
for item in items
.iter()
.filter(|item| {
// Item must be a selected directory
item.selected && item.metadata.is_dir()
})
.chain(&self.parent_item_opt)
{
// Item must have a path
let Some(path) = item.path_opt().map(|path| path.to_path_buf()) else {
continue;
};
subscriptions.push(Subscription::run_with_id(
("dir_size", path.clone()),
stream::channel(1, |mut output| async move {
let total_size = calculate_dir_size(&path);
if preview {
// Load directory size for selected items
for item in items
.iter()
.filter(|item| {
// Item must be a selected directory
item.selected && item.metadata.is_dir()
})
.chain(&self.parent_item_opt)
{
// Item must have a path
let Some(path) = item.path_opt().map(|path| path.to_path_buf()) else {
continue;
};
subscriptions.push(Subscription::run_with_id(
("dir_size", path.clone()),
stream::channel(1, |mut output| async move {
let message = {
let path = path.clone();
tokio::task::spawn_blocking(move || {
let start = Instant::now();
let total_size = calculate_dir_size(&path);
log::debug!(
"calculated directory size of {:?} in {:?}",
path,
start.elapsed()
);
Message::DirectorySize(path.clone(), total_size)
})
.await
.unwrap()
};
match output
.send(Message::DirectorySize(path.clone(), total_size))
.await
{
Ok(()) => {}
Err(err) => {
log::warn!("failed to send dirsize for {:?}: {}", &path, err);
match output.send(message).await {
Ok(()) => {}
Err(err) => {
log::warn!(
"failed to send directory size for {:?}: {}",
&path,
err
);
}
}
}
std::future::pending().await
}),
));
// Only calculate size for one directory
break;
std::future::pending().await
}),
));
// Only calculate size for one directory
break;
}
}
// Load search items incrementally