masscollaborationlabs/p2pfs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Move code to common method.

Browse source
This commit is contained in:
Yuxin Wang 2018-10-10 21:19:45 -04:00
parent 8e5d81cea5
commit 23420f4304
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
p2pfs/core/peer.py
View file

@ -139,6 +139,9 @@ class Peer(MessageServer):
assert MessageType(message['type']) == MessageType.REPLY_FILE_LOCATION
fileinfo, chunkinfo = message['fileinfo'], message['chunkinfo']
logger.debug('{}: {} ==> {}'.format(filename, fileinfo, chunkinfo))
# cancel out self registration
if self._server_address in chunkinfo:
del chunkinfo[json.dumps(self._server_address)]
return fileinfo, chunkinfo
async def _test_peer_rtt(self, peers):
@ -304,8 +307,7 @@ class Peer(MessageServer):
# update chunkinfo to see if new peers have registered and update downloading plan
assert not self._tracker_writer.is_closing()
_, chunkinfo = await self._request_chunkinfo(file)
# cancel out self registration
del chunkinfo[json.dumps(self._server_address)]
for address, possessed_chunks in chunkinfo.items():
# if the chunkinfo hasn't been updated with peer_address removed
if address == peer_address: