mirror of https://github.com/zulip/zulip.git
33 lines
935 B
Python
33 lines
935 B
Python
|
from __future__ import absolute_import
|
||
|
|
|
||
|
|
from django.core.management.base import BaseCommand
|
||
|
|
from django.core.management import CommandError
|
||
|
|
from django.conf import settings
|
||
|
|
from zerver.worker.queue_processors import get_worker
|
||
|
|
import sys
|
||
|
|
import signal
|
||
|
|
import logging
|
||
|
|
|
||
|
|
class Command(BaseCommand):
|
||
|
|
args = "<queue name>"
|
||
|
|
help = "Runs a queue processing worker"
|
||
|
|
def handle(self, *args, **options):
|
||
|
|
logging.basicConfig()
|
||
|
|
logger = logging.getLogger('process_queue')
|
||
|
|
|
||
|
|
if len(args) != 1:
|
||
|
|
raise CommandError("Wrong number of arguments")
|
||
|
|
|
||
|
|
def signal_handler(signal, frame):
|
||
|
|
logger.info("Disconnecting from queue")
|
||
|
|
worker.stop()
|
||
|
|
sys.exit(0)
|
||
|
|
|
||
|
|
signal.signal(signal.SIGTERM, signal_handler)
|
||
|
|
signal.signal(signal.SIGINT, signal_handler)
|
||
|
|
|
||
|
|
logger.info("Connecting to queue")
|
||
|
|
worker = get_worker(args[0])
|
||
|
|
worker.start()
|
||
|
|
|