kill process when the worker crashesrelease/1.5.0

1 files changed, 4 insertions, 6 deletions

diff --git a/api.py b/api.py
index ddf891d..afc08aa 100644
--- a/api.py
+++ b/api.py
@@ -283,15 +283,13 @@ class Analyzer(joblib.worker.Worker):
                      object_type, object_id)
 
 async def startup():
-    for _ in range(1):
-        worker = Analyzer()
-        await worker.connect(db_config, queue_db)
-        await worker.setup()
-        await worker.start(batchlimit=1000)
+    worker = Analyzer()
+    await worker.connect(db_config, queue_db)
+    await worker.setup()
+    await worker.run(batchlimit=1000)
 
 
 logging.basicConfig(level=logging.DEBUG)
 
 loop = asyncio.get_event_loop()
 loop.run_until_complete(startup())
-loop.run_forever()