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

ensure hash_announcer exists before announcing

Browse source
This commit is contained in:
Job Evers-Meltzer 2017-01-06 08:00:22 -06:00
parent 30af8a0e20
commit f37808859f
1 changed files with 15 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
lbrynet/core/BlobManager.py
View file

@ -73,6 +73,11 @@ class BlobManager(DHTHashSupplier):
def add_blob_to_upload_history(self, blob_hash, host, rate): def add_blob_to_upload_history(self, blob_hash, host, rate):
pass pass
def _immediate_announce(self, blob_hashes):
if self.hash_announcer:
return self.hash_announcer.immediate_announce(blob_hashes)
class DiskBlobManager(BlobManager): class DiskBlobManager(BlobManager):
"""This class stores blobs on the hard disk""" """This class stores blobs on the hard disk"""
@ -145,7 +150,7 @@ class DiskBlobManager(BlobManager):
next_announce_time = time.time() + self.hash_reannounce_time next_announce_time = time.time() + self.hash_reannounce_time
d = self._add_completed_blob(blob.blob_hash, blob.length, d = self._add_completed_blob(blob.blob_hash, blob.length,
time.time(), next_announce_time) time.time(), next_announce_time)
d.addCallback(lambda _: self.hash_announcer.immediate_announce([blob.blob_hash])) d.addCallback(lambda _: self._immediate_announce([blob.blob_hash]))
return d return d
def completed_blobs(self, blobs_to_check): def completed_blobs(self, blobs_to_check):
@ -163,12 +168,9 @@ class DiskBlobManager(BlobManager):
new_blob = self.blob_type(self.blob_dir, blob_creator.blob_hash, True, blob_creator.length) new_blob = self.blob_type(self.blob_dir, blob_creator.blob_hash, True, blob_creator.length)
new_blob.verified = True new_blob.verified = True
self.blobs[blob_creator.blob_hash] = new_blob self.blobs[blob_creator.blob_hash] = new_blob
if self.hash_announcer is not None: self._immediate_announce([blob_creator.blob_hash])
self.hash_announcer.immediate_announce([blob_creator.blob_hash]) next_announce_time = time.time() + self.hash_reannounce_time
next_announce_time = time.time() + self.hash_reannounce_time d = self.blob_completed(new_blob, next_announce_time)
d = self.blob_completed(new_blob, next_announce_time)
else:
d = self.blob_completed(new_blob)
return d return d
def delete_blobs(self, blob_hashes): def delete_blobs(self, blob_hashes):
@ -181,7 +183,7 @@ class DiskBlobManager(BlobManager):
def immediate_announce_all_blobs(self): def immediate_announce_all_blobs(self):
d = self._get_all_verified_blob_hashes() d = self._get_all_verified_blob_hashes()
d.addCallback(self.hash_announcer.immediate_announce) d.addCallback(self._immediate_announce)
return d return d
def get_blob_length(self, blob_hash): def get_blob_length(self, blob_hash):
@ -498,12 +500,9 @@ class TempBlobManager(BlobManager):
new_blob.data_buffer = blob_creator.data_buffer new_blob.data_buffer = blob_creator.data_buffer
new_blob.length = blob_creator.length new_blob.length = blob_creator.length
self.blobs[blob_creator.blob_hash] = new_blob self.blobs[blob_creator.blob_hash] = new_blob
if self.hash_announcer is not None: self._immediate_announce([blob_creator.blob_hash])
self.hash_announcer.immediate_announce([blob_creator.blob_hash]) next_announce_time = time.time() + self.hash_reannounce_time
next_announce_time = time.time() + self.hash_reannounce_time d = self.blob_completed(new_blob, next_announce_time)
d = self.blob_completed(new_blob, next_announce_time)
else:
d = self.blob_completed(new_blob)
d.addCallback(lambda _: new_blob) d.addCallback(lambda _: new_blob)
return d return d
@ -519,7 +518,8 @@ class TempBlobManager(BlobManager):
return defer.fail(NoSuchBlobError(blob_hash)) return defer.fail(NoSuchBlobError(blob_hash))
def immediate_announce_all_blobs(self): def immediate_announce_all_blobs(self):
return self.hash_announcer.immediate_announce(self.blobs.iterkeys()) if self.hash_announcer:
return self.hash_announcer.immediate_announce(self.blobs.iterkeys())
def _manage(self): def _manage(self):
from twisted.internet import reactor from twisted.internet import reactor