X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=o2ims%2Fservice%2Fwatcher%2Fpserver_cpu_watcher.py;h=c6c601f24853aa053adf13fc448cdf48e83d1edd;hb=576ad21504bb3e50546b349c63ebec7bd0c805ab;hp=51a5a76e1715f54343f88f3c43e1f0457800ef82;hpb=f73c8e3b01b8f5b7438ba544870e06d8f30cdea0;p=pti%2Fo2.git

diff --git a/o2ims/service/watcher/pserver_cpu_watcher.py b/o2ims/service/watcher/pserver_cpu_watcher.py
index 51a5a76..c6c601f 100644
--- a/o2ims/service/watcher/pserver_cpu_watcher.py
+++ b/o2ims/service/watcher/pserver_cpu_watcher.py
@@ -1,38 +1,42 @@
-# Copyright (C) 2021 Wind River Systems, Inc.
-#
-#  Licensed under the Apache License, Version 2.0 (the "License");
-#  you may not use this file except in compliance with the License.
-#  You may obtain a copy of the License at
-#
-#      http://www.apache.org/licenses/LICENSE-2.0
-#
-#  Unless required by applicable law or agreed to in writing, software
-#  distributed under the License is distributed on an "AS IS" BASIS,
-#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-#  See the License for the specific language governing permissions and
-#  limitations under the License.
-
-from o2ims.domain.stx_object import StxGenericModel
-from o2common.service.client.base_client import BaseClient
-# from o2common.service.unit_of_work import AbstractUnitOfWork
-from o2ims.service.watcher.resource_watcher import ResourceWatcher
-from o2ims.domain import commands
-from o2common.service.messagebus import MessageBus
-
-from o2common.helper import o2logging
-logger = o2logging.get_logger(__name__)
-
-
-class PServerCpuWatcher(ResourceWatcher):
-    def __init__(self, client: BaseClient,
-                 bus: MessageBus) -> None:
-        super().__init__(client, bus)
-
-    def _targetname(self):
-        return "pserver_cpu"
-
-    def _probe(self, parent: StxGenericModel):
-        hostid = parent.id
-        newmodels = self._client.list(hostid=hostid)
-        return [commands.UpdatePserverCpu(data=m, parentid=hostid)
-                for m in newmodels]
+# Copyright (C) 2021 Wind River Systems, Inc.
+#
+#  Licensed under the Apache License, Version 2.0 (the "License");
+#  you may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#  See the License for the specific language governing permissions and
+#  limitations under the License.
+
+from o2ims.domain.stx_object import StxGenericModel
+from o2common.service.client.base_client import BaseClient
+# from o2common.service.unit_of_work import AbstractUnitOfWork
+from o2ims.service.watcher.resource_watcher import ResourceWatcher
+from o2ims.domain import commands
+from o2common.service.messagebus import MessageBus
+
+from o2common.helper import o2logging
+logger = o2logging.get_logger(__name__)
+
+
+class PServerCpuWatcher(ResourceWatcher):
+    def __init__(self, client: BaseClient,
+                 bus: MessageBus) -> None:
+        super().__init__(client, bus)
+
+    def _targetname(self):
+        return "pserver_cpu"
+
+    def _probe(self, parent: StxGenericModel, tags):
+        # Set a tag for children resource
+        self._tags.pool = tags.pool
+        self._set_respool_client()
+
+        hostid = parent.id
+        newmodels = self._client.list(hostid=hostid)
+        return [commands.UpdatePserverCpu(data=m, parentid=hostid)
+                for m in newmodels]