X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=o2common%2Fservice%2Fwatcher%2Fworker.py;fp=o2common%2Fservice%2Fwatcher%2Fworker.py;h=64d189d736f5fcbfc720b42031b9baa4c872a54b;hb=84867b76a65efc4e9add52c86eae7e451c4a038d;hp=0000000000000000000000000000000000000000;hpb=19ee63847f8eb032009e63634b90eb594f3c0408;p=pti%2Fo2.git

diff --git a/o2common/service/watcher/worker.py b/o2common/service/watcher/worker.py
new file mode 100644
index 0000000..64d189d
--- /dev/null
+++ b/o2common/service/watcher/worker.py
@@ -0,0 +1,61 @@
+# Copyright (C) 2021 Wind River Systems, Inc.
+#
+#  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.
+
+import time
+import sched
+from o2common.service.watcher.base import WatcherTree
+
+from o2common.helper import o2logging
+logger = o2logging.get_logger(__name__)
+
+
+class PollWorker(object):
+    def __init__(self, interval=10) -> None:
+        super().__init__()
+        self.watchers = []
+        self.schedinstance = sched.scheduler(time.time, time.sleep)
+        self.schedinterval = interval
+        self._stopped = True
+
+    def set_interval(self, interval):
+        if interval > 0:
+            self.schedinterval = interval
+        else:
+            raise Exception("Invalid interval:" + interval)
+
+    def add_watcher(self, watcher: WatcherTree):
+        self.watchers.append(watcher)
+
+    def _repeat(self):
+        logger.debug("_repeat started")
+        if self._stopped:
+            return
+        for w in self.watchers:
+            try:
+                # logger.debug("about to probe:"+w)
+                w.probe(None)
+            except Exception as ex:
+                logger.warning("Worker raises exception:" + str(ex))
+                continue
+        self.schedinstance.enter(self.schedinterval, 1, self._repeat)
+
+    # note the sched run will block current thread
+    def start(self):
+        self._stopped = False
+        logger.debug('about to start sched task')
+        self.schedinstance.enter(self.schedinterval, 1, self._repeat)
+        self.schedinstance.run()
+
+    def stop(self):
+        self._stopped = True