Update src/client.rs

Co-authored-by: Maciej Zwoliński <[email protected]>
eigerco · Jan 25, 2024 · 995eed5 · 995eed5
1 parent 405464f
commit 995eed5
Showing 1 changed file with 7 additions and 13 deletions.
diff --git a/src/client.rs b/src/client.rs
@@ -125,19 +125,13 @@ where
     }
 
     pub(crate) fn new_connection_handler(&mut self, peer: PeerId) -> ClientConnectionHandler<S> {
-        match self.peers.entry(peer) {
-            hash_map::Entry::Occupied(mut entry) => {
-                entry.get_mut().established_connections_num += 1;
-            }
-            hash_map::Entry::Vacant(entry) => {
-                entry.insert(PeerState {
-                    established_connections_num: 1,
-                    sending: Arc::new(Mutex::new(SendingState::Ready)),
-                    wantlist: WantlistState::new(),
-                    send_full: true,
-                });
-            }
-        }
+         let peer = self.peers.entry(peer).or_insert_with(|| PeerState {
+            established_connections_num: 0,
+            sending: Arc::new(Mutex::new(SendingState::Ready)),
+            wantlist: WantlistState::new(),
+            send_full: true,
+        });
+        peer.established_connections_num += 1;
 
         ClientConnectionHandler {
             protocol: self.protocol.clone(),