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

defer db insert during downloads

Browse source
This commit is contained in:
Victor Shyba 2019-02-08 03:42:31 -03:00
parent af1619ebfb
commit e2b06677b5
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
lbrynet/stream/assembler.py
View file

@ -86,6 +86,7 @@ class StreamAssembler:
) )
await self.blob_manager.blob_completed(self.sd_blob) await self.blob_manager.blob_completed(self.sd_blob)
written_blobs = None written_blobs = None
save_tasks = []
try: try:
with open(self.output_path, 'wb') as stream_handle: with open(self.output_path, 'wb') as stream_handle:
self.stream_handle = stream_handle self.stream_handle = stream_handle
@ -101,7 +102,7 @@ class StreamAssembler:
await self.blob_manager.delete_blobs([blob_info.blob_hash]) await self.blob_manager.delete_blobs([blob_info.blob_hash])
continue continue
if await self._decrypt_blob(blob, blob_info, self.descriptor.key): if await self._decrypt_blob(blob, blob_info, self.descriptor.key):
await self.blob_manager.blob_completed(blob) save_tasks.append(asyncio.ensure_future(self.blob_manager.blob_completed(blob)))
written_blobs = i written_blobs = i
if not self.wrote_bytes_event.is_set(): if not self.wrote_bytes_event.is_set():
self.wrote_bytes_event.set() self.wrote_bytes_event.set()
@ -115,6 +116,8 @@ class StreamAssembler:
self.descriptor.sd_hash) self.descriptor.sd_hash)
continue continue
finally: finally:
if save_tasks:
await asyncio.wait(save_tasks)
if written_blobs == len(self.descriptor.blobs) - 2: if written_blobs == len(self.descriptor.blobs) - 2:
log.debug("finished decrypting and assembling stream") log.debug("finished decrypting and assembling stream")
await self.after_finished() await self.after_finished()