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.
19 from lcm.pub.database.models import NfInstModel
20 from lcm.pub.database.models import NSInstModel
21 from lcm.ns.vnfs.const import VNF_STATUS
22 from lcm.pub.msapi import catalog
25 logger = logging.getLogger(__name__)
26 SCALE_TYPE = ("SCALE_NS", "SCALE_VNF")
28 scale_vnf_data_mapping = {
38 def ignorcase_get(args, key):
46 if old_key.upper() == key.upper():
51 def mapping_conv(keyword_map, rest_return):
53 for param in keyword_map:
54 if keyword_map[param]:
55 if isinstance(keyword_map[param], dict):
56 resp_data[param] = mapping_conv(
57 keyword_map[param], ignorcase_get(
60 resp_data[param] = ignorcase_get(rest_return, param)
64 def get_vnf_scale_info(filename, ns_instanceId, aspect, step):
65 json_data = get_json_data(filename)
66 scale_options = ignorcase_get(json_data, "scale_options")
67 for i in range(scale_options.__len__()):
68 ns_scale_option = scale_options[i]
69 if (ignorcase_get(ns_scale_option, "ns_instanceId") == ns_instanceId) \
70 and (ignorcase_get(ns_scale_option, "ns_scale_aspect") == aspect):
71 ns_scale_info_list = ignorcase_get(
72 ns_scale_option, "ns_scale_info_list")
73 for j in range(ns_scale_info_list.__len__()):
74 ns_scale_info = ns_scale_info_list[j]
75 if ns_scale_info["step"] == step:
76 return ns_scale_info["vnf_scale_info"]
81 def get_vnf_instance_id_list(vnfd_id):
83 kwargs['package_id'] = vnfd_id
84 kwargs['status'] = VNF_STATUS.ACTIVE
86 nf_model_list = NfInstModel.objects.filter(**kwargs)
87 vnf_instance_id_list = list()
88 nf_model_len = nf_model_list.__len__()
90 logger.error("No VNF instances found(vnfd_id=%s)" % vnfd_id)
92 for i in range(nf_model_len):
93 vnf_instance_id_list.append(nf_model_list[i].nfinstid)
95 return vnf_instance_id_list
98 def get_json_data(filename):
101 data = json.JSONDecoder().decode(json_str)
106 def check_scale_list(vnf_scale_list, ns_instanceId, aspect, step):
107 if vnf_scale_list is None or vnf_scale_list.__len__() == 0:
109 "The scaling option[ns=%s, aspect=%s, step=%s] does not exist. Pls check the config file." %
110 (ns_instanceId, aspect, step))
112 "The scaling option[ns=%s, aspect=%s, step=%s] does not exist. Pls check the config file." %
113 (ns_instanceId, aspect, step))
115 return vnf_scale_list
118 def get_scale_vnf_data_list(filename, ns_instanceId, aspect, step, scale_type):
120 vnf_scale_list = get_vnf_scale_info(filename, ns_instanceId, aspect, step)
121 check_scale_list(vnf_scale_list, ns_instanceId, aspect, step)
122 scaleVnfDataList = set_scaleVnfData_type(vnf_scale_list, scale_type)
123 logger.debug("scaleVnfDataList = %s" % scaleVnfDataList)
124 return scaleVnfDataList
127 # Get the nsd id according to the ns instance id.
128 def get_nsdId(ns_instanceId):
129 if NSInstModel.objects.filter(id=ns_instanceId):
130 nsd_id = NSInstModel.objects.filter(id=ns_instanceId)[0].nsd_id
136 def get_and_check_params(scaleNsData, ns_InstanceId):
137 if scaleNsData is None:
138 raise Exception("Error! scaleNsData in the request is Empty!")
140 scaleNsByStepsData = scaleNsData["scaleNsByStepsData"][0]
141 if scaleNsByStepsData is None:
142 raise Exception("Error! scaleNsByStepsData in the request is Empty!")
144 aspect = scaleNsByStepsData["aspectId"]
145 numberOfSteps = scaleNsByStepsData["numberOfSteps"]
146 scale_type = scaleNsByStepsData["scalingDirection"]
148 return aspect, numberOfSteps, scale_type
151 def get_scale_vnf_data(scaleNsData, ns_InstanceId):
152 curdir_path = os.path.dirname(
155 os.path.abspath(__file__))))
156 filename = curdir_path + "/ns/data/scalemapping.json"
157 logger.debug("filename = %s" % filename)
158 aspect, numberOfSteps, scale_type = get_and_check_params(
159 scaleNsData, ns_InstanceId)
160 return get_scale_vnf_data_list(
168 # Get scaling vnf data info list according to the ns instance id and request ScaleNsData.
169 def get_scale_vnf_data_info_list(scaleNsData, ns_InstanceId):
170 # Gets the nsd id accordign to the ns instance id.
171 nsd_id = get_nsdId(ns_InstanceId)
173 # Gets the scalingmap json data from the package according to the ns instance id.
174 scalingmap_json = catalog.get_scalingmap_json_package(ns_InstanceId)
176 # Gets and checks the values of parameters.
177 aspect, numberOfSteps, scale_type = get_and_check_params(
178 scaleNsData, ns_InstanceId)
180 # Firstly, gets the scaling vnf data info list from the scaling map json data.
181 scale_vnf_data_info_list_from_json = get_scale_vnf_data_from_json(scalingmap_json, nsd_id, aspect, numberOfSteps)
182 check_scale_list(scale_vnf_data_info_list_from_json, ns_InstanceId, aspect, numberOfSteps)
184 # Secondly, adds the property of vnfInstanceId to the list according to the vnfd id.
185 scale_vnf_data_info_list = set_scacle_vnf_instance_id(scale_vnf_data_info_list_from_json)
186 check_scale_list(scale_vnf_data_info_list, ns_InstanceId, aspect, numberOfSteps)
188 # Lastly, adds the property of type to the list acoording to the request ScaleNsData.
189 scale_vnf_data_info_list = set_scaleVnfData_type(scale_vnf_data_info_list, scale_type)
190 check_scale_list(scale_vnf_data_info_list, ns_InstanceId, aspect, numberOfSteps)
192 return scale_vnf_data_info_list
195 # Get the vnf scaling info from the scaling_map.json according to the ns package id.
196 def get_scale_vnf_data_from_json(scalingmap_json, nsd_id, aspect, step):
197 scale_options = ignorcase_get(scalingmap_json, "scale_options")
198 for i in range(scale_options.__len__()):
199 ns_scale_option = scale_options[i]
200 if (ignorcase_get(ns_scale_option, "nsd_id") == nsd_id) and (
201 ignorcase_get(ns_scale_option, "ns_scale_aspect") == aspect):
202 ns_scale_info_list = ignorcase_get(
203 ns_scale_option, "ns_scale_info")
204 for j in range(ns_scale_info_list.__len__()):
205 ns_scale_info = ns_scale_info_list[j]
206 if ns_scale_info["step"] == step:
207 vnf_scale_info_list = ns_scale_info["vnf_scale_info"]
209 return vnf_scale_info_list
211 logger.error("get_scale_vnf_data_from_json method retuan null")
215 # Gets the vnf instance id according to the vnfd_id and modify the list of scaling vnf info accrodingly.
216 def set_scacle_vnf_instance_id(vnf_scale_info_list):
217 scale_vnf_data_info_list = []
218 for i in range(vnf_scale_info_list.__len__()):
219 vnf_scale_info = vnf_scale_info_list[i]
220 vnfd_id = vnf_scale_info["vnfd_id"]
221 vnf_instance_id_list = get_vnf_instance_id_list(vnfd_id)
223 while index < vnf_instance_id_list.__len__():
224 copy_vnf_scale_info = copy.deepcopy(vnf_scale_info)
225 copy_vnf_scale_info.pop("vnfd_id")
226 copy_vnf_scale_info["vnfInstanceId"] = vnf_instance_id_list[index]
228 scale_vnf_data_info_list.append(copy_vnf_scale_info)
230 return scale_vnf_data_info_list
233 # Sets the scaling type of vnf data info list.
234 def set_scaleVnfData_type(vnf_scale_list, scale_type):
236 "vnf_scale_list = %s, type = %s" %
237 (vnf_scale_list, scale_type))
238 scaleVnfDataList = []
239 if vnf_scale_list is not None:
240 for i in range(vnf_scale_list.__len__()):
241 scaleVnfData = copy.deepcopy(scale_vnf_data_mapping)
242 scaleVnfData["vnfInstanceId"] = vnf_scale_list[i]["vnfInstanceId"]
243 scaleVnfData["scaleByStepData"]["type"] = scale_type
244 scaleVnfData["scaleByStepData"]["aspectId"] = vnf_scale_list[i]["vnf_scaleAspectId"]
245 scaleVnfData["scaleByStepData"]["numberOfSteps"] = vnf_scale_list[i]["numberOfSteps"]
246 scaleVnfDataList.append(scaleVnfData)
247 logger.debug("scaleVnfDataList = %s" % scaleVnfDataList)
248 return scaleVnfDataList