Fix more bugs in directory size calculation

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

View file

@ -2453,7 +2453,7 @@ impl Application for App {
} }
} }
Message::PendingPauseAll(pause) => { Message::PendingPauseAll(pause) => {
for (id, (_, _, controller)) in self.pending_operations.iter() { for (_id, (_, _, controller)) in self.pending_operations.iter() {
if pause { if pause {
controller.pause(); controller.pause();
} else { } else {
@ -4437,10 +4437,16 @@ impl Application for App {
)); ));
} }
let mut selected_preview = None;
if self.core.window.show_context {
if let ContextPage::Preview(entity_opt, PreviewKind::Selected) = self.context_page {
selected_preview = Some(entity_opt.unwrap_or_else(|| self.tab_model.active()));
}
}
for entity in self.tab_model.iter() { for entity in self.tab_model.iter() {
if let Some(tab) = self.tab_model.data::<Tab>(entity) { if let Some(tab) = self.tab_model.data::<Tab>(entity) {
subscriptions.push( subscriptions.push(
tab.subscription() tab.subscription(selected_preview == Some(entity))
.with(entity) .with(entity)
.map(|(entity, tab_msg)| Message::TabMessage(Some(entity), tab_msg)), .map(|(entity, tab_msg)| Message::TabMessage(Some(entity), tab_msg)),
); );

View file

@ -1677,7 +1677,15 @@ impl Application for App {
std::future::pending().await std::future::pending().await
}), }),
), ),
self.tab.subscription().map(Message::TabMessage), self.tab
.subscription(
self.core.window.show_context
&& matches!(
self.context_page,
ContextPage::Preview(_, PreviewKind::Selected)
),
)
.map(Message::TabMessage),
]; ];
for (key, mounter) in MOUNTERS.iter() { for (key, mounter) in MOUNTERS.iter() {

View file

@ -4386,7 +4386,7 @@ impl Tab {
widget::responsive(|size| self.view_responsive(key_binds, size)).into() 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 { let Some(items) = &self.items_opt else {
return Subscription::none(); return Subscription::none();
}; };
@ -4470,6 +4470,7 @@ impl Tab {
} }
} }
if preview {
// Load directory size for selected items // Load directory size for selected items
for item in items for item in items
.iter() .iter()
@ -4486,15 +4487,30 @@ impl Tab {
subscriptions.push(Subscription::run_with_id( subscriptions.push(Subscription::run_with_id(
("dir_size", path.clone()), ("dir_size", path.clone()),
stream::channel(1, |mut output| async move { 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); let total_size = calculate_dir_size(&path);
log::debug!(
match output "calculated directory size of {:?} in {:?}",
.send(Message::DirectorySize(path.clone(), total_size)) path,
start.elapsed()
);
Message::DirectorySize(path.clone(), total_size)
})
.await .await
{ .unwrap()
};
match output.send(message).await {
Ok(()) => {} Ok(()) => {}
Err(err) => { Err(err) => {
log::warn!("failed to send dirsize for {:?}: {}", &path, err); log::warn!(
"failed to send directory size for {:?}: {}",
&path,
err
);
} }
} }
@ -4504,6 +4520,7 @@ impl Tab {
// Only calculate size for one directory // Only calculate size for one directory
break; break;
} }
}
// Load search items incrementally // Load search items incrementally
if let Location::Search(path, term, show_hidden, start) = &self.location { if let Location::Search(path, term, show_hidden, start) = &self.location {