External-Mirrors/awx
2
0
Fork 0
mirror of https://github.com/ansible/awx.git synced 2026-02-24 14:36:00 -03:30
Code Issues Packages Projects Releases Wiki Activity

catch SIGTERM or SIGINT and send offline message

Browse Source 
Signed-off-by: jessicamack <jmack@redhat.com>
This commit is contained in:
jessicamack
2023-04-13 15:21:59 -04:00
parent 7fda4b0675
commit 919d1e5d40
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
awx/main/management/commands/run_heartbeet.py
View File

@@ -2,6 +2,7 @@ import json
import logging
import os
import time
import signal
from django.core.management.base import BaseCommand
from django.conf import settings
@@ -56,6 +57,8 @@ class Command(BaseCommand):
logger.debug('Sending heartbeat')
conn.notify('web_heartbeet', self.construct_payload())
time.sleep(settings.BROADCAST_WEBSOCKET_BEACON_FROM_WEB_RATE_SECONDS)
signal.signal(signal.SIGTERM, conn.notify('web_heartbeet', self.construct_payload(action='offline')))
signal.signal(signal.SIGINT, conn.notify('web_heartbeet', self.construct_payload(action='offline')))
# TODO: Send a message with action=offline if we notice a SIGTERM or SIGINT
# (wsrelay can use this to remove the node quicker)