error catching for sockets

src/modules/socket_handler.rs

@@ -82,16 +82,29 @@ impl Socket {
 
     pub fn send(&self, message: String) {
         let mut streams = self.socket_txs.lock().unwrap();
+        let mut removes = Vec::<usize>::new();
         if streams.len() == 0 {return}
         for i in 0..streams.len(){
-            let mut tx = streams.get(i).unwrap().as_ref();
+            match streams.get(i) {
-            
+                None => {
+                    removes.push(i);
+                }, 
+                Some(strm) => {
+                    let mut tx = strm.as_ref();
                     match tx.write(message.clone().as_bytes()) {
-                Err(_) => {streams.remove(i); println!("removed a socket"); continue;},
+                        Err(_) => {
-                Ok(_) => {},
+                            removes.push(i)
-            }
+                        },
+                        Ok(_) => {
                             tx.write(b"\n").unwrap();
                             tx.flush().unwrap();
                         }
                     }
                 }
+            }
+        }
+        for i in removes.iter() {
+            streams.remove(*i);
+        }
+    }
+}