CoinStoreProtocol

Author: richardkiss

chia/full_node/consensus_store_sqlite3.py

@@ -14,6 +14,7 @@
 from chia.consensus.block_height_map import BlockHeightMap
 from chia.full_node.block_store import BlockStore
 from chia.full_node.coin_store import CoinStore
+from chia.full_node.coin_store_protocol import CoinStoreProtocol
 from chia.types.blockchain_format.coin import Coin
 from chia.types.coin_record import CoinRecord
 from chia.util.db_wrapper import DBWrapper2
@@ -22,7 +23,7 @@
 @dataclass
 class ConsensusStoreSQLite3Writer:
     block_store: BlockStore
-    coin_store: CoinStore
+    coin_store: CoinStoreProtocol
 
     async def add_full_block(self, header_hash: bytes32, block: FullBlock, block_record: BlockRecord) -> None:
         await self.block_store.add_full_block(header_hash, block, block_record)
@@ -62,7 +63,7 @@ class ConsensusStoreSQLite3:
     """
 
     block_store: BlockStore
-    coin_store: CoinStore
+    coin_store: CoinStoreProtocol
     height_map: BlockHeightMap
 
     # Writer context and writer facade for transactional writes (re-entrant via depth counter)

chia/full_node/full_node.py

@@ -52,6 +52,7 @@
 from chia.full_node.block_store import BlockStore
 from chia.full_node.check_fork_next_block import check_fork_next_block
 from chia.full_node.coin_store import CoinStore
+from chia.full_node.coin_store_protocol import CoinStoreProtocol
 from chia.full_node.consensus_store_sqlite3 import ConsensusStoreSQLite3
 from chia.full_node.full_node_api import FullNodeAPI
 from chia.full_node.full_node_store import FullNodeStore, FullNodeStorePeakResult, UnfinishedBlockEntry
@@ -159,7 +160,7 @@ class FullNode:
     _db_wrapper: Optional[DBWrapper2] = None
     _hint_store: Optional[HintStore] = None
     _block_store: Optional[BlockStore] = None
-    _coin_store: Optional[CoinStore] = None
+    _coin_store: Optional[CoinStoreProtocol] = None
     _mempool_manager: Optional[MempoolManager] = None
     _init_weight_proof: Optional[asyncio.Task[None]] = None
     _blockchain: Optional[Blockchain] = None
@@ -421,7 +422,7 @@ def blockchain(self) -> Blockchain:
         return self._blockchain
 
     @property
-    def coin_store(self) -> CoinStore:
+    def coin_store(self) -> CoinStoreProtocol:
         assert self._coin_store is not None
         return self._coin_store