do not limit DHT results by K, respect max_results

This commit is contained in:
Victor Shyba 2021-12-03 16:48:22 -03:00 committed by Victor Shyba
parent 3d5c9cc1c2
commit 54d6fb9da4

View file

@ -286,7 +286,7 @@ class IterativeNodeFinder(IterativeFinder):
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)