Display some server process information (memory, cpu usage, uptime)

[valheim_web.git] / backend / src / valheim_controller.rs

diff --git a/backend/src/valheim_controller.rs b/backend/src/valheim_controller.rs
index 3fd4970..2ce5ea4 100644 (file)
--- a/backend/src/valheim_controller.rs
+++ b/backend/src/valheim_controller.rs
@@ -1,15 +1,66 @@
+use std::cmp::min;\r
+\r
 use sysinfo::{ ProcessExt, SystemExt };\r
 \r
-pub fn is_valheim_running() -> bool {\r
+#[derive(Debug)]\r
+pub struct ValheimExe {\r
+    memory: u64, // [kB].\r
+    cpu_usage: f32, // [%].\r
+    uptime: u64, // [s].\r
+}\r
+\r
+impl ValheimExe {\r
+    pub fn format_memory(&self) -> String {\r
+        format_byte_size(self.memory * 1024, 2)\r
+    }\r
+    pub fn format_cpu_usage(&self) -> String {\r
+        format!("{:.2} %", self.cpu_usage)\r
+    }\r
+    pub fn format_uptime(&self) -> String {\r
+        let mins = self.uptime / 60;\r
+        let hours = mins / 60;\r
+        format!("{}h{}min", hours, mins - 60 * hours)\r
+    }\r
+}\r
+\r
+const BINARY_PREFIXES: [&str; 8] = ["B", "KiB", "MiB", "GiB", "TiB", "PiB", "EiB", "ZiB"];\r
+\r
+fn format_byte_size(bytes: u64, precision: usize) -> String {\r
+    for i in 0 .. 8 {\r
+        let mut size: u64 = 1;\r
+        for j in 0 .. i {\r
+            size *= 1024;\r
+        }\r
+\r
+        if bytes < 1024 {\r
+            return format!("{} {}", std::cmp::max(0u64, bytes), BINARY_PREFIXES[i]);\r
+        }\r
+        else if bytes < 1024 * size {\r
+            return format!("{:.prec$} {}", bytes as f64 / size as f64, BINARY_PREFIXES[i], prec = precision);\r
+        }\r
+    }\r
+\r
+    String::from("")\r
+}\r
+\r
+const VALHEIM_PROCESS_NAME: &str = "valheim_server";\r
+\r
+pub fn get_valheim_executable_information() -> Option<ValheimExe> {\r
     let mut system = sysinfo::System::new_all();\r
     system.refresh_all();\r
-    let process = system.get_process_by_name("valheim_server");\r
+    let processes = system.get_process_by_name(VALHEIM_PROCESS_NAME);\r
+\r
+    if processes.len() >= 1 {\r
+        let process = processes.first().unwrap();\r
 \r
-    if process.len() >= 1 {\r
-        println!("Valheim running");\r
-        true\r
+        Some(\r
+            ValheimExe {\r
+                memory: process.memory(),\r
+                cpu_usage: process.cpu_usage(),\r
+                uptime: std::time::SystemTime::now().duration_since(std::time::UNIX_EPOCH).unwrap().as_secs() - process.start_time()\r
+            }\r
+        )\r
     } else {\r
-        println!("Valheim not running");\r
-        false\r
+        None\r
     }\r
 }
\ No newline at end of file