1 # Copyright 2017 ZTE Corporation.
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
7 # http://www.apache.org/licenses/LICENSE-2.0
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.
20 from lcm.ns.biz.scale_aspect import get_scale_vnf_data_info_list
21 from lcm.ns.enum import NS_INST_STATUS
22 from lcm.pub.database.models import JobModel, NSInstModel
23 from lcm.pub.exceptions import NSLCMException
24 from lcm.pub.utils.jobutil import JobUtil
25 from lcm.jobs.enum import JOB_MODEL_STATUS, JOB_PROGRESS
26 from lcm.pub.utils.values import ignore_case_get
27 from lcm.ns_vnfs.biz.scale_vnfs import NFManualScaleService
28 from lcm.ns.biz.ns_lcm_op_occ import NsLcmOpOcc
31 SCALE_TYPE = ("SCALE_NS", "SCALE_VNF")
32 logger = logging.getLogger(__name__)
35 class NSManualScaleService(threading.Thread):
40 def __init__(self, ns_instance_id, request_data, job_id):
41 super(NSManualScaleService, self).__init__()
42 self.ns_instance_id = ns_instance_id
43 self.request_data = request_data
45 self.occ_id = NsLcmOpOcc.create(ns_instance_id, "SCALE", "PROCESSING", False, request_data)
46 self.scale_vnf_data = ''
51 except NSLCMException as e:
52 JobUtil.add_job_status(self.job_id, JOB_PROGRESS.ERROR, e.args[0])
53 NsLcmOpOcc.update(self.occ_id, operationState="FAILED", error=e.args[0])
54 except Exception as e:
55 logger.error(e.args[0])
56 logger.error(traceback.format_exc())
57 JobUtil.add_job_status(self.job_id, JOB_PROGRESS.ERROR, 'ns scale fail')
58 NsLcmOpOcc.update(self.occ_id, operationState="FAILED", error=e.args[0])
60 self.update_ns_status(NS_INST_STATUS.ACTIVE)
63 self.update_job(JOB_PROGRESS.STARTED, desc='ns scale start')
64 self.update_ns_status(NS_INST_STATUS.SCALING)
65 self.check_and_set_params()
67 self.update_job(JOB_PROGRESS.FINISHED, desc='ns scale success')
68 NsLcmOpOcc.update(self.occ_id, "COMPLETED")
70 def check_and_set_params(self):
71 scale_type = ignore_case_get(self.request_data, 'scaleType')
72 if scale_type != SCALE_TYPE[0]:
73 raise NSLCMException('scaleType should be SCALE_NS.')
75 scale_ns_data = ignore_case_get(self.request_data, 'scaleNsData')
76 self.scale_vnf_data = get_scale_vnf_data_info_list(
77 scale_ns_data, self.ns_instance_id)
78 logger.debug('scale_vnf_data = %s' % self.scale_vnf_data)
79 if not self.scale_vnf_data:
80 raise NSLCMException('Failed to get scaleVnfData parameter')
82 def do_vnfs_scale(self):
87 for i in range(len(self.scale_vnf_data)):
88 vnf_scale_params = self.prepare_vnf_scale_params(
89 self.scale_vnf_data[i])
90 count = len(self.scale_vnf_data)
91 progress_range = [11 + 80 / count * i, 10 + 80 / count * (i + 1)]
92 status = self.do_vnf_scale(vnf_scale_params, progress_range)
93 if status is JOB_MODEL_STATUS.FINISHED:
95 'nf[%s] scale handle end' %
96 vnf_scale_params.get('vnfInstanceId'))
99 desc='nf[%s] scale handle end' %
100 vnf_scale_params.get('vnfInstanceId'))
102 raise NSLCMException('VNF scale failed')
104 def prepare_vnf_scale_params(self, vnf_data):
106 Prepare parameters for VNF instance scale
111 "vnfInstanceId": ignore_case_get(vnf_data, 'vnfInstanceId'),
112 "scaleVnfData": ignore_case_get(vnf_data, 'scaleByStepData'),
113 "nsInstanceId": self.ns_instance_id
116 def do_vnf_scale(self, vnf_scale_params, progress_range):
119 :param vnf_scale_params:
120 :param progress_range:
123 nf_inst_id = vnf_scale_params.get('vnfInstanceId')
124 nf_service = NFManualScaleService(nf_inst_id, vnf_scale_params)
128 desc='nf[%s] scale handle start' %
130 status = self.wait_job_finish(nf_service.job_id)
134 def wait_job_finish(sub_job_id, timeout=3600):
136 start_time = end_time = datetime.datetime.now()
137 while (end_time - start_time).seconds < timeout:
138 job_result = JobModel.objects.get(jobid=sub_job_id)
139 time.sleep(query_interval)
140 end_time = datetime.datetime.now()
141 if job_result.progress == JOB_PROGRESS.FINISHED:
142 return JOB_MODEL_STATUS.FINISHED
143 if job_result.progress > JOB_PROGRESS.FINISHED:
144 return JOB_MODEL_STATUS.ERROR
145 return JOB_MODEL_STATUS.TIMEOUT
147 def update_job(self, progress, desc=''):
149 Update the information of job
154 JobUtil.add_job_status(self.job_id, progress, desc)
156 def update_ns_status(self, status):
158 Update NS instance status
162 NSInstModel.objects.filter(
163 id=self.ns_instance_id).update(