paulmataruso/zulip
1
0
Fork 0
mirror of https://github.com/zulip/zulip.git synced 2025-11-19 14:08:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

Prepare process_queue to be used with supervisor's numprocs

Browse Source 
(imported from commit 5a652b93f1e8b32b5ed89d622035161abaedfb11)
This commit is contained in:
Zev Benjamin
2013-09-06 15:58:15 -04:00
parent 90ee96d456
commit 871afde142
1 changed files with 11 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
zerver/management/commands/process_queue.py
View File

@@ -9,24 +9,30 @@ import signal
import logging import logging
class Command(BaseCommand): class Command(BaseCommand):
args = "<queue name>" args = "<queue name> [<worker number>]"
help = "Runs a queue processing worker" help = "Runs a queue processing worker"
def handle(self, *args, **options): def handle(self, *args, **options):
logging.basicConfig() logging.basicConfig()
logger = logging.getLogger('process_queue') logger = logging.getLogger('process_queue')
if len(args) != 1: if len(args) not in (1, 2):
raise CommandError("Wrong number of arguments") raise CommandError("Wrong number of arguments")
queue_name = args[0]
if len(args) > 1:
worker_num = int(args[1])
else:
worker_num = 0
def signal_handler(signal, frame): def signal_handler(signal, frame):
logger.info("Disconnecting from queue") logger.info("Worker %d disconnecting from queue %s" % (worker_num, queue_name))
worker.stop() worker.stop()
sys.exit(0) sys.exit(0)
signal.signal(signal.SIGTERM, signal_handler) signal.signal(signal.SIGTERM, signal_handler)
signal.signal(signal.SIGINT, signal_handler) signal.signal(signal.SIGINT, signal_handler)
logger.info("Connecting to queue") logger.info("Worker %d connecting to queue %s" % (worker_num, queue_name))
worker = get_worker(args[0]) worker = get_worker(queue_name)
worker.start() worker.start()