Modify code of query_inst_resource
[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
21 from lcm.pub.exceptions import NFLCMException
22 from lcm.pub.msapi.nfvolcm import apply_grant_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
27 logger = logging.getLogger(__name__)
28
29
30 class TermVnf(Thread):
31     def __init__(self, data, nf_inst_id, job_id):
32         super(TermVnf, self).__init__()
33         self.data = data
34         self.nf_inst_id = nf_inst_id
35         self.job_id = job_id
36         self.terminationType = ignore_case_get(self.data, "terminationType")
37         self.gracefulTerminationTimeout = ignore_case_get(self.data, "gracefulTerminationTimeout")
38         self.inst_resource = {'volumn': [],  # [{"vim_id": ignore_case_get(ret, "vim_id")},{}]
39                               'network': [],
40                               'subnet': [],
41                               'port': [],
42                               'flavor': [],
43                               'vm': [],
44                               }
45
46     def run(self):
47         try:
48             self.term_pre()
49             self.grant_resource()
50             self.query_inst_resource()
51             # self.delete_resource()
52             # self.lcm_notify()
53             JobUtil.add_job_status(self.job_id, 100, "Terminate Vnf success.")
54         except NFLCMException as e:
55             self.vnf_term_failed_handle(e.message)
56         except:
57             logger.error(traceback.format_exc())
58             self.vnf_term_failed_handle(traceback.format_exc())
59
60     def term_pre(self):
61         vnf_insts = NfInstModel.objects.filter(nfinstid=self.nf_inst_id)
62         if not vnf_insts.exists():
63             raise NFLCMException('VnfInst(%s) does not exist' % self.nf_inst_id)
64         sel_vnf = vnf_insts[0]
65         if sel_vnf.status != 'VNF_INSTANTIATED':
66             raise NFLCMException("Don't allow to delete vnf(status:[%s])" % sel_vnf.status)
67         if self.terminationType == 'GRACEFUL' and not self.gracefulTerminationTimeout:
68             raise NFLCMException("Graceful termination must set timeout")
69
70         NfInstModel.objects.filter(nfinstid=self.nf_inst_id).update(status=VNF_STATUS.TERMINATING)
71         JobUtil.add_job_status(self.job_id, 10, 'Nf terminating pre-check finish')
72         logger.info("Nf terminating pre-check finish")
73
74     def grant_resource(self):
75         logger.info("nf_cancel_task grant_resource begin")
76         content_args = {'vnfInstanceId': self.nf_inst_id, 'vnfDescriptorId': '',
77                         'lifecycleOperation': 'Instantiate', 'jobId': self.job_id,
78                         'addResource': [], 'removeResource': [],
79                         'placementConstraint': [], 'additionalParam': {}}
80
81         vdus = VmInstModel.objects.filter(instid=self.nf_inst_id, is_predefined=1)
82         res_index = 1
83         for vdu in vdus:
84             res_def = {'type': 'VDU',
85                        'resDefId': str(res_index),
86                        'resDesId': vdu.resouceid}
87             content_args['removeResource'].append(res_def)
88             res_index += 1
89
90         logger.info('content_args=%s' % content_args)
91         self.apply_result = apply_grant_to_nfvo(content_args)
92         vim_info = ignore_case_get(self.apply_result, "vim")
93         logger.info("nf_cancel_task grant_resource end")
94         JobUtil.add_job_status(self.job_id, 20, 'Nf terminating grant_resource finish')
95
96     def query_inst_resource(self):
97         logger.info('[query_resource begin]:inst_id=%s' % self.nf_inst_id)
98         # query_volumn_resource
99         vol_list = StorageInstModel.objects.filter(instid=self.nf_inst_id, is_predefined=1)
100         for vol in vol_list:
101             vol_info = {}
102             if not vol.resouceid:
103                 continue
104             vol_info["vim_id"] = vol.vimid
105             vol_info["tenant_id"] = vol.tenant
106             vol_info["res_id"] = vol.resouceid
107             self.inst_resource['volumn'].append(vol_info)
108         logger.info('[query_volumn_resource]:ret_volumns=%s' % self.inst_resource['volumn'])
109
110         # query_network_resource
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         # query_subnetwork_resource
123         subnetwork_list = SubNetworkInstModel.objects.filter(instid=self.nf_inst_id, is_predefined=1)
124         for subnetwork in subnetwork_list:
125             subnetwork_info = {}
126             if not subnetwork.resouceid:
127                 continue
128             subnetwork_info["vim_id"] = subnetwork.vimid
129             subnetwork_info["tenant_id"] = subnetwork.tenant
130             subnetwork_info["res_id"] = subnetwork.resouceid
131             self.inst_resource['subnet'].append(subnetwork_info)
132         logger.info('[query_subnetwork_resource]:ret_networks=%s' % self.inst_resource['subnet'])
133
134         # query_port_resource
135         port_list = PortInstModel.objects.filter(instid=self.nf_inst_id, is_predefined=1)
136         for port in port_list:
137             port_info = {}
138             if not port.resouceid:
139                 continue
140             port_info["vim_id"] = port.vimid
141             port_info["tenant_id"] = port.tenant
142             port_info["res_id"] = port.resouceid
143             self.inst_resource['port'].append(port_info)
144         logger.info('[query_port_resource]:ret_networks=%s' % self.inst_resource['port'])
145
146         # query_flavor_resource
147         flavor_list = FlavourInstModel.objects.filter(instid=self.nf_inst_id, is_predefined=1)
148         for flavor in flavor_list:
149             flavor_info = {}
150             if not flavor.resouceid:
151                 continue
152             flavor_info["vim_id"] = flavor.vimid
153             flavor_info["tenant_id"] = flavor.tenant
154             flavor_info["res_id"] = flavor.resouceid
155             self.inst_resource['flavor'].append(flavor_info)
156         logger.info('[query_flavor_resource]:ret_networks=%s' % self.inst_resource['flavor'])
157
158         # query_vm_resource
159         vm_list = VmInstModel.objects.filter(instid=self.nf_inst_id, is_predefined=1)
160         for vm in vm_list:
161             vm_info = {}
162             if not vm.resouceid:
163                 continue
164             vm_info["vim_id"] = vm.vimid
165             vm_info["tenant_id"] = vm.tenant
166             vm_info["res_id"] = vm.resouceid
167             self.inst_resource['vm'].append(vm_info)
168         logger.info('[query_vm_resource]:ret_vms=%s' % self.inst_resource['vm'])
169
170     def delete_resource(self):
171         pass
172
173     def lcm_notify(self):
174         pass
175
176     # def load_nfvo_config(self):
177     #     logger.info("[NF instantiation]get nfvo connection info start")
178     #     reg_info = NfvoRegInfoModel.objects.filter(vnfminstid='vnfm111').first()
179     #     if reg_info:
180     #         self.vnfm_inst_id = reg_info.vnfminstid
181     #         self.nfvo_inst_id = reg_info.nfvoid
182     #         logger.info("[NF instantiation] Registered nfvo id is [%s]" % self.nfvo_inst_id)
183     #     else:
184     #         raise NFLCMException("Nfvo was not registered")
185     #     logger.info("[NF instantiation]get nfvo connection info end")
186
187     def vnf_term_failed_handle(self, error_msg):
188         logger.error('VNF termination failed, detail message: %s' % error_msg)
189         NfInstModel.objects.filter(nfinstid=self.nf_inst_id).update(status='failed', lastuptime=now_time())
190         JobUtil.add_job_status(self.job_id, 255, error_msg)