Ping candidates every minute

TorrentDam · Sep 15, 2024 · 8fff212 · 8fff212
1 parent f03d4d5
commit 8fff212
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/dht/src/main/scala/com/github/torrentdam/bittorrent/dht/Node.scala b/dht/src/main/scala/com/github/torrentdam/bittorrent/dht/Node.scala
@@ -86,9 +86,17 @@ object Node {
           case Left(_) => IO.unit
         }
     }
-    
+
   private def pingCandidates(nodes: Queue[IO, NodeInfo], client: Client, routingTable: RoutingTable[IO])(using Logger[IO]) =
-    nodes.take.flatMap(pingCandidate(_, client, routingTable).attempt.void).foreverM
+    nodes
+      .tryTakeN(none)
+      .flatMap(candidates => 
+        candidates
+          .distinct
+          .traverse_(pingCandidate(_, client, routingTable).attempt.void)
+      )
+      .productR(IO.sleep(1.minute))
+      .foreverM
 
 
   private def reportingQueryHandler(queue: Queue[IO, NodeInfo], next: QueryHandler[IO]): QueryHandler[IO] = (address, query) =>