From e0526ac8286feceafe36f0d5557b73cf380a1a6d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Joakim=20Frosteg=C3=A5rd?= <joakim.frostegard@gmail.com>
Date: Wed, 6 May 2020 00:44:23 +0200
Subject: [PATCH] aquatic handle_announce_requests: reorder code, remove branch

---
 aquatic/src/lib/handlers.rs | 28 ++++++++++++++--------------
 1 file changed, 14 insertions(+), 14 deletions(-)

diff --git a/aquatic/src/lib/handlers.rs b/aquatic/src/lib/handlers.rs
index 2b2902c..99823ac 100644
--- a/aquatic/src/lib/handlers.rs
+++ b/aquatic/src/lib/handlers.rs
@@ -219,28 +219,23 @@ pub fn handle_announce_requests(
             .entry(request.info_hash)
             .or_default();
         
-        let opt_removed_peer_status = if peer_status == PeerStatus::Stopped {
-            torrent_data.peers.remove(&peer_key).map(|peer| peer.status)
-        } else {
-            torrent_data.peers.insert(peer_key, peer).map(|peer| peer.status)
-        };
-
-        let max_num_peers_to_take = calc_max_num_peers_to_take(
-            config,
-            request.peers_wanted.0
-        );
-
-        match peer_status {
+        let opt_removed_peer = match peer_status {
             PeerStatus::Leeching => {
                 torrent_data.num_leechers += 1;
+
+                torrent_data.peers.insert(peer_key, peer)
             },
             PeerStatus::Seeding => {
                 torrent_data.num_seeders += 1;
+
+                torrent_data.peers.insert(peer_key, peer)
             },
-            PeerStatus::Stopped => {}
+            PeerStatus::Stopped => {
+                torrent_data.peers.remove(&peer_key)
+            }
         };
 
-        match opt_removed_peer_status {
+        match opt_removed_peer.map(|peer| peer.status){
             Some(PeerStatus::Leeching) => {
                 torrent_data.num_leechers -= 1;
             },
@@ -250,6 +245,11 @@ pub fn handle_announce_requests(
             _ => {}
         }
 
+        let max_num_peers_to_take = calc_max_num_peers_to_take(
+            config,
+            request.peers_wanted.0
+        );
+
         let response_peers = extract_response_peers(
             rng,
             &torrent_data.peers,