Fix vfc-vnflcm unittests
[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
15 import json
16 import logging
17 import traceback
18 from threading import Thread
19
20 from lcm.nf.vnfs.const import VNF_STATUS
21 from lcm.pub.database.models import NfInstModel, VmInstModel, NetworkInstModel, StorageInstModel, \
22     FlavourInstModel, PortInstModel, SubNetworkInstModel, VNFCInstModel, NfvoRegInfoModel
23 from lcm.pub.exceptions import NFLCMException
24 from lcm.pub.msapi.gvnfmdriver import apply_grant_to_nfvo, notify_lcm_to_nfvo
25 from lcm.pub.utils.jobutil import JobUtil
26 from lcm.pub.utils.timeutil import now_time
27 from lcm.pub.utils.values import ignore_case_get
28 from lcm.pub.vimapi import adaptor
29
30 logger = logging.getLogger(__name__)
31
32
33 class TermVnf(Thread):
34     def __init__(self, data, nf_inst_id, job_id):
35         super(TermVnf, self).__init__()
36         self.data = data
37         self.nf_inst_id = nf_inst_id
38         self.job_id = job_id
39         self.terminationType = ignore_case_get(self.data, "terminationType")
40         self.gracefulTerminationTimeout = ignore_case_get(self.data, "gracefulTerminationTimeout")
41         self.apply_result = None
42         self.notify_data = None
43         self.inst_resource = {
44             'volumn': [],
45             'network': [],
46             'subnet': [],
47             'port': [],
48             'flavor': [],
49             'vm': [],
50         }
51
52     def run(self):
53         try:
54             if self.term_pre():
55                 self.grant_resource()
56                 self.query_inst_resource()
57                 self.query_notify_data()
58                 self.delete_resource()
59                 self.lcm_notify()
60             JobUtil.add_job_status(self.job_id, 100, "Terminate Vnf success.")
61         except NFLCMException as e:
62             self.vnf_term_failed_handle(e.message)
63         except:
64             self.vnf_term_failed_handle(traceback.format_exc())
65
66     def term_pre(self):
67         vnf_insts = NfInstModel.objects.filter(nfinstid=self.nf_inst_id)
68         if not vnf_insts.exists():
69             logger.warn('VnfInst(%s) does not exist' % self.nf_inst_id)
70             return False
71             # raise NFLCMException('VnfInst(%s) does not exist' % self.nf_inst_id)
72         # sel_vnf = vnf_insts[0]
73         # if sel_vnf.status != 'VNF_INSTANTIATED':
74         #    raise NFLCMException("Don't allow to terminate vnf(status:[%s])" % sel_vnf.status)
75         if self.terminationType == 'GRACEFUL' and not self.gracefulTerminationTimeout:
76             logger.warn("Set Graceful default termination timeout = 60")
77             self.gracefulTerminationTimeout = 60
78             # raise NFLCMException("Graceful termination must set timeout")
79
80         NfInstModel.objects.filter(nfinstid=self.nf_inst_id).update(status=VNF_STATUS.TERMINATING)
81         JobUtil.add_job_status(self.job_id, 10, 'Nf terminating pre-check finish')
82         logger.info("Nf terminating pre-check finish")
83         return True
84
85     def grant_resource(self):
86         logger.info("nf_cancel_task grant_resource begin")
87         content_args = {'vnfInstanceId': self.nf_inst_id, 'vnfDescriptorId': '',
88                         'lifecycleOperation': 'Terminate', 'jobId': self.job_id,
89                         'addResource': [], 'removeResource': [],
90                         'placementConstraint': [], 'additionalParam': {}}
91
92         vdus = VmInstModel.objects.filter(instid=self.nf_inst_id, is_predefined=1)
93         res_index = 1
94         for vdu in vdus:
95             res_def = {'type': 'VDU',
96                        'resDefId': str(res_index),
97                        'resDesId': vdu.resouceid}
98             content_args['removeResource'].append(res_def)
99             res_index += 1
100
101         vnfmInfo = NfvoRegInfoModel.objects.filter(nfvoid=self.nf_inst_id)
102         if len(vnfmInfo) == 0:
103             raise NFLCMException('nf_inst_id(%s) does not exist in NfvoRegInfoModel' % self.nf_inst_id)
104         content_args['additionalParam']['vnfmid'] = vnfmInfo[0].vnfminstid
105         content_args['additionalParam']['vimid'] = vnfmInfo[0].apiurl
106         logger.info('content_args=%s' % content_args)
107         self.apply_result = apply_grant_to_nfvo(json.dumps(content_args))
108         logger.info("nf_cancel_task grant_resource end")
109         JobUtil.add_job_status(self.job_id, 20, 'Nf terminating grant_resource finish')
110
111     def query_inst_resource(self):
112         logger.info('[query_resource begin]:inst_id=%s' % self.nf_inst_id)
113         vol_list = StorageInstModel.objects.filter(instid=self.nf_inst_id)
114         for vol in vol_list:
115             vol_info = {}
116             if not vol.resouceid:
117                 continue
118             vol_info["vim_id"] = vol.vimid
119             vol_info["tenant_id"] = vol.tenant
120             vol_info["res_id"] = vol.resouceid
121             vol_info["is_predefined"] = vol.is_predefined
122             self.inst_resource['volumn'].append(vol_info)
123         logger.info('[query_volumn_resource]:ret_volumns=%s' % self.inst_resource['volumn'])
124
125         network_list = NetworkInstModel.objects.filter(instid=self.nf_inst_id)
126         for network in network_list:
127             network_info = {}
128             if not network.resouceid:
129                 continue
130             network_info["vim_id"] = network.vimid
131             network_info["tenant_id"] = network.tenant
132             network_info["res_id"] = network.resouceid
133             network_info["is_predefined"] = network.is_predefined
134             self.inst_resource['network'].append(network_info)
135         logger.info('[query_network_resource]:ret_networks=%s' % self.inst_resource['network'])
136
137         subnetwork_list = SubNetworkInstModel.objects.filter(instid=self.nf_inst_id)
138         for subnetwork in subnetwork_list:
139             subnetwork_info = {}
140             if not subnetwork.resouceid:
141                 continue
142             subnetwork_info["vim_id"] = subnetwork.vimid
143             subnetwork_info["tenant_id"] = subnetwork.tenant
144             subnetwork_info["res_id"] = subnetwork.resouceid
145             subnetwork_info["is_predefined"] = subnetwork.is_predefined
146             self.inst_resource['subnet'].append(subnetwork_info)
147         logger.info('[query_subnetwork_resource]:ret_networks=%s' % self.inst_resource['subnet'])
148
149         port_list = PortInstModel.objects.filter(instid=self.nf_inst_id)
150         for port in port_list:
151             port_info = {}
152             if not port.resouceid:
153                 continue
154             port_info["vim_id"] = port.vimid
155             port_info["tenant_id"] = port.tenant
156             port_info["res_id"] = port.resouceid
157             port_info["is_predefined"] = port.is_predefined
158             self.inst_resource['port'].append(port_info)
159         logger.info('[query_port_resource]:ret_networks=%s' % self.inst_resource['port'])
160
161         flavor_list = FlavourInstModel.objects.filter(instid=self.nf_inst_id)
162         for flavor in flavor_list:
163             flavor_info = {}
164             if not flavor.resouceid:
165                 continue
166             flavor_info["vim_id"] = flavor.vimid
167             flavor_info["tenant_id"] = flavor.tenant
168             flavor_info["res_id"] = flavor.resouceid
169             flavor_info["is_predefined"] = flavor.is_predefined
170             self.inst_resource['flavor'].append(flavor_info)
171         logger.info('[query_flavor_resource]:ret_networks=%s' % self.inst_resource['flavor'])
172
173         vm_list = VmInstModel.objects.filter(instid=self.nf_inst_id)
174         for vm in vm_list:
175             vm_info = {}
176             if not vm.resouceid:
177                 continue
178             vm_info["vim_id"] = vm.vimid
179             vm_info["tenant_id"] = vm.tenant
180             vm_info["res_id"] = vm.resouceid
181             vm_info["is_predefined"] = vm.is_predefined
182             self.inst_resource['vm'].append(vm_info)
183         logger.info('[query_vm_resource]:ret_vms=%s' % self.inst_resource['vm'])
184
185     def query_notify_data(self):
186         logger.info('[NF terminate] send notify request to nfvo start')
187         affected_vnfc = []
188         vnfcs = VNFCInstModel.objects.filter(instid=self.nf_inst_id)
189         for vnfc in vnfcs:
190             vm_resource = {}
191             if vnfc.vmid:
192                 vm = VmInstModel.objects.filter(vmid=vnfc.vmid)
193                 if vm:
194                     vm_resource = {'vimId': vm[0].vimid, 'resourceId': vm[0].resouceid,
195                                    'resourceName': vm[0].vmname, 'resourceType': 'vm'}
196             affected_vnfc.append(
197                 {'vnfcInstanceId': vnfc.vnfcinstanceid,
198                  'vduId': vnfc.vduid,
199                  'changeType': 'removed',
200                  'computeResource': vm_resource})
201         affected_vl = []
202         networks = NetworkInstModel.objects.filter(instid=self.nf_inst_id)
203         for network in networks:
204             network_resource = {'vimId': network.vimid, 'resourceId': network.resouceid,
205                                 'resourceName': network.name, 'resourceType': 'network'}
206             affected_vl.append(
207                 {'vlInstanceId': network.networkid,
208                  'vldid': network.nodeId,
209                  'changeType': 'removed',
210                  'networkResource': network_resource})
211         affected_cp = []
212         ports = PortInstModel.objects.filter(instid=self.nf_inst_id)
213         for port in ports:
214             affected_cp.append(
215                 {'vsInstanceId': port.portid,
216                  'cpdid': port.nodeId,
217                  'changeType': 'removed',
218                  'storageResource': {'vimId': port.vimid, 'resourceId': port.resouceid,
219                                      'resourceName': port.name, 'resourceType': 'port'}})
220         affected_vs = []
221         vss = StorageInstModel.objects.filter(instid=self.nf_inst_id)
222         for vs in vss:
223             affected_vs.append(
224                 {'vsInstanceId': vs.storageid,
225                  'vsdId': vs.nodeId,
226                  'changeType': 'removed',
227                  'storageResource': {'vimId': vs.vimid, 'resourceId': vs.resouceid,
228                                      'resourceName': vs.name, 'resourceType': 'volume'}})
229         self.notify_data = {
230             "status": 'result',
231             "vnfInstanceId": self.nf_inst_id,
232             "operation": 'Terminal',
233             "jobId": self.job_id,
234             'affectedVnfc': affected_vnfc,
235             'affectedVirtualLink': affected_vl,
236             'affectedVirtualStorage': affected_vs,
237             'affectedCp': affected_cp}
238
239         vnfmInfo = NfvoRegInfoModel.objects.filter(nfvoid=self.nf_inst_id)
240         if len(vnfmInfo) == 0:
241             raise NFLCMException('nf_inst_id(%s) does not exist in NfvoRegInfoModel' % self.nf_inst_id)
242         self.notify_data['VNFMID'] = vnfmInfo[0].vnfminstid
243         logger.info('content_args=%s' % self.notify_data)
244
245     def delete_resource(self):
246         logger.info('rollback resource begin')
247         adaptor.delete_vim_res(self.inst_resource, self.do_notify_delete)
248         logger.info('rollback resource complete')
249
250     def do_notify_delete(self, res_type, res_id):
251         logger.error('Deleting [%s] resource:resourceid [%s]' % (res_type, res_id))
252         if res_type == adaptor.RES_VM:
253             VmInstModel.objects.filter(instid=self.nf_inst_id, resouceid=res_id).delete()
254         elif res_type == adaptor.RES_FLAVOR:
255             FlavourInstModel.objects.filter(instid=self.nf_inst_id, resouceid=res_id).delete()
256         elif res_type == adaptor.RES_PORT:
257             PortInstModel.objects.filter(instid=self.nf_inst_id, resouceid=res_id).delete()
258         elif res_type == adaptor.RES_SUBNET:
259             SubNetworkInstModel.objects.filter(instid=self.nf_inst_id, resouceid=res_id).delete()
260         elif res_type == adaptor.RES_NETWORK:
261             NetworkInstModel.objects.filter(instid=self.nf_inst_id, resouceid=res_id).delete()
262         elif res_type == adaptor.RES_VOLUME:
263             StorageInstModel.objects.filter(instid=self.nf_inst_id, resouceid=res_id).delete()
264
265     def lcm_notify(self):
266         NfInstModel.objects.filter(nfinstid=self.nf_inst_id).update(status='NOT_INSTANTIATED', lastuptime=now_time())
267         logger.info('[NF termination] send notify request to nfvo end')
268         resp = notify_lcm_to_nfvo(json.dumps(self.notify_data))
269         logger.info('[NF termination] get lcm response %s' % resp)
270         logger.info('[NF termination] send notify request to nfvo end')
271
272     def vnf_term_failed_handle(self, error_msg):
273         logger.error('VNF termination failed, detail message: %s' % error_msg)
274         NfInstModel.objects.filter(nfinstid=self.nf_inst_id).update(status='failed', lastuptime=now_time())
275         JobUtil.add_job_status(self.job_id, 255, error_msg)