lcm/ns/ns_manual_scale.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 datetime
  15 import logging
  16 import threading
  17 import time
  18 import traceback
  19
  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
  27
  28 JOB_ERROR = 255
  29 SCALE_TYPE = ("SCALE_NS", "SCALE_VNF")
  30 logger = logging.getLogger(__name__)
  31
  32
  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
  38         self.job_id = job_id
  39         self.scale_type = ''
  40         self.scale_vnf_data = ''
  41         self.scale_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         except:
  49             logger.error(traceback.format_exc())
  50             JobUtil.add_job_status(self.job_id, JOB_ERROR, 'ns scale fail')
  51         finally:
  52             self.update_ns_status(NS_INST_STATUS.ACTIVE)
  53
  54     def do_biz(self):
  55         self.update_job(1, desc='ns scale start')
  56         self.update_ns_status(NS_INST_STATUS.SCALING)
  57         self.get_and_check_params()
  58         self.do_vnfs_scale()
  59         self.update_job(100, desc='ns scale success')
  60
  61     def get_and_check_params(self):
  62         self.scale_type = ignore_case_get(self.request_data, 'scaleType')
  63         if not self.scale_type or self.scale_type != SCALE_TYPE[0]:
  64             logger.error('scaleType parameter does not exist or value is incorrect. It must be SCALE_NS.')
  65             raise NSLCMException('scaleType parameter does not exist or value incorrect. It must be SCALE_NS.')
  66
  67         # Get data if SCALE_NS
  68         self.scale_ns_data = ignore_case_get(self.request_data, 'scaleNsData')
  69         self.scale_vnf_data = get_scale_vnf_data(self.scale_ns_data,self.ns_instance_id)
  70         logger.debug('scale_vnf_data = %s' % self.scale_vnf_data)
  71         # Get data if SCALE_VNF
  72         if not self.scale_vnf_data:
  73             logger.error('scaleVnfData parameter does not exist or value incorrect')
  74             raise NSLCMException('scaleVnfData parameter does not exist or value incorrect')
  75
  76     def do_vnfs_scale(self):
  77         for i in range(len(self.scale_vnf_data)):
  78             vnf_scale_params = self.prepare_vnf_scale_params(self.scale_vnf_data[i])
  79             count = len(self.scale_vnf_data)
  80             progress_range = [11 + 80 / count * i, 10 + 80 / count * (i + 1)]
  81             status = self.do_vnf_scale(vnf_scale_params, progress_range)
  82             if status is JOB_MODEL_STATUS.FINISHED:
  83                 logger.info('nf[%s] scale handle end' % vnf_scale_params.get('vnfInstanceId'))
  84                 self.update_job(progress_range[1],
  85                                 desc='nf[%s] scale handle end' % vnf_scale_params.get('vnfInstanceId'))
  86             else:
  87                 logger.error('nf scale failed')
  88                 raise NSLCMException('nf scale failed')
  89
  90     def prepare_vnf_scale_params(self, vnf_data):
  91         vnf_instance_id = ignore_case_get(vnf_data, 'vnfInstanceId')
  92         scale_by_step_data = ignore_case_get(vnf_data, 'scaleByStepData')
  93         result = {
  94             "vnfInstanceId": vnf_instance_id,
  95             "scaleByStepData": scale_by_step_data,
  96             "nsInstanceId": self.ns_instance_id
  97         }
  98         return result
  99
 100     def do_vnf_scale(self, vnf_scale_params, progress_range):
 101         nf_inst_id = vnf_scale_params.get('vnfInstanceId')
 102         nf_service = NFManualScaleService(nf_inst_id, vnf_scale_params)
 103         nf_service.start()
 104         self.update_job(progress_range[0], desc='nf[%s] scale handle start' % nf_inst_id)
 105         status = self.wait_job_finish(nf_service.job_id)
 106         return status
 107
 108     @staticmethod
 109     def wait_job_finish(sub_job_id, timeout=3600):
 110         query_interval = 2
 111         start_time = end_time = datetime.datetime.now()
 112         while (end_time - start_time).seconds < timeout:
 113             job_result = JobModel.objects.get(jobid=sub_job_id)
 114             time.sleep(query_interval)
 115             end_time = datetime.datetime.now()
 116             if job_result.progress == 100:
 117                 return JOB_MODEL_STATUS.FINISHED
 118             elif job_result.progress > 100:
 119                 return JOB_MODEL_STATUS.ERROR
 120             else:
 121                 continue
 122         return JOB_MODEL_STATUS.TIMEOUT
 123
 124     def update_job(self, progress, desc=''):
 125         JobUtil.add_job_status(self.job_id, progress, desc)
 126
 127     def update_ns_status(self, status):
 128         NSInstModel.objects.filter(id=self.ns_instance_id).update(status=status)