X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=o2ims%2Fservice%2Fauditor%2Fpserver_mem_handler.py;fp=o2ims%2Fservice%2Fauditor%2Fpserver_mem_handler.py;h=42bccb745910d52abd7b64f02321dbfadc9ec3a5;hb=f73c8e3b01b8f5b7438ba544870e06d8f30cdea0;hp=0000000000000000000000000000000000000000;hpb=71f8155b3b60befb21aabaffe0b0cf4d940eda61;p=pti%2Fo2.git

diff --git a/o2ims/service/auditor/pserver_mem_handler.py b/o2ims/service/auditor/pserver_mem_handler.py
new file mode 100644
index 0000000..42bccb7
--- /dev/null
+++ b/o2ims/service/auditor/pserver_mem_handler.py
@@ -0,0 +1,116 @@
+# 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.
+
+# pylint: disable=unused-argument
+from __future__ import annotations
+import uuid
+# import json
+
+from o2ims.domain import commands
+from o2ims.domain.stx_object import StxGenericModel
+from o2common.service.unit_of_work import AbstractUnitOfWork
+from o2ims.domain.resource_type import MismatchedModel
+from o2ims.domain.ocloud import Resource, ResourceType
+
+from o2common.helper import o2logging
+logger = o2logging.get_logger(__name__)
+
+
+class InvalidResourceType(Exception):
+    pass
+
+
+def update_pserver_mem(
+    cmd: commands.UpdatePserverMem,
+    uow: AbstractUnitOfWork
+):
+    stxobj = cmd.data
+    with uow:
+        p_resource = uow.resources.get(cmd.parentid)
+        resourcepool = uow.resource_pools.get(p_resource.resourcePoolId)
+
+        res = uow.session.execute(
+            '''
+            SELECT "resourceTypeId", "oCloudId", "name"
+            FROM resourcetype
+            WHERE "resourceTypeEnum" = :resource_type_enum
+            ''',
+            dict(resource_type_enum=stxobj.type.name)
+        )
+        first = res.first()
+        if first is None:
+            resourcetype_id = str(uuid.uuid4())
+            uow.resource_types.add(ResourceType(
+                resourcetype_id,
+                'pserver_mem', stxobj.type,
+                resourcepool.oCloudId))
+        else:
+            resourcetype_id = first['resourceTypeId']
+
+        resource = uow.resources.get(stxobj.id)
+        if not resource:
+            logger.info("add the memory of pserver:" + stxobj.name
+                        + " update_at: " + str(stxobj.updatetime)
+                        + " id: " + str(stxobj.id)
+                        + " hash: " + str(stxobj.hash))
+            localmodel = create_by(stxobj, p_resource, resourcetype_id)
+            uow.resources.add(localmodel)
+
+            logger.info("Add the memory of pserver: " + stxobj.id
+                        + ", name: " + stxobj.name)
+        else:
+            localmodel = resource
+            if is_outdated(localmodel, stxobj):
+                logger.info("update memory of pserver:" + stxobj.name
+                            + " update_at: " + str(stxobj.updatetime)
+                            + " id: " + str(stxobj.id)
+                            + " hash: " + str(stxobj.hash))
+                update_by(localmodel, stxobj, p_resource)
+                uow.resources.update(localmodel)
+
+            logger.info("Update the memory of pserver: " + stxobj.id
+                        + ", name: " + stxobj.name)
+        uow.commit()
+
+
+def is_outdated(resource: Resource, stxobj: StxGenericModel):
+    return True if resource.hash != stxobj.hash else False
+
+
+def create_by(stxobj: StxGenericModel, parent: Resource, resourcetype_id: str)\
+        -> Resource:
+    # content = json.loads(stxobj.content)
+    resourcetype_id = resourcetype_id
+    resourcepool_id = parent.resourcePoolId
+    parent_id = parent.resourceId
+    gAssetId = ''  # TODO: global ID
+    description = "A memory resource of the physical server"
+    resource = Resource(stxobj.id, resourcetype_id, resourcepool_id,
+                        parent_id, gAssetId, stxobj.content, description)
+    resource.createtime = stxobj.createtime
+    resource.updatetime = stxobj.updatetime
+    resource.hash = stxobj.hash
+
+    return resource
+
+
+def update_by(target: Resource, stxobj: StxGenericModel,
+              parentid: str) -> None:
+    if target.resourceId != stxobj.id:
+        raise MismatchedModel("Mismatched Id")
+    target.createtime = stxobj.createtime
+    target.updatetime = stxobj.updatetime
+    target.hash = stxobj.hash
+    target.version_number = target.version_number + 1
+    target.events = []