mirror of
https://github.com/aljazceru/python-teos.git
synced 2025-12-18 14:44:21 +01:00
89181e6a7e
Data obtained from the database was not properly handled in some cases: - If some appointments are accepted and no block is received and the tower is restarted, the data was not loaded. The tower was checking that the Watcher/Responder had a last known block, which may not have been the case. Now best_tip is set as last_known_block on a fresh bootstrap. - The Watcher/ Responder were not being awaken when loading data from the database. - The block queues were not properly populated
121 lines
4.8 KiB
Python
121 lines
4.8 KiB
Python
from getopt import getopt
|
|
from sys import argv, exit
|
|
from signal import signal, SIGINT, SIGQUIT, SIGTERM
|
|
|
|
from common.logger import Logger
|
|
|
|
from pisa import config, LOG_PREFIX
|
|
from pisa.api import API
|
|
from pisa.watcher import Watcher
|
|
from pisa.builder import Builder
|
|
from pisa.db_manager import DBManager
|
|
from pisa.chain_monitor import ChainMonitor
|
|
from pisa.block_processor import BlockProcessor
|
|
from pisa.tools import can_connect_to_bitcoind, in_correct_network
|
|
|
|
logger = Logger(actor="Daemon", log_name_prefix=LOG_PREFIX)
|
|
|
|
|
|
def handle_signals(signal_received, frame):
|
|
logger.info("Closing connection with appointments db")
|
|
db_manager.db.close()
|
|
chain_monitor.terminate = True
|
|
|
|
logger.info("Shutting down PISA")
|
|
exit(0)
|
|
|
|
|
|
def main():
|
|
global db_manager, chain_monitor
|
|
|
|
signal(SIGINT, handle_signals)
|
|
signal(SIGTERM, handle_signals)
|
|
signal(SIGQUIT, handle_signals)
|
|
|
|
logger.info("Starting PISA")
|
|
db_manager = DBManager(config.get("DB_PATH"))
|
|
|
|
if not can_connect_to_bitcoind():
|
|
logger.error("Can't connect to bitcoind. Shutting down")
|
|
|
|
elif not in_correct_network(config.get("BTC_NETWORK")):
|
|
logger.error("bitcoind is running on a different network, check conf.py and bitcoin.conf. Shutting down")
|
|
|
|
else:
|
|
try:
|
|
# Create the chain monitor and start monitoring the chain
|
|
chain_monitor = ChainMonitor()
|
|
chain_monitor.monitor_chain()
|
|
|
|
watcher_appointments_data = db_manager.load_watcher_appointments()
|
|
responder_trackers_data = db_manager.load_responder_trackers()
|
|
|
|
with open(config.get("PISA_SECRET_KEY"), "rb") as key_file:
|
|
secret_key_der = key_file.read()
|
|
|
|
watcher = Watcher(db_manager, chain_monitor, secret_key_der, config)
|
|
chain_monitor.attach_watcher(watcher.block_queue, watcher.asleep)
|
|
chain_monitor.attach_responder(watcher.responder.block_queue, watcher.responder.asleep)
|
|
|
|
if len(watcher_appointments_data) == 0 and len(responder_trackers_data) == 0:
|
|
logger.info("Fresh bootstrap")
|
|
|
|
# Set the current tip as the last known block for both on a fresh start
|
|
db_manager.store_last_block_hash_watcher(BlockProcessor.get_best_block_hash())
|
|
db_manager.store_last_block_hash_responder(BlockProcessor.get_best_block_hash())
|
|
|
|
else:
|
|
logger.info("Bootstrapping from backed up data")
|
|
block_processor = BlockProcessor()
|
|
|
|
last_block_watcher = db_manager.load_last_block_hash_watcher()
|
|
last_block_responder = db_manager.load_last_block_hash_responder()
|
|
|
|
# FIXME: 32-reorgs-offline dropped txs are not used at this point.
|
|
missed_blocks_responder = None
|
|
|
|
# Build Responder with backed up data if found
|
|
if len(responder_trackers_data) != 0:
|
|
last_common_ancestor_responder, dropped_txs_responder = block_processor.find_last_common_ancestor(
|
|
last_block_responder
|
|
)
|
|
missed_blocks_responder = block_processor.get_missed_blocks(last_common_ancestor_responder)
|
|
|
|
watcher.responder.trackers, watcher.responder.tx_tracker_map = Builder.build_trackers(
|
|
responder_trackers_data
|
|
)
|
|
Builder.populate_block_queue(watcher.responder.block_queue, missed_blocks_responder)
|
|
watcher.responder.awake()
|
|
|
|
# Build Watcher. If the blocks of both match we don't perform the search twice.
|
|
if len(watcher_appointments_data) != 0:
|
|
if last_block_watcher == last_block_responder and missed_blocks_responder is not None:
|
|
missed_blocks_watcher = missed_blocks_responder
|
|
else:
|
|
last_common_ancestor_watcher, dropped_txs_watcher = block_processor.find_last_common_ancestor(
|
|
last_block_watcher
|
|
)
|
|
missed_blocks_watcher = block_processor.get_missed_blocks(last_common_ancestor_watcher)
|
|
|
|
watcher.appointments, watcher.locator_uuid_map = Builder.build_appointments(
|
|
watcher_appointments_data
|
|
)
|
|
Builder.populate_block_queue(watcher.block_queue, missed_blocks_watcher)
|
|
watcher.awake()
|
|
|
|
# Fire the API
|
|
API(watcher, config=config).start()
|
|
|
|
except Exception as e:
|
|
logger.error("An error occurred: {}. Shutting down".format(e))
|
|
exit(1)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
opts, _ = getopt(argv[1:], "", [""])
|
|
for opt, arg in opts:
|
|
# FIXME: Leaving this here for future option/arguments
|
|
pass
|
|
|
|
main()
|