From f86d940f208f73c0ea1634fa8f894c4d7800d8eb Mon Sep 17 00:00:00 2001
From: Jack Robison <jackrobison@lbry.io>
Date: Tue, 1 Dec 2020 11:28:27 -0500
Subject: [PATCH] TX_COUNTS_STATE

---
 lbry/wallet/server/leveldb.py | 23 +++++++++++++----------
 1 file changed, 13 insertions(+), 10 deletions(-)

diff --git a/lbry/wallet/server/leveldb.py b/lbry/wallet/server/leveldb.py
index a3ff2a0ff..7cee52bd3 100644
--- a/lbry/wallet/server/leveldb.py
+++ b/lbry/wallet/server/leveldb.py
@@ -50,6 +50,7 @@ HASHX_HISTORY_PREFIX = b'x'
 UNDO_PREFIX = b'U'
 UTXO_STATE = b'state-utxo'
 HIST_STATE = b'state-hist'
+TX_COUNTS_STATE = b'state-tx-counts'
 
 
 @dataclass
@@ -71,10 +72,11 @@ class RocksReaderContext:
         self.db.open(self.name, create=False, read_only=True)
 
     def update_state(self):
-        self.tx_counts = [
-            unpack_be_uint64(tx_count)
-            for tx_count in self.db.iterator(prefix=TX_COUNT_PREFIX, include_key=False)
-        ]
+        tx_counts = array.array("L")
+        counts = self.db.get(TX_COUNTS_STATE)
+        if counts:
+            tx_counts.frombytes(counts)
+        self.tx_counts = tx_counts.tolist()
 
     def ctx_tx_hash(self, tx_num):
         tx_height = bisect_right(self.tx_counts, tx_num)
@@ -195,10 +197,9 @@ def lookup_hashXs_utxos(prevouts):
 
 
 def get_counts():
-    return tuple(
-        util.unpack_be_uint64(tx_count)
-        for tx_count in proc_ctx.get().db.iterator(prefix=TX_COUNT_PREFIX, include_key=False)
-    )
+    ctx = proc_ctx.get()
+    return ctx.tx_counts
+
 
 
 def read_txids():
@@ -393,7 +394,7 @@ class LevelDB:
 
         tx_counts = await asyncio.get_event_loop().run_in_executor(self.executor, get_counts)
         assert len(tx_counts) == self.db_height + 1, f"{len(tx_counts)} vs {self.db_height + 1}"
-        self.tx_counts = array.array('I', tx_counts)
+        self.tx_counts = tx_counts
 
         if self.tx_counts:
             assert self.db_tx_count == self.tx_counts[-1], \
@@ -590,6 +591,9 @@ class LevelDB:
                     tx_num += 1
                     offset += 32
 
+            batch_put(TX_COUNTS_STATE, b''.join(map(util.pack_le_int64, self.tx_counts)))
+
+
         flush_data.block_txs.clear()
         flush_data.block_hashes.clear()
 
@@ -766,7 +770,6 @@ class LevelDB:
 
     def fs_block_hashes(self, height, count):
         if height + count > len(self.headers):
-            print("boom")
             raise self.DBError(f'only got {len(self.headers) - height:,d} headers starting at {height:,d}, not {count:,d}')
         return [self.coin.header_hash(header) for header in self.headers[height:height + count]]