lcm/ns/biz/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.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
  29
  30 JOB_ERROR = 255
  31 SCALE_TYPE = ("SCALE_NS", "SCALE_VNF")
  32 logger = logging.getLogger(__name__)
  33
  34
  35 class NSManualScaleService(threading.Thread):
  36     """
  37     Scale the NS instance
  38     """
  39
  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
  44         self.job_id = job_id
  45         self.occ_id = NsLcmOpOcc.create(ns_instance_id, "SCALE", "PROCESSING", False, request_data)
  46         self.scale_vnf_data = ''
  47
  48     def run(self):
  49         try:
  50             self.do_biz()
  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])
  59         finally:
  60             self.update_ns_status(NS_INST_STATUS.ACTIVE)
  61
  62     def do_biz(self):
  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()
  66         self.do_vnfs_scale()
  67         self.update_job(JOB_PROGRESS.FINISHED, desc='ns scale success')
  68         NsLcmOpOcc.update(self.occ_id, "COMPLETED")
  69
  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.')
  74
  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')
  81
  82     def do_vnfs_scale(self):
  83         """
  84         Scale VNF instance
  85         :return:
  86         """
  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:
  94                 logger.info(
  95                     'nf[%s] scale handle end' %
  96                     vnf_scale_params.get('vnfInstanceId'))
  97                 self.update_job(
  98                     progress_range[1],
  99                     desc='nf[%s] scale handle end' %
 100                          vnf_scale_params.get('vnfInstanceId'))
 101             else:
 102                 raise NSLCMException('VNF scale failed')
 103
 104     def prepare_vnf_scale_params(self, vnf_data):
 105         """
 106         Prepare parameters for VNF instance scale
 107         :param vnf_data:
 108         :return:
 109         """
 110         return {
 111             "vnfInstanceId": ignore_case_get(vnf_data, 'vnfInstanceId'),
 112             "scaleVnfData": ignore_case_get(vnf_data, 'scaleByStepData'),
 113             "nsInstanceId": self.ns_instance_id
 114         }
 115
 116     def do_vnf_scale(self, vnf_scale_params, progress_range):
 117         """
 118         Scale VNF instance
 119         :param vnf_scale_params:
 120         :param progress_range:
 121         :return:
 122         """
 123         nf_inst_id = vnf_scale_params.get('vnfInstanceId')
 124         nf_service = NFManualScaleService(nf_inst_id, vnf_scale_params)
 125         nf_service.start()
 126         self.update_job(
 127             progress_range[0],
 128             desc='nf[%s] scale handle start' %
 129                  nf_inst_id)
 130         status = self.wait_job_finish(nf_service.job_id)
 131         return status
 132
 133     @staticmethod
 134     def wait_job_finish(sub_job_id, timeout=3600):
 135         query_interval = 2
 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
 146
 147     def update_job(self, progress, desc=''):
 148         """
 149         Update the information of job
 150         :param progress:
 151         :param desc:
 152         :return:
 153         """
 154         JobUtil.add_job_status(self.job_id, progress, desc)
 155
 156     def update_ns_status(self, status):
 157         """
 158         Update NS instance status
 159         :param status:
 160         :return:
 161         """
 162         NSInstModel.objects.filter(
 163             id=self.ns_instance_id).update(
 164             status=status)