Refactor vnflcm termination do_notify_delete
[vfc/gvnfm/vnflcm.git] / lcm / lcm / nf / 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.const import VNF_STATUS
21 from lcm.pub.database.models import NfInstModel, VmInstModel, NetworkInstModel, StorageInstModel, \
22     PortInstModel, VNFCInstModel, NfvoRegInfoModel, FlavourInstModel, SubNetworkInstModel
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 = {'volumn': [], 'network': [], 'subnet': [], 'port': [], 'flavor': [], 'vm': []}
44         self.resource_map = {'Storage': 'volumn', 'Network': 'network', 'SubNetwork': 'subnet', 'Port': 'port',
45                              'Flavour': 'flavor', 'Vm': 'vm'}
46
47     def run(self):
48         try:
49             if self.term_pre():
50                 self.grant_resource()
51                 self.query_inst_resource()
52                 self.query_notify_data()
53                 self.delete_resource()
54                 self.lcm_notify()
55             JobUtil.add_job_status(self.job_id, 100, "Terminate Vnf success.")
56         except NFLCMException as e:
57             self.vnf_term_failed_handle(e.message)
58         except Exception as e:
59             logger.error(e.message)
60             self.vnf_term_failed_handle(traceback.format_exc())
61
62     def term_pre(self):
63         vnf_insts = NfInstModel.objects.filter(nfinstid=self.nf_inst_id)
64         if not vnf_insts.exists():
65             logger.warn('VnfInst(%s) does not exist' % self.nf_inst_id)
66             return False
67             # raise NFLCMException('VnfInst(%s) does not exist' % self.nf_inst_id)
68         # sel_vnf = vnf_insts[0]
69         # if sel_vnf.status != 'VNF_INSTANTIATED':
70         #    raise NFLCMException("Don't allow to terminate vnf(status:[%s])" % sel_vnf.status)
71         if self.terminationType == 'GRACEFUL' and not self.gracefulTerminationTimeout:
72             logger.warn("Set Graceful default termination timeout = 60")
73             self.gracefulTerminationTimeout = 60
74             # raise NFLCMException("Graceful termination must set timeout")
75
76         NfInstModel.objects.filter(nfinstid=self.nf_inst_id).update(status=VNF_STATUS.TERMINATING)
77         JobUtil.add_job_status(self.job_id, 10, 'Nf terminating pre-check finish')
78         logger.info("Nf terminating pre-check finish")
79         return True
80
81     def grant_resource(self):
82         logger.info("nf_cancel_task grant_resource begin")
83         content_args = {
84             'vnfInstanceId': self.nf_inst_id,
85             'vnfDescriptorId': '',
86             'lifecycleOperation': 'Terminate',
87             'jobId': self.job_id,
88             'addResource': [],
89             'removeResource': [],
90             'placementConstraint': [],
91             'additionalParam': {}
92         }
93
94         vdus = VmInstModel.objects.filter(instid=self.nf_inst_id, is_predefined=1)
95         res_index = 1
96         for vdu in vdus:
97             res_def = {
98                 'type': 'VDU',
99                 'resDefId': str(res_index),
100                 'resDesId': vdu.resouceid}
101             content_args['removeResource'].append(res_def)
102             res_index += 1
103
104         vnfmInfo = NfvoRegInfoModel.objects.filter(nfvoid=self.nf_inst_id)
105         if len(vnfmInfo) == 0:
106             raise NFLCMException('nf_inst_id(%s) does not exist in NfvoRegInfoModel' % self.nf_inst_id)
107         content_args['additionalParam']['vnfmid'] = vnfmInfo[0].vnfminstid
108         content_args['additionalParam']['vimid'] = vnfmInfo[0].apiurl
109         logger.info('content_args=%s' % content_args)
110         self.apply_result = apply_grant_to_nfvo(json.dumps(content_args))
111         logger.info("nf_cancel_task grant_resource end")
112         JobUtil.add_job_status(self.job_id, 20, 'Nf terminating grant_resource finish')
113
114     def query_inst_resource(self):
115         logger.info('[query_resource begin]:inst_id=%s' % self.nf_inst_id)
116         for resource_type in self.resource_map.keys():
117             resource_table = globals().get(resource_type + 'InstModel')
118             resource_insts = resource_table.objects.filter(instid=self.nf_inst_id)
119             for resource_inst in resource_insts:
120                 if not resource_inst.resouceid:
121                     continue
122                 self.inst_resource[self.resource_map.get(resource_type)].append(self.get_resource(resource_inst))
123         logger.info('[query_resource]:ret_resource=%s' % self.inst_resource)
124
125     def get_resource(self, resource):
126         return {
127             "vim_id": resource.vimid,
128             "tenant_id": resource.tenant,
129             "res_id": resource.resouceid,
130             "is_predefined": resource.is_predefined
131         }
132
133     def query_notify_data(self):
134         logger.info('[NF terminate] send notify request to nfvo start')
135         affected_vnfc = []
136         vnfcs = VNFCInstModel.objects.filter(instid=self.nf_inst_id)
137         for vnfc in vnfcs:
138             vm_resource = {}
139             if vnfc.vmid:
140                 vm = VmInstModel.objects.filter(vmid=vnfc.vmid)
141                 if vm:
142                     vm_resource = {'vimId': vm[0].vimid, 'resourceId': vm[0].resouceid,
143                                    'resourceName': vm[0].vmname, 'resourceType': 'vm'}
144             affected_vnfc.append(
145                 {'vnfcInstanceId': vnfc.vnfcinstanceid,
146                  'vduId': vnfc.vduid,
147                  'changeType': 'removed',
148                  'computeResource': vm_resource})
149         affected_vl = []
150         networks = NetworkInstModel.objects.filter(instid=self.nf_inst_id)
151         for network in networks:
152             network_resource = {
153                 'vimId': network.vimid,
154                 'resourceId': network.resouceid,
155                 'resourceName': network.name,
156                 'resourceType': 'network'}
157             affected_vl.append(
158                 {'vlInstanceId': network.networkid,
159                  'vldid': network.nodeId,
160                  'changeType': 'removed',
161                  'networkResource': network_resource})
162         affected_cp = []
163         ports = PortInstModel.objects.filter(instid=self.nf_inst_id)
164         for port in ports:
165             affected_cp.append(
166                 {'vsInstanceId': port.portid,
167                  'cpdid': port.nodeId,
168                  'changeType': 'removed',
169                  'storageResource': {'vimId': port.vimid, 'resourceId': port.resouceid,
170                                      'resourceName': port.name, 'resourceType': 'port'}})
171         affected_vs = []
172         vss = StorageInstModel.objects.filter(instid=self.nf_inst_id)
173         for vs in vss:
174             affected_vs.append(
175                 {'vsInstanceId': vs.storageid,
176                  'vsdId': vs.nodeId,
177                  'changeType': 'removed',
178                  'storageResource': {'vimId': vs.vimid, 'resourceId': vs.resouceid,
179                                      'resourceName': vs.name, 'resourceType': 'volume'}})
180         FlavourInstModel.objects.filter(instid=self.nf_inst_id)
181         SubNetworkInstModel.objects.filter(instid=self.nf_inst_id)
182         self.notify_data = {
183             "status": 'result',
184             "vnfInstanceId": self.nf_inst_id,
185             "operation": 'Terminal',
186             "jobId": self.job_id,
187             'affectedVnfc': affected_vnfc,
188             'affectedVirtualLink': affected_vl,
189             'affectedVirtualStorage': affected_vs,
190             'affectedCp': affected_cp}
191
192         vnfmInfo = NfvoRegInfoModel.objects.filter(nfvoid=self.nf_inst_id)
193         if len(vnfmInfo) == 0:
194             raise NFLCMException('nf_inst_id(%s) does not exist in NfvoRegInfoModel' % self.nf_inst_id)
195         self.notify_data['VNFMID'] = vnfmInfo[0].vnfminstid
196         logger.info('content_args=%s' % self.notify_data)
197
198     def delete_resource(self):
199         logger.info('rollback resource begin')
200         adaptor.delete_vim_res(self.inst_resource, self.do_notify_delete)
201         logger.info('rollback resource complete')
202
203     def do_notify_delete(self, res_type, res_id):
204         logger.error('Deleting [%s] resource:resourceid [%s]' % (res_type, res_id))
205         resource_type = self.resource_map.keys()[self.resource_map.values().index(res_type)]
206         resource_table = globals().get(resource_type + 'InstModel')
207         resource_table.objects.filter(instid=self.nf_inst_id, resouceid=res_id).delete()
208
209     def lcm_notify(self):
210         NfInstModel.objects.filter(nfinstid=self.nf_inst_id).update(status='NOT_INSTANTIATED', lastuptime=now_time())
211         logger.info('[NF termination] send notify request to nfvo end')
212         resp = notify_lcm_to_nfvo(json.dumps(self.notify_data))
213         logger.info('[NF termination] get lcm response %s' % resp)
214         logger.info('[NF termination] send notify request to nfvo end')
215
216     def vnf_term_failed_handle(self, error_msg):
217         logger.error('VNF termination failed, detail message: %s' % error_msg)
218         NfInstModel.objects.filter(nfinstid=self.nf_inst_id).update(status='failed', lastuptime=now_time())
219         JobUtil.add_job_status(self.job_id, 255, error_msg)