1 files changed, 14 insertions, 0 deletions
diff --git a/src/server/connection.rs b/src/server/connection.rs
index 662e33d..9275b67 100644
--- a/src/server/connection.rs
+++ b/src/server/connection.rs
@@ -57,3 +57,17 @@ impl ServerConnection {
         }
     }
 }
+
+pub fn receive(buff_r: &mut BufReader<TcpStream>) -> Option<String> {
+    let mut recv = String::new();
+    match buff_r.read_line(&mut recv) {
+        Ok(result) => {
+            if result == 0 {
+                None
+            } else {
+                Some(recv.replace("\n", ""))
+            }
+        }
+        Err(_) => Some(String::new()),
+    }
+}