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

simple address subs instead of batch

Browse source
This commit is contained in:
Victor Shyba 2019-08-22 16:32:14 -03:00
parent 5e8d169c50
commit 6c3147fcf4
2 changed files with 9 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
torba/torba/client/baseledger.py
View file

@ -395,9 +395,13 @@ class BaseLedger(metaclass=LedgerRegistry):
async def subscribe_addresses(self, address_manager: baseaccount.AddressManager, addresses: List[str]): async def subscribe_addresses(self, address_manager: baseaccount.AddressManager, addresses: List[str]):
if self.network.is_connected and addresses: if self.network.is_connected and addresses:
async for address, remote_status in self.network.subscribe_address(*addresses): await asyncio.wait([
# subscribe isnt a retriable call as it happens right after a connection is made self.subscribe_address(address_manager, address) for address in addresses
self._update_tasks.add(self.update_history(address, remote_status, address_manager)) ])
async def subscribe_address(self, address_manager: baseaccount.AddressManager, address: str):
remote_status = await self.network.subscribe_address(address)
self._update_tasks.add(self.update_history(address, remote_status, address_manager))
def process_status_update(self, update): def process_status_update(self, update):
address, remote_status = update address, remote_status = update

8
torba/torba/client/basenetwork.py
View file

@ -242,12 +242,8 @@ class BaseNetwork:
def subscribe_headers(self): def subscribe_headers(self):
return self.rpc('blockchain.headers.subscribe', [True], session=self.client) return self.rpc('blockchain.headers.subscribe', [True], session=self.client)
async def subscribe_address(self, *addresses): def subscribe_address(self, address):
async with self.client.send_batch() as batch: return self.rpc('blockchain.address.subscribe', [address], session=self.client)
for address in addresses:
batch.add_request('blockchain.address.subscribe', [address])
for address, status in zip(addresses, batch.results):
yield address, status
class SessionPool: class SessionPool: