Fix for issue#14964 (#15212) (#15595)

This commit is contained in:
mssonicbld 2023-06-29 21:51:19 +08:00 committed by GitHub
parent 2fc98cd8fc
commit d0126e679b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -11,6 +11,7 @@ from sonic_py_common.logger import Logger
from . import utils from . import utils
from sonic_py_common.task_base import ProcessTaskBase from sonic_py_common.task_base import ProcessTaskBase
from .config import Config from .config import Config
import signal
SYSLOG_IDENTIFIER = "system#monitor" SYSLOG_IDENTIFIER = "system#monitor"
REDIS_TIMEOUT_MS = 0 REDIS_TIMEOUT_MS = 0
@ -117,6 +118,8 @@ class Sysmonitor(ProcessTaskBase):
self.state_db = None self.state_db = None
self.config_db = None self.config_db = None
self.config = Config() self.config = Config()
self.mpmgr = multiprocessing.Manager()
self.myQ = self.mpmgr.Queue()
#Sets system ready status to state db #Sets system ready status to state db
def post_system_status(self, state): def post_system_status(self, state):
@ -428,13 +431,11 @@ class Sysmonitor(ProcessTaskBase):
self.state_db = swsscommon.SonicV2Connector(use_unix_socket_path=True) self.state_db = swsscommon.SonicV2Connector(use_unix_socket_path=True)
self.state_db.connect(self.state_db.STATE_DB) self.state_db.connect(self.state_db.STATE_DB)
mpmgr = multiprocessing.Manager()
myQ = mpmgr.Queue()
try: try:
monitor_system_bus = MonitorSystemBusTask(myQ) monitor_system_bus = MonitorSystemBusTask(self.myQ)
monitor_system_bus.task_run() monitor_system_bus.task_run()
monitor_statedb_table = MonitorStateDbTask(myQ) monitor_statedb_table = MonitorStateDbTask(self.myQ)
monitor_statedb_table.task_run() monitor_statedb_table.task_run()
except Exception as e: except Exception as e:
@ -448,7 +449,7 @@ class Sysmonitor(ProcessTaskBase):
# Queue to receive the STATEDB and Systemd state change event # Queue to receive the STATEDB and Systemd state change event
while not self.task_stopping_event.is_set(): while not self.task_stopping_event.is_set():
try: try:
msg = myQ.get(timeout=QUEUE_TIMEOUT) msg = self.myQ.get(timeout=QUEUE_TIMEOUT)
event = msg["unit"] event = msg["unit"]
event_src = msg["evt_src"] event_src = msg["evt_src"]
event_time = msg["time"] event_time = msg["time"]
@ -472,5 +473,24 @@ class Sysmonitor(ProcessTaskBase):
return return
self.system_service() self.system_service()
def task_stop(self):
# Signal the process to stop
self.task_stopping_event.set()
#Clear the resources of mpmgr- Queue
self.mpmgr.shutdown()
# Wait for the process to exit
self._task_process.join(self._stop_timeout_secs)
# If the process didn't exit, attempt to kill it
if self._task_process.is_alive():
logger.log_notice("Attempting to kill sysmon main process with pid {}".format(self._task_process.pid))
os.kill(self._task_process.pid, signal.SIGKILL)
if self._task_process.is_alive():
logger.log_error("Sysmon main process with pid {} could not be killed".format(self._task_process.pid))
return False
return True