Optimized code and testcase of vnflcm
[vfc/gvnfm/vnflcm.git] / lcm / lcm / nf / vnfs / vnf_cancel / term_vnf.py
1 # Copyright 2017 ZTE Corporation.
2 #
3 # Licensed under the Apache License, Version 2.0 (the "License");
4 # you may not use this file except in compliance with the License.
5 # You may obtain a copy of the License at
6 #
7 #         http://www.apache.org/licenses/LICENSE-2.0
8 #
9 # Unless required by applicable law or agreed to in writing, software
10 # distributed under the License is distributed on an "AS IS" BASIS,
11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 # See the License for the specific language governing permissions and
13 # limitations under the License.
14 import logging
15 import traceback
16 from threading import Thread
17
18 from lcm.nf.vnfs.const import VNF_STATUS
19 from lcm.pub.database.models import NfInstModel, VmInstModel, NetworkInstModel, StorageInstModel, \
20     FlavourInstModel, PortInstModel, SubNetworkInstModel, VNFCInstModel
21 from lcm.pub.exceptions import NFLCMException
22 from lcm.pub.msapi.nfvolcm import apply_grant_to_nfvo, notify_lcm_to_nfvo
23 from lcm.pub.utils.jobutil import JobUtil
24 from lcm.pub.utils.timeutil import now_time
25 from lcm.pub.utils.values import ignore_case_get
26 from lcm.pub.vimapi import adaptor
27
28 logger = logging.getLogger(__name__)
29
30
31 class TermVnf(Thread):
32     def __init__(self, data, nf_inst_id, job_id):
33         super(TermVnf, self).__init__()
34         self.data = data
35         self.nf_inst_id = nf_inst_id
36         self.job_id = job_id
37         self.terminationType = ignore_case_get(self.data, "terminationType")
38         self.gracefulTerminationTimeout = ignore_case_get(self.data, "gracefulTerminationTimeout")
39         self.apply_result = None
40         self.notify_data = None
41         self.inst_resource = {'volumn': [],
42                               'network': [],
43                               'subnet': [],
44                               'port': [],
45                               'flavor': [],
46                               'vm': [],
47                               }
48
49     def run(self):
50         try:
51             self.term_pre()
52             self.grant_resource()
53             self.query_inst_resource()
54             self.query_notify_data()
55             self.delete_resource()
56             self.lcm_notify()
57             JobUtil.add_job_status(self.job_id, 100, "Terminate Vnf success.")
58         except NFLCMException as e:
59             self.vnf_term_failed_handle(e.message)
60         except:
61             self.vnf_term_failed_handle(traceback.format_exc())
62
63     def term_pre(self):
64         vnf_insts = NfInstModel.objects.filter(nfinstid=self.nf_inst_id)
65         if not vnf_insts.exists():
66             raise NFLCMException('VnfInst(%s) does not exist' % self.nf_inst_id)
67         sel_vnf = vnf_insts[0]
68         if sel_vnf.status != 'VNF_INSTANTIATED':
69             raise NFLCMException("Don't allow to terminate vnf(status:[%s])" % sel_vnf.status)
70         if self.terminationType == 'GRACEFUL' and not self.gracefulTerminationTimeout:
71             raise NFLCMException("Graceful termination must set timeout")
72
73         NfInstModel.objects.filter(nfinstid=self.nf_inst_id).update(status=VNF_STATUS.TERMINATING)
74         JobUtil.add_job_status(self.job_id, 10, 'Nf terminating pre-check finish')
75         logger.info("Nf terminating pre-check finish")
76
77     def grant_resource(self):
78         logger.info("nf_cancel_task grant_resource begin")
79         content_args = {'vnfInstanceId': self.nf_inst_id, 'vnfDescriptorId': '',
80                         'lifecycleOperation': 'Instantiate', 'jobId': self.job_id,
81                         'addResource': [], 'removeResource': [],
82                         'placementConstraint': [], 'additionalParam': {}}
83
84         vdus = VmInstModel.objects.filter(instid=self.nf_inst_id, is_predefined=1)
85         res_index = 1
86         for vdu in vdus:
87             res_def = {'type': 'VDU',
88                        'resDefId': str(res_index),
89                        'resDesId': vdu.resouceid}
90             content_args['removeResource'].append(res_def)
91             res_index += 1
92
93         logger.info('content_args=%s' % content_args)
94         self.apply_result = apply_grant_to_nfvo(content_args)
95         logger.info("nf_cancel_task grant_resource end")
96         JobUtil.add_job_status(self.job_id, 20, 'Nf terminating grant_resource finish')
97
98     def query_inst_resource(self):
99         logger.info('[query_resource begin]:inst_id=%s' % self.nf_inst_id)
100         vol_list = StorageInstModel.objects.filter(instid=self.nf_inst_id, is_predefined=1)
101         for vol in vol_list:
102             vol_info = {}
103             if not vol.resouceid:
104                 continue
105             vol_info["vim_id"] = vol.vimid
106             vol_info["tenant_id"] = vol.tenant
107             vol_info["res_id"] = vol.resouceid
108             self.inst_resource['volumn'].append(vol_info)
109         logger.info('[query_volumn_resource]:ret_volumns=%s' % self.inst_resource['volumn'])
110
111         network_list = NetworkInstModel.objects.filter(instid=self.nf_inst_id, is_predefined=1)
112         for network in network_list:
113             network_info = {}
114             if not network.resouceid:
115                 continue
116             network_info["vim_id"] = network.vimid
117             network_info["tenant_id"] = network.tenant
118             network_info["res_id"] = network.resouceid
119             self.inst_resource['network'].append(network_info)
120         logger.info('[query_network_resource]:ret_networks=%s' % self.inst_resource['network'])
121
122         subnetwork_list = SubNetworkInstModel.objects.filter(instid=self.nf_inst_id, is_predefined=1)
123         for subnetwork in subnetwork_list:
124             subnetwork_info = {}
125             if not subnetwork.resouceid:
126                 continue
127             subnetwork_info["vim_id"] = subnetwork.vimid
128             subnetwork_info["tenant_id"] = subnetwork.tenant
129             subnetwork_info["res_id"] = subnetwork.resouceid
130             self.inst_resource['subnet'].append(subnetwork_info)
131         logger.info('[query_subnetwork_resource]:ret_networks=%s' % self.inst_resource['subnet'])
132
133         port_list = PortInstModel.objects.filter(instid=self.nf_inst_id, is_predefined=1)
134         for port in port_list:
135             port_info = {}
136             if not port.resouceid:
137                 continue
138             port_info["vim_id"] = port.vimid
139             port_info["tenant_id"] = port.tenant
140             port_info["res_id"] = port.resouceid
141             self.inst_resource['port'].append(port_info)
142         logger.info('[query_port_resource]:ret_networks=%s' % self.inst_resource['port'])
143
144         flavor_list = FlavourInstModel.objects.filter(instid=self.nf_inst_id, is_predefined=1)
145         for flavor in flavor_list:
146             flavor_info = {}
147             if not flavor.resouceid:
148                 continue
149             flavor_info["vim_id"] = flavor.vimid
150             flavor_info["tenant_id"] = flavor.tenant
151             flavor_info["res_id"] = flavor.resouceid
152             self.inst_resource['flavor'].append(flavor_info)
153         logger.info('[query_flavor_resource]:ret_networks=%s' % self.inst_resource['flavor'])
154
155         vm_list = VmInstModel.objects.filter(instid=self.nf_inst_id, is_predefined=1)
156         for vm in vm_list:
157             vm_info = {}
158             if not vm.resouceid:
159                 continue
160             vm_info["vim_id"] = vm.vimid
161             vm_info["tenant_id"] = vm.tenant
162             vm_info["res_id"] = vm.resouceid
163             self.inst_resource['vm'].append(vm_info)
164         logger.info('[query_vm_resource]:ret_vms=%s' % self.inst_resource['vm'])
165
166     def query_notify_data(self):
167         logger.info('[NF instantiation] send notify request to nfvo start')
168         affected_vnfc = []
169         vnfcs = VNFCInstModel.objects.filter(instid=self.nf_inst_id)
170         for vnfc in vnfcs:
171             vm_resource = {}
172             if vnfc.vmid:
173                 vm = VmInstModel.objects.filter(vmid=vnfc.vmid)
174                 if vm:
175                     vm_resource = {'vimId': vm[0].vimid, 'resourceId': vm[0].resouceid,
176                                    'resourceName': vm[0].vmname, 'resourceType': 'vm'}
177             affected_vnfc.append(
178                 {'vnfcInstanceId': vnfc.vnfcinstanceid,
179                  'vduId': vnfc.vduid,
180                  'changeType': 'removed',
181                  'computeResource': vm_resource})
182         affected_vl = []
183         networks = NetworkInstModel.objects.filter(instid=self.nf_inst_id)
184         for network in networks:
185             network_resource = {'vimId': network.vimid, 'resourceId': network.resouceid,
186                                 'resourceName': network.name, 'resourceType': 'network'}
187             affected_vl.append(
188                 {'vlInstanceId': network.networkid,
189                  'vldid': network.nodeId,
190                  'changeType': 'removed',
191                  'networkResource': network_resource})
192         affected_cp = []
193         ports = PortInstModel.objects.filter(instid=self.nf_inst_id)
194         for port in ports:
195             affected_cp.append(
196                 {'vsInstanceId': port.portid,
197                  'cpdid': port.nodeId,
198                  'changeType': 'removed',
199                  'storageResource': {'vimId': port.vimid, 'resourceId': port.resouceid,
200                                      'resourceName': port.name, 'resourceType': 'port'}})
201         affected_vs = []
202         vss = StorageInstModel.objects.filter(instid=self.nf_inst_id)
203         for vs in vss:
204             affected_vs.append(
205                 {'vsInstanceId': vs.storageid,
206                  'vsdId': vs.nodeId,
207                  'changeType': 'removed',
208                  'storageResource': {'vimId': vs.vimid, 'resourceId': vs.resouceid,
209                                      'resourceName': vs.name, 'resourceType': 'volume'}})
210         self.notify_data = {
211             "status": 'result',
212             "nfInstanceId": self.nf_inst_id,
213             "operation": 'Terminal',
214             "jobId": self.job_id,
215             'affectedVnfc': affected_vnfc,
216             'affectedVirtualLink': affected_vl,
217             'affectedVirtualStorage': affected_vs,
218             'affectedCp': affected_cp
219             }
220         logger.info('content_args=%s' % self.notify_data)
221
222     def delete_resource(self):
223         logger.error('rollback resource begin')
224         adaptor.delete_vim_res(self.inst_resource, self.do_notify_delete)
225         logger.error('rollback resource complete')
226
227     def do_notify_delete(self, res_type, res_id):
228         logger.error('Deleting [%s] resource:resourceid [%s]' % (res_type, res_id))
229         if res_type == adaptor.RES_VM:
230             VmInstModel.objects.filter(instid=self.nf_inst_id, resouceid=res_id).delete()
231         elif res_type == adaptor.RES_FLAVOR:
232             FlavourInstModel.objects.filter(instid=self.nf_inst_id, resouceid=res_id).delete()
233         elif res_type == adaptor.RES_PORT:
234             PortInstModel.objects.filter(instid=self.nf_inst_id, resouceid=res_id).delete()
235         elif res_type == adaptor.RES_SUBNET:
236             SubNetworkInstModel.objects.filter(instid=self.nf_inst_id, resouceid=res_id).delete()
237         elif res_type == adaptor.RES_NETWORK:
238             NetworkInstModel.objects.filter(instid=self.nf_inst_id, resouceid=res_id).delete()
239         elif res_type == adaptor.RES_VOLUME:
240             StorageInstModel.objects.filter(instid=self.nf_inst_id, resouceid=res_id).delete()
241
242     def lcm_notify(self):
243         NfInstModel.objects.filter(nfinstid=self.nf_inst_id).update(status='NOT_INSTANTIATED', lastuptime=now_time())
244         logger.info('[NF termination] send notify request to nfvo end')
245         resp = notify_lcm_to_nfvo(self.notify_data)
246         logger.info('[NF termination] get lcm response %s' % resp)
247         logger.info('[NF termination] send notify request to nfvo end')
248
249     def vnf_term_failed_handle(self, error_msg):
250         logger.error('VNF termination failed, detail message: %s' % error_msg)
251         NfInstModel.objects.filter(nfinstid=self.nf_inst_id).update(status='failed', lastuptime=now_time())
252         JobUtil.add_job_status(self.job_id, 255, error_msg)