mirror of
https://github.com/clearlinux/uwsgi.git
synced 2026-08-25 15:36:07 +00:00
Curtis
46 lines
1.0 KiB
Python
46 lines
1.0 KiB
Python
from uwsgidecorators import spool
|
|
import Queue
|
|
from threading import Thread
|
|
|
|
queues = {}
|
|
|
|
|
|
class queueconsumer(object):
|
|
|
|
def __init__(self, name, num=1, **kwargs):
|
|
self.name = name
|
|
self.num = num
|
|
self.queue = Queue.Queue()
|
|
self.threads = []
|
|
self.func = None
|
|
queues[self.name] = self
|
|
|
|
@staticmethod
|
|
def consumer(self):
|
|
while True:
|
|
req = self.queue.get()
|
|
print req
|
|
self.func(req)
|
|
self.queue.task_done()
|
|
|
|
def __call__(self, f):
|
|
self.func = f
|
|
for i in range(self.num):
|
|
t = Thread(target=self.consumer, args=(self,))
|
|
self.threads.append(t)
|
|
t.daemon = True
|
|
t.start()
|
|
|
|
|
|
@spool
|
|
def spooler_enqueuer(arguments):
|
|
if 'queue' in arguments:
|
|
queue = arguments['queue']
|
|
queues[queue].queue.put(arguments)
|
|
else:
|
|
raise Exception("You have to specify a queue name")
|
|
|
|
|
|
def enqueue(*args, **kwargs):
|
|
return spooler_enqueuer.spool(*args, **kwargs)
|