Pārlūkot izejas kodu

add multiple workers

David 9 gadi atpakaļ
vecāks
revīzija
c8e5741b44
1 mainītis faili ar 8 papildinājumiem un 2 dzēšanām
  1. 8 2
      back/worker.py

+ 8 - 2
back/worker.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 import os
+from multiprocessing import Pool
 
 import redis
 from rq import Worker, Queue, Connection
@@ -11,8 +12,13 @@ redis_url = 'redis://localhost:6379'
 conn = redis.from_url(redis_url)
 q = Queue('default', connection=conn)
 
-if __name__ == '__main__':
-    with Connection(conn):
+def do_work(elem):
+    with Connection(conn) as c:
+        print("working")
         worker = Worker([q])
         worker.work()
+        print("finished")
 
+if __name__ == '__main__':
+    p = Pool(2)
+    p.map(do_work, [1,2])