LBRYCommunity/lbry-sdk
1
0
Fork 1
Code Issues Pull requests Projects Releases 282 Packages Wiki Activity

simplify bucket refresh loop

Browse source
This commit is contained in:
Victor Shyba 2022-08-11 21:14:56 -03:00
parent e887453aa5
commit d61accea1a
1 changed files with 2 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
lbry/dht/protocol/routing_table.py
View file

@ -235,13 +235,9 @@ class TreeRoutingTable:
return self.buckets[self._kbucket_index(contact_id)].get_peer(contact_id) return self.buckets[self._kbucket_index(contact_id)].get_peer(contact_id)
def get_refresh_list(self, start_index: int = 0, force: bool = False) -> typing.List[bytes]: def get_refresh_list(self, start_index: int = 0, force: bool = False) -> typing.List[bytes]:
bucket_index = start_index
refresh_ids = [] refresh_ids = []
now = int(self._loop.time()) for offset, _ in enumerate(self.buckets[start_index:]):
for bucket in self.buckets[start_index:]: refresh_ids.append(self._midpoint_id_in_bucket_range(start_index + offset))
to_search = self._midpoint_id_in_bucket_range(bucket_index)
refresh_ids.append(to_search)
bucket_index += 1
# if we have 3 or fewer populated buckets get two random ids in the range of each to try and # if we have 3 or fewer populated buckets get two random ids in the range of each to try and
# populate/split the buckets further # populate/split the buckets further
buckets_with_contacts = self.buckets_with_contacts() buckets_with_contacts = self.buckets_with_contacts()