diff options
author | 2020-03-25 10:05:35 +0000 | |
---|---|---|
committer | 2020-03-25 11:38:36 +0000 | |
commit | 03e058dbd9e298eda2ce4fcb3a5babaf08434cf5 (patch) | |
tree | b3ba5a1251f50cdd369ad22439b8745e0aa6c182 /components/pm-subscription-handler/pmsh_service/mod/exit_handler.py | |
parent | 8bfed3e53e55f711b1eb7d9cca696b3519658842 (diff) |
Handle graceful exit of PMSH1.0.2
Issue-ID: DCAEGEN2-1832
Change-Id: If0362e1927f7013d25f0cf23ade5ce9d2bdea8e3
Signed-off-by: emartin <ephraim.martin@est.tech>
Diffstat (limited to 'components/pm-subscription-handler/pmsh_service/mod/exit_handler.py')
-rwxr-xr-x | components/pm-subscription-handler/pmsh_service/mod/exit_handler.py | 48 |
1 files changed, 48 insertions, 0 deletions
diff --git a/components/pm-subscription-handler/pmsh_service/mod/exit_handler.py b/components/pm-subscription-handler/pmsh_service/mod/exit_handler.py new file mode 100755 index 00000000..adc4941c --- /dev/null +++ b/components/pm-subscription-handler/pmsh_service/mod/exit_handler.py @@ -0,0 +1,48 @@ +# ============LICENSE_START=================================================== +# Copyright (C) 2020 Nordix Foundation. +# ============================================================================ +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +# +# SPDX-License-Identifier: Apache-2.0 +# ============LICENSE_END===================================================== + +from mod.pmsh_utils import logger +from mod.subscription import AdministrativeState + + +class ExitHandler: + """ Handles PMSH graceful exit when a SIGTERM signal is received. + + Args: + periodic_tasks (List[PeriodicTask]): PeriodicTasks that needs to be cancelled. + subscription_handler (SubscriptionHandler): The subscription handler instance. + """ + + shutdown_signal_received = False + + def __init__(self, *, periodic_tasks, subscription_handler): + self.periodic_tasks = periodic_tasks + self.subscription_handler = subscription_handler + + def __call__(self, sig_num, frame): + logger.debug(f'ExitHandler was called with signal number: {sig_num}.') + current_sub = self.subscription_handler.current_sub + if current_sub and current_sub.administrativeState == AdministrativeState.UNLOCKED.value: + for thread in self.periodic_tasks: + logger.debug(f'Cancelling periodic task with thread name: {thread.name}.') + thread.cancel() + current_sub.administrativeState = AdministrativeState.LOCKED.value + current_sub.process_subscription(self.subscription_handler.current_nfs, + self.subscription_handler.mr_pub, + self.subscription_handler.app_conf) + ExitHandler.shutdown_signal_received = True |