1 # Copyright (C) 2021 Wind River Systems, Inc.
\r
3 # Licensed under the Apache License, Version 2.0 (the "License");
\r
4 # you may not use this file except in compliance with the License.
\r
5 # You may obtain a copy of the License at
\r
7 # http://www.apache.org/licenses/LICENSE-2.0
\r
9 # Unless required by applicable law or agreed to in writing, software
\r
10 # distributed under the License is distributed on an "AS IS" BASIS,
\r
11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
\r
12 # See the License for the specific language governing permissions and
\r
13 # limitations under the License.
\r
17 from o2ims.service.watcher.base import WatcherTree
\r
19 from o2common.helper import o2logging
\r
20 logger = o2logging.get_logger(__name__)
\r
23 class PollWorker(object):
\r
24 def __init__(self, interval=10) -> None:
\r
27 self.schedinstance = sched.scheduler(time.time, time.sleep)
\r
28 self.schedinterval = interval
\r
29 self._stopped = True
\r
31 def set_interval(self, interval):
\r
33 self.schedinterval = interval
\r
35 raise Exception("Invalid interval:" + interval)
\r
37 def add_watcher(self, watcher: WatcherTree):
\r
38 self.watchers.append(watcher)
\r
41 logger.debug("_repeat started")
\r
44 for w in self.watchers:
\r
46 # logger.debug("about to probe:"+w)
\r
48 except Exception as ex:
\r
49 logger.warning("Worker raises exception:" + str(ex))
\r
51 self.schedinstance.enter(self.schedinterval, 1, self._repeat)
\r
53 # note the sched run will block current thread
\r
55 self._stopped = False
\r
56 logger.debug('about to start sched task')
\r
57 self.schedinstance.enter(self.schedinterval, 1, self._repeat)
\r
58 self.schedinstance.run()
\r
61 self._stopped = True
\r