Madiator2011/lbry-sdk
1
0
Fork 0
forked from LBRYCommunity/lbry-sdk
Code Pull requests Activity

exclude self and querying node from closest contacts

Browse source
This commit is contained in:
Jack Robison 2018-07-02 14:44:56 -04:00
parent b6289d101d
commit 929a0cc5af
No known key found for this signature in database
GPG key ID: DF25C68FE0239BB2
1 changed files with 6 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
lbrynet/dht/routingtable.py
View file

@ -161,12 +161,15 @@ class TreeRoutingTable(object):
node is returning all of the contacts that it knows of. node is returning all of the contacts that it knows of.
@rtype: list @rtype: list
""" """
exclude = [self._parentNodeID]
if sender_node_id:
exclude.append(sender_node_id)
if key in exclude:
exclude.remove(key)
count = count or constants.k count = count or constants.k
sender_node_id = sender_node_id or self._parentNodeID
distance = Distance(key) distance = Distance(key)
contacts = self.get_contacts() contacts = self.get_contacts()
contacts = [c for c in contacts if c.id != sender_node_id] contacts = [c for c in contacts if c.id not in exclude]
contacts.sort(key=lambda c: distance(c.id)) contacts.sort(key=lambda c: distance(c.id))
return contacts[:min(count, len(contacts))] return contacts[:min(count, len(contacts))]