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
21 logger = logging.getLogger(__name__)
\r
24 class InvalidOcloudState(Exception):
\r
28 class BaseWatcher(object):
\r
29 def __init__(self, client: BaseClient) -> None:
\r
31 self._client = client
\r
33 def targetname(self) -> str:
\r
34 return self._targetname()
\r
40 raise NotImplementedError
\r
42 def _targetname(self):
\r
43 raise NotImplementedError
\r
46 class OcloudWatcher(BaseWatcher):
\r
47 def __init__(self, ocloud_client: BaseClient,
\r
48 uow: AbstractUnitOfWork) -> None:
\r
49 super().__init__(ocloud_client)
\r
52 def _targetname(self):
\r
56 ocloudmodel = self._client.get(None)
\r
58 self._compare_and_update(ocloudmodel)
\r
60 def _compare_and_update(self, ocloudmodel: StxGenericModel) -> bool:
\r
62 # localmodel = self._uow.stxobjects.get(str(ocloudmodel.id))
\r
63 oclouds = self._uow.stxobjects.list(ResourceTypeEnum.OCLOUD)
\r
64 if len(oclouds) > 1:
\r
65 raise InvalidOcloudState("More than 1 ocloud is found")
\r
66 if len(oclouds) == 0:
\r
67 logger.info("add ocloud:" + ocloudmodel.name
\r
68 + " update_at: " + str(ocloudmodel.updatetime)
\r
69 + " id: " + str(ocloudmodel.id)
\r
70 + " hash: " + str(ocloudmodel.hash))
\r
71 self._uow.stxobjects.add(ocloudmodel)
\r
73 localmodel = oclouds.pop()
\r
74 if localmodel.is_outdated(ocloudmodel):
\r
75 logger.info("update ocloud:" + ocloudmodel.name
\r
76 + " update_at: " + str(ocloudmodel.updatetime)
\r
77 + " id: " + str(ocloudmodel.id)
\r
78 + " hash: " + str(ocloudmodel.hash))
\r
79 localmodel.update_by(ocloudmodel)
\r
80 self._uow.stxobjects.update(localmodel)
\r
84 class DmsWatcher(BaseWatcher):
\r
85 def __init__(self, client: BaseClient,
\r
86 uow: AbstractUnitOfWork) -> None:
\r
87 super().__init__(client)
\r
90 def _targetname(self):
\r
94 ocloudmodel = self._client.get(None)
\r
96 self._compare_and_update(ocloudmodel)
\r
98 def _compare_and_update(self, newmodel: StxGenericModel) -> bool:
\r
100 # localmodel = self._uow.stxobjects.get(ocloudmodel.id)
\r
101 localmodel = self._uow.stxobjects.get(str(newmodel.id))
\r
103 logger.info("add dms:" + newmodel.name)
\r
104 self._uow.stxobjects.add(newmodel)
\r
105 elif localmodel.is_outdated(newmodel):
\r
106 logger.info("update dms:" + newmodel.name)
\r
107 localmodel.update_by(newmodel)
\r
108 self._uow.stxobjects.update(newmodel)
\r
112 class ResourcePoolWatcher(BaseWatcher):
\r
113 def __init__(self, client: BaseClient,
\r
114 uow: AbstractUnitOfWork) -> None:
\r
118 def _targetname(self):
\r
119 return "resourcepool"
\r
122 ocloudmodel = self._client.get(None)
\r
124 logger.info("detect ocloudmodel:" + ocloudmodel.name)
\r
125 self._compare_and_update(ocloudmodel)
\r
127 def _compare_and_update(self, newmodel: StxGenericModel) -> bool:
\r
129 # localmodel = self._uow.stxobjects.get(ocloudmodel.id)
\r
130 localmodel = self._uow.stxobjects.get(str(newmodel.id))
\r
132 self._uow.stxobjects.add(newmodel)
\r
133 elif localmodel.is_outdated(newmodel):
\r
134 localmodel.update_by(newmodel)
\r
135 self._uow.stxobjects.update(newmodel)
\r
139 class ResourceWatcher(BaseWatcher):
\r
140 def __init__(self, client: BaseClient,
\r
141 uow: AbstractUnitOfWork) -> None:
\r
145 def _targetname(self):
\r
149 ocloudmodel = self._client.get(None)
\r
151 self._compare_and_update(ocloudmodel)
\r
153 def _compare_and_update(self, newmodel: StxGenericModel) -> bool:
\r
155 # localmodel = self._repo.get(ocloudmodel.id)
\r
156 localmodel = self._uow.stxobjects.get(str(newmodel.id))
\r
158 self._uow.stxobjects.add(newmodel)
\r
159 elif localmodel.is_outdated(newmodel):
\r
160 localmodel.update_by(newmodel)
\r
161 self._uow.stxobjects.update(newmodel)
\r