From 8c4c4d1d3634d0aa1787b816d109dcf95b8f6310 Mon Sep 17 00:00:00 2001
From: Bastian Gruber <foreach@me.com>
Date: Mon, 1 May 2023 10:38:14 +0200
Subject: [PATCH] Full workaround

---
 problem_04/bin/server.rs | 17 +++++++++++------
 1 file changed, 11 insertions(+), 6 deletions(-)

diff --git a/problem_04/bin/server.rs b/problem_04/bin/server.rs
index f33698b..0d72cb8 100644
--- a/problem_04/bin/server.rs
+++ b/problem_04/bin/server.rs
@@ -13,7 +13,7 @@ async fn main() -> io::Result<()> {
 
     tokio::spawn(async move {
         while let Some((bytes, addr)) = rx.recv().await {
-            let len = s.send_to(&bytes, &addr).await.unwrap();
+            let _ = s.send_to(&bytes, &addr).await.unwrap();
         }
     });
 
@@ -22,17 +22,22 @@ async fn main() -> io::Result<()> {
         let (len, addr) = r.recv_from(&mut buf).await?;
         let message = str::from_utf8(&buf[..len]).unwrap().trim_matches('\n');
 
-        if message.contains("=") {
-            let (key, value) = message.split_once('=').unwrap();
-
+        if message.contains("version") {
+            let message = format!("version=gruberb 1.0");
+            tx.send((message.as_bytes().to_vec(), addr)).await.unwrap();
+        } else if message.contains("=") {
+            let (mut key, value) = message.split_once('=').unwrap();
+            if key.is_empty() {
+                key = " ";
+            }
             storage
                 .lock()
                 .unwrap()
                 .insert(key.to_string(), value.to_string());
-            // continue;
         } else {
             let value = storage.lock().unwrap().get(message).unwrap().clone();
-            tx.send((value.as_bytes().to_vec(), addr)).await.unwrap();
+            let message = format!("{message}={value}");
+            tx.send((message.as_bytes().to_vec(), addr)).await.unwrap();
         }
 
         buf.fill(0);