Modify code and testcase of jobs status
[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 JobStatusModel, 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': [],  # [{"vim_id": ignore_case_get(ret, "vim_id")},{}]
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         # query_volumn_resource
101         vol_list = StorageInstModel.objects.filter(instid=self.nf_inst_id, is_predefined=1)
102         for vol in vol_list:
103             vol_info = {}
104             if not vol.resouceid:
105                 continue
106             vol_info["vim_id"] = vol.vimid
107             vol_info["tenant_id"] = vol.tenant
108             vol_info["res_id"] = vol.resouceid
109             self.inst_resource['volumn'].append(vol_info)
110         logger.info('[query_volumn_resource]:ret_volumns=%s' % self.inst_resource['volumn'])
111
112         # query_network_resource
113         network_list = NetworkInstModel.objects.filter(instid=self.nf_inst_id, is_predefined=1)
114         for network in network_list:
115             network_info = {}
116             if not network.resouceid:
117                 continue
118             network_info["vim_id"] = network.vimid
119             network_info["tenant_id"] = network.tenant
120             network_info["res_id"] = network.resouceid
121             self.inst_resource['network'].append(network_info)
122         logger.info('[query_network_resource]:ret_networks=%s' % self.inst_resource['network'])
123
124         # query_subnetwork_resource
125         subnetwork_list = SubNetworkInstModel.objects.filter(instid=self.nf_inst_id, is_predefined=1)
126         for subnetwork in subnetwork_list:
127             subnetwork_info = {}
128             if not subnetwork.resouceid:
129                 continue
130             subnetwork_info["vim_id"] = subnetwork.vimid
131             subnetwork_info["tenant_id"] = subnetwork.tenant
132             subnetwork_info["res_id"] = subnetwork.resouceid
133             self.inst_resource['subnet'].append(subnetwork_info)
134         logger.info('[query_subnetwork_resource]:ret_networks=%s' % self.inst_resource['subnet'])
135
136         # query_port_resource
137         port_list = PortInstModel.objects.filter(instid=self.nf_inst_id, is_predefined=1)
138         for port in port_list:
139             port_info = {}
140             if not port.resouceid:
141                 continue
142             port_info["vim_id"] = port.vimid
143             port_info["tenant_id"] = port.tenant
144             port_info["res_id"] = port.resouceid
145             self.inst_resource['port'].append(port_info)
146         logger.info('[query_port_resource]:ret_networks=%s' % self.inst_resource['port'])
147
148         # query_flavor_resource
149         flavor_list = FlavourInstModel.objects.filter(instid=self.nf_inst_id, is_predefined=1)
150         for flavor in flavor_list:
151             flavor_info = {}
152             if not flavor.resouceid:
153                 continue
154             flavor_info["vim_id"] = flavor.vimid
155             flavor_info["tenant_id"] = flavor.tenant
156             flavor_info["res_id"] = flavor.resouceid
157             self.inst_resource['flavor'].append(flavor_info)
158         logger.info('[query_flavor_resource]:ret_networks=%s' % self.inst_resource['flavor'])
159
160         # query_vm_resource
161         vm_list = VmInstModel.objects.filter(instid=self.nf_inst_id, is_predefined=1)
162         for vm in vm_list:
163             vm_info = {}
164             if not vm.resouceid:
165                 continue
166             vm_info["vim_id"] = vm.vimid
167             vm_info["tenant_id"] = vm.tenant
168             vm_info["res_id"] = vm.resouceid
169             self.inst_resource['vm'].append(vm_info)
170         logger.info('[query_vm_resource]:ret_vms=%s' % self.inst_resource['vm'])
171
172     def query_notify_data(self):
173         logger.info('[NF instantiation] send notify request to nfvo start')
174         affected_vnfc = []
175         vnfcs = VNFCInstModel.objects.filter(instid=self.nf_inst_id)
176         for vnfc in vnfcs:
177             vm_resource = {}
178             if vnfc.vmid:
179                 vm = VmInstModel.objects.filter(vmid=vnfc.vmid)
180                 if vm:
181                     vm_resource = {'vimId': vm[0].vimid, 'resourceId': vm[0].resouceid,
182                                    'resourceName': vm[0].vmname, 'resourceType': 'vm'}
183             affected_vnfc.append(
184                 {'vnfcInstanceId': vnfc.vnfcinstanceid,
185                  'vduId': vnfc.vduid,
186                  'changeType': 'removed',
187                  'computeResource': vm_resource})
188         affected_vl = []
189         networks = NetworkInstModel.objects.filter(instid=self.nf_inst_id)
190         for network in networks:
191             network_resource = {'vimId': network.vimid, 'resourceId': network.resouceid,
192                                 'resourceName': network.name, 'resourceType': 'network'}
193             affected_vl.append(
194                 {'vlInstanceId': network.networkid,
195                  'vldid': network.nodeId,
196                  'changeType': 'removed',
197                  'networkResource': network_resource})
198         affected_cp = []
199         ports = PortInstModel.objects.filter(instid=self.nf_inst_id)
200         for port in ports:
201             affected_cp.append(
202                 {'vsInstanceId': port.portid,
203                  'cpdid': port.nodeId,
204                  'changeType': 'removed',
205                  'storageResource': {'vimId': port.vimid, 'resourceId': port.resouceid,
206                                      'resourceName': port.name, 'resourceType': 'port'}})
207         affected_vs = []
208         vss = StorageInstModel.objects.filter(instid=self.nf_inst_id)
209         for vs in vss:
210             affected_vs.append(
211                 {'vsInstanceId': vs.storageid,
212                  'vsdId': vs.nodeId,
213                  'changeType': 'removed',
214                  'storageResource': {'vimId': vs.vimid, 'resourceId': vs.resouceid,
215                                      'resourceName': vs.name, 'resourceType': 'volume'}})
216         self.notify_data = {
217             "status": 'result',
218             "nfInstanceId": self.nf_inst_id,
219             "operation": 'Terminal',
220             "jobId": self.job_id,
221             'affectedVnfc': affected_vnfc,
222             'affectedVirtualLink': affected_vl,
223             'affectedVirtualStorage': affected_vs,
224             # "vnfdmodule": allocate_data,
225             # "additionalParam": addition_param,
226             # "nfvoInstanceId": self.nfvo_inst_id,
227             # "vnfmInstanceId": self.vnfm_inst_id,
228             # 'affectedcapacity': affectedcapacity,
229             # 'affectedService': [],
230             'affectedCp': affected_cp
231             }
232         logger.info('content_args=%s' % self.notify_data)
233
234     def delete_resource(self):
235         logger.error('rollback resource begin')
236         adaptor.delete_vim_res(self.inst_resource, self.do_notify_delete)
237         logger.error('rollback resource complete')
238
239     def do_notify_delete(self, res_type, res_id):
240         logger.error('Deleting [%s] resource:resourceid [%s]' % (res_type, res_id))
241         if res_type == adaptor.RES_VM:
242             VmInstModel.objects.filter(instid=self.nf_inst_id, resouceid=res_id).delete()
243         elif res_type == adaptor.RES_FLAVOR:
244             FlavourInstModel.objects.filter(instid=self.nf_inst_id, resouceid=res_id).delete()
245         elif res_type == adaptor.RES_PORT:
246             PortInstModel.objects.filter(instid=self.nf_inst_id, resouceid=res_id).delete()
247         elif res_type == adaptor.RES_SUBNET:
248             SubNetworkInstModel.objects.filter(instid=self.nf_inst_id, resouceid=res_id).delete()
249         elif res_type == adaptor.RES_NETWORK:
250             NetworkInstModel.objects.filter(instid=self.nf_inst_id, resouceid=res_id).delete()
251         elif res_type == adaptor.RES_VOLUME:
252             StorageInstModel.objects.filter(instid=self.nf_inst_id, resouceid=res_id).delete()
253
254     def lcm_notify(self):
255         NfInstModel.objects.filter(nfinstid=self.nf_inst_id).update(status='NOT_INSTANTIATED', lastuptime=now_time())
256         logger.info('[NF termination] send notify request to nfvo end')
257         resp = notify_lcm_to_nfvo(self.notify_data)
258         logger.info('[NF termination] get lcm response %s' % resp)
259         logger.info('[NF termination] send notify request to nfvo end')
260
261     def vnf_term_failed_handle(self, error_msg):
262         logger.error('VNF termination failed, detail message: %s' % error_msg)
263         NfInstModel.objects.filter(nfinstid=self.nf_inst_id).update(status='failed', lastuptime=now_time())
264         JobUtil.add_job_status(self.job_id, 255, error_msg)