chainsyncer

commit a4ad3feb2f17005c5758a30cc80a21b6f7f23ffa
parent d837602394e0c9f4a5451119faa93788e94022ab
Author: nolash <dev@holbrook.no>
Date:   Mon, 22 Feb 2021 23:38:42 +0100

Add graceful shutdown

Diffstat:
M
chainsyncer/driver.py
 | 
23
++++++++++++++++++++++-
M
chainsyncer/filter.py
 | 
3
++-

2 files changed, 24 insertions(+), 2 deletions(-)
diff --git a/chainsyncer/driver.py b/chainsyncer/driver.py
@@ -2,12 +2,14 @@
 import uuid
 import logging
 import time
+import signal
 
 # external imports
 from chainlib.eth.block import (
         block_by_number,
         Block,
         )
+from chainlib.eth.tx import receipt
 
 # local imports
 from chainsyncer.filter import SyncFilter
@@ -23,6 +25,7 @@ class Syncer:
 
     running_global = True
     yield_delay=0.005
+    signal_set = False
 
     def __init__(self, backend, loop_callback=noop_callback, progress_callback=noop_callback):
         self.cursor = None
@@ -31,6 +34,22 @@ class Syncer:
         self.filter = SyncFilter(backend)
         self.progress_callback = progress_callback
         self.loop_callback = loop_callback
+        if not Syncer.signal_set:
+            signal.signal(signal.SIGINT, Syncer.__sig_terminate)
+            signal.signal(signal.SIGTERM, Syncer.__sig_terminate)
+            Syncer.signal_set = True
+
+
+    @staticmethod
+    def __sig_terminate(sig, frame):
+        logg.warning('got signal {}'.format(sig))
+        Syncer.terminate()
+
+
+    @staticmethod
+    def terminate():
+        logg.info('termination requested!')
+        Syncer.running_global = False
 
 
     def chain(self):
@@ -61,7 +80,7 @@ class BlockPollSyncer(Syncer):
         while self.running and Syncer.running_global:
             if self.loop_callback != None:
                 self.loop_callback(last_block, last_tx)
-            while True:
+            while True and Syncer.running_global:
                 try:
                     block = self.get(conn)
                 except Exception as e:
@@ -89,6 +108,8 @@ class HeadSyncer(BlockPollSyncer):
         while True:
             try:
                 tx = block.tx(i)
+                rcpt = conn.do(receipt(tx.hash))
+                tx.apply_receipt(rcpt)
                 self.progress_callback(block.number, i, 'processing {}'.format(repr(tx)))
                 self.backend.set(block.number, i)
                 self.filter.apply(conn, block, tx)
diff --git a/chainsyncer/filter.py b/chainsyncer/filter.py
@@ -39,7 +39,8 @@ class SyncFilter:
             logg.debug('applying filter {}'.format(str(f)))
             f.filter(conn, block, tx, self.backend.db_session)
             self.backend.complete_filter(i)
-        self.backend.disconnect()
+        if session != None:
+            self.backend.disconnect()
 
 
 class NoopFilter: