+# Copyright (C) 2021 Wind River Systems, Inc.\r
+#\r
+# Licensed under the Apache License, Version 2.0 (the "License");\r
+# you may not use this file except in compliance with the License.\r
+# You may obtain a copy of the License at\r
+#\r
+# http://www.apache.org/licenses/LICENSE-2.0\r
+#\r
+# Unless required by applicable law or agreed to in writing, software\r
+# distributed under the License is distributed on an "AS IS" BASIS,\r
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
+# See the License for the specific language governing permissions and\r
+# limitations under the License.\r
+\r
+import time\r
+import sched\r
+from o2ims.service.watcher.base import BaseWatcher\r
+\r
+import logging\r
+logger = logging.getLogger(__name__)\r
+\r
+\r
+class PollWorker(object):\r
+ def __init__(self, interval=10) -> None:\r
+ super().__init__()\r
+ self.watchers = {}\r
+ self.schedinstance = sched.scheduler(time.time, time.sleep)\r
+ self.schedinterval = interval\r
+ self._stopped = True\r
+\r
+ def set_interval(self, interval):\r
+ if interval > 0:\r
+ self.schedinterval = interval\r
+ else:\r
+ raise Exception("Invalid interval:" + interval)\r
+\r
+ def add_watcher(self, watcher: BaseWatcher):\r
+ self.watchers[watcher.targetname()] = watcher\r
+\r
+ def _repeat(self):\r
+ logger.debug("_repeat started")\r
+ if self._stopped:\r
+ return\r
+ for w in self.watchers.keys():\r
+ try:\r
+ logger.debug("about to probe:"+w)\r
+ self.watchers[w].probe()\r
+ except Exception as ex:\r
+ logger.warning(ex.message)\r
+ continue\r
+ self.schedinstance.enter(self.schedinterval, 1, self._repeat)\r
+\r
+ # note the sched run will block current thread\r
+ def start(self):\r
+ self._stopped = False\r
+ logger.debug('about to start sched task')\r
+ self.schedinstance.enter(self.schedinterval, 1, self._repeat)\r
+ self.schedinstance.run()\r
+\r
+ def stop(self):\r
+ self._stopped = True\r
+\r
+\r
+defaultworker = PollWorker()\r