kill process when the worker crashesrelease/1.5.0

1 files changed, 12 insertions, 14 deletions

diff --git a/api.py b/api.py
index cec9dc3..71ec6a3 100644
--- a/api.py
+++ b/api.py
@@ -71,19 +71,18 @@ class Apigrabber(joblib.worker.Worker):
 
 
 async def startup():
-    for _ in range(1):
-        worker = Apigrabber(
-            apitoken=os.environ["VAINSOCIAL_APITOKEN"]
-        )
-        await worker.connect(
-            host=os.environ["POSTGRESQL_HOST"],
-            port=os.environ["POSTGRESQL_PORT"],
-            user=os.environ["POSTGRESQL_USER"],
-            password=os.environ["POSTGRESQL_PASSWORD"],
-            database=os.environ["POSTGRESQL_DB"]
-        )
-        await worker.setup()
-        await worker.start()
+    worker = Apigrabber(
+        apitoken=os.environ["VAINSOCIAL_APITOKEN"]
+    )
+    await worker.connect(
+        host=os.environ["POSTGRESQL_HOST"],
+        port=os.environ["POSTGRESQL_PORT"],
+        user=os.environ["POSTGRESQL_USER"],
+        password=os.environ["POSTGRESQL_PASSWORD"],
+        database=os.environ["POSTGRESQL_DB"]
+    )
+    await worker.setup()
+    await worker.run(batchlimit=1)
 
 
 if __name__ == "__main__":
@@ -91,4 +90,3 @@ if __name__ == "__main__":
 
     loop = asyncio.get_event_loop()
     loop.run_until_complete(startup())
-    loop.run_forever()