mirror of
https://github.com/ansible/awx.git
synced 2026-03-05 18:51:06 -03: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
43 lines
1.1 KiB
Python
43 lines
1.1 KiB
Python
# Copyright (c) 2018 Ansible by Red Hat
|
|
# All Rights Reserved.
|
|
|
|
import os
|
|
import logging
|
|
import signal
|
|
import time
|
|
|
|
from django import db
|
|
|
|
|
|
from awx.main.utils.redis import get_redis_client
|
|
from awx.main.dispatch.pool import WorkerPool
|
|
|
|
logger = logging.getLogger('awx.main.commands.run_callback_receiver')
|
|
|
|
|
|
def signame(sig):
|
|
return dict((k, v) for v, k in signal.__dict__.items() if v.startswith('SIG') and not v.startswith('SIG_'))[sig]
|
|
|
|
|
|
class AWXConsumerRedis(object):
|
|
|
|
def __init__(self, name, worker):
|
|
self.name = name
|
|
self.pool = WorkerPool()
|
|
self.pool.init_workers(worker.work_loop)
|
|
self.redis = get_redis_client()
|
|
|
|
def run(self):
|
|
signal.signal(signal.SIGINT, self.stop)
|
|
signal.signal(signal.SIGTERM, self.stop)
|
|
|
|
logger.info(f'Callback receiver started with pid={os.getpid()}')
|
|
db.connection.close() # logs use database, so close connection
|
|
|
|
while True:
|
|
time.sleep(60)
|
|
|
|
def stop(self, signum, frame):
|
|
logger.warning('received {}, stopping'.format(signame(signum)))
|
|
raise SystemExit()
|