lcm/ns/vnfs/scale_vnfs.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 json
  15 import logging
  16 import threading
  17 import traceback
  18
  19 from lcm.ns.vnfs.const import VNF_STATUS
  20 from lcm.ns.vnfs.wait_job import wait_job_finish
  21 from lcm.pub.database.models import NfInstModel
  22 from lcm.pub.exceptions import NSLCMException
  23 from lcm.pub.msapi.vnfmdriver import send_nf_scaling_request
  24 from lcm.pub.utils.jobutil import JobUtil, JOB_TYPE, JOB_MODEL_STATUS
  25 from lcm.pub.utils.values import ignore_case_get
  26
  27 JOB_ERROR = 255
  28 SCALE_TYPE = ("SCALE_OUT", "SCALE_IN")
  29 logger = logging.getLogger(__name__)
  30
  31
  32 class NFManualScaleService(threading.Thread):
  33     def __init__(self, vnf_instance_id, data):
  34         super(NFManualScaleService, self).__init__()
  35         self.vnf_instance_id = vnf_instance_id
  36         self.data = data
  37         self.job_id = JobUtil.create_job("NF", JOB_TYPE.MANUAL_SCALE_VNF, vnf_instance_id)
  38         self.scale_vnf_data = ''
  39         self.nf_model = {}
  40         self.nf_scale_params = []
  41         self.m_nf_inst_id = ''
  42         self.vnfm_inst_id = ''
  43
  44     def run(self):
  45         try:
  46             self.do_biz()
  47         except NSLCMException as e:
  48             JobUtil.add_job_status(self.job_id, JOB_ERROR, e.message)
  49         except Exception as ex:
  50             logger.error(ex.message)
  51             logger.error(traceback.format_exc())
  52             JobUtil.add_job_status(self.job_id, JOB_ERROR, 'nf scale fail')
  53         finally:
  54             self.update_nf_status()
  55
  56     def do_biz(self):
  57         self.update_job(1, desc='nf scale start')
  58         self.update_nf_status(VNF_STATUS.SCALING)
  59         self.get_and_check_params()
  60         self.send_nf_scaling_requests()
  61         self.update_job(100, desc='nf scale success')
  62
  63     def get_and_check_params(self):
  64         nf_info = NfInstModel.objects.filter(nfinstid=self.vnf_instance_id)
  65         if not nf_info:
  66             logger.error('NF instance[id=%s] does not exist' % self.vnf_instance_id)
  67             raise NSLCMException('NF instance[id=%s] does not exist' % self.vnf_instance_id)
  68         logger.debug('vnfd_model = %s, vnf_instance_id = %s' % (nf_info[0].vnfd_model, self.vnf_instance_id))
  69         self.nf_model = json.loads(nf_info[0].vnfd_model)
  70         self.m_nf_inst_id = nf_info[0].mnfinstid
  71         self.vnfm_inst_id = nf_info[0].vnfm_inst_id
  72         self.scale_vnf_data = ignore_case_get(self.data, 'scaleVnfData')
  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')
  76         for vnf_data in self.scale_vnf_data:
  77             scale_type = ignore_case_get(vnf_data, 'type')
  78             aspect_id = ignore_case_get(vnf_data, 'aspectId')
  79             number_of_steps = ignore_case_get(vnf_data, 'numberOfSteps')
  80             self.nf_scale_params.append({
  81                 'type': scale_type,
  82                 'aspectId': aspect_id,
  83                 'numberOfSteps': number_of_steps,
  84                 'additionalParam': {'vnfdModel': self.nf_model}
  85             })
  86
  87     def send_nf_scaling_requests(self):
  88         for i in range(len(self.nf_scale_params)):
  89             progress_range = [10 + 80 / len(self.nf_scale_params) * i, 10 + 80 / len(self.nf_scale_params) * (i + 1)]
  90             self.send_nf_scaling_request(self.nf_scale_params[i], progress_range)
  91
  92     def send_nf_scaling_request(self, scale_param, progress_range):
  93         req_param = json.JSONEncoder().encode(scale_param)
  94         rsp = send_nf_scaling_request(self.vnfm_inst_id, self.m_nf_inst_id, req_param)
  95         vnfm_job_id = ignore_case_get(rsp, 'jobId')
  96         ret = wait_job_finish(self.vnfm_inst_id, self.job_id, vnfm_job_id, progress_range=progress_range, timeout=1200,
  97                               mode='1')
  98         if ret != JOB_MODEL_STATUS.FINISHED:
  99             logger.error('[NF scale] nf scale failed')
 100             raise NSLCMException("nf scale failed")
 101
 102     def update_nf_status(self, status=VNF_STATUS.ACTIVE):
 103         NfInstModel.objects.filter(nfinstid=self.vnf_instance_id).update(status=status)
 104
 105     def update_job(self, progress, desc=''):
 106         JobUtil.add_job_status(self.job_id, progress, desc)