diff options
| -rw-r--r-- | src/display/ui_state.rs | 17 |
1 files changed, 10 insertions, 7 deletions
diff --git a/src/display/ui_state.rs b/src/display/ui_state.rs index de5fd9a..c6d9b41 100644 --- a/src/display/ui_state.rs +++ b/src/display/ui_state.rs @@ -134,20 +134,23 @@ impl UIState { total_bytes_downloaded += connection_info.total_bytes_downloaded; total_bytes_uploaded += connection_info.total_bytes_uploaded; + let data_for_process: &mut NetworkData; + if let Some(process_name) = UIState::get_proc_name(&connections_to_procs, &connection.local_socket) { - let data_for_process = processes.entry(process_name.clone()).or_default(); - data_for_process.total_bytes_downloaded += - connection_info.total_bytes_downloaded; - data_for_process.total_bytes_uploaded += connection_info.total_bytes_uploaded; - if !connection_previously_seen { - data_for_process.connection_count += 1; - } + data_for_process = processes.entry(process_name.clone()).or_default(); connection_data.process_name = process_name.clone(); } else { + data_for_process = processes.entry(String::from("<UNKNOWN>")).or_default(); connection_data.process_name = String::from("<UNKNOWN>"); } + + data_for_process.total_bytes_downloaded += connection_info.total_bytes_downloaded; + data_for_process.total_bytes_uploaded += connection_info.total_bytes_uploaded; + if !connection_previously_seen { + data_for_process.connection_count += 1; + } } } let divide_by = if self.utilization_data.is_empty() { |