Merge "move json from code to indepandent file"
[vfc/nfvo/lcm.git] / lcm / ns / biz / ns_heal.py
1 # Copyright 2017 Intel 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 datetime
16 import logging
17 import threading
18 import time
19 import traceback
20
21 from lcm.ns.enum import NS_INST_STATUS
22 from lcm.pub.database.models import JobModel, NSInstModel, NfInstModel, VNFCInstModel, VmInstModel
23 from lcm.pub.exceptions import NSLCMException
24 from lcm.pub.utils.jobutil import JobUtil, JOB_MODEL_STATUS
25 from lcm.pub.utils.values import ignore_case_get
26 from lcm.ns_vnfs.biz.heal_vnfs import NFHealService
27 from lcm.ns.biz.ns_lcm_op_occ import NsLcmOpOcc
28
29 JOB_ERROR = 255
30 logger = logging.getLogger(__name__)
31
32
33 class NSHealService(threading.Thread):
34     def __init__(self, ns_instance_id, request_data, job_id):
35         super(NSHealService, self).__init__()
36         self.ns_instance_id = ns_instance_id
37         self.request_data = request_data
38         self.job_id = job_id
39         self.occ_id = NsLcmOpOcc.create(ns_instance_id, "HEAL", "PROCESSING", False, request_data)
40         self.heal_vnf_data = ''
41         self.heal_ns_data = ''
42
43     def run(self):
44         try:
45             self.do_biz()
46         except NSLCMException as e:
47             JobUtil.add_job_status(self.job_id, JOB_ERROR, e.message)
48             NsLcmOpOcc.update(self.occ_id, operationState="FAILED", error=e.message)
49         except Exception as e:
50             logger.error(traceback.format_exc())
51             JobUtil.add_job_status(self.job_id, JOB_ERROR, 'ns heal fail')
52             NsLcmOpOcc.update(self.occ_id, operationState="FAILED", error=e.message)
53
54     def do_biz(self):
55         self.update_job(1, desc='ns heal start')
56         self.get_and_check_params()
57         self.update_ns_status(NS_INST_STATUS.HEALING)
58         self.do_heal()
59         self.update_ns_status(NS_INST_STATUS.ACTIVE)
60         self.update_job(100, desc='ns heal success')
61         NsLcmOpOcc.update(self.occ_id, "COMPLETED")
62
63     def get_and_check_params(self):
64         ns_info = NSInstModel.objects.filter(id=self.ns_instance_id)
65         if not ns_info:
66             errmsg = 'NS [id=%s] does not exist' % self.ns_instance_id
67             logger.error(errmsg)
68             raise NSLCMException(errmsg)
69
70         self.heal_ns_data = ignore_case_get(self.request_data, 'healNsData')
71         self.heal_vnf_data = ignore_case_get(self.request_data, 'healVnfData')
72
73         if self.heal_ns_data and self.heal_vnf_data:
74             errmsg = 'healNsData and healVnfData can not exist together'
75             logger.error(errmsg)
76             raise NSLCMException(errmsg)
77
78         if not self.heal_ns_data and not self.heal_vnf_data:
79             errmsg = 'healNsData and healVnfData parameters does not exist or value is incorrect.'
80             logger.error(errmsg)
81             raise NSLCMException(errmsg)
82
83     def do_heal(self):
84         if self.heal_vnf_data:
85             vnf_heal_params = self.prepare_vnf_heal_params(self.heal_vnf_data)
86             status = self.do_vnf_or_ns_heal(vnf_heal_params, 15)
87             if status is JOB_MODEL_STATUS.FINISHED:
88                 logger.info('nf[%s] heal handle end' %
89                             vnf_heal_params.get('vnfInstanceId'))
90                 self.update_job(90,
91                                 desc='nf[%s] heal handle end' % vnf_heal_params.get('vnfInstanceId'))
92             else:
93                 errmsg = 'nf heal failed'
94                 logger.error(errmsg)
95                 raise NSLCMException(errmsg)
96         else:
97             ns_heal_params = self.prepare_ns_heal_params(self.heal_ns_data)
98             for ns_heal_param in ns_heal_params:
99                 status = self.do_vnf_or_ns_heal(ns_heal_param, 15)
100                 if status is JOB_MODEL_STATUS.FINISHED:
101                     logger.info('nf[%s] heal handle end' %
102                                 ns_heal_param.get('vnfInstanceId'))
103                     self.update_job(90,
104                                     desc='nf[%s] heal handle end' % ns_heal_param.get('vnfInstanceId'))
105                 else:
106                     errmsg = 'nf heal failed'
107                     logger.error(errmsg)
108                     raise NSLCMException(errmsg)
109
110     def do_vnf_or_ns_heal(self, heal_param, progress):
111         instance_id = heal_param.get('vnfInstanceId')
112         nf_service = NFHealService(instance_id, heal_param)
113         nf_service.start()
114         self.update_job(
115             progress, desc='nf[%s] heal handle start' % instance_id)
116         status = self.wait_job_finish(nf_service.job_id)
117         return status
118
119     def prepare_ns_heal_params(self, ns_data):
120         degree_healing = ignore_case_get(ns_data, 'degreeHealing')
121         if not degree_healing:
122             errmsg = 'degreeHealing does not exist.'
123             logger.error(errmsg)
124             raise NSLCMException(errmsg)
125         ns_instance_id = self.ns_instance_id
126         cause = ''
127         action = ignore_case_get(ns_data, 'actionsHealing')
128         if degree_healing == "HEAL_RESTORE":
129             ns_inst_infos = NfInstModel.objects.filter(
130                 ns_inst_id=self.ns_instance_id)
131             if not ns_inst_infos.exists():
132                 raise NSLCMException(
133                     'NSInsts(%s) does not exist' % self.ns_instance_id)
134
135             result_arr = []
136             for ns_inst_info in ns_inst_infos:
137                 vnfc_insts = VNFCInstModel.objects.filter(
138                     nfinstid=ns_inst_info.nfinstid)
139                 # If a condition is not met, will it all terminate?
140                 if not vnfc_insts.exists():
141                     raise NSLCMException(
142                         'vnfcinsts(%s) does not exist' % ns_inst_info.nfinstid)
143                 for vnfc_inst in vnfc_insts:
144                     vm_id = vnfc_inst.vmid
145                     vdu_id = vnfc_inst.vduid
146                     vm_inst_info = VmInstModel.objects.filter(vmid=vm_id)
147                     if not vm_inst_info.exists():
148                         raise NSLCMException(
149                             'vminstinfo(%s) does not exist' % vm_id)
150                     vm_name = vm_inst_info[0].vmname
151
152                     result = {
153                         "vnfInstanceId": ns_instance_id,
154                         "cause": cause,
155                         "additionalParams": {
156                             "action": action,
157                             "actionvminfo": {
158                                 "vmid": vm_id,
159                                 "vduid": vdu_id,
160                                 "vmname": vm_name
161                             }
162                         }
163                     }
164                     result_arr.append(result)
165             return result_arr
166         else:
167             errmsg = 'The degree of healing dose not exist or value is incorrect.'
168             logger.error(errmsg)
169             raise NSLCMException(errmsg)
170
171     def prepare_vnf_heal_params(self, vnf_data):
172         vnf_instance_id = ignore_case_get(vnf_data, 'vnfInstanceId')
173         if not vnf_instance_id:
174             errmsg = 'vnfinstanceid does not exist or value is incorrect.'
175             logger.error(errmsg)
176             raise NSLCMException(errmsg)
177         cause = ignore_case_get(vnf_data, 'cause')
178         additional_params = ignore_case_get(vnf_data, 'additionalParams')
179         action = ignore_case_get(additional_params, 'action')
180         action_vm_info = ignore_case_get(additional_params, 'actionvminfo')
181         vm_id = ignore_case_get(action_vm_info, 'vmid')
182         vdu_id = ignore_case_get(action_vm_info, 'vduid')
183         vm_name = ignore_case_get(action_vm_info, 'vmname')
184
185         result = {
186             "vnfInstanceId": vnf_instance_id,
187             "cause": cause,
188             "additionalParams": {
189                 "action": action,
190                 "actionvminfo": {
191                     "vmid": vm_id,
192                     "vduid": vdu_id,
193                     "vmname": vm_name
194                 }
195             }
196         }
197         return result
198
199     @staticmethod
200     def wait_job_finish(sub_job_id, timeout=3600):
201         query_interval = 2
202         start_time = end_time = datetime.datetime.now()
203         while (end_time - start_time).seconds < timeout:
204             job_result = JobModel.objects.get(jobid=sub_job_id)
205             time.sleep(query_interval)
206             end_time = datetime.datetime.now()
207             if job_result.progress == 100:
208                 return JOB_MODEL_STATUS.FINISHED
209             elif job_result.progress > 100:
210                 return JOB_MODEL_STATUS.ERROR
211             else:
212                 continue
213         return JOB_MODEL_STATUS.TIMEOUT
214
215     def update_job(self, progress, desc=''):
216         JobUtil.add_job_status(self.job_id, progress, desc)
217
218     def update_ns_status(self, status):
219         NSInstModel.objects.filter(
220             id=self.ns_instance_id).update(status=status)