o2ims/service/watcher/ocloud_watcher.py

   1 # Copyright (C) 2021 Wind River Systems, Inc.
   2 #
   3 #  Licensed under the Apache License, Version 2.0 (the "License");
   4 #  you may not use this file except in compliance with the License.
   5 #  You may obtain a copy of the License at
   6 #
   7 #      http://www.apache.org/licenses/LICENSE-2.0
   8 #
   9 #  Unless required by applicable law or agreed to in writing, software
  10 #  distributed under the License is distributed on an "AS IS" BASIS,
  11 #  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12 #  See the License for the specific language governing permissions and
  13 #  limitations under the License.
  14
  15 # from o2ims.domain.resource_type import ResourceTypeEnum
  16 from o2common.service.client.base_client import BaseClient
  17 from o2ims.domain.stx_object import StxGenericModel
  18 # from o2common.service.unit_of_work import AbstractUnitOfWork
  19 from o2common.service.watcher.base import BaseWatcher
  20 from o2ims.domain import commands
  21 from o2common.service.messagebus import MessageBus
  22
  23 from o2common.helper import o2logging
  24 logger = o2logging.get_logger(__name__)
  25
  26
  27 class OcloudWatcher(BaseWatcher):
  28     def __init__(self, ocloud_client: BaseClient,
  29                  bus: MessageBus) -> None:
  30         super().__init__(ocloud_client, bus)
  31
  32     def _targetname(self):
  33         return "ocloud"
  34
  35     def _probe(self, parent: object = None, tags: object = None):
  36         newmodel = self._client.get(None)
  37         if newmodel:
  38             logger.debug("found ocloud: " + newmodel.name)
  39         else:
  40             logger.warning("Failed to find out any ocloud")
  41         #     self._compare_and_update(ocloudmodel)
  42         return [commands.UpdateOCloud(newmodel)] if newmodel else []
  43
  44 # def _compare_and_update(self, ocloudmodel: StxGenericModel) -> bool:
  45 #     with self._uow:
  46 #         # localmodel = self._uow.stxobjects.get(str(ocloudmodel.id))
  47 #         oclouds = self._uow.stxobjects.list(ResourceTypeEnum.OCLOUD)
  48 #         if len(oclouds) > 1:
  49 #             raise InvalidOcloudState("More than 1 ocloud is found")
  50 #         if len(oclouds) == 0:
  51 #             logger.info("add ocloud:" + ocloudmodel.name
  52 #                         + " update_at: " + str(ocloudmodel.updatetime)
  53 #                         + " id: " + str(ocloudmodel.id)
  54 #                         + " hash: " + str(ocloudmodel.hash))
  55 #             self._uow.stxobjects.add(ocloudmodel)
  56 #         else:
  57 #             localmodel = oclouds.pop()
  58 #             if localmodel.is_outdated(ocloudmodel):
  59 #                 logger.info("update ocloud:" + ocloudmodel.name
  60 #                             + " update_at: " + str(ocloudmodel.updatetime)
  61 #                             + " id: " + str(ocloudmodel.id)
  62 #                             + " hash: " + str(ocloudmodel.hash))
  63 #                 localmodel.update_by(ocloudmodel)
  64 #                 self._uow.stxobjects.update(localmodel)
  65 #         self._uow.commit()
  66
  67
  68 class DmsWatcher(BaseWatcher):
  69     def __init__(self, client: BaseClient,
  70                  bus: MessageBus) -> None:
  71         super().__init__(client, bus)
  72
  73     def _targetname(self):
  74         return "dms"
  75
  76     def _probe(self, parent: StxGenericModel, tags: object = None):
  77         ocloudid = parent.id
  78         newmodels = self._client.list(ocloudid=ocloudid)
  79         # for newmodel in newmodels:
  80         #     super()._compare_and_update(newmodel)
  81         # return newmodels
  82         return [commands.UpdateDms(data=m, parentid=ocloudid)
  83                 for m in newmodels]