aljaz/python-teos
1
0
Fork 0
mirror of https://github.com/aljazceru/python-teos.git synced 2025-12-17 22:24:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #123 from talaia-labs/117-last-block-on-restart

Browse Source 
Stores last block on db on fresh bootstrap for Watcher and Responder
This commit is contained in:
Sergi Delgado Segura
2020-04-20 15:11:54 +02:00
committed by GitHub
parent f472d6ccc2 c165318940
commit e2ae29b4fe
2 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
teos/responder.py
View File

@@ -259,6 +259,7 @@ class Responder:
# Distinguish fresh bootstraps from bootstraps from db # Distinguish fresh bootstraps from bootstraps from db
if self.last_known_block is None: if self.last_known_block is None:
self.last_known_block = self.block_processor.get_best_block_hash() self.last_known_block = self.block_processor.get_best_block_hash()
self.db_manager.store_last_block_hash_responder(self.last_known_block)
while True: while True:
block_hash = self.block_queue.get() block_hash = self.block_queue.get()

8
teos/watcher.py
View File

@@ -54,6 +54,7 @@ class Watcher:
signing_key (:mod:`PrivateKey`): a private key used to sign accepted appointments. signing_key (:mod:`PrivateKey`): a private key used to sign accepted appointments.
max_appointments (:obj:`int`): the maximum amount of appointments accepted by the ``Watcher`` at the same time. max_appointments (:obj:`int`): the maximum amount of appointments accepted by the ``Watcher`` at the same time.
expiry_delta (:obj:`int`): the additional time the ``Watcher`` will keep an expired appointment around. expiry_delta (:obj:`int`): the additional time the ``Watcher`` will keep an expired appointment around.
last_known_block (:obj:`str`): the last block known by the ``Watcher``.
Raises: Raises:
ValueError: if `teos_sk_file` is not found. ValueError: if `teos_sk_file` is not found.
@@ -70,6 +71,7 @@ class Watcher:
self.max_appointments = max_appointments self.max_appointments = max_appointments
self.expiry_delta = expiry_delta self.expiry_delta = expiry_delta
self.signing_key = Cryptographer.load_private_key_der(sk_der) self.signing_key = Cryptographer.load_private_key_der(sk_der)
self.last_known_block = db_manager.load_last_block_hash_watcher()
def awake(self): def awake(self):
"""Starts a new thread to monitor the blockchain for channel breaches""" """Starts a new thread to monitor the blockchain for channel breaches"""
@@ -166,6 +168,11 @@ class Watcher:
:obj:`Responder <teos.responder.Responder>` upon detecting a breach. :obj:`Responder <teos.responder.Responder>` upon detecting a breach.
""" """
# Distinguish fresh bootstraps from bootstraps from db
if self.last_known_block is None:
self.last_known_block = self.block_processor.get_best_block_hash()
self.db_manager.store_last_block_hash_watcher(self.last_known_block)
while True: while True:
block_hash = self.block_queue.get() block_hash = self.block_queue.get()
block = self.block_processor.get_block(block_hash) block = self.block_processor.get_block(block_hash)
@@ -228,6 +235,7 @@ class Watcher:
# Register the last processed block for the watcher # Register the last processed block for the watcher
self.db_manager.store_last_block_hash_watcher(block_hash) self.db_manager.store_last_block_hash_watcher(block_hash)
self.last_known_block = block.get("hash")
self.block_queue.task_done() self.block_queue.task_done()
def get_breaches(self, txids): def get_breaches(self, txids):