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

adding immediate announce option to hash announce

Browse source
This commit is contained in:
Kay Kurokawa 2017-02-09 10:09:58 -05:00
parent aa45b0e121
commit 34f5bc93ae
2 changed files with 14 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
lbrynet/core/server/DHTHashAnnouncer.py
View file

@ -39,7 +39,7 @@ class DHTHashAnnouncer(object):
def immediate_announce(self, blob_hashes):
if self.peer_port is not None:
return self._announce_hashes(blob_hashes)
return self._announce_hashes(blob_hashes, immediate=True)
else:
return defer.succeed(False)
@ -56,7 +56,7 @@ class DHTHashAnnouncer(object):
dl = defer.DeferredList(ds)
return dl
def _announce_hashes(self, hashes):
def _announce_hashes(self, hashes, immediate=False):
if not hashes:
return
log.debug('Announcing %s hashes', len(hashes))
@ -67,6 +67,9 @@ class DHTHashAnnouncer(object):
for h in hashes:
announce_deferred = defer.Deferred()
ds.append(announce_deferred)
if immediate:
self.hash_queue.appendleft((h, announce_deferred))
else:
self.hash_queue.append((h, announce_deferred))
log.debug('There are now %s hashes remaining to be announced', self.hash_queue_size())

9
tests/unit/core/server/test_DHTHashAnnouncer.py
View file

@ -42,9 +42,16 @@ class DHTHashAnnouncerTest(unittest.TestCase):
def test_basic(self):
self.announcer._announce_available_hashes()
self.assertEqual(self.announcer.hash_queue_size(),self.announcer.CONCURRENT_ANNOUNCERS)
self.clock.advance(1)
self.assertEqual(self.dht_node.blobs_announced, self.num_blobs)
self.assertEqual(self.announcer.hash_queue_size(), 0)
def test_immediate_announce(self):
# Test that immediate announce puts a hash at the front of the queue
self.announcer._announce_available_hashes()
blob_hash = binascii.b2a_hex(os.urandom(32))
self.announcer.immediate_announce([blob_hash])
self.assertEqual(self.announcer.hash_queue_size(),self.announcer.CONCURRENT_ANNOUNCERS+1)
self.assertEqual(blob_hash, self.announcer.hash_queue[0][0])