Merge pull request #23 from vainglorygame/multiple-workers

support multiple threads per worker

1 files changed, 10 insertions, 7 deletions

diff --git a/api.py b/api.py
index 5c7528d..885795e 100644
--- a/api.py
+++ b/api.py
@@ -129,15 +129,18 @@ class Worker(object):
                 logging.info("nothing to do, idling")
                 await asyncio.sleep(10)
 
+    async def start(self, number=1):
+        """Start jobs in background."""
+        for _ in range(number):
+            asyncio.ensure_future(self.run())
 
 async def startup():
-    for _ in range(1):
-        worker = Worker()
-        await worker.connect(
-            source_db, dest_db
-        )
-        await worker.setup()
-        await worker.run()
+    worker = Worker()
+    await worker.connect(
+        source_db, dest_db
+    )
+    await worker.setup()
+    await worker.start(10)
 
 logging.basicConfig(level=logging.DEBUG)
 loop = asyncio.get_event_loop()