poller: save state before waiting for workers

If workers hang stopping them will fail and we'll never save
the processing state. Looks like this has be broken for a while.

Signed-off-by: Jakub Kicinski <[email protected]>

Author: kuba-moo

pw_poller.py

@@ -249,6 +249,12 @@ def run(self, life) -> None:
         except KeyboardInterrupt:
             pass  # finally will still run, but don't splat
         finally:
+            # Dump state before trying to stop workers, in case they hang
+            self._state['last_poll'] = prev_big_scan.timestamp()
+            self._state['done_series'] = list(self.seen_series)
+            with open('poller.state', 'w') as f:
+                json.dump(self._state, f)
+
             log_open_sec(f"Stopping threads")
             for worker in self._workers:
                 worker.should_die = True
@@ -258,12 +264,6 @@ def run(self, life) -> None:
                 worker.join()
             log_end_sec()
 
-            self._state['last_poll'] = prev_big_scan.timestamp()
-            self._state['done_series'] = list(self.seen_series)
-            # Dump state
-            with open('poller.state', 'w') as f:
-                json.dump(self._state, f)
-
 
 if __name__ == "__main__":
     os.umask(0o002)