Merge "Deal with VnfLcmOocNotification related stuffs."

3 files changed, 214 insertions, 5 deletions

diff --git a/lcm/v2/handle_vnflcmooc_notification.py b/lcm/v2/handle_vnflcmooc_notification.py
new file mode 100644
index 00000000..76082ed6
--- /dev/null
+++ b/lcm/v2/handle_vnflcmooc_notification.py
@@ -0,0 +1,192 @@
+# Copyright 2016 ZTE Corporation.
+#
+# 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.
+
+import logging
+import traceback
+
+from rest_framework import status
+from rest_framework.response import Response
+from lcm.ns.vnfs.const import INST_TYPE
+from lcm.pub.config.config import REPORT_TO_AAI
+from lcm.pub.exceptions import NSLCMException
+from lcm.pub.database.models import VNFCInstModel, VLInstModel, NfInstModel, VmInstModel
+from lcm.pub.msapi.aai import query_vserver_aai, \
+    delete_vserver_aai
+from lcm.pub.utils.values import ignore_case_get
+from lcm.pub.msapi.extsys import split_vim_to_owner_region, get_vim_by_id
+from lcm.pub.msapi.aai import create_vserver_aai
+
+
+logger = logging.getLogger(__name__)
+
+
+class HandleVnfLcmOocNotification(object):
+    def __init__(self, vnfmid, vnfInstanceId, data):
+        logger.debug("[Notify LCM] vnfmid=%s, vnfInstanceId=%s, data=%s" % (vnfmid, vnfInstanceId, data))
+        self.vnf_instid = ''
+        self.vnfmid = vnfmid
+        self.m_vnfInstanceId = vnfInstanceId
+        self.operation = ignore_case_get(data, 'operation')
+        self.affectedVnfcs = ignore_case_get(data, 'affectedVnfcs')
+        self.affectedVls = ignore_case_get(data, 'affectedVls')
+        self.affectedCps = ignore_case_get(data, 'changedExtConnectivity')
+        self.affectedVirtualStorage = ignore_case_get(data, 'affectedVirtualStorages')
+
+    def do_biz(self):
+        try:
+            self.vnf_instid = self.get_vnfinstid(self.m_vnfInstanceId, self.vnfmid)
+            self.update_Vnfc()
+            self.update_Vl()
+            # self.update_Cp()
+            # self.update_Storage()
+            # if REPORT_TO_AAI:
+            #    self.update_network_in_aai()
+            logger.debug("notify lcm end")
+        except NSLCMException as e:
+            self.exception(e.message)
+        except Exception:
+            logger.error(traceback.format_exc())
+            self.exception('unexpected exception')
+
+    def get_vnfinstid(self, mnfinstid, vnfm_inst_id):
+        logger.debug("vnfinstid in vnfm is:%s,vnfmid is:%s", mnfinstid, vnfm_inst_id)
+        logger.debug("mnfinstid=%s, vnfm_inst_id=%s", mnfinstid, vnfm_inst_id)
+        nfinst = NfInstModel.objects.filter(mnfinstid=mnfinstid, vnfm_inst_id=vnfm_inst_id).first()
+        if nfinst:
+            return nfinst.nfinstid
+        raise NSLCMException("vnfinstid not exist")
+
+    def exception(self, error_msg):
+        logger.error('Notify Lcm failed, detail message: %s' % error_msg)
+        return Response(data={'error': '%s' % error_msg}, status=status.HTTP_409_CONFLICT)
+
+    def update_Vnfc(self):
+        for vnfc in self.affectedVnfcs:
+            vnfcInstanceId = ignore_case_get(vnfc, 'id')
+            vduId = ignore_case_get(vnfc, 'vduId')
+            changeType = ignore_case_get(vnfc, 'changeType')
+            computeResource = ignore_case_get(vnfc, 'computeResource')
+            vimId = ignore_case_get(computeResource, "vimConnectionId")
+            vmId = ignore_case_get(computeResource, 'resourceId')
+            vmName = ignore_case_get(computeResource, 'resourceId')  # replaced with resouceId temporarily
+
+            if changeType == 'added':
+                VNFCInstModel(vnfcinstanceid=vnfcInstanceId, vduid=vduId,
+                              nfinstid=self.vnf_instid, vmid=vmId).save()
+                VmInstModel(vmid=vmId, vimid=vimId, resouceid=vmId, insttype=INST_TYPE.VNF,
+                            instid=self.vnf_instid, vmname=vmName, hostid='1').save()
+                if REPORT_TO_AAI:
+                    self.create_vserver_in_aai(vimId, vmId, vmName)
+            elif changeType == 'removed':
+                if REPORT_TO_AAI:
+                    self.delete_vserver_in_aai(vimId, vmId, vmName)
+                VNFCInstModel.objects.filter(vnfcinstanceid=vnfcInstanceId).delete()
+            elif changeType == 'modified':
+                VNFCInstModel.objects.filter(vnfcinstanceid=vnfcInstanceId).update(vduid=vduId,
+                                                                                   nfinstid=self.vnf_instid,
+                                                                                   vmid=vmId)
+            else:
+                self.exception('affectedVnfc struct error: changeType not in {added,removed,modified}')
+        logger.debug("Success to update all vserver to aai.")
+
+    def update_Vl(self):
+        for vl in self.affectedVls:
+            vlInstanceId = ignore_case_get(vl, 'id')
+            vldid = ignore_case_get(vl, 'virtualLinkDescId')
+            changeType = ignore_case_get(vl, 'changeType')
+            networkResource = ignore_case_get(vl, 'networkResource')
+            resourceType = ignore_case_get(networkResource, 'vimLevelResourceType')
+            resourceId = ignore_case_get(networkResource, 'resourceId')
+            resourceName = ignore_case_get(networkResource, 'resourceId')  # replaced with resouceId temporarily
+
+            if resourceType != 'network':
+                self.exception('affectedVl struct error: resourceType not euqal network')
+
+            ownerId = self.get_vnfinstid(self.m_vnfInstanceId, self.vnfmid)
+
+            if changeType == 'added':
+                VLInstModel(vlinstanceid=vlInstanceId, vldid=vldid, vlinstancename=resourceName, ownertype=0,
+                            ownerid=ownerId, relatednetworkid=resourceId, vltype=0).save()
+            elif changeType == 'removed':
+                VLInstModel.objects.filter(vlinstanceid=vlInstanceId).delete()
+            elif changeType == 'modified':
+                VLInstModel.objects.filter(vlinstanceid=vlInstanceId)\
+                    .update(vldid=vldid, vlinstancename=resourceName, ownertype=0, ownerid=ownerId,
+                            relatednetworkid=resourceId, vltype=0)
+            else:
+                self.exception('affectedVl struct error: changeType not in {added,removed,modified}')
+
+    def create_vserver_in_aai(self, vim_id, vserver_id, vserver_name):
+        logger.debug("NotifyLcm::create_vserver_in_aai::report vserver instance to aai.")
+        try:
+            cloud_owner, cloud_region_id = split_vim_to_owner_region(vim_id)
+
+            # query vim_info from aai
+            vim_info = get_vim_by_id(vim_id)
+            tenant_id = vim_info["tenantId"]
+            data = {
+                "vserver-id": vserver_id,
+                "vserver-name": vserver_name,
+                "prov-status": "ACTIVE",
+                "vserver-selflink": "",
+                "in-maint": True,
+                "is-closed-loop-disabled": False,
+                "relationship-list": {
+                    "relationship": [
+                        {
+                            "related-to": "generic-vnf",
+                            "relationship-data": [
+                                {
+                                    "relationship-key": "generic-vnf.vnf-id",
+                                    "relationship-value": self.vnf_instid
+                                }
+                            ]
+                        }
+                    ]
+                }
+            }
+
+            # create vserver instance in aai
+            resp_data, resp_status = create_vserver_aai(cloud_owner, cloud_region_id, tenant_id, vserver_id, data)
+            logger.debug("Success to create vserver[%s] to aai, vnf instance=[%s], resp_status: [%s]."
+                         % (vserver_id, self.vnf_instid, resp_status))
+        except NSLCMException as e:
+            logger.debug("Fail to create vserver to aai, vnf instance=[%s], detail message: %s"
+                         % (self.vnf_instid, e.message))
+        except:
+            logger.error(traceback.format_exc())
+
+    def delete_vserver_in_aai(self, vim_id, vserver_id, vserver_name):
+        logger.debug("delete_vserver_in_aai start![%s]", vserver_name)
+        try:
+            cloud_owner, cloud_region_id = split_vim_to_owner_region(vim_id)
+            # query vim_info from aai, get tenant
+            vim_info = get_vim_by_id(vim_id)
+            tenant_id = vim_info["tenantId"]
+
+            # query vserver instance in aai, get resource_version
+            vserver_info = query_vserver_aai(cloud_owner, cloud_region_id, tenant_id, vserver_id)
+            resource_version = vserver_info["resource-version"]
+
+            # delete vserver instance from aai
+            resp_data, resp_status = delete_vserver_aai(cloud_owner, cloud_region_id,
+                                                        tenant_id, vserver_id, resource_version)
+            logger.debug(
+                "Success to delete vserver instance[%s] from aai, resp_status: [%s]." %
+                (vserver_id, resp_status))
+            logger.debug("delete_vserver_in_aai end!")
+        except NSLCMException as e:
+            logger.debug("Fail to delete vserver from aai, detail message: %s" % e.message)
+        except:
+            logger.error(traceback.format_exc())
diff --git a/lcm/v2/tests.py b/lcm/v2/tests.py
index 406edadd..55fa2f1d 100644
--- a/lcm/v2/tests.py
+++ b/lcm/v2/tests.py
@@ -112,12 +112,19 @@ class VnfGrantViewTest(unittest.TestCase):
             "isAutomaticInvocation": True,
             "vnfLcmOpOccId": "string",
             "affectedVnfcs": [{
-                "vnfcInstanceId": "string",
+                "id": "string",
                 "vduId": "string",
                 "changeType": "added",
-                "vimId": "string",
-                "vmId": "string",
-                "vmName": "string"
+                "computeResource": {
+                    "vimConnectionId": "string",
+                    "resourceProviderId": "string",
+                    "resouceId": "string",
+                    "vimLevelResourceType": "string"
+                },
+                "metadata": {},
+                "affectedVnfcCpIds": [],
+                "addedStorageResourceIds": [],
+                "removedStorageResourceIds": [],
             }],
             "affectedVirtualLinks": [{
                 "vlInstanceId": "string",
diff --git a/lcm/v2/views.py b/lcm/v2/views.py
index e216022d..a3ce25a7 100644
--- a/lcm/v2/views.py
+++ b/lcm/v2/views.py
@@ -23,6 +23,7 @@ from lcm.v2.serializers import GrantRequestSerializer
 from lcm.v2.serializers import GrantSerializer
 from lcm.v2.serializers import VnfLcmOperationOccurrenceNotificationSerializer
 from lcm.v2.grant_vnf import GrantVnf
+from lcm.v2.handle_vnflcmooc_notification import HandleVnfLcmOocNotification
 
 logger = logging.getLogger(__name__)
 
@@ -70,7 +71,16 @@ class VnfNotifyView(APIView):
     def post(self, request, vnfmId, vnfInstanceId):
         logger.debug("VnfNotifyView post: %s" % request.data)
         logger.debug("vnfmId: %s vnfInstanceId: %s", vnfmId, vnfInstanceId)
-        return Response(data={}, status=status.HTTP_204_NO_CONTENT)
+        try:
+            vnfLcmOocNotificationSerializer = VnfLcmOperationOccurrenceNotificationSerializer(data=request.data)
+            if not vnfLcmOocNotificationSerializer.is_valid():
+                raise Exception(vnfLcmOocNotificationSerializer.errors)
+            HandleVnfLcmOocNotification(vnfmId, vnfInstanceId, vnfLcmOocNotificationSerializer.data).do_biz()
+            return Response(data={}, status=status.HTTP_204_NO_CONTENT)
+        except Exception as e:
+            logger.error(traceback.format_exc())
+            logger.error("Exception in VnfLcmOoc Notification: %s", e.message)
+            return Response(data={'error': e.message}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
 
     @swagger_auto_schema(
         responses={