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

make sure that we don't call HashBlobWriter.finished_cb multiple times when calling close()

Browse source
This commit is contained in:
Kay Kurokawa 2017-09-13 14:01:20 -04:00 • committed by Jack Robison
parent 468a16af58
commit 39c4db3471
No known key found for this signature in database
GPG key ID: 284699E7404E3CFF
1 changed files with 7 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
lbrynet/core/HashBlob.py
View file

@ -47,6 +47,7 @@ class HashBlobWriter(object):
self.write_handle = BytesIO()
self.length_getter = length_getter
self.finished_cb = finished_cb
self.finished_cb_d = None
self._hashsum = get_lbry_hash_obj()
self.len_so_far = 0
@ -62,7 +63,7 @@ class HashBlobWriter(object):
self._hashsum.update(data)
self.len_so_far += len(data)
if self.len_so_far > self.length_getter():
self.finished_cb(
self.finished_cb_d = self.finished_cb(
self,
Failure(InvalidDataError("Length so far is greater than the expected length."
" %s to %s" % (self.len_so_far,
@ -70,7 +71,7 @@ class HashBlobWriter(object):
else:
self.write_handle.write(data)
if self.len_so_far == self.length_getter():
self.finished_cb(self)
self.finished_cb_d = self.finished_cb(self)
def close_handle(self):
if self.write_handle is not None:
@ -78,12 +79,13 @@ class HashBlobWriter(object):
self.write_handle = None
def close(self, reason=None):
# we've already closed, so do nothing
if self.write_handle is None:
# if we've already called finished_cb because we either finished writing
# or closed already, do nothing
if self.finished_cb_d is not None:
return
if reason is None:
reason = Failure(DownloadCanceledError())
self.finished_cb(self, reason)
self.finished_cb_d = self.finished_cb(self, reason)
class HashBlob(object):