Refactor vnflcm termination do_notify_delete
[vfc/gvnfm/vnflcm.git] / lcm / lcm / nf / vnf_cancel / term_vnf.py
index 8678030..2b6f2e6 100644 (file)
@@ -19,7 +19,7 @@ from threading import Thread
 
 from lcm.nf.const import VNF_STATUS
 from lcm.pub.database.models import NfInstModel, VmInstModel, NetworkInstModel, StorageInstModel, \
-    FlavourInstModel, PortInstModel, SubNetworkInstModel, VNFCInstModel, NfvoRegInfoModel
+    PortInstModel, VNFCInstModel, NfvoRegInfoModel, FlavourInstModel, SubNetworkInstModel
 from lcm.pub.exceptions import NFLCMException
 from lcm.pub.msapi.gvnfmdriver import apply_grant_to_nfvo, notify_lcm_to_nfvo
 from lcm.pub.utils.jobutil import JobUtil
@@ -40,14 +40,9 @@ class TermVnf(Thread):
         self.gracefulTerminationTimeout = ignore_case_get(self.data, "gracefulTerminationTimeout")
         self.apply_result = None
         self.notify_data = None
-        self.inst_resource = {
-            'volumn': [],
-            'network': [],
-            'subnet': [],
-            'port': [],
-            'flavor': [],
-            'vm': [],
-        }
+        self.inst_resource = {'volumn': [], 'network': [], 'subnet': [], 'port': [], 'flavor': [], 'vm': []}
+        self.resource_map = {'Storage': 'volumn', 'Network': 'network', 'SubNetwork': 'subnet', 'Port': 'port',
+                             'Flavour': 'flavor', 'Vm': 'vm'}
 
     def run(self):
         try:
@@ -85,17 +80,24 @@ class TermVnf(Thread):
 
     def grant_resource(self):
         logger.info("nf_cancel_task grant_resource begin")
-        content_args = {'vnfInstanceId': self.nf_inst_id, 'vnfDescriptorId': '',
-                        'lifecycleOperation': 'Terminate', 'jobId': self.job_id,
-                        'addResource': [], 'removeResource': [],
-                        'placementConstraint': [], 'additionalParam': {}}
+        content_args = {
+            'vnfInstanceId': self.nf_inst_id,
+            'vnfDescriptorId': '',
+            'lifecycleOperation': 'Terminate',
+            'jobId': self.job_id,
+            'addResource': [],
+            'removeResource': [],
+            'placementConstraint': [],
+            'additionalParam': {}
+        }
 
         vdus = VmInstModel.objects.filter(instid=self.nf_inst_id, is_predefined=1)
         res_index = 1
         for vdu in vdus:
-            res_def = {'type': 'VDU',
-                       'resDefId': str(res_index),
-                       'resDesId': vdu.resouceid}
+            res_def = {
+                'type': 'VDU',
+                'resDefId': str(res_index),
+                'resDesId': vdu.resouceid}
             content_args['removeResource'].append(res_def)
             res_index += 1
 
@@ -111,77 +113,22 @@ class TermVnf(Thread):
 
     def query_inst_resource(self):
         logger.info('[query_resource begin]:inst_id=%s' % self.nf_inst_id)
