From 60a8aac5f7acc61432fc6a59843e2c91d5df6a35 Mon Sep 17 00:00:00 2001
From: schneefux <schneefux+commit@schneefux.xyz>
Date: Sun, 19 Mar 2017 15:37:54 +0100
Subject: kill process when the worker crashes

---
 api.py | 26 ++++++++++++--------------
 1 file 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()
-- 
cgit v1.3.1