mirror of
https://github.com/ansible/awx.git
synced 2026-06-22 07:07:47 -02:30
f80bbc57d8
* Additional dispatcher removal simplifications and waiting repear updates * Fix double call and logging message * Implement bugbot comment, should reap running on lost instances * Add test case for new pending behavior
51 lines
1.7 KiB
Python
51 lines
1.7 KiB
Python
# Copyright (c) 2015 Ansible, Inc.
|
|
# All Rights Reserved.
|
|
|
|
import redis
|
|
|
|
from django.core.management.base import BaseCommand, CommandError
|
|
import redis.exceptions
|
|
|
|
from awx.main.analytics.subsystem_metrics import CallbackReceiverMetricsServer
|
|
from awx.main.dispatch.worker import AWXConsumerRedis, CallbackBrokerWorker
|
|
from awx.main.utils.redis import get_redis_client
|
|
|
|
|
|
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(self.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())
|
|
consumer.run()
|
|
except KeyboardInterrupt:
|
|
print('Terminating Callback Receiver')
|
|
if consumer:
|
|
consumer.stop()
|
|
|
|
def status(self, *args, **kwargs):
|
|
r = get_redis_client()
|
|
workers = []
|
|
for key in r.keys('awx_callback_receiver_statistics_*'):
|
|
workers.append(r.get(key).decode('utf-8'))
|
|
return '\n'.join(workers)
|