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.const import NS_INST_STATUS
21 from lcm.ns.vnfs.scale_vnfs import NFManualScaleService
22 from lcm.pub.database.models import JobModel, NSInstModel
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.pub.utils.scaleaspect import get_scale_vnf_data_package
29 SCALE_TYPE = ("SCALE_NS", "SCALE_VNF")
30 logger = logging.getLogger(__name__)
33 class NSManualScaleService(threading.Thread):
34 def __init__(self, ns_instance_id, request_data, job_id):
35 super(NSManualScaleService, self).__init__()
36 self.ns_instance_id = ns_instance_id
37 self.request_data = request_data
40 self.scale_vnf_data = ''
41 self.scale_ns_data = ''
46 except NSLCMException as e:
47 JobUtil.add_job_status(self.job_id, JOB_ERROR, e.message)
48 except Exception as ex:
49 logger.error(ex.message)
50 logger.error(traceback.format_exc())
51 JobUtil.add_job_status(self.job_id, JOB_ERROR, 'ns scale fail')
53 self.update_ns_status(NS_INST_STATUS.ACTIVE)
56 self.update_job(1, desc='ns scale start')
57 self.update_ns_status(NS_INST_STATUS.SCALING)
58 self.get_and_check_params()
60 self.update_job(100, desc='ns scale success')
62 def get_and_check_params(self):
63 self.scale_type = ignore_case_get(self.request_data, 'scaleType')
64 if not self.scale_type or self.scale_type != SCALE_TYPE[0]:
65 logger.error('scaleType parameter does not exist or value is incorrect. It must be SCALE_NS.')
66 raise NSLCMException('scaleType parameter does not exist or value incorrect. It must be SCALE_NS.')
68 # Get data if SCALE_NS
69 self.scale_ns_data = ignore_case_get(self.request_data, 'scaleNsByStepsData')
70 self.scale_vnf_data = get_scale_vnf_data_package(self.scale_ns_data, self.ns_instance_id)
71 logger.debug('scale_vnf_data = %s' % self.scale_vnf_data)
72 # Get data if SCALE_VNF
73 if not self.scale_vnf_data:
74 logger.error('scaleVnfData parameter does not exist or value incorrect')
75 raise NSLCMException('scaleVnfData parameter does not exist or value incorrect')
77 def do_vnfs_scale(self):
78 for i in range(len(self.scale_vnf_data)):
79 vnf_scale_params = self.prepare_vnf_scale_params(self.scale_vnf_data[i])
80 count = len(self.scale_vnf_data)
81 progress_range = [11 + 80 / count * i, 10 + 80 / count * (i + 1)]
82 status = self.do_vnf_scale(vnf_scale_params, progress_range)
83 if status is JOB_MODEL_STATUS.FINISHED:
84 logger.info('nf[%s] scale handle end' % vnf_scale_params.get('vnfInstanceId'))
85 self.update_job(progress_range[1],
86 desc='nf[%s] scale handle end' % vnf_scale_params.get('vnfInstanceId'))
88 logger.error('nf scale failed')
89 raise NSLCMException('nf scale failed')
91 def prepare_vnf_scale_params(self, vnf_data):
92 vnf_instance_id = ignore_case_get(vnf_data, 'vnfInstanceId')
93 scale_by_step_data = ignore_case_get(vnf_data, 'scaleByStepData')
95 "vnfInstanceId": vnf_instance_id,
96 "scaleVnfData": scale_by_step_data,
97 "nsInstanceId": self.ns_instance_id
101 def do_vnf_scale(self, vnf_scale_params, progress_range):
102 nf_inst_id = vnf_scale_params.get('vnfInstanceId')
103 nf_service = NFManualScaleService(nf_inst_id, vnf_scale_params)
105 self.update_job(progress_range[0], desc='nf[%s] scale handle start' % nf_inst_id)
106 status = self.wait_job_finish(nf_service.job_id)
110 def wait_job_finish(sub_job_id, timeout=3600):
112 start_time = end_time = datetime.datetime.now()
113 while (end_time - start_time).seconds < timeout:
114 job_result = JobModel.objects.get(jobid=sub_job_id)
115 time.sleep(query_interval)
116 end_time = datetime.datetime.now()
117 if job_result.progress == 100:
118 return JOB_MODEL_STATUS.FINISHED
119 elif job_result.progress > 100:
120 return JOB_MODEL_STATUS.ERROR
123 return JOB_MODEL_STATUS.TIMEOUT
125 def update_job(self, progress, desc=''):
126 JobUtil.add_job_status(self.job_id, progress, desc)
128 def update_ns_status(self, status):
129 NSInstModel.objects.filter(id=self.ns_instance_id).update(status=status)