Change CLI args

Add p core selector
Fix socket argument parsing
Better budget calculation

1 files changed, 3 insertions, 3 deletions

diff --git a/src/socket.rs b/src/socket.rs
index 2bb0dbb..d851eed 100644
--- a/src/socket.rs
+++ b/src/socket.rs
@@ -41,8 +41,8 @@ impl LoggingSocketService {
                         break;
                     }
                     let (command, args) = line.split_once(' ').unwrap_or((line.trim(), ""));
-                    let output = match dbg!(command) {
-                        "pid" => self.get_process(args.parse().unwrap_or_default()),
+                    let output = match command {
+                        "pid" => self.get_process(args.trim().parse().unwrap_or_default()),
                         "list" => self.list_processes(),
                         "limit" => self.set_power_limit(args),
                         _ => "Unrecognized command#".into(),
@@ -64,7 +64,7 @@ impl LoggingSocketService {
     }
 
     fn set_power_limit(&self, args: &str) -> String {
-        if let Ok(power_limit) = args.parse() {
+        if let Ok(power_limit) = args.trim().parse() {
             self.power_limit
                 .store(power_limit, std::sync::atomic::Ordering::Relaxed);
             "#".into()