External-Mirrors/awx
2
0
Fork 0
mirror of https://github.com/ansible/awx.git synced 2026-01-10 15:32:07 -03:30
Code Issues Packages Projects Releases Wiki Activity

graceful killing of receiver worker processes

Browse Source
This commit is contained in:
AlanCoding 2017-08-14 17:13:32 -04:00
parent 8be1bea33a
commit 9ad34edc38
No known key found for this signature in database
GPG Key ID: FD2C3C012A72926B
1 changed files with 14 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
awx/main/management/commands/run_callback_receiver.py
View File

@ -27,6 +27,17 @@ from awx.main.models import * # noqa
logger = logging.getLogger('awx.main.commands.run_callback_receiver')
class WorkerSignalHandler:
def __init__(self):
self.kill_now = False
signal.signal(signal.SIGINT, self.exit_gracefully)
signal.signal(signal.SIGTERM, self.exit_gracefully)
def exit_gracefully(self, *args, **kwargs):
self.kill_now = True
class CallbackBrokerWorker(ConsumerMixin):
def __init__(self, connection, use_workers=True):
self.connection = connection
@ -102,13 +113,14 @@ class CallbackBrokerWorker(ConsumerMixin):
return None
def callback_worker(self, queue_actual, idx):
while True:
signal_handler = WorkerSignalHandler()
while not signal_handler.kill_now:
try:
body = queue_actual.get(block=True, timeout=1)
except QueueEmpty:
continue
except Exception as e:
logger.info("Exception on worker thread, restarting: " + str(e))
logger.error("Exception on worker thread, restarting: " + str(e))
continue
try:
if 'job_id' not in body and 'ad_hoc_command_id' not in body: