1 # Copyright 2016-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.
21 from rest_framework import status
23 from lcm.pub.database.models import DefPkgMappingModel, ServiceBaseInfoModel, InputParamMappingModel
24 from lcm.pub.database.models import NSInstModel, NfPackageModel, VNFFGInstModel, WFPlanModel
25 from lcm.pub.msapi.catalog import get_process_id, query_rawdata_from_catalog
26 from lcm.pub.msapi.catalog import get_servicetemplate_id, get_servicetemplate
27 from lcm.pub.msapi.sdc_run_catalog import query_vnfpackage_by_id
28 from lcm.pub.msapi.wso2bpel import workflow_run
29 from lcm.pub.msapi.extsys import select_vnfm
30 from lcm.pub.utils.jobutil import JobUtil
31 from lcm.pub.utils import toscautil
32 from lcm.pub.utils.values import ignore_case_get
33 from lcm.pub.exceptions import NSLCMException
34 from lcm.pub.config.config import WORKFLOW_OPTION
35 from lcm.workflows import build_in
36 from lcm.pub.msapi import activiti
37 from lcm.pub.msapi import sdc_run_catalog
39 logger = logging.getLogger(__name__)
42 class InstantNSService(object):
43 def __init__(self, ns_inst_id, plan_content):
44 self.ns_inst_id = ns_inst_id
45 self.req_data = plan_content
49 job_id = JobUtil.create_job("NS", "NS_INST", self.ns_inst_id)
50 logger.debug('ns-instant(%s) workflow starting...' % self.ns_inst_id)
51 logger.debug('req_data=%s' % self.req_data)
52 ns_inst = NSInstModel.objects.get(id=self.ns_inst_id)
55 for key, val in self.req_data['additionalParamForNs'].items():
56 input_parameters.append({"key": key, "value": val})
59 if 'location' in self.req_data['additionalParamForNs']:
60 vim_id = self.req_data['additionalParamForNs']['location']
61 location_constraints = []
62 if 'locationConstraints' in self.req_data:
63 location_constraints = self.req_data['locationConstraints']
65 JobUtil.add_job_status(job_id, 5, 'Start query nsd(%s)' % ns_inst.nspackage_id)
67 if WORKFLOW_OPTION == "wso2":
68 src_plan = query_rawdata_from_catalog(ns_inst.nspackage_id, input_parameters)
69 dst_plan = toscautil.convert_nsd_model(src_plan["rawData"])
71 dst_plan = sdc_run_catalog.parse_nsd(ns_inst.nspackage_id, input_parameters)
72 logger.debug('tosca plan dest:%s' % dst_plan)
74 NSInstModel.objects.filter(id=self.ns_inst_id).update(nsd_model=dst_plan)
76 params_json = json.JSONEncoder().encode(self.req_data["additionalParamForNs"])
79 plan_dict = json.JSONDecoder().decode(dst_plan)
80 for vnf in ignore_case_get(plan_dict, "vnfs"):
81 vnfd_id = vnf['properties']['id']
82 # vnfd = NfPackageModel.objects.get(vnfdid=vnfd_id)
83 vnfpackage_info = query_vnfpackage_by_id(vnfd_id)
84 vnfd_model = json.JSONDecoder().decode(ignore_case_get(vnfpackage_info["packageInfo"], "vnfdModel"))
85 vnfm_type = vnfd_model["metadata"].get("vnfmType", "ztevmanagerdriver")
86 vimid = self.get_vnf_vim_id(vim_id, location_constraints, vnfd_id)
87 vnfm_info = select_vnfm(vnfm_type=vnfm_type, vim_id=vimid)
89 "vnfProfileId": vnf["vnf_id"],
92 "vnfmInstanceId": vnfm_info["vnfmId"],
93 "vnfmType": vnfm_type,
99 self.set_vl_vim_id(vim_id, location_constraints, plan_dict)
100 dst_plan = json.JSONEncoder().encode(plan_dict)
101 logger.debug('tosca plan dest add vimid:%s' % dst_plan)
102 NSInstModel.objects.filter(id=self.ns_inst_id).update(nsd_model=dst_plan)
104 vnf_params_json = json.JSONEncoder().encode(params_vnf)
105 plan_input = {'jobId': job_id,
106 'nsInstanceId': self.req_data["nsInstanceId"],
107 'object_context': dst_plan,
108 'object_additionalParamForNs': params_json,
109 'object_additionalParamForVnf': vnf_params_json}
110 plan_input.update(**self.get_model_count(dst_plan))
111 plan_input["sdnControllerId"] = ignore_case_get(
112 self.req_data['additionalParamForNs'], "sdncontroller")
114 ServiceBaseInfoModel(service_id=self.ns_inst_id,
115 service_name=ns_inst.name,
117 description=ns_inst.description,
119 status=ns_inst.status,
121 create_time=int(time.time() * 1000)).save()
123 if WORKFLOW_OPTION == "wso2":
124 service_tpl = get_servicetemplate(ns_inst.nsd_id)
125 DefPkgMappingModel(service_id=self.ns_inst_id,
126 service_def_id=service_tpl['csarId'],
127 template_name=service_tpl['templateName'],
128 template_id=service_tpl['serviceTemplateId']).save()
133 for key, val in self.req_data['additionalParamForNs'].items():
134 InputParamMappingModel(service_id=self.ns_inst_id,
135 input_key=key, input_value=val).save()
137 for vnffg in ignore_case_get(plan_dict, "vnffgs"):
138 VNFFGInstModel(vnffgdid=vnffg["vnffg_id"],
139 vnffginstid=str(uuid.uuid4()),
140 nsinstid=self.ns_inst_id,
141 endpointnumber=0).save()
143 if WORKFLOW_OPTION == "wso2":
144 return self.start_wso2_workflow(job_id, ns_inst, plan_input)
145 elif WORKFLOW_OPTION == "activiti":
146 return self.start_activiti_workflow()
148 return self.start_buildin_workflow(job_id, plan_input)
150 except Exception as e:
151 logger.error(traceback.format_exc())
152 logger.error("ns-instant(%s) workflow error:%s" % (self.ns_inst_id, e.message))
153 JobUtil.add_job_status(job_id, 255, 'NS instantiation failed: %s' % e.message)
154 return dict(data={'error': e.message}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
156 def start_wso2_workflow(self, job_id, ns_inst, plan_input):
157 servicetemplate_id = get_servicetemplate_id(ns_inst.nsd_id)
158 process_id = get_process_id('init', servicetemplate_id)
159 data = {"processId": process_id, "params": {"planInput": plan_input}}
160 logger.debug('ns-instant(%s) workflow data:%s' % (self.ns_inst_id, data))
162 ret = workflow_run(data)
163 logger.info("ns-instant(%s) workflow result:%s" % (self.ns_inst_id, ret))
164 JobUtil.add_job_status(job_id, 10, 'NS inst(%s) wso2 workflow started: %s' % (
165 self.ns_inst_id, ret.get('status')))
166 if ret.get('status') == 1:
167 return dict(data={'jobId': job_id}, status=status.HTTP_200_OK)
168 return dict(data={'error': ret['message']}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
170 def start_activiti_workflow(self, job_id, plan_input):
171 plans = WFPlanModel.objects.filter()
173 raise NSLCMException("No plan is found, you should deploy plan first!")
175 "processId": plans[0].process_id,
178 ret = activiti.exec_workflow(data)
179 logger.info("ns-instant(%s) workflow result:%s" % (self.ns_inst_id, ret))
180 JobUtil.add_job_status(job_id, 10, 'NS inst(%s) activiti workflow started: %s' % (
181 self.ns_inst_id, ret.get('status')))
182 if ret.get('status') == 1:
183 return dict(data={'jobId': job_id}, status=status.HTTP_200_OK)
184 return dict(data={'error': ret['message']}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
186 def start_buildin_workflow(self, job_id, plan_input):
187 JobUtil.add_job_status(job_id, 10, 'NS inst(%s) buildin workflow started.' %
189 build_in.run_ns_instantiate(plan_input)
190 return dict(data={'jobId': job_id}, status=status.HTTP_200_OK)
192 def get_vnf_vim_id(self, vim_id, location_constraints, vnfdid):
193 for location in location_constraints:
194 if "vnfProfileId" in location and vnfdid == location["vnfProfileId"]:
195 return location["locationConstraints"]["vimId"]
198 raise NSLCMException("No Vim info is found for vnf(%s)." % vnfdid)
200 def set_vl_vim_id(self, vim_id, location_constraints, plan_dict):
201 if "vls" not in plan_dict:
202 logger.debug("No vl is found in nsd.")
205 for vnf in ignore_case_get(plan_dict, "vnfs"):
206 if "dependencies" in vnf:
207 for depend in vnf["dependencies"]:
208 vl_vnf[depend["vl_id"]] = vnf['properties']['id']
210 for location in location_constraints:
211 if "vnfProfileId" in location:
212 vnfd_id = location["vnfProfileId"]
213 vnf_vim[vnfd_id] = location["locationConstraints"]["vimId"]
214 for vl in plan_dict["vls"]:
215 vnfdid = ignore_case_get(vl_vnf, vl["vl_id"])
216 vimid = ignore_case_get(vnf_vim, vnfdid)
220 raise NSLCMException("No Vim info for vl(%s) of vnf(%s)." % (vl["vl_id"], vnfdid))
221 if "location_info" not in vl["properties"]:
222 vl["properties"]["location_info"] = {}
223 vl["properties"]["location_info"]["vimid"] = vimid
226 def get_model_count(context):
227 data = json.JSONDecoder().decode(context)
228 vls = len(data.get('vls', []))
229 sfcs = len(data.get('fps', []))
230 vnfs = len(data.get('vnfs', []))
231 return {'vlCount': str(vls), 'sfcCount': str(sfcs), 'vnfCount': str(vnfs)}