Graceful shutdown of driver

2022-04-02 11:21:58 +00:00 · 2022-04-02 11:21:58 +00:00 · 5f298cb804
commit 5f298cb804
parent f4c6936517
1 changed files with 4 additions and 0 deletions
--- a/chainsyncer/driver/base.py
+++ b/chainsyncer/driver/base.py
@ -40,6 +40,10 @@ class SyncDriver:
        self.clock_id = time.CLOCK_MONOTONIC_RAW
        self.store.connect()
        self.store.start(offset=offset, target=target)
+        if not SyncDriver.signal_set:
+            for sig in SyncDriver.signal_request:
+                signal.signal(sig, self.__sig_terminate)
+            SyncDriver.signal_set = True


    def __sig_terminate(self, sig, frame):