feat(api): log errors from background jobs

ssube · Feb 4, 2023 · 53f4924 · 53f4924
1 parent f5039d8
commit 53f4924
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/api/onnx_web/device_pool.py b/api/onnx_web/device_pool.py
@@ -125,5 +125,14 @@ def submit(self, key: str, fn: Callable[..., None], /, *args, **kwargs) -> None:
         job = Job(key, future, context)
         self.jobs.append(job)
 
+        def job_done(f: Future):
+            try:
+                f.result()
+                logger.info('job %s finished successfully', key)
+            except Exception as err:
+                logger.warn('job %s failed with an error: %s', key, err)
+
+        future.add_done_callback(job_done)
+
     def status(self) -> Dict[str, Tuple[bool, int]]:
         return [(job.key, job.future.done(), job.get_progress()) for job in self.jobs]