bring back removed code

Alex6323 · Sep 16, 2020 · 404e454 · 404e454
1 parent 370b095
commit 404e454
Showing 1 changed file with 18 additions and 4 deletions.
diff --git a/bee-protocol/src/worker/requester/transaction.rs b/bee-protocol/src/worker/requester/transaction.rs
@@ -67,6 +67,23 @@ impl TransactionRequesterWorker {
 
         let guard = Protocol::get().peer_manager.handshaked_peers_keys.read().await;
 
+        for _ in 0..guard.len() {
+            let epid = &guard[self.counter % guard.len()];
+
+            self.counter += 1;
+
+            if let Some(peer) = Protocol::get().peer_manager.handshaked_peers.get(epid) {
+                if peer.has_data(index) {
+                    let hash = hash.as_trits().encode::<T5B1Buf>();
+                    Sender::<TransactionRequest>::send(
+                        epid,
+                        TransactionRequest::new(cast_slice(hash.as_i8_slice())),
+                    );
+                    return true;
+                }
+            }
+        }
+
         for _ in 0..guard.len() {
             let epid = &guard[self.counter % guard.len()];
 
@@ -92,10 +109,7 @@ impl TransactionRequesterWorker {
             if let Some(peer) = Protocol::get().peer_manager.handshaked_peers.get(epid) {
                 if peer.maybe_has_data(index) {
                     let hash = hash.as_trits().encode::<T5B1Buf>();
-                    Sender::<TransactionRequest>::send(
-                        epid,
-                        TransactionRequest::new(cast_slice(hash.as_i8_slice())),
-                    );
+                    Sender::<TransactionRequest>::send(epid, TransactionRequest::new(cast_slice(hash.as_i8_slice())));
                     return true;
                 }
             }