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

group cache hits in a single dict

Browse source
This commit is contained in:
Victor Shyba 2020-12-23 21:37:20 -03:00
parent 270da80d64
commit adb418aafc
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
lbry/wallet/ledger.py
View file

@ -610,13 +610,13 @@ class Ledger(metaclass=LedgerRegistry):
async def request_transactions(self, to_request: Tuple[Tuple[str, int], ...], cached=False):
batches = [[]]
remote_heights = {}
cache_hits = set()
for txid, height in sorted(to_request, key=lambda x: x[1]):
if cached:
if txid in self._tx_cache:
if self._tx_cache[txid].tx is not None and self._tx_cache[txid].tx.is_verified:
tx = self._tx_cache[txid].tx
yield {tx.id: self._tx_cache[txid].tx}
cache_hits.add(txid)
continue
else:
self._tx_cache[txid] = TransactionCacheItem()
@ -626,6 +626,8 @@ class Ledger(metaclass=LedgerRegistry):
batches[-1].append(txid)
if not batches[-1]:
batches.pop()
if cached and cache_hits:
yield {txid: self._tx_cache[txid].tx for txid in cache_hits}
for batch in batches:
txs = await self._single_batch(batch, remote_heights)