-        vol_list = StorageInstModel.objects.filter(instid=self.nf_inst_id)
-        for vol in vol_list:
-            vol_info = {}
-            if not vol.resouceid:
-                continue
-            vol_info["vim_id"] = vol.vimid
-            vol_info["tenant_id"] = vol.tenant
-            vol_info["res_id"] = vol.resouceid
-            vol_info["is_predefined"] = vol.is_predefined
-            self.inst_resource['volumn'].append(vol_info)
-        logger.info('[query_volumn_resource]:ret_volumns=%s' % self.inst_resource['volumn'])
-
-        network_list = NetworkInstModel.objects.filter(instid=self.nf_inst_id)
-        for network in network_list:
-            network_info = {}
-            if not network.resouceid:
-                continue
-            network_info["vim_id"] = network.vimid
-            network_info["tenant_id"] = network.tenant
-            network_info["res_id"] = network.resouceid
-            network_info["is_predefined"] = network.is_predefined
-            self.inst_resource['network'].append(network_info)
-        logger.info('[query_network_resource]:ret_networks=%s' % self.inst_resource['network'])
-
-        subnetwork_list = SubNetworkInstModel.objects.filter(instid=self.nf_inst_id)
-        for subnetwork in subnetwork_list:
-            subnetwork_info = {}
-            if not subnetwork.resouceid:
-                continue
-            subnetwork_info["vim_id"] = subnetwork.vimid
-            subnetwork_info["tenant_id"] = subnetwork.tenant
-            subnetwork_info["res_id"] = subnetwork.resouceid
-            subnetwork_info["is_predefined"] = subnetwork.is_predefined
-            self.inst_resource['subnet'].append(subnetwork_info)
-        logger.info('[query_subnetwork_resource]:ret_networks=%s' % self.inst_resource['subnet'])
-
-        port_list = PortInstModel.objects.filter(instid=self.nf_inst_id)
-        for port in port_list:
-            port_info = {}
-            if not port.resouceid:
-                continue
-            port_info["vim_id"] = port.vimid
-            port_info["tenant_id"] = port.tenant
-            port_info["res_id"] = port.resouceid
-            port_info["is_predefined"] = port.is_predefined
-            self.inst_resource['port'].append(port_info)
-        logger.info('[query_port_resource]:ret_networks=%s' % self.inst_resource['port'])
-
-        flavor_list = FlavourInstModel.objects.filter(instid=self.nf_inst_id)
-        for flavor in flavor_list:
-            flavor_info = {}
-            if not flavor.resouceid:
-                continue
-            flavor_info["vim_id"] = flavor.vimid
-            flavor_info["tenant_id"] = flavor.tenant
-            flavor_info["res_id"] = flavor.resouceid
-            flavor_info["is_predefined"] = flavor.is_predefined
-            self.inst_resource['flavor'].append(flavor_info)
-        logger.info('[query_flavor_resource]:ret_networks=%s' % self.inst_resource['flavor'])
-
-        vm_list = VmInstModel.objects.filter(instid=self.nf_inst_id)
-        for vm in vm_list:
-            vm_info = {}
-            if not vm.resouceid:
-                continue
-            vm_info["vim_id"] = vm.vimid
-            vm_info["tenant_id"] = vm.tenant
-            vm_info["res_id"] = vm.resouceid
-            vm_info["is_predefined"] = vm.is_predefined
-            self.inst_resource['vm'].append(vm_info)
-        logger.info('[query_vm_resource]:ret_vms=%s' % self.inst_resource['vm'])
+        for resource_type in self.resource_map.keys():
+            resource_table = globals().get(resource_type + 'InstModel')
+            resource_insts = resource_table.objects.filter(instid=self.nf_inst_id)
+            for resource_inst in resource_insts:
+                if not resource_inst.resouceid:
+                    continue
+                self.inst_resource[self.resource_map.get(resource_type)].append(self.get_resource(resource_inst))
+        logger.info('[query_resource]:ret_resource=%s' % self.inst_resource)
+
+    def get_resource(self, resource):
+        return {
+            "vim_id": resource.vimid,
+            "tenant_id": resource.tenant,
+            "res_id": resource.resouceid,
+            "is_predefined": resource.is_predefined
+        }
 
     def query_notify_data(self):
         logger.info('[NF terminate] send notify request to nfvo start')
@@ -202,8 +149,11 @@ class TermVnf(Thread):
         affected_vl = []
         networks = NetworkInstModel.objects.filter(instid=self.nf_inst_id)
         for network in networks:
-            network_resource = {'vimId': network.vimid, 'resourceId': network.resouceid,
-                                'resourceName': network.name, 'resourceType': 'network'}
+            network_resource = {
+                'vimId': network.vimid,
+                'resourceId': network.resouceid,
+                'resourceName': network.name,
+                'resourceType': 'network'}
             affected_vl.append(
                 {'vlInstanceId': network.networkid,
                  'vldid': network.nodeId,
@@ -227,6 +177,8 @@ class TermVnf(Thread):
                  'changeType': 'removed',
                  'storageResource': {'vimId': vs.vimid, 'resourceId': vs.resouceid,
                                      'resourceName': vs.name, 'resourceType': 'volume'}})
+        FlavourInstModel.objects.filter(instid=self.nf_inst_id)
+        SubNetworkInstModel.objects.filter(instid=self.nf_inst_id)
         self.notify_data = {
             "status": 'result',
             "vnfInstanceId": self.nf_inst_id,
@@ -250,18 +202,9 @@ class TermVnf(Thread):
 
     def do_notify_delete(self, res_type, res_id):
         logger.error('Deleting [%s] resource:resourceid [%s]' % (res_type, res_id))
-        if res_type == adaptor.RES_VM:
-            VmInstModel.objects.filter(instid=self.nf_inst_id, resouceid=res_id).delete()
-        elif res_type == adaptor.RES_FLAVOR:
-            FlavourInstModel.objects.filter(instid=self.nf_inst_id, resouceid=res_id).delete()
-        elif res_type == adaptor.RES_PORT:
-            PortInstModel.objects.filter(instid=self.nf_inst_id, resouceid=res_id).delete()
-        elif res_type == adaptor.RES_SUBNET:
-            SubNetworkInstModel.objects.filter(instid=self.nf_inst_id, resouceid=res_id).delete()
-        elif res_type == adaptor.RES_NETWORK:
-            NetworkInstModel.objects.filter(instid=self.nf_inst_id, resouceid=res_id).delete()
-        elif res_type == adaptor.RES_VOLUME:
-            StorageInstModel.objects.filter(instid=self.nf_inst_id, resouceid=res_id).delete()
+        resource_type = self.resource_map.keys()[self.resource_map.values().index(res_type)]
+        resource_table = globals().get(resource_type + 'InstModel')
+        resource_table.objects.filter(instid=self.nf_inst_id, resouceid=res_id).delete()
 
     def lcm_notify(self):
         NfInstModel.objects.filter(nfinstid=self.nf_inst_id).update(status='NOT_INSTANTIATED', lastuptime=now_time())