From 5a5f744651970b7b27c19a585d2bafb72b6bcafe Mon Sep 17 00:00:00 2001
From: hsiang41 <rick.chen@prophetstor.com>
Date: Tue, 20 Nov 2018 11:03:03 +0800
Subject: [PATCH] mgr: Diskprediction local activate when config changes

Trigger diskprediction local action when receive notify configuration changed.

Signed-off-by: Rick Chen <rick.chen@prophetstor.com>
---
 src/pybind/mgr/diskprediction_local/module.py | 25 +++++++++++++------
 1 file changed, 18 insertions(+), 7 deletions(-)

diff --git a/src/pybind/mgr/diskprediction_local/module.py b/src/pybind/mgr/diskprediction_local/module.py
index fd7784886df..8662e2f75fb 100644
--- a/src/pybind/mgr/diskprediction_local/module.py
+++ b/src/pybind/mgr/diskprediction_local/module.py
@@ -3,6 +3,7 @@ diskprediction with local predictor
 """
 import json
 import datetime
+import _strptime
 from threading import Event
 import time
 
@@ -34,8 +35,17 @@ class Module(MgrModule):
         for opt in self.OPTIONS:
             setattr(self, opt['name'], opt['default'])
         # other
-        self.run = True
-        self.event = Event()
+        self._run = True
+        self._event = Event()
+
+    def config_notify(self):
+        for opt in self.OPTIONS:
+            setattr(self,
+                    opt['name'],
+                    self.get_config(opt['name']) or opt['default'])
+            self.log.debug(' %s = %s', opt['name'], getattr(self, opt['name']))
+        if self.get_option('device_failure_prediction_mode') == 'local':
+            self._event.set()
 
     def refresh_config(self):
         for opt in self.OPTIONS:
@@ -55,6 +65,7 @@ class Module(MgrModule):
 
     def serve(self):
         self.log.info('Starting diskprediction local module')
+        self.config_notify()
         last_predicted = None
         ls = self.get_store('last_predicted')
         if ls:
@@ -64,7 +75,7 @@ class Module(MgrModule):
                 pass
         self.log.debug('Last predicted %s', last_predicted)
 
-        while self.run:
+        while self._run:
             self.refresh_config()
             mode = self.get_option('device_failure_prediction_mode')
             if mode == 'local':
@@ -91,13 +102,13 @@ class Module(MgrModule):
 
             sleep_interval = int(self.sleep_interval) or 60
             self.log.debug('Sleeping for %d seconds', sleep_interval)
-            self.event.wait(sleep_interval)
-            self.event.clear()
+            self._event.wait(sleep_interval)
+            self._event.clear()
 
     def shutdown(self):
         self.log.info('Stopping')
-        self.run = False
-        self.event.set()
+        self._run = False
+        self._event.set()
 
     @staticmethod
     def _convert_timestamp(predicted_timestamp, life_expectancy_day):
-- 
2.39.5