Add o2dms api endpoint

[pti/o2.git] / o2common / service / watcher / base.py

diff --git a/o2common/service/watcher/base.py b/o2common/service/watcher/base.py
new file mode 100644 (file)
index 0000000..0daf8d4
--- /dev/null
+++ b/o2common/service/watcher/base.py
@@ -0,0 +1,98 @@
+# 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
+# from logging import exception\r
+# from cgtsclient import exc\r
+from o2ims.service.client.base_client import BaseClient\r
+# from o2ims.domain.stx_object import StxGenericModel\r
+# from o2ims.service.unit_of_work import AbstractUnitOfWork\r
+from o2ims.domain import commands\r
+from o2ims.service.messagebus import MessageBus\r
+from o2common.helper import o2logging\r
+logger = o2logging.get_logger(__name__)\r
+\r
+\r
+class BaseWatcher(object):\r
+    def __init__(self, client: BaseClient,\r
+                 bus: MessageBus) -> None:\r
+        super().__init__()\r
+        self._client = client\r
+        self._bus = bus\r
+        # self._uow = bus.uow\r
+\r
+    def targetname(self) -> str:\r
+        return self._targetname()\r
+\r
+    def probe(self, parent: commands.UpdateStxObject = None):\r
+        try:\r
+            cmds = self._probe(parent.data if parent else None)\r
+            for cmd in cmds:\r
+                self._bus.handle(cmd)\r
+\r
+            # return self._probe(parent)\r
+            return cmds\r
+        except Exception as ex:\r
+            logger.warning("Failed to probe resource due to: " + str(ex))\r
+            return []\r
+\r
+    def _probe(self, parent: object = None) -> commands.UpdateStxObject:\r
+        raise NotImplementedError\r
+\r
+    def _targetname(self):\r
+        raise NotImplementedError\r
+\r
+    # def _compare_and_update(self, newmodel: StxGenericModel) -> bool:\r
+    #     with self._uow:\r
+    #         # localmodel = self._uow.stxobjects.get(ocloudmodel.id)\r
+    #         localmodel = self._uow.stxobjects.get(str(newmodel.id))\r
+    #         if not localmodel:\r
+    #             logger.info("add entry:" + newmodel.name)\r
+    #             self._uow.stxobjects.add(newmodel)\r
+    #         elif localmodel.is_outdated(newmodel):\r
+    #             logger.info("update entry:" + newmodel.name)\r
+    #             localmodel.update_by(newmodel)\r
+    #             self._uow.stxobjects.update(localmodel)\r
+    #         self._uow.commit()\r
+\r
+\r
+# node to organize watchers in tree hierachy\r
+class WatcherTree(object):\r
+    def __init__(self, watcher: BaseWatcher) -> None:\r
+        super().__init__()\r
+        self.watcher = watcher\r
+        self.children = {}\r
+\r
+    def addchild(self, watcher: BaseWatcher) -> object:\r
+        child = WatcherTree(watcher)\r
+        self.children[watcher.targetname()] = child\r
+        return child\r
+\r
+    def removechild(self, targetname: str) -> object:\r
+        return self.children.pop(targetname)\r
+\r
+    # probe all resources by parent, depth = 0 for indefinite recursive\r
+    def probe(self, parentresource=None, depth: int = 0):\r
+        logger.debug("probe resources with watcher: "\r
+                     + self.watcher.targetname())\r
+        childdepth = depth - 1 if depth > 0 else 0\r
+        resources = self.watcher.probe(parentresource)\r
+        logger.debug("probe returns " + str(len(resources)) + " resources")\r
+\r
+        if depth == 1:\r
+            # stop recursive\r
+            return\r
+\r
+        for res in resources:\r
+            for targetname in self.children.keys():\r
+                self.children[targetname].probe(res, childdepth)\r