do not limit DHT results by K, respect max_results

lbryio · Dec 8, 2021 · 8667bf5 · 8667bf5
1 parent 442326f
commit 8667bf5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lbry/dht/protocol/iterative_find.py b/lbry/dht/protocol/iterative_find.py
@@ -285,7 +285,7 @@ def put_result(self, from_iter: typing.Iterable['KademliaPeer'], finish=False):
             and self.peer_manager.peer_is_good(peer) is not False
         ]
         not_yet_yielded.sort(key=lambda peer: self.distance(peer.node_id))
-        to_yield = not_yet_yielded[:min(constants.K, len(not_yet_yielded))]
+        to_yield = not_yet_yielded[:max(constants.K, self.max_results)]
         if to_yield:
             self.yielded_peers.update(to_yield)
             self.iteration_queue.put_nowait(to_yield)