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
15 from o2ims.domain.resource_type import ResourceTypeEnum
\r
16 from o2ims.service.client.base_client import BaseClient
\r
17 from o2ims.domain.stx_object import StxGenericModel
\r
18 from o2ims.service.unit_of_work import AbstractUnitOfWork
\r
19 from o2ims.service.watcher.base import BaseWatcher
\r
22 logger = logging.getLogger(__name__)
\r
25 class InvalidOcloudState(Exception):
\r
29 class OcloudWatcher(BaseWatcher):
\r
30 def __init__(self, ocloud_client: BaseClient,
\r
31 uow: AbstractUnitOfWork) -> None:
\r
32 super().__init__(ocloud_client, uow)
\r
34 def _targetname(self):
\r
37 def _probe(self, parent: object = None):
\r
38 ocloudmodel = self._client.get(None)
\r
40 self._compare_and_update(ocloudmodel)
\r
41 return [ocloudmodel]
\r
43 def _compare_and_update(self, ocloudmodel: StxGenericModel) -> bool:
\r
45 # localmodel = self._uow.stxobjects.get(str(ocloudmodel.id))
\r
46 oclouds = self._uow.stxobjects.list(ResourceTypeEnum.OCLOUD)
\r
47 if len(oclouds) > 1:
\r
48 raise InvalidOcloudState("More than 1 ocloud is found")
\r
49 if len(oclouds) == 0:
\r
50 logger.info("add ocloud:" + ocloudmodel.name
\r
51 + " update_at: " + str(ocloudmodel.updatetime)
\r
52 + " id: " + str(ocloudmodel.id)
\r
53 + " hash: " + str(ocloudmodel.hash))
\r
54 self._uow.stxobjects.add(ocloudmodel)
\r
56 localmodel = oclouds.pop()
\r
57 if localmodel.is_outdated(ocloudmodel):
\r
58 logger.info("update ocloud:" + ocloudmodel.name
\r
59 + " update_at: " + str(ocloudmodel.updatetime)
\r
60 + " id: " + str(ocloudmodel.id)
\r
61 + " hash: " + str(ocloudmodel.hash))
\r
62 localmodel.update_by(ocloudmodel)
\r
63 self._uow.stxobjects.update(localmodel)
\r
67 class DmsWatcher(BaseWatcher):
\r
68 def __init__(self, client: BaseClient,
\r
69 uow: AbstractUnitOfWork) -> None:
\r
70 super().__init__(client, uow)
\r
72 def _targetname(self):
\r
75 def _probe(self, parent: object = None):
\r
76 ocloudid = parent.id if parent else None
\r
77 newmodels = self._client.list(ocloudid=ocloudid)
\r
78 for newmodel in newmodels:
\r
79 super()._compare_and_update(newmodel)
\r