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

Check if same file is being downloaded.

Browse source
This commit is contained in:
Yuxin Wang 2018-09-26 22:18:56 -04:00
parent b3ad70a229
commit 0d77334bff
1 changed files with 7 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
p2pfs/core/peer.py
View file

@ -77,17 +77,22 @@ class Peer(MessageServer):
with self._file_list_lock:
if self._file_list is None or file not in self._file_list.keys():
return False, 'Requested file {} does not exist, try list_file?'.format(file)
with self._download_lock:
if file in self._download_results:
return False, 'Download {} already in progress.'.format(file)
self._download_results[file] = Queue()
self._write_message(self._server_sock, {
'type': MessageType.REQUEST_FILE_LOCATION,
'filename': file
})
self._download_results[file] = Queue()
# wait until reply is ready
fileinfo, chunkinfo = self._download_results[file].get()
logger.debug('{}: {} ==> {}'.format(file, fileinfo, chunkinfo))
with self._download_lock:
del self._download_results[file]
return True, 'File {} dowloaded to {}'.format(file, destination)
def exit(self):