mirror of
https://github.com/ansible/awx.git
synced 2026-02-16 10:40:01 -03:30
c3ee0c2d8a
* Sensible log behavior when redis is unavailable * Consistent behavior with dispatcher and callback
49 lines
1.6 KiB
Python
49 lines
1.6 KiB
Python
# Copyright (c) 2015 Ansible, Inc.
|
|
# All Rights Reserved.
|
|
|
|
import redis
|
|
|
|
from django.conf import settings
|
|
from django.core.management.base import BaseCommand, CommandError
|
|
import redis.exceptions
|
|
|
|
from awx.main.analytics.subsystem_metrics import CallbackReceiverMetricsServer
|
|
from awx.main.dispatch.control import Control
|
|
from awx.main.dispatch.worker import AWXConsumerRedis, CallbackBrokerWorker
|
|
|
|
|
|
class Command(BaseCommand):
|
|
"""
|
|
Save Job Callback receiver
|
|
Runs as a management command and receives job save events. It then hands
|
|
them off to worker processors (see Worker) which writes them to the database
|
|
"""
|
|
|
|
help = 'Launch the job callback receiver'
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument('--status', dest='status', action='store_true', help='print the internal state of any running dispatchers')
|
|
|
|
def handle(self, *arg, **options):
|
|
if options.get('status'):
|
|
print(Control('callback_receiver').status())
|
|
return
|
|
consumer = None
|
|
|
|
try:
|
|
CallbackReceiverMetricsServer().start()
|
|
except redis.exceptions.ConnectionError as exc:
|
|
raise CommandError(f'Callback receiver could not connect to redis, error: {exc}')
|
|
|
|
try:
|
|
consumer = AWXConsumerRedis(
|
|
'callback_receiver',
|
|
CallbackBrokerWorker(),
|
|
queues=[getattr(settings, 'CALLBACK_QUEUE', '')],
|
|
)
|
|
consumer.run()
|
|
except KeyboardInterrupt:
|
|
print('Terminating Callback Receiver')
|
|
if consumer:
|
|
consumer.stop()
|