X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=o2ims%2Fservice%2Fwatcher%2Fbase.py;h=0daf8d43a7fbf37721a1354e3bef2c71d648972f;hb=62f8863960ebd439c714b0ceed204731d9b31266;hp=c3ff3d4fd1176d97110d4473929350dc93dbb15a;hpb=8be81dfad35b08c4de77168e885bb18253069771;p=pti%2Fo2.git diff --git a/o2ims/service/watcher/base.py b/o2ims/service/watcher/base.py index c3ff3d4..0daf8d4 100644 --- a/o2ims/service/watcher/base.py +++ b/o2ims/service/watcher/base.py @@ -12,151 +12,87 @@ # See the License for the specific language governing permissions and # limitations under the License. -from o2ims.domain.resource_type import ResourceTypeEnum +# from logging import exception +# from cgtsclient import exc from o2ims.service.client.base_client import BaseClient -from o2ims.domain.stx_object import StxGenericModel -from o2ims.service.unit_of_work import AbstractUnitOfWork - -import logging -logger = logging.getLogger(__name__) - - -class InvalidOcloudState(Exception): - pass +# from o2ims.domain.stx_object import StxGenericModel +# from o2ims.service.unit_of_work import AbstractUnitOfWork +from o2ims.domain import commands +from o2ims.service.messagebus import MessageBus +from o2common.helper import o2logging +logger = o2logging.get_logger(__name__) class BaseWatcher(object): - def __init__(self, client: BaseClient) -> None: + def __init__(self, client: BaseClient, + bus: MessageBus) -> None: super().__init__() self._client = client + self._bus = bus + # self._uow = bus.uow def targetname(self) -> str: return self._targetname() - def probe(self): - self._probe() + def probe(self, parent: commands.UpdateStxObject = None): + try: + cmds = self._probe(parent.data if parent else None) + for cmd in cmds: + self._bus.handle(cmd) - def _probe(self): - raise NotImplementedError + # return self._probe(parent) + return cmds + except Exception as ex: + logger.warning("Failed to probe resource due to: " + str(ex)) + return [] - def _targetname(self): + def _probe(self, parent: object = None) -> commands.UpdateStxObject: raise NotImplementedError - -class OcloudWatcher(BaseWatcher): - def __init__(self, ocloud_client: BaseClient, - uow: AbstractUnitOfWork) -> None: - super().__init__(ocloud_client) - self._uow = uow - - def _targetname(self): - return "ocloud" - - def _probe(self): - ocloudmodel = self._client.get(None) - if ocloudmodel: - self._compare_and_update(ocloudmodel) - - def _compare_and_update(self, ocloudmodel: StxGenericModel) -> bool: - with self._uow: - # localmodel = self._uow.stxobjects.get(str(ocloudmodel.id)) - oclouds = self._uow.stxobjects.list(ResourceTypeEnum.OCLOUD) - if len(oclouds) > 1: - raise InvalidOcloudState("More than 1 ocloud is found") - if len(oclouds) == 0: - logger.info("add ocloud:" + ocloudmodel.name - + " update_at: " + str(ocloudmodel.updatetime) - + " id: " + str(ocloudmodel.id) - + " hash: " + str(ocloudmodel.hash)) - self._uow.stxobjects.add(ocloudmodel) - else: - localmodel = oclouds.pop() - if localmodel.is_outdated(ocloudmodel): - logger.info("update ocloud:" + ocloudmodel.name - + " update_at: " + str(ocloudmodel.updatetime) - + " id: " + str(ocloudmodel.id) - + " hash: " + str(ocloudmodel.hash)) - localmodel.update_by(ocloudmodel) - self._uow.stxobjects.update(localmodel) - self._uow.commit() - - -class DmsWatcher(BaseWatcher): - def __init__(self, client: BaseClient, - uow: AbstractUnitOfWork) -> None: - super().__init__(client) - self._uow = uow - def _targetname(self): - return "dms" - - def _probe(self): - ocloudmodel = self._client.get(None) - if ocloudmodel: - self._compare_and_update(ocloudmodel) - - def _compare_and_update(self, newmodel: StxGenericModel) -> bool: - with self._uow: - # localmodel = self._uow.stxobjects.get(ocloudmodel.id) - localmodel = self._uow.stxobjects.get(str(newmodel.id)) - if not localmodel: - logger.info("add dms:" + newmodel.name) - self._uow.stxobjects.add(newmodel) - elif localmodel.is_outdated(newmodel): - logger.info("update dms:" + newmodel.name) - localmodel.update_by(newmodel) - self._uow.stxobjects.update(newmodel) - self._uow.commit() - - -class ResourcePoolWatcher(BaseWatcher): - def __init__(self, client: BaseClient, - uow: AbstractUnitOfWork) -> None: - super().__init__() - self._uow = uow + raise NotImplementedError - def _targetname(self): - return "resourcepool" - - def _probe(self): - ocloudmodel = self._client.get(None) - if ocloudmodel: - logger.info("detect ocloudmodel:" + ocloudmodel.name) - self._compare_and_update(ocloudmodel) - - def _compare_and_update(self, newmodel: StxGenericModel) -> bool: - with self._uow: - # localmodel = self._uow.stxobjects.get(ocloudmodel.id) - localmodel = self._uow.stxobjects.get(str(newmodel.id)) - if not localmodel: - self._uow.stxobjects.add(newmodel) - elif localmodel.is_outdated(newmodel): - localmodel.update_by(newmodel) - self._uow.stxobjects.update(newmodel) - self._uow.commit() - - -class ResourceWatcher(BaseWatcher): - def __init__(self, client: BaseClient, - uow: AbstractUnitOfWork) -> None: + # def _compare_and_update(self, newmodel: StxGenericModel) -> bool: + # with self._uow: + # # localmodel = self._uow.stxobjects.get(ocloudmodel.id) + # localmodel = self._uow.stxobjects.get(str(newmodel.id)) + # if not localmodel: + # logger.info("add entry:" + newmodel.name) + # self._uow.stxobjects.add(newmodel) + # elif localmodel.is_outdated(newmodel): + # logger.info("update entry:" + newmodel.name) + # localmodel.update_by(newmodel) + # self._uow.stxobjects.update(localmodel) + # self._uow.commit() + + +# node to organize watchers in tree hierachy +class WatcherTree(object): + def __init__(self, watcher: BaseWatcher) -> None: super().__init__() - self._uow = uow - - def _targetname(self): - return "resource" - - def _probe(self): - ocloudmodel = self._client.get(None) - if ocloudmodel: - self._compare_and_update(ocloudmodel) - - def _compare_and_update(self, newmodel: StxGenericModel) -> bool: - with self._uow: - # localmodel = self._repo.get(ocloudmodel.id) - localmodel = self._uow.stxobjects.get(str(newmodel.id)) - if not localmodel: - self._uow.stxobjects.add(newmodel) - elif localmodel.is_outdated(newmodel): - localmodel.update_by(newmodel) - self._uow.stxobjects.update(newmodel) - self._uow.commit() + self.watcher = watcher + self.children = {} + + def addchild(self, watcher: BaseWatcher) -> object: + child = WatcherTree(watcher) + self.children[watcher.targetname()] = child + return child + + def removechild(self, targetname: str) -> object: + return self.children.pop(targetname) + + # probe all resources by parent, depth = 0 for indefinite recursive + def probe(self, parentresource=None, depth: int = 0): + logger.debug("probe resources with watcher: " + + self.watcher.targetname()) + childdepth = depth - 1 if depth > 0 else 0 + resources = self.watcher.probe(parentresource) + logger.debug("probe returns " + str(len(resources)) + " resources") + + if depth == 1: + # stop recursive + return + + for res in resources: + for targetname in self.children.keys(): + self.children[targetname].probe(res, childdepth)