Implement etop

1 files changed, 24 insertions, 15 deletions

diff --git a/src/socket.rs b/src/socket.rs
index 316c662..e2ecdbf 100644
--- a/src/socket.rs
+++ b/src/socket.rs
@@ -35,21 +35,9 @@ impl LoggingSocketService {
                         break;
                     }
                     if let Ok(pid) = line.trim().parse::<Pid>() {
-                        if let Some(info) = self.process_info.read().unwrap().get(&pid).clone() {
-                            socket
-                                .write_all(
-                                    format!(
-                                        "pid: {pid} process: {}J process tree: {}J\n",
-                                        info.energy, info.tree_energy
-                                    )
-                                    .as_bytes(),
-                                )
-                                .unwrap();
-                        } else {
-                            socket
-                                .write_all(format!("Unknown pid: {pid}\n").as_bytes())
-                                .unwrap();
-                        }
+                        socket
+                            .write_all(self.handle_request(pid).as_bytes())
+                            .unwrap();
                     } else {
                         socket
                             .write_all(
@@ -63,6 +51,27 @@ impl LoggingSocketService {
             }
         });
     }
+
+    fn handle_request(&self, pid: i32) -> String {
+        let mut output = String::new();
+        use std::fmt::Write;
+        if pid == -1 {
+            for (pid, info) in self.process_info.read().unwrap().iter() {
+                writeln!(&mut output, "{pid},{},{}", info.energy, info.tree_energy).unwrap();
+            }
+            writeln!(&mut output, "#",).unwrap();
+            return output;
+        }
+
+        if let Some(info) = self.process_info.read().unwrap().get(&pid) {
+            format!(
+                "pid: {pid} process: {}J process tree: {}J\n",
+                info.energy, info.tree_energy
+            )
+        } else {
+            format!("Unknown pid: {pid}\n")
+        }
+    }
 }
 
 pub fn start_logging_socket_service(