Selaa lähdekoodia

Allow enabling and disabling a client.

Jing Yang 5 vuotta sitten
vanhempi
commit
2a4522ae9a
1 muutettua tiedostoa jossa 7 lisäystä ja 2 poistoa
  1. 7 2
      src/network.rs

+ 7 - 2
src/network.rs

@@ -49,11 +49,12 @@ impl Network {
         self.keep_running = false;
     }
 
-    pub fn make_connection(
-        &self,
+    pub fn make_client(
+        &mut self,
         client: ClientIdentifier,
         server: ServerIdentifier,
     ) -> Client {
+        self.clients.insert(client.clone(), (true, server.clone()));
         Client {
             client,
             server,
@@ -61,6 +62,10 @@ impl Network {
         }
     }
 
+    pub fn set_enable_client(&mut self, client: &ClientIdentifier, yes: bool) {
+        self.clients.get_mut(client).map(|pair| pair.0 = yes);
+    }
+
     pub fn add_server(
         &mut self,
         server_name: ServerIdentifier,