Bug 240 - NS Scaling
[osm/SO.git] / rwlaunchpad / plugins / rwnsm / rift / tasklets / rwnsmtasklet / rwnsmtasklet.py
1 #
2 # Copyright 2016 RIFT.IO Inc
3 #
4 # Licensed under the Apache License, Version 2.0 (the "License");
5 # you may not use this file except in compliance with the License.
6 # You may obtain a copy of the License at
7 #
8 # http://www.apache.org/licenses/LICENSE-2.0
9 #
10 # Unless required by applicable law or agreed to in writing, software
11 # distributed under the License is distributed on an "AS IS" BASIS,
12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 # See the License for the specific language governing permissions and
14 # limitations under the License.
15 #
16
17 # vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4
18 import asyncio
19 import ncclient
20 import ncclient.asyncio_manager
21 import os
22 import shutil
23 import sys
24 import tempfile
25 import time
26 import uuid
27 import yaml
28 import requests
29 import json
30
31
32 from collections import deque
33 from collections import defaultdict
34 from enum import Enum
35
36 import gi
37 gi.require_version('RwYang', '1.0')
38 gi.require_version('RwNsdYang', '1.0')
39 gi.require_version('RwDts', '1.0')
40 gi.require_version('RwNsmYang', '1.0')
41 gi.require_version('RwNsrYang', '1.0')
42 gi.require_version('RwTypes', '1.0')
43 gi.require_version('RwVlrYang', '1.0')
44 gi.require_version('RwVnfrYang', '1.0')
45 from gi.repository import (
46 RwYang,
47 RwNsrYang,
48 NsrYang,
49 NsdYang,
50 RwVlrYang,
51 VnfrYang,
52 RwVnfrYang,
53 RwNsmYang,
54 RwsdnalYang,
55 RwDts as rwdts,
56 RwTypes,
57 ProtobufC,
58 )
59
60 import rift.tasklets
61 import rift.mano.ncclient
62 import rift.mano.config_data.config
63 import rift.mano.dts as mano_dts
64
65 from . import rwnsm_conman as conman
66 from . import cloud
67 from . import publisher
68 from . import xpath
69 from . import config_value_pool
70 from . import rwvnffgmgr
71 from . import scale_group
72
73
74 class NetworkServiceRecordState(Enum):
75 """ Network Service Record State """
76 INIT = 101
77 VL_INIT_PHASE = 102
78 VNF_INIT_PHASE = 103
79 VNFFG_INIT_PHASE = 104
80 RUNNING = 106
81 SCALING_OUT = 107
82 SCALING_IN = 108
83 TERMINATE = 109
84 TERMINATE_RCVD = 110
85 VL_TERMINATE_PHASE = 111
86 VNF_TERMINATE_PHASE = 112
87 VNFFG_TERMINATE_PHASE = 113
88 TERMINATED = 114
89 FAILED = 115
90 VL_INSTANTIATE = 116
91 VL_TERMINATE = 117
92
93
94 class NetworkServiceRecordError(Exception):
95 """ Network Service Record Error """
96 pass
97
98
99 class NetworkServiceDescriptorError(Exception):
100 """ Network Service Descriptor Error """
101 pass
102
103
104 class VirtualNetworkFunctionRecordError(Exception):
105 """ Virtual Network Function Record Error """
106 pass
107
108
109 class NetworkServiceDescriptorNotFound(Exception):
110 """ Cannot find Network Service Descriptor"""
111 pass
112
113
114 class NetworkServiceDescriptorRefCountExists(Exception):
115 """ Network Service Descriptor reference count exists """
116 pass
117
118
119 class NetworkServiceDescriptorUnrefError(Exception):
120 """ Failed to unref a network service descriptor """
121 pass
122
123
124 class NsrInstantiationFailed(Exception):
125 """ Failed to instantiate network service """
126 pass
127
128
129 class VnfInstantiationFailed(Exception):
130 """ Failed to instantiate virtual network function"""
131 pass
132
133
134 class VnffgInstantiationFailed(Exception):
135 """ Failed to instantiate virtual network function"""
136 pass
137
138
139 class VnfDescriptorError(Exception):
140 """Failed to instantiate virtual network function"""
141 pass
142
143
144 class ScalingOperationError(Exception):
145 pass
146
147
148 class ScaleGroupMissingError(Exception):
149 pass
150
151
152 class PlacementGroupError(Exception):
153 pass
154
155
156 class NsrNsdUpdateError(Exception):
157 pass
158
159
160 class NsrVlUpdateError(NsrNsdUpdateError):
161 pass
162
163
164 class VlRecordState(Enum):
165 """ VL Record State """
166 INIT = 101
167 INSTANTIATION_PENDING = 102
168 ACTIVE = 103
169 TERMINATE_PENDING = 104
170 TERMINATED = 105
171 FAILED = 106
172
173
174 class VnffgRecordState(Enum):
175 """ VNFFG Record State """
176 INIT = 101
177 INSTANTIATION_PENDING = 102
178 ACTIVE = 103
179 TERMINATE_PENDING = 104
180 TERMINATED = 105
181 FAILED = 106
182
183
184 class VnffgRecord(object):
185 """ Vnffg Records class"""
186 SFF_DP_PORT = 4790
187 SFF_MGMT_PORT = 5000
188 def __init__(self, dts, log, loop, vnffgmgr, nsr, nsr_name, vnffgd_msg, sdn_account_name):
189
190 self._dts = dts
191 self._log = log
192 self._loop = loop
193 self._vnffgmgr = vnffgmgr
194 self._nsr = nsr
195 self._nsr_name = nsr_name
196 self._vnffgd_msg = vnffgd_msg
197 if sdn_account_name is None:
198 self._sdn_account_name = ''
199 else:
200 self._sdn_account_name = sdn_account_name
201
202 self._vnffgr_id = str(uuid.uuid4())
203 self._vnffgr_rsp_id = list()
204 self._vnffgr_state = VnffgRecordState.INIT
205
206 @property
207 def id(self):
208 """ VNFFGR id """
209 return self._vnffgr_id
210
211 @property
212 def state(self):
213 """ state of this VNF """
214 return self._vnffgr_state
215
216 def fetch_vnffgr(self):
217 """
218 Get VNFFGR message to be published
219 """
220
221 if self._vnffgr_state == VnffgRecordState.INIT:
222 vnffgr_dict = {"id": self._vnffgr_id,
223 "vnffgd_id_ref": self._vnffgd_msg.id,
224 "vnffgd_name_ref": self._vnffgd_msg.name,
225 "sdn_account": self._sdn_account_name,
226 "operational_status": 'init',
227 }
228 vnffgr = NsrYang.YangData_Nsr_NsInstanceOpdata_Nsr_Vnffgr.from_dict(vnffgr_dict)
229 elif self._vnffgr_state == VnffgRecordState.TERMINATED:
230 vnffgr_dict = {"id": self._vnffgr_id,
231 "vnffgd_id_ref": self._vnffgd_msg.id,
232 "vnffgd_name_ref": self._vnffgd_msg.name,
233 "sdn_account": self._sdn_account_name,
234 "operational_status": 'terminated',
235 }
236 vnffgr = NsrYang.YangData_Nsr_NsInstanceOpdata_Nsr_Vnffgr.from_dict(vnffgr_dict)
237 else:
238 try:
239 vnffgr = self._vnffgmgr.fetch_vnffgr(self._vnffgr_id)
240 except Exception:
241 self._log.exception("Fetching VNFFGR for VNFFG with id %s failed", self._vnffgr_id)
242 self._vnffgr_state = VnffgRecordState.FAILED
243 vnffgr_dict = {"id": self._vnffgr_id,
244 "vnffgd_id_ref": self._vnffgd_msg.id,
245 "vnffgd_name_ref": self._vnffgd_msg.name,
246 "sdn_account": self._sdn_account_name,
247 "operational_status": 'failed',
248 }
249 vnffgr = NsrYang.YangData_Nsr_NsInstanceOpdata_Nsr_Vnffgr.from_dict(vnffgr_dict)
250
251 return vnffgr
252
253 @asyncio.coroutine
254 def vnffgr_create_msg(self):
255 """ Virtual Link Record message for Creating VLR in VNS """
256 vnffgr_dict = {"id": self._vnffgr_id,
257 "vnffgd_id_ref": self._vnffgd_msg.id,
258 "vnffgd_name_ref": self._vnffgd_msg.name,
259 "sdn_account": self._sdn_account_name,
260 }
261 vnffgr = NsrYang.YangData_Nsr_NsInstanceOpdata_Nsr_Vnffgr.from_dict(vnffgr_dict)
262 for rsp in self._vnffgd_msg.rsp:
263 vnffgr_rsp = vnffgr.rsp.add()
264 vnffgr_rsp.id = str(uuid.uuid4())
265 vnffgr_rsp.name = self._nsr.name + '.' + rsp.name
266 self._vnffgr_rsp_id.append(vnffgr_rsp.id)
267 vnffgr_rsp.vnffgd_rsp_id_ref = rsp.id
268 vnffgr_rsp.vnffgd_rsp_name_ref = rsp.name
269 for rsp_cp_ref in rsp.vnfd_connection_point_ref:
270 vnfd = [vnfr.vnfd for vnfr in self._nsr.vnfrs.values() if vnfr.vnfd.id == rsp_cp_ref.vnfd_id_ref]
271 self._log.debug("VNFD message during VNFFG instantiation is %s",vnfd)
272 if len(vnfd) > 0 and vnfd[0].has_field('service_function_type'):
273 self._log.debug("Service Function Type for VNFD ID %s is %s",rsp_cp_ref.vnfd_id_ref, vnfd[0].service_function_type)
274 else:
275 self._log.error("Service Function Type not available for VNFD ID %s; Skipping in chain",rsp_cp_ref.vnfd_id_ref)
276 continue
277
278 vnfr_cp_ref = vnffgr_rsp.vnfr_connection_point_ref.add()
279 vnfr_cp_ref.member_vnf_index_ref = rsp_cp_ref.member_vnf_index_ref
280 vnfr_cp_ref.hop_number = rsp_cp_ref.order
281 vnfr_cp_ref.vnfd_id_ref =rsp_cp_ref.vnfd_id_ref
282 vnfr_cp_ref.service_function_type = vnfd[0].service_function_type
283 for nsr_vnfr in self._nsr.vnfrs.values():
284 if (nsr_vnfr.vnfd.id == vnfr_cp_ref.vnfd_id_ref and
285 nsr_vnfr.member_vnf_index == vnfr_cp_ref.member_vnf_index_ref):
286 vnfr_cp_ref.vnfr_id_ref = nsr_vnfr.id
287 vnfr_cp_ref.vnfr_name_ref = nsr_vnfr.name
288 vnfr_cp_ref.vnfr_connection_point_ref = rsp_cp_ref.vnfd_connection_point_ref
289
290 vnfr = yield from self._nsr.fetch_vnfr(nsr_vnfr.xpath)
291 self._log.debug(" Received VNFR is %s", vnfr)
292 while vnfr.operational_status != 'running':
293 self._log.info("Received vnf op status is %s; retrying",vnfr.operational_status)
294 if vnfr.operational_status == 'failed':
295 self._log.error("Fetching VNFR for %s failed", vnfr.id)
296 raise NsrInstantiationFailed("Failed NS %s instantiation due to VNFR %s failure" % (self.id, vnfr.id))
297 yield from asyncio.sleep(2, loop=self._loop)
298 vnfr = yield from self._nsr.fetch_vnfr(nsr_vnfr.xpath)
299 self._log.debug("Received VNFR is %s", vnfr)
300
301 vnfr_cp_ref.connection_point_params.mgmt_address = vnfr.mgmt_interface.ip_address
302 for cp in vnfr.connection_point:
303 if cp.name == vnfr_cp_ref.vnfr_connection_point_ref:
304 vnfr_cp_ref.connection_point_params.port_id = cp.connection_point_id
305 vnfr_cp_ref.connection_point_params.name = self._nsr.name + '.' + cp.name
306 for vdu in vnfr.vdur:
307 for ext_intf in vdu.external_interface:
308 if ext_intf.name == vnfr_cp_ref.vnfr_connection_point_ref:
309 vnfr_cp_ref.connection_point_params.vm_id = vdu.vim_id
310 self._log.debug("VIM ID for CP %s in VNFR %s is %s",cp.name,nsr_vnfr.id,
311 vnfr_cp_ref.connection_point_params.vm_id)
312 break
313
314 vnfr_cp_ref.connection_point_params.address = cp.ip_address
315 vnfr_cp_ref.connection_point_params.port = VnffgRecord.SFF_DP_PORT
316
317 for vnffgd_classifier in self._vnffgd_msg.classifier:
318 _rsp = [rsp for rsp in vnffgr.rsp if rsp.vnffgd_rsp_id_ref == vnffgd_classifier.rsp_id_ref]
319 if len(_rsp) > 0:
320 rsp_id_ref = _rsp[0].id
321 rsp_name = _rsp[0].name
322 else:
323 self._log.error("RSP with ID %s not found during classifier creation for classifier id %s",vnffgd_classifier.rsp_id_ref,vnffgd_classifier.id)
324 continue
325 vnffgr_classifier = vnffgr.classifier.add()
326 vnffgr_classifier.id = vnffgd_classifier.id
327 vnffgr_classifier.name = self._nsr.name + '.' + vnffgd_classifier.name
328 _rsp[0].classifier_name = vnffgr_classifier.name
329 vnffgr_classifier.rsp_id_ref = rsp_id_ref
330 vnffgr_classifier.rsp_name = rsp_name
331 for nsr_vnfr in self._nsr.vnfrs.values():
332 if (nsr_vnfr.vnfd.id == vnffgd_classifier.vnfd_id_ref and
333 nsr_vnfr.member_vnf_index == vnffgd_classifier.member_vnf_index_ref):
334 vnffgr_classifier.vnfr_id_ref = nsr_vnfr.id
335 vnffgr_classifier.vnfr_name_ref = nsr_vnfr.name
336 vnffgr_classifier.vnfr_connection_point_ref = vnffgd_classifier.vnfd_connection_point_ref
337
338 if nsr_vnfr.vnfd.service_function_chain == 'CLASSIFIER':
339 vnffgr_classifier.sff_name = nsr_vnfr.name
340
341 vnfr = yield from self._nsr.fetch_vnfr(nsr_vnfr.xpath)
342 self._log.debug(" Received VNFR is %s", vnfr)
343 while vnfr.operational_status != 'running':
344 self._log.info("Received vnf op status is %s; retrying",vnfr.operational_status)
345 if vnfr.operational_status == 'failed':
346 self._log.error("Fetching VNFR for %s failed", vnfr.id)
347 raise NsrInstantiationFailed("Failed NS %s instantiation due to VNFR %s failure" % (self.id, vnfr.id))
348 yield from asyncio.sleep(2, loop=self._loop)
349 vnfr = yield from self._nsr.fetch_vnfr(nsr_vnfr.xpath)
350 self._log.debug("Received VNFR is %s", vnfr)
351
352 for cp in vnfr.connection_point:
353 if cp.name == vnffgr_classifier.vnfr_connection_point_ref:
354 vnffgr_classifier.port_id = cp.connection_point_id
355 vnffgr_classifier.ip_address = cp.ip_address
356 for vdu in vnfr.vdur:
357 for ext_intf in vdu.external_interface:
358 if ext_intf.name == vnffgr_classifier.vnfr_connection_point_ref:
359 vnffgr_classifier.vm_id = vdu.vim_id
360 self._log.debug("VIM ID for CP %s in VNFR %s is %s",cp.name,nsr_vnfr.id,
361 vnfr_cp_ref.connection_point_params.vm_id)
362 break
363
364 self._log.info("VNFFGR msg to be sent is %s", vnffgr)
365 return vnffgr
366
367 @asyncio.coroutine
368 def vnffgr_nsr_sff_list(self):
369 """ SFF List for VNFR """
370 sff_list = {}
371 sf_list = [nsr_vnfr.name for nsr_vnfr in self._nsr.vnfrs.values() if nsr_vnfr.vnfd.service_function_chain == 'SF']
372
373 for nsr_vnfr in self._nsr.vnfrs.values():
374 if (nsr_vnfr.vnfd.service_function_chain == 'CLASSIFIER' or nsr_vnfr.vnfd.service_function_chain == 'SFF'):
375 vnfr = yield from self._nsr.fetch_vnfr(nsr_vnfr.xpath)
376 self._log.debug(" Received VNFR is %s", vnfr)
377 while vnfr.operational_status != 'running':
378 self._log.info("Received vnf op status is %s; retrying",vnfr.operational_status)
379 if vnfr.operational_status == 'failed':
380 self._log.error("Fetching VNFR for %s failed", vnfr.id)
381 raise NsrInstantiationFailed("Failed NS %s instantiation due to VNFR %s failure" % (self.id, vnfr.id))
382 yield from asyncio.sleep(2, loop=self._loop)
383 vnfr = yield from self._nsr.fetch_vnfr(nsr_vnfr.xpath)
384 self._log.debug("Received VNFR is %s", vnfr)
385
386 sff = RwsdnalYang.VNFFGSff()
387 sff_list[nsr_vnfr.vnfd.id] = sff
388 sff.name = nsr_vnfr.name
389 sff.function_type = nsr_vnfr.vnfd.service_function_chain
390
391 sff.mgmt_address = vnfr.mgmt_interface.ip_address
392 sff.mgmt_port = VnffgRecord.SFF_MGMT_PORT
393 for cp in vnfr.connection_point:
394 sff_dp = sff.dp_endpoints.add()
395 sff_dp.name = self._nsr.name + '.' + cp.name
396 sff_dp.address = cp.ip_address
397 sff_dp.port = VnffgRecord.SFF_DP_PORT
398 if nsr_vnfr.vnfd.service_function_chain == 'SFF':
399 for sf_name in sf_list:
400 _sf = sff.vnfr_list.add()
401 _sf.vnfr_name = sf_name
402
403 return sff_list
404
405 @asyncio.coroutine
406 def instantiate(self):
407 """ Instantiate this VNFFG """
408
409 self._log.info("Instaniating VNFFGR with vnffgd %s",
410 self._vnffgd_msg)
411
412
413 vnffgr_request = yield from self.vnffgr_create_msg()
414 vnffg_sff_list = yield from self.vnffgr_nsr_sff_list()
415
416 try:
417 vnffgr = self._vnffgmgr.create_vnffgr(vnffgr_request,self._vnffgd_msg.classifier,vnffg_sff_list)
418 except Exception as e:
419 self._log.exception("VNFFG instantiation failed: %s", str(e))
420 self._vnffgr_state = VnffgRecordState.FAILED
421 raise NsrInstantiationFailed("Failed NS %s instantiation due to VNFFGR %s failure" % (self.id, vnffgr_request.id))
422
423 self._vnffgr_state = VnffgRecordState.INSTANTIATION_PENDING
424
425 self._log.info("Instantiated VNFFGR :%s", vnffgr)
426 self._vnffgr_state = VnffgRecordState.ACTIVE
427
428 self._log.info("Invoking update_state to update NSR state for NSR ID: %s", self._nsr.id)
429 yield from self._nsr.update_state()
430
431 def vnffgr_in_vnffgrm(self):
432 """ Is there a VNFR record in VNFM """
433 if (self._vnffgr_state == VnffgRecordState.ACTIVE or
434 self._vnffgr_state == VnffgRecordState.INSTANTIATION_PENDING or
435 self._vnffgr_state == VnffgRecordState.FAILED):
436 return True
437
438 return False
439
440 @asyncio.coroutine
441 def terminate(self):
442 """ Terminate this VNFFGR """
443 if not self.vnffgr_in_vnffgrm():
444 self._log.error("Ignoring terminate request for id %s in state %s",
445 self.id, self._vnffgr_state)
446 return
447
448 self._log.info("Terminating VNFFGR id:%s", self.id)
449 self._vnffgr_state = VnffgRecordState.TERMINATE_PENDING
450
451 self._vnffgmgr.terminate_vnffgr(self._vnffgr_id)
452
453 self._vnffgr_state = VnffgRecordState.TERMINATED
454 self._log.debug("Terminated VNFFGR id:%s", self.id)
455
456
457 class VirtualLinkRecord(object):
458 """ Virtual Link Records class"""
459 XPATH = "D,/vlr:vlr-catalog/vlr:vlr"
460 @staticmethod
461 @asyncio.coroutine
462 def create_record(dts, log, loop, nsr_name, vld_msg, cloud_account_name, om_datacenter, ip_profile, nsr_id, restart_mode=False):
463 """Creates a new VLR object based on the given data.
464
465 If restart mode is enabled, then we look for existing records in the
466 DTS and create a VLR records using the exiting data(ID)
467
468 Returns:
469 VirtualLinkRecord
470 """
471 vlr_obj = VirtualLinkRecord(
472 dts,
473 log,
474 loop,
475 nsr_name,
476 vld_msg,
477 cloud_account_name,
478 om_datacenter,
479 ip_profile,
480 nsr_id,
481 )
482
483 if restart_mode:
484 res_iter = yield from dts.query_read(
485 "D,/vlr:vlr-catalog/vlr:vlr",
486 rwdts.XactFlag.MERGE)
487
488 for fut in res_iter:
489 response = yield from fut
490 vlr = response.result
491
492 # Check if the record is already present, if so use the ID of
493 # the existing record. Since the name of the record is uniquely
494 # formed we can use it as a search key!
495 if vlr.name == vlr_obj.name:
496 vlr_obj.reset_id(vlr.id)
497 break
498
499 return vlr_obj
500
501 def __init__(self, dts, log, loop, nsr_name, vld_msg, cloud_account_name, om_datacenter, ip_profile, nsr_id):
502 self._dts = dts
503 self._log = log
504 self._loop = loop
505 self._nsr_name = nsr_name
506 self._vld_msg = vld_msg
507 self._cloud_account_name = cloud_account_name
508 self._om_datacenter_name = om_datacenter
509 self._assigned_subnet = None
510 self._nsr_id = nsr_id
511 self._ip_profile = ip_profile
512 self._vlr_id = str(uuid.uuid4())
513 self._state = VlRecordState.INIT
514 self._prev_state = None
515 self._create_time = int(time.time())
516
517 @property
518 def xpath(self):
519 """ path for this object """
520 return "D,/vlr:vlr-catalog/vlr:vlr[vlr:id = '{}']".format(self._vlr_id)
521
522 @property
523 def id(self):
524 """ VLR id """
525 return self._vlr_id
526
527 @property
528 def nsr_name(self):
529 """ Get NSR name for this VL """
530 return self.nsr_name
531
532 @property
533 def vld_msg(self):
534 """ Virtual Link Desciptor """
535 return self._vld_msg
536
537 @property
538 def assigned_subnet(self):
539 """ Subnet assigned to this VL"""
540 return self._assigned_subnet
541
542 @property
543 def name(self):
544 """
545 Get the name for this VLR.
546 VLR name is "nsr name:VLD name"
547 """
548 if self.vld_msg.vim_network_name:
549 return self.vld_msg.vim_network_name
550 elif self.vld_msg.name == "multisite":
551 # This is a temporary hack to identify manually provisioned inter-site network
552 return self.vld_msg.name
553 else:
554 return self._nsr_name + "." + self.vld_msg.name
555
556 @property
557 def cloud_account_name(self):
558 """ Cloud account that this VLR should be created in """
559 return self._cloud_account_name
560
561 @property
562 def om_datacenter_name(self):
563 """ Datacenter that this VLR should be created in """
564 return self._om_datacenter_name
565
566 @staticmethod
567 def vlr_xpath(vlr):
568 """ Get the VLR path from VLR """
569 return (VirtualLinkRecord.XPATH + "[vlr:id = '{}']").format(vlr.id)
570
571 @property
572 def state(self):
573 """ VLR state """
574 return self._state
575
576 @state.setter
577 def state(self, value):
578 """ VLR set state """
579 self._state = value
580
581 @property
582 def prev_state(self):
583 """ VLR previous state """
584 return self._prev_state
585
586 @prev_state.setter
587 def prev_state(self, value):
588 """ VLR set previous state """
589 self._prev_state = value
590
591 @property
592 def vlr_msg(self):
593 """ Virtual Link Record message for Creating VLR in VNS """
594 vld_fields = ["short_name",
595 "vendor",
596 "description",
597 "version",
598 "type_yang",
599 "vim_network_name",
600 "provider_network"]
601
602 vld_copy_dict = {k: v for k, v in self.vld_msg.as_dict().items()
603 if k in vld_fields}
604
605 vlr_dict = {"id": self._vlr_id,
606 "nsr_id_ref": self._nsr_id,
607 "vld_ref": self.vld_msg.id,
608 "name": self.name,
609 "create_time": self._create_time,
610 "cloud_account": self.cloud_account_name,
611 "om_datacenter": self.om_datacenter_name,
612 }
613
614 if self._ip_profile and self._ip_profile.has_field('ip_profile_params'):
615 vlr_dict['ip_profile_params' ] = self._ip_profile.ip_profile_params.as_dict()
616
617 vlr_dict.update(vld_copy_dict)
618 vlr = RwVlrYang.YangData_Vlr_VlrCatalog_Vlr.from_dict(vlr_dict)
619 return vlr
620
621 def reset_id(self, vlr_id):
622 self._vlr_id = vlr_id
623
624 def create_nsr_vlr_msg(self, vnfrs):
625 """ The VLR message"""
626 nsr_vlr = RwNsrYang.YangData_Nsr_NsInstanceOpdata_Nsr_Vlr()
627 nsr_vlr.vlr_ref = self._vlr_id
628 nsr_vlr.assigned_subnet = self.assigned_subnet
629 nsr_vlr.cloud_account = self.cloud_account_name
630 nsr_vlr.om_datacenter = self.om_datacenter_name
631
632 for conn in self.vld_msg.vnfd_connection_point_ref:
633 for vnfr in vnfrs:
634 if (vnfr.vnfd.id == conn.vnfd_id_ref and
635 vnfr.member_vnf_index == conn.member_vnf_index_ref and
636 self.cloud_account_name == vnfr.cloud_account_name and
637 self.om_datacenter_name == vnfr.om_datacenter_name):
638 cp_entry = nsr_vlr.vnfr_connection_point_ref.add()
639 cp_entry.vnfr_id = vnfr.id
640 cp_entry.connection_point = conn.vnfd_connection_point_ref
641
642 return nsr_vlr
643
644 @asyncio.coroutine
645 def instantiate(self):
646 """ Instantiate this VL """
647 self._log.debug("Instaniating VLR key %s, vld %s",
648 self.xpath, self._vld_msg)
649 vlr = None
650 self._state = VlRecordState.INSTANTIATION_PENDING
651 self._log.debug("Executing VL create path:%s msg:%s",
652 self.xpath, self.vlr_msg)
653
654 with self._dts.transaction(flags=0) as xact:
655 block = xact.block_create()
656 block.add_query_create(self.xpath, self.vlr_msg)
657 self._log.debug("Executing VL create path:%s msg:%s",
658 self.xpath, self.vlr_msg)
659 res_iter = yield from block.execute(now=True)
660 for ent in res_iter:
661 res = yield from ent
662 vlr = res.result
663
664 if vlr is None:
665 self._state = VlRecordState.FAILED
666 raise NsrInstantiationFailed("Failed NS %s instantiation due to empty response" % self.id)
667
668 if vlr.operational_status == 'failed':
669 self._log.debug("NS Id:%s VL creation failed for vlr id %s", self.id, vlr.id)
670 self._state = VlRecordState.FAILED
671 raise NsrInstantiationFailed("Failed VL %s instantiation (%s)" % (vlr.id, vlr.operational_status_details))
672
673 self._log.info("Instantiated VL with xpath %s and vlr:%s",
674 self.xpath, vlr)
675 self._state = VlRecordState.ACTIVE
676 self._assigned_subnet = vlr.assigned_subnet
677
678 def vlr_in_vns(self):
679 """ Is there a VLR record in VNS """
680 if (self._state == VlRecordState.ACTIVE or
681 self._state == VlRecordState.INSTANTIATION_PENDING or
682 self._state == VlRecordState.TERMINATE_PENDING or
683 self._state == VlRecordState.FAILED):
684 return True
685
686 return False
687
688 @asyncio.coroutine
689 def terminate(self):
690 """ Terminate this VL """
691 if not self.vlr_in_vns():
692 self._log.debug("Ignoring terminate request for id %s in state %s",
693 self.id, self._state)
694 return
695
696 self._log.debug("Terminating VL id:%s", self.id)
697 self._state = VlRecordState.TERMINATE_PENDING
698
699 with self._dts.transaction(flags=0) as xact:
700 block = xact.block_create()
701 block.add_query_delete(self.xpath)
702 yield from block.execute(flags=0, now=True)
703
704 self._state = VlRecordState.TERMINATED
705 self._log.debug("Terminated VL id:%s", self.id)
706
707
708 class VnfRecordState(Enum):
709 """ Vnf Record State """
710 INIT = 101
711 INSTANTIATION_PENDING = 102
712 ACTIVE = 103
713 TERMINATE_PENDING = 104
714 TERMINATED = 105
715 FAILED = 106
716
717
718 class VirtualNetworkFunctionRecord(object):
719 """ Virtual Network Function Record class"""
720 XPATH = "D,/vnfr:vnfr-catalog/vnfr:vnfr"
721
722 @staticmethod
723 @asyncio.coroutine
724 def create_record(dts, log, loop, vnfd, const_vnfd_msg, nsd_id, nsr_name,
725 cloud_account_name, om_datacenter_name, nsr_id, group_name, group_instance_id,
726 placement_groups, restart_mode=False):
727 """Creates a new VNFR object based on the given data.
728
729 If restart mode is enabled, then we look for existing records in the
730 DTS and create a VNFR records using the exiting data(ID)
731
732 Returns:
733 VirtualNetworkFunctionRecord
734 """
735 vnfr_obj = VirtualNetworkFunctionRecord(
736 dts,
737 log,
738 loop,
739 vnfd,
740 const_vnfd_msg,
741 nsd_id,
742 nsr_name,
743 cloud_account_name,
744 om_datacenter_name,
745 nsr_id,
746 group_name,
747 group_instance_id,
748 placement_groups,
749 restart_mode=restart_mode)
750
751 if restart_mode:
752 res_iter = yield from dts.query_read(
753 "D,/vnfr:vnfr-catalog/vnfr:vnfr",
754 rwdts.XactFlag.MERGE)
755
756 for fut in res_iter:
757 response = yield from fut
758 vnfr = response.result
759
760 if vnfr.name == vnfr_obj.name:
761 vnfr_obj.reset_id(vnfr.id)
762 break
763
764 return vnfr_obj
765
766 def __init__(self,
767 dts,
768 log,
769 loop,
770 vnfd,
771 const_vnfd_msg,
772 nsd_id,
773 nsr_name,
774 cloud_account_name,
775 om_datacenter_name,
776 nsr_id,
777 group_name=None,
778 group_instance_id=None,
779 placement_groups = [],
780 restart_mode = False):
781 self._dts = dts
782 self._log = log
783 self._loop = loop
784 self._vnfd = vnfd
785 self._const_vnfd_msg = const_vnfd_msg
786 self._nsd_id = nsd_id
787 self._nsr_name = nsr_name
788 self._nsr_id = nsr_id
789 self._cloud_account_name = cloud_account_name
790 self._om_datacenter_name = om_datacenter_name
791 self._group_name = group_name
792 self._group_instance_id = group_instance_id
793 self._placement_groups = placement_groups
794 self._config_status = NsrYang.ConfigStates.INIT
795 self._create_time = int(time.time())
796
797 self._prev_state = VnfRecordState.INIT
798 self._state = VnfRecordState.INIT
799 self._state_failed_reason = None
800
801 self.config_store = rift.mano.config_data.config.ConfigStore(self._log)
802 self.configure()
803
804 self._vnfr_id = str(uuid.uuid4())
805 self._name = None
806 self._vnfr_msg = self.create_vnfr_msg()
807 self._log.debug("Set VNFR {} config type to {}".
808 format(self.name, self.config_type))
809 self.restart_mode = restart_mode
810
811
812 if group_name is None and group_instance_id is not None:
813 raise ValueError("Group instance id must not be provided with an empty group name")
814
815 @property
816 def id(self):
817 """ VNFR id """
818 return self._vnfr_id
819
820 @property
821 def xpath(self):
822 """ VNFR xpath """
823 return "D,/vnfr:vnfr-catalog/vnfr:vnfr[vnfr:id = '{}']".format(self.id)
824
825 @property
826 def vnfr_msg(self):
827 """ VNFR message """
828 return self._vnfr_msg
829
830 @property
831 def const_vnfr_msg(self):
832 """ VNFR message """
833 return RwNsrYang.YangData_Nsr_NsInstanceOpdata_Nsr_ConstituentVnfrRef(vnfr_id=self.id,cloud_account=self.cloud_account_name,om_datacenter=self._om_datacenter_name)
834
835 @property
836 def vnfd(self):
837 """ vnfd """
838 return self._vnfd
839
840 @property
841 def cloud_account_name(self):
842 """ Cloud account that this VNF should be created in """
843 return self._cloud_account_name
844
845 @property
846 def om_datacenter_name(self):
847 """ Datacenter that this VNF should be created in """
848 return self._om_datacenter_name
849
850
851 @property
852 def active(self):
853 """ Is this VNF actve """
854 return True if self._state == VnfRecordState.ACTIVE else False
855
856 @property
857 def state(self):
858 """ state of this VNF """
859 return self._state
860
861 @property
862 def state_failed_reason(self):
863 """ Error message in case this VNF is in failed state """
864 return self._state_failed_reason
865
866 @property
867 def member_vnf_index(self):
868 """ Member VNF index """
869 return self._const_vnfd_msg.member_vnf_index
870
871 @property
872 def nsr_name(self):
873 """ NSR name"""
874 return self._nsr_name
875
876 @property
877 def name(self):
878 """ Name of this VNFR """
879 if self._name is not None:
880 return self._name
881
882 name_tags = [self._nsr_name]
883
884 if self._group_name is not None:
885 name_tags.append(self._group_name)
886
887 if self._group_instance_id is not None:
888 name_tags.append(str(self._group_instance_id))
889
890 name_tags.extend([self.vnfd.name, str(self.member_vnf_index)])
891
892 self._name = "__".join(name_tags)
893
894 return self._name
895
896 @staticmethod
897 def vnfr_xpath(vnfr):
898 """ Get the VNFR path from VNFR """
899 return (VirtualNetworkFunctionRecord.XPATH + "[vnfr:id = '{}']").format(vnfr.id)
900
901 @property
902 def config_type(self):
903 cfg_types = ['netconf', 'juju', 'script']
904 for method in cfg_types:
905 if self._vnfd.vnf_configuration.has_field(method):
906 return method
907 return 'none'
908
909 @property
910 def config_status(self):
911 """Return the config status as YANG ENUM string"""
912 self._log.debug("Map VNFR {} config status {} ({})".
913 format(self.name, self._config_status, self.config_type))
914 if self.config_type == 'none':
915 return 'config_not_needed'
916 elif self._config_status == NsrYang.ConfigStates.CONFIGURED:
917 return 'configured'
918 elif self._config_status == NsrYang.ConfigStates.FAILED:
919 return 'failed'
920
921 return 'configuring'
922
923 def set_state(self, state):
924 """ set the state of this object """
925 self._prev_state = self._state
926 self._state = state
927
928 def reset_id(self, vnfr_id):
929 self._vnfr_id = vnfr_id
930 self._vnfr_msg = self.create_vnfr_msg()
931
932 def configure(self):
933 self.config_store.merge_vnfd_config(
934 self._nsd_id,
935 self._vnfd,
936 self.member_vnf_index,
937 )
938
939 def create_vnfr_msg(self):
940 """ VNFR message for this VNFR """
941 vnfd_fields = [
942 "short_name",
943 "vendor",
944 "description",
945 "version",
946 "type_yang",
947 ]
948 vnfd_copy_dict = {k: v for k, v in self._vnfd.as_dict().items() if k in vnfd_fields}
949 vnfr_dict = {
950 "id": self.id,
951 "nsr_id_ref": self._nsr_id,
952 "name": self.name,
953 "cloud_account": self._cloud_account_name,
954 "om_datacenter": self._om_datacenter_name,
955 "config_status": self.config_status
956 }
957 vnfr_dict.update(vnfd_copy_dict)
958
959 vnfr = RwVnfrYang.YangData_Vnfr_VnfrCatalog_Vnfr.from_dict(vnfr_dict)
960 vnfr.vnfd = VnfrYang.YangData_Vnfr_VnfrCatalog_Vnfr_Vnfd.from_dict(self.vnfd.as_dict(),
961 ignore_missing_keys=True)
962 vnfr.member_vnf_index_ref = self.member_vnf_index
963 vnfr.vnf_configuration.from_dict(self._vnfd.vnf_configuration.as_dict())
964
965 if self._vnfd.mgmt_interface.has_field("port"):
966 vnfr.mgmt_interface.port = self._vnfd.mgmt_interface.port
967
968 for group_info in self._placement_groups:
969 group = vnfr.placement_groups_info.add()
970 group.from_dict(group_info.as_dict())
971
972 # UI expects the monitoring param field to exist
973 vnfr.monitoring_param = []
974
975 self._log.debug("Get vnfr_msg for VNFR {} : {}".format(self.name, vnfr))
976 return vnfr
977
978 @asyncio.coroutine
979 def update_vnfm(self):
980 self._log.debug("Send an update to VNFM for VNFR {} with {}".
981 format(self.name, self.vnfr_msg))
982 yield from self._dts.query_update(
983 self.xpath,
984 rwdts.XactFlag.TRACE,
985 self.vnfr_msg
986 )
987
988 def get_config_status(self):
989 """Return the config status as YANG ENUM"""
990 return self._config_status
991
992 @asyncio.coroutine
993 def set_config_status(self, status):
994
995 def status_to_string(status):
996 status_dc = {
997 NsrYang.ConfigStates.INIT : 'init',
998 NsrYang.ConfigStates.CONFIGURING : 'configuring',
999 NsrYang.ConfigStates.CONFIG_NOT_NEEDED : 'config_not_needed',
1000 NsrYang.ConfigStates.CONFIGURED : 'configured',
1001 NsrYang.ConfigStates.FAILED : 'failed',
1002 }
1003
1004 return status_dc[status]
1005
1006 self._log.debug("Update VNFR {} from {} ({}) to {}".
1007 format(self.name, self._config_status,
1008 self.config_type, status))
1009 if self._config_status == NsrYang.ConfigStates.CONFIGURED:
1010 self._log.error("Updating already configured VNFR {}".
1011 format(self.name))
1012 return
1013
1014 if self._config_status != status:
1015 try:
1016 self._config_status = status
1017 # I don't think this is used. Original implementor can check.
1018 # Caused Exception, so corrected it by status_to_string
1019 # But not sure whats the use of this variable?
1020 self.vnfr_msg.config_status = status_to_string(status)
1021 except Exception as e:
1022 self._log.error("Exception=%s", str(e))
1023 pass
1024
1025 self._log.debug("Updated VNFR {} status to {}".format(self.name, status))
1026
1027 if self._config_status != NsrYang.ConfigStates.INIT:
1028 try:
1029 # Publish only after VNFM has the VNFR created
1030 yield from self.update_vnfm()
1031 except Exception as e:
1032 self._log.error("Exception updating VNFM with new status {} of VNFR {}: {}".
1033 format(status, self.name, e))
1034 self._log.exception(e)
1035
1036 def is_configured(self):
1037 if self.config_type == 'none':
1038 return True
1039
1040 if self._config_status == NsrYang.ConfigStates.CONFIGURED:
1041 return True
1042
1043 return False
1044
1045 @asyncio.coroutine
1046 def instantiate(self, nsr):
1047 """ Instantiate this VNFR"""
1048
1049 self._log.debug("Instaniating VNFR key %s, vnfd %s",
1050 self.xpath, self._vnfd)
1051
1052 self._log.debug("Create VNF with xpath %s and vnfr %s",
1053 self.xpath, self.vnfr_msg)
1054
1055 self.set_state(VnfRecordState.INSTANTIATION_PENDING)
1056
1057 def find_vlr_for_cp(conn):
1058 """ Find VLR for the given connection point """
1059 for vlr in nsr.vlrs:
1060 for vnfd_cp in vlr.vld_msg.vnfd_connection_point_ref:
1061 if (vnfd_cp.vnfd_id_ref == self._vnfd.id and
1062 vnfd_cp.vnfd_connection_point_ref == conn.name and
1063 vnfd_cp.member_vnf_index_ref == self.member_vnf_index and
1064 vlr.cloud_account_name == self.cloud_account_name):
1065 self._log.debug("Found VLR for cp_name:%s and vnf-index:%d",
1066 conn.name, self.member_vnf_index)
1067 return vlr
1068 return None
1069
1070 # For every connection point in the VNFD fill in the identifier
1071 for conn_p in self._vnfd.connection_point:
1072 cpr = VnfrYang.YangData_Vnfr_VnfrCatalog_Vnfr_ConnectionPoint()
1073 cpr.name = conn_p.name
1074 cpr.type_yang = conn_p.type_yang
1075 if conn_p.has_field('port_security_enabled'):
1076 cpr.port_security_enabled = conn_p.port_security_enabled
1077
1078 vlr_ref = find_vlr_for_cp(conn_p)
1079 if vlr_ref is None:
1080 msg = "Failed to find VLR for cp = %s" % conn_p.name
1081 self._log.debug("%s", msg)
1082 # raise VirtualNetworkFunctionRecordError(msg)
1083 continue
1084
1085 cpr.vlr_ref = vlr_ref.id
1086 self.vnfr_msg.connection_point.append(cpr)
1087 self._log.debug("Connection point [%s] added, vnf id=%s vnfd id=%s",
1088 cpr, self.vnfr_msg.id, self.vnfr_msg.vnfd.id)
1089
1090 if not self.restart_mode:
1091 yield from self._dts.query_create(self.xpath,
1092 0, # this is sub
1093 self.vnfr_msg)
1094 else:
1095 yield from self._dts.query_update(self.xpath,
1096 0,
1097 self.vnfr_msg)
1098
1099 self._log.info("Created VNF with xpath %s and vnfr %s",
1100 self.xpath, self.vnfr_msg)
1101
1102 self._log.info("Instantiated VNFR with xpath %s and vnfd %s, vnfr %s",
1103 self.xpath, self._vnfd, self.vnfr_msg)
1104
1105 @asyncio.coroutine
1106 def update_state(self, vnfr_msg):
1107 """ Update this VNFR"""
1108 if vnfr_msg.operational_status == "running":
1109 if self.vnfr_msg.operational_status != "running":
1110 yield from self.is_active()
1111 elif vnfr_msg.operational_status == "failed":
1112 yield from self.instantiation_failed(failed_reason=vnfr_msg.operational_status_details)
1113
1114 @asyncio.coroutine
1115 def is_active(self):
1116 """ This VNFR is active """
1117 self._log.debug("VNFR %s is active", self._vnfr_id)
1118 self.set_state(VnfRecordState.ACTIVE)
1119
1120 @asyncio.coroutine
1121 def instantiation_failed(self, failed_reason=None):
1122 """ This VNFR instantiation failed"""
1123 self._log.error("VNFR %s instantiation failed", self._vnfr_id)
1124 self.set_state(VnfRecordState.FAILED)
1125 self._state_failed_reason = failed_reason
1126
1127 def vnfr_in_vnfm(self):
1128 """ Is there a VNFR record in VNFM """
1129 if (self._state == VnfRecordState.ACTIVE or
1130 self._state == VnfRecordState.INSTANTIATION_PENDING or
1131 self._state == VnfRecordState.FAILED):
1132 return True
1133
1134 return False
1135
1136 @asyncio.coroutine
1137 def terminate(self):
1138 """ Terminate this VNF """
1139 if not self.vnfr_in_vnfm():
1140 self._log.debug("Ignoring terminate request for id %s in state %s",
1141 self.id, self._state)
1142 return
1143
1144 self._log.debug("Terminating VNF id:%s", self.id)
1145 self.set_state(VnfRecordState.TERMINATE_PENDING)
1146 with self._dts.transaction(flags=0) as xact:
1147 block = xact.block_create()
1148 block.add_query_delete(self.xpath)
1149 yield from block.execute(flags=0)
1150 self.set_state(VnfRecordState.TERMINATED)
1151 self._log.debug("Terminated VNF id:%s", self.id)
1152
1153
1154 class NetworkServiceStatus(object):
1155 """ A class representing the Network service's status """
1156 MAX_EVENTS_RECORDED = 10
1157 """ Network service Status class"""
1158 def __init__(self, dts, log, loop):
1159 self._dts = dts
1160 self._log = log
1161 self._loop = loop
1162
1163 self._state = NetworkServiceRecordState.INIT
1164 self._events = deque([])
1165
1166 @asyncio.coroutine
1167 def create_notification(self, evt, evt_desc, evt_details):
1168 xp = "N,/rw-nsr:nsm-notification"
1169 notif = RwNsrYang.YangNotif_RwNsr_NsmNotification()
1170 notif.event = evt
1171 notif.description = evt_desc
1172 notif.details = evt_details if evt_details is not None else None
1173
1174 yield from self._dts.query_create(xp, rwdts.XactFlag.ADVISE, notif)
1175 self._log.info("Notification called by creating dts query: %s", notif)
1176
1177 def record_event(self, evt, evt_desc, evt_details):
1178 """ Record an event """
1179 self._log.debug("Recording event - evt %s, evt_descr %s len = %s",
1180 evt, evt_desc, len(self._events))
1181 if len(self._events) >= NetworkServiceStatus.MAX_EVENTS_RECORDED:
1182 self._events.popleft()
1183 self._events.append((int(time.time()), evt, evt_desc,
1184 evt_details if evt_details is not None else None))
1185
1186 self._loop.create_task(self.create_notification(evt,evt_desc,evt_details))
1187
1188 def set_state(self, state):
1189 """ set the state of this status object """
1190 self._state = state
1191
1192 def yang_str(self):
1193 """ Return the state as a yang enum string """
1194 state_to_str_map = {"INIT": "init",
1195 "VL_INIT_PHASE": "vl_init_phase",
1196 "VNF_INIT_PHASE": "vnf_init_phase",
1197 "VNFFG_INIT_PHASE": "vnffg_init_phase",
1198 "SCALING_GROUP_INIT_PHASE": "scaling_group_init_phase",
1199 "RUNNING": "running",
1200 "SCALING_OUT": "scaling_out",
1201 "SCALING_IN": "scaling_in",
1202 "TERMINATE_RCVD": "terminate_rcvd",
1203 "TERMINATE": "terminate",
1204 "VL_TERMINATE_PHASE": "vl_terminate_phase",
1205 "VNF_TERMINATE_PHASE": "vnf_terminate_phase",
1206 "VNFFG_TERMINATE_PHASE": "vnffg_terminate_phase",
1207 "TERMINATED": "terminated",
1208 "FAILED": "failed",
1209 "VL_INSTANTIATE": "vl_instantiate",
1210 "VL_TERMINATE": "vl_terminate",
1211 }
1212 return state_to_str_map[self._state.name]
1213
1214 @property
1215 def state(self):
1216 """ State of this status object """
1217 return self._state
1218
1219 @property
1220 def msg(self):
1221 """ Network Service Record as a message"""
1222 event_list = []
1223 idx = 1
1224 for entry in self._events:
1225 event = RwNsrYang.YangData_Nsr_NsInstanceOpdata_Nsr_OperationalEvents()
1226 event.id = idx
1227 idx += 1
1228 event.timestamp, event.event, event.description, event.details = entry
1229 event_list.append(event)
1230 return event_list
1231
1232
1233 class NetworkServiceRecord(object):
1234 """ Network service record """
1235 XPATH = "D,/nsr:ns-instance-opdata/nsr:nsr"
1236
1237 def __init__(self, dts, log, loop, nsm, nsm_plugin, nsr_cfg_msg, sdn_account_name, key_pairs, restart_mode=False,
1238 vlr_handler=None):
1239 self._dts = dts
1240 self._log = log
1241 self._loop = loop
1242 self._nsm = nsm
1243 self._nsr_cfg_msg = nsr_cfg_msg
1244 self._nsm_plugin = nsm_plugin
1245 self._sdn_account_name = sdn_account_name
1246 self._vlr_handler = vlr_handler
1247
1248 self._nsd = None
1249 self._nsr_msg = None
1250 self._nsr_regh = None
1251 self._key_pairs = key_pairs
1252 self._vlrs = []
1253 self._vnfrs = {}
1254 self._vnfds = {}
1255 self._vnffgrs = {}
1256 self._param_pools = {}
1257 self._scaling_groups = {}
1258 self._create_time = int(time.time())
1259 self._op_status = NetworkServiceStatus(dts, log, loop)
1260 self._config_status = NsrYang.ConfigStates.CONFIGURING
1261 self._config_status_details = None
1262 self._job_id = 0
1263 self.restart_mode = restart_mode
1264 self.config_store = rift.mano.config_data.config.ConfigStore(self._log)
1265 self._debug_running = False
1266 self._is_active = False
1267 self._vl_phase_completed = False
1268 self._vnf_phase_completed = False
1269 self.vlr_uptime_tasks = {}
1270
1271
1272 # Initalise the state to init
1273 # The NSR moves through the following transitions
1274 # 1. INIT -> VLS_READY once all the VLs in the NSD are created
1275 # 2. VLS_READY - VNFS_READY when all the VNFs in the NSD are created
1276 # 3. VNFS_READY - READY when the NSR is published
1277
1278 self.set_state(NetworkServiceRecordState.INIT)
1279
1280 self.substitute_input_parameters = InputParameterSubstitution(self._log)
1281
1282 @property
1283 def nsm_plugin(self):
1284 """ NSM Plugin """
1285 return self._nsm_plugin
1286
1287 def set_state(self, state):
1288 """ Set state for this NSR"""
1289 self._log.debug("Setting state to %s", state)
1290 # We are in init phase and is moving to the next state
1291 # The new state could be a FAILED state or VNF_INIIT_PHASE
1292 if self.state == NetworkServiceRecordState.VL_INIT_PHASE:
1293 self._vl_phase_completed = True
1294
1295 if self.state == NetworkServiceRecordState.VNF_INIT_PHASE:
1296 self._vnf_phase_completed = True
1297
1298 self._op_status.set_state(state)
1299 self._nsm_plugin.set_state(self.id, state)
1300
1301 @property
1302 def id(self):
1303 """ Get id for this NSR"""
1304 return self._nsr_cfg_msg.id
1305
1306 @property
1307 def name(self):
1308 """ Name of this network service record """
1309 return self._nsr_cfg_msg.name
1310
1311 @property
1312 def cloud_account_name(self):
1313 return self._nsr_cfg_msg.cloud_account
1314
1315 @property
1316 def om_datacenter_name(self):
1317 if self._nsr_cfg_msg.has_field('om_datacenter'):
1318 return self._nsr_cfg_msg.om_datacenter
1319 return None
1320
1321 @property
1322 def state(self):
1323 """State of this NetworkServiceRecord"""
1324 return self._op_status.state
1325
1326 @property
1327 def active(self):
1328 """ Is this NSR active ?"""
1329 return True if self._op_status.state == NetworkServiceRecordState.RUNNING else False
1330
1331 @property
1332 def vlrs(self):
1333 """ VLRs associated with this NSR"""
1334 return self._vlrs
1335
1336 @property
1337 def vnfrs(self):
1338 """ VNFRs associated with this NSR"""
1339 return self._vnfrs
1340
1341 @property
1342 def vnffgrs(self):
1343 """ VNFFGRs associated with this NSR"""
1344 return self._vnffgrs
1345
1346 @property
1347 def scaling_groups(self):
1348 """ Scaling groups associated with this NSR """
1349 return self._scaling_groups
1350
1351 @property
1352 def param_pools(self):
1353 """ Parameter value pools associated with this NSR"""
1354 return self._param_pools
1355
1356 @property
1357 def nsr_cfg_msg(self):
1358 return self._nsr_cfg_msg
1359
1360 @nsr_cfg_msg.setter
1361 def nsr_cfg_msg(self, msg):
1362 self._nsr_cfg_msg = msg
1363
1364 @property
1365 def nsd_msg(self):
1366 """ NSD Protobuf for this NSR """
1367 if self._nsd is not None:
1368 return self._nsd
1369 self._nsd = self._nsr_cfg_msg.nsd
1370 return self._nsd
1371
1372 @property
1373 def nsd_id(self):
1374 """ NSD ID for this NSR """
1375 return self.nsd_msg.id
1376
1377 @property
1378 def job_id(self):
1379 ''' Get a new job id for config primitive'''
1380 self._job_id += 1
1381 return self._job_id
1382
1383 @property
1384 def config_status(self):
1385 """ Config status for NSR """
1386 return self._config_status
1387
1388 def resolve_placement_group_cloud_construct(self, input_group):
1389 """
1390 Returns the cloud specific construct for placement group
1391 """
1392 copy_dict = ['name', 'requirement', 'strategy']
1393
1394 for group_info in self._nsr_cfg_msg.nsd_placement_group_maps:
1395 if group_info.placement_group_ref == input_group.name:
1396 group = VnfrYang.YangData_Vnfr_VnfrCatalog_Vnfr_PlacementGroupsInfo()
1397 group_dict = {k:v for k,v in
1398 group_info.as_dict().items() if k != 'placement_group_ref'}
1399 for param in copy_dict:
1400 group_dict.update({param: getattr(input_group, param)})
1401 group.from_dict(group_dict)
1402 return group
1403 return None
1404
1405
1406 def __str__(self):
1407 return "NSR(name={}, nsd_id={}, cloud_account={})".format(
1408 self.name, self.nsd_id, self.cloud_account_name
1409 )
1410
1411 def _get_vnfd(self, vnfd_id, config_xact):
1412 """ Fetch vnfd msg for the passed vnfd id """
1413 return self._nsm.get_vnfd(vnfd_id, config_xact)
1414
1415 def _get_vnfd_cloud_account(self, vnfd_member_index):
1416 """ Fetch Cloud Account for the passed vnfd id """
1417 if self._nsr_cfg_msg.vnf_cloud_account_map:
1418 vim_accounts = [(vnf.cloud_account,vnf.om_datacenter) for vnf in self._nsr_cfg_msg.vnf_cloud_account_map \
1419 if vnfd_member_index == vnf.member_vnf_index_ref]
1420 if vim_accounts and vim_accounts[0]:
1421 return vim_accounts[0]
1422 return (self.cloud_account_name,self.om_datacenter_name)
1423
1424 def _get_constituent_vnfd_msg(self, vnf_index):
1425 for const_vnfd in self.nsd_msg.constituent_vnfd:
1426 if const_vnfd.member_vnf_index == vnf_index:
1427 return const_vnfd
1428
1429 raise ValueError("Constituent VNF index %s not found" % vnf_index)
1430
1431 def record_event(self, evt, evt_desc, evt_details=None, state=None):
1432 """ Record an event """
1433 self._op_status.record_event(evt, evt_desc, evt_details)
1434 if state is not None:
1435 self.set_state(state)
1436
1437 def scaling_trigger_str(self, trigger):
1438 SCALING_TRIGGER_STRS = {
1439 NsdYang.ScalingTrigger.PRE_SCALE_IN : 'pre-scale-in',
1440 NsdYang.ScalingTrigger.POST_SCALE_IN : 'post-scale-in',
1441 NsdYang.ScalingTrigger.PRE_SCALE_OUT : 'pre-scale-out',
1442 NsdYang.ScalingTrigger.POST_SCALE_OUT : 'post-scale-out',
1443 }
1444 try:
1445 return SCALING_TRIGGER_STRS[trigger]
1446 except Exception as e:
1447 self._log.error("Scaling trigger mapping error for {} : {}".
1448 format(trigger, e))
1449 self._log.exception(e)
1450 return "Unknown trigger"
1451
1452 @asyncio.coroutine
1453 def instantiate_vls(self):
1454 """
1455 This function instantiates VLs for every VL in this Network Service
1456 """
1457 self._log.debug("Instantiating %d VLs in NSD id %s", len(self._vlrs),
1458 self.id)
1459 for vlr in self._vlrs:
1460 yield from self.nsm_plugin.instantiate_vl(self, vlr)
1461 vlr.state = VlRecordState.ACTIVE
1462 self.vlr_uptime_tasks[vlr.id] = self._loop.create_task(self.vlr_uptime_update(vlr))
1463
1464
1465 def vlr_uptime_update(self, vlr):
1466 try:
1467
1468 vlr_ = RwVlrYang.YangData_Vlr_VlrCatalog_Vlr.from_dict({'id': vlr.id})
1469 while True:
1470 vlr_.uptime = int(time.time()) - vlr._create_time
1471 yield from self._vlr_handler.update(None, VirtualLinkRecord.vlr_xpath(vlr), vlr_)
1472 yield from asyncio.sleep(2, loop=self._loop)
1473 except asyncio.CancelledError:
1474 self._log.debug("Received cancellation request for vlr_uptime_update task")
1475 yield from self._vlr_handler.delete(None, VirtualLinkRecord.vlr_xpath(vlr))
1476
1477
1478 @asyncio.coroutine
1479 def create(self, config_xact):
1480 """ Create this network service"""
1481 # Create virtual links for all the external vnf
1482 # connection points in this NS
1483 yield from self.create_vls()
1484
1485 # Create VNFs in this network service
1486 yield from self.create_vnfs(config_xact)
1487
1488 # Create VNFFG for network service
1489 self.create_vnffgs()
1490
1491 # Create Scaling Groups for each scaling group in NSD
1492 self.create_scaling_groups()
1493
1494 # Create Parameter Pools
1495 self.create_param_pools()
1496
1497 @asyncio.coroutine
1498 def apply_scale_group_config_script(self, script, group, scale_instance, trigger, vnfrs=None):
1499 """ Apply config based on script for scale group """
1500
1501 @asyncio.coroutine
1502 def add_vnfrs_data(vnfrs_list):
1503 """ Add as a dict each of the VNFRs data """
1504 vnfrs_data = []
1505 for vnfr in vnfrs_list:
1506 self._log.debug("Add VNFR {} data".format(vnfr))
1507 vnfr_data = dict()
1508 vnfr_data['name'] = vnfr.name
1509 if trigger in [NsdYang.ScalingTrigger.PRE_SCALE_IN, NsdYang.ScalingTrigger.POST_SCALE_OUT]:
1510 # Get VNF management and other IPs, etc
1511 opdata = yield from self.fetch_vnfr(vnfr.xpath)
1512 self._log.debug("VNFR {} op data: {}".format(vnfr.name, opdata))
1513 try:
1514 vnfr_data['rw_mgmt_ip'] = opdata.mgmt_interface.ip_address
1515 vnfr_data['rw_mgmt_port'] = opdata.mgmt_interface.port
1516 except Exception as e:
1517 self._log.error("Unable to get management IP for vnfr {}:{}".
1518 format(vnfr.name, e))
1519
1520 try:
1521 vnfr_data['connection_points'] = []
1522 for cp in opdata.connection_point:
1523 con_pt = dict()
1524 con_pt['name'] = cp.name
1525 con_pt['ip_address'] = cp.ip_address
1526 vnfr_data['connection_points'].append(con_pt)
1527 except Exception as e:
1528 self._log.error("Exception getting connections points for VNFR {}: {}".
1529 format(vnfr.name, e))
1530
1531 vnfrs_data.append(vnfr_data)
1532 self._log.debug("VNFRs data: {}".format(vnfrs_data))
1533
1534 return vnfrs_data
1535
1536 def add_nsr_data(nsr):
1537 nsr_data = dict()
1538 nsr_data['name'] = nsr.name
1539 return nsr_data
1540
1541 if script is None or len(script) == 0:
1542 self._log.error("Script not provided for scale group config: {}".format(group.name))
1543 return False
1544
1545 if script[0] == '/':
1546 path = script
1547 else:
1548 path = os.path.join(os.environ['RIFT_INSTALL'], "usr/bin", script)
1549 if not os.path.exists(path):
1550 self._log.error("Config faled for scale group {}: Script does not exist at {}".
1551 format(group.name, path))
1552 return False
1553
1554 # Build a YAML file with all parameters for the script to execute
1555 # The data consists of 5 sections
1556 # 1. Trigger
1557 # 2. Scale group config
1558 # 3. VNFRs in the scale group
1559 # 4. VNFRs outside scale group
1560 # 5. NSR data
1561 data = dict()
1562 data['trigger'] = group.trigger_map(trigger)
1563 data['config'] = group.group_msg.as_dict()
1564
1565 if vnfrs:
1566 data["vnfrs_in_group"] = yield from add_vnfrs_data(vnfrs)
1567 else:
1568 data["vnfrs_in_group"] = yield from add_vnfrs_data(scale_instance.vnfrs)
1569
1570 data["vnfrs_others"] = yield from add_vnfrs_data(self.vnfrs.values())
1571 data["nsr"] = add_nsr_data(self)
1572
1573 tmp_file = None
1574 with tempfile.NamedTemporaryFile(delete=False) as tmp_file:
1575 tmp_file.write(yaml.dump(data, default_flow_style=True)
1576 .encode("UTF-8"))
1577
1578 self._log.debug("Creating a temp file: {} with input data: {}".
1579 format(tmp_file.name, data))
1580
1581 cmd = "{} {}".format(path, tmp_file.name)
1582 self._log.debug("Running the CMD: {}".format(cmd))
1583 proc = yield from asyncio.create_subprocess_shell(cmd, loop=self._loop)
1584 rc = yield from proc.wait()
1585 if rc:
1586 self._log.error("The script {} for scale group {} config returned: {}".
1587 format(script, group.name, rc))
1588 return False
1589
1590 # Success
1591 return True
1592
1593
1594 @asyncio.coroutine
1595 def apply_scaling_group_config(self, trigger, group, scale_instance, vnfrs=None):
1596 """ Apply the config for the scaling group based on trigger """
1597 if group is None or scale_instance is None:
1598 return False
1599
1600 @asyncio.coroutine
1601 def update_config_status(success=True, err_msg=None):
1602 self._log.debug("Update %s config status to %r : %s",
1603 scale_instance, success, err_msg)
1604 if (scale_instance.config_status == "failed"):
1605 # Do not update the config status if it is already in failed state
1606 return
1607
1608 if scale_instance.config_status == "configured":
1609 # Update only to failed state an already configured scale instance
1610 if not success:
1611 scale_instance.config_status = "failed"
1612 scale_instance.config_err_msg = err_msg
1613 yield from self.update_state()
1614 else:
1615 # We are in configuring state
1616 # Only after post scale out mark instance as configured
1617 if trigger == NsdYang.ScalingTrigger.POST_SCALE_OUT:
1618 if success:
1619 scale_instance.config_status = "configured"
1620 else:
1621 scale_instance.config_status = "failed"
1622 scale_instance.config_err_msg = err_msg
1623 yield from self.update_state()
1624
1625 config = group.trigger_config(trigger)
1626 if config is None:
1627 return True
1628
1629 self._log.debug("Scaling group {} config: {}".format(group.name, config))
1630 if config.has_field("ns_config_primitive_name_ref"):
1631 config_name = config.ns_config_primitive_name_ref
1632 nsd_msg = self.nsd_msg
1633 config_primitive = None
1634 for ns_cfg_prim in nsd_msg.service_primitive:
1635 if ns_cfg_prim.name == config_name:
1636 config_primitive = ns_cfg_prim
1637 break
1638
1639 if config_primitive is None:
1640 raise ValueError("Could not find ns_cfg_prim %s in nsr %s" % (config_name, self.name))
1641
1642 self._log.debug("Scaling group {} config primitive: {}".format(group.name, config_primitive))
1643 if config_primitive.has_field("user_defined_script"):
1644 rc = yield from self.apply_scale_group_config_script(config_primitive.user_defined_script,
1645 group, scale_instance, trigger, vnfrs)
1646 err_msg = None
1647 if not rc:
1648 err_msg = "Failed config for trigger {} using config script '{}'". \
1649 format(self.scaling_trigger_str(trigger),
1650 config_primitive.user_defined_script)
1651 yield from update_config_status(success=rc, err_msg=err_msg)
1652 return rc
1653 else:
1654 err_msg = "Failed config for trigger {} as config script is not specified". \
1655 format(self.scaling_trigger_str(trigger))
1656 yield from update_config_status(success=False, err_msg=err_msg)
1657 raise NotImplementedError("Only script based config support for scale group for now: {}".
1658 format(group.name))
1659 else:
1660 err_msg = "Failed config for trigger {} as config primitive is not specified".\
1661 format(self.scaling_trigger_str(trigger))
1662 yield from update_config_status(success=False, err_msg=err_msg)
1663 self._log.error("Config primitive not specified for config action in scale group %s" %
1664 (group.name))
1665 return False
1666
1667 def create_scaling_groups(self):
1668 """ This function creates a NSScalingGroup for every scaling
1669 group defined in he NSD"""
1670
1671 for scaling_group_msg in self.nsd_msg.scaling_group_descriptor:
1672 self._log.debug("Found scaling_group %s in nsr id %s",
1673 scaling_group_msg.name, self.id)
1674
1675 group_record = scale_group.ScalingGroup(
1676 self._log,
1677 scaling_group_msg
1678 )
1679
1680 self._scaling_groups[group_record.name] = group_record
1681
1682 @asyncio.coroutine
1683 def create_scale_group_instance(self, group_name, index, config_xact, is_default=False):
1684 group = self._scaling_groups[group_name]
1685 scale_instance = group.create_instance(index, is_default)
1686
1687 @asyncio.coroutine
1688 def create_vnfs():
1689 self._log.debug("Creating %u VNFs associated with NS id %s scaling group %s",
1690 len(self.nsd_msg.constituent_vnfd), self.id, self)
1691
1692 vnfrs = []
1693 for vnf_index, count in group.vnf_index_count_map.items():
1694 const_vnfd_msg = self._get_constituent_vnfd_msg(vnf_index)
1695 vnfd_msg = self._get_vnfd(const_vnfd_msg.vnfd_id_ref, config_xact)
1696
1697 cloud_account_name, om_datacenter_name = self._get_vnfd_cloud_account(const_vnfd_msg.member_vnf_index)
1698 if cloud_account_name is None:
1699 cloud_account_name = self.cloud_account_name
1700 for _ in range(count):
1701 vnfr = yield from self.create_vnf_record(vnfd_msg, const_vnfd_msg, cloud_account_name, om_datacenter_name, group_name, index)
1702 scale_instance.add_vnfr(vnfr)
1703 vnfrs.append(vnfr)
1704 return vnfrs
1705
1706 @asyncio.coroutine
1707 def instantiate_instance():
1708 self._log.debug("Creating %s VNFRS", scale_instance)
1709 vnfrs = yield from create_vnfs()
1710 yield from self.publish()
1711
1712 self._log.debug("Instantiating %s VNFRS for %s", len(vnfrs), scale_instance)
1713 scale_instance.operational_status = "vnf_init_phase"
1714 yield from self.update_state()
1715
1716 try:
1717 rc = yield from self.apply_scaling_group_config(NsdYang.ScalingTrigger.PRE_SCALE_OUT,
1718 group, scale_instance, vnfrs)
1719 if not rc:
1720 self._log.error("Pre scale out config for scale group {} ({}) failed".
1721 format(group.name, index))
1722 scale_instance.operational_status = "failed"
1723 else:
1724 yield from self.instantiate_vnfs(vnfrs, scaleout=True)
1725
1726
1727 except Exception as e:
1728 self._log.exception("Failed to begin instantiatiation of vnfs for scale group {}: {}".
1729 format(group.name, e))
1730 self._log.exception(e)
1731 scale_instance.operational_status = "failed"
1732
1733 yield from self.update_state()
1734
1735 yield from instantiate_instance()
1736
1737 @asyncio.coroutine
1738 def delete_scale_group_instance(self, group_name, index):
1739 group = self._scaling_groups[group_name]
1740 scale_instance = group.get_instance(index)
1741 if scale_instance.is_default:
1742 raise ScalingOperationError("Cannot terminate a default scaling group instance")
1743
1744 scale_instance.operational_status = "terminate"
1745 yield from self.update_state()
1746
1747 @asyncio.coroutine
1748 def terminate_instance():
1749 self._log.debug("Terminating %s VNFRS" % scale_instance)
1750 rc = yield from self.apply_scaling_group_config(NsdYang.ScalingTrigger.PRE_SCALE_IN,
1751 group, scale_instance)
1752 if not rc:
1753 self._log.error("Pre scale in config for scale group {} ({}) failed".
1754 format(group.name, index))
1755
1756 # Going ahead with terminate, even if there is an error in pre-scale-in config
1757 # as this could be result of scale out failure and we need to cleanup this group
1758 yield from self.terminate_vnfrs(scale_instance.vnfrs)
1759 group.delete_instance(index)
1760
1761 scale_instance.operational_status = "vnf_terminate_phase"
1762 yield from self.update_state()
1763
1764 yield from terminate_instance()
1765
1766 @asyncio.coroutine
1767 def _update_scale_group_instances_status(self):
1768 @asyncio.coroutine
1769 def post_scale_out_task(group, instance):
1770 # Apply post scale out config once all VNFRs are active
1771 rc = yield from self.apply_scaling_group_config(NsdYang.ScalingTrigger.POST_SCALE_OUT,
1772 group, instance)
1773 instance.operational_status = "running"
1774 if rc:
1775 self._log.debug("Scale out for group {} and instance {} succeeded".
1776 format(group.name, instance.instance_id))
1777 else:
1778 self._log.error("Post scale out config for scale group {} ({}) failed".
1779 format(group.name, instance.instance_id))
1780
1781 yield from self.update_state()
1782
1783 group_instances = {group: group.instances for group in self._scaling_groups.values()}
1784 for group, instances in group_instances.items():
1785 self._log.debug("Updating %s instance status", group)
1786 for instance in instances:
1787 instance_vnf_state_list = [vnfr.state for vnfr in instance.vnfrs]
1788 self._log.debug("Got vnfr instance states: %s", instance_vnf_state_list)
1789 if instance.operational_status == "vnf_init_phase":
1790 if all([state == VnfRecordState.ACTIVE for state in instance_vnf_state_list]):
1791 instance.operational_status = "running"
1792
1793 # Create a task for post scale out to allow us to sleep before attempting
1794 # to configure newly created VM's
1795 self._loop.create_task(post_scale_out_task(group, instance))
1796
1797 elif any([state == VnfRecordState.FAILED for state in instance_vnf_state_list]):
1798 self._log.debug("Scale out for group {} and instance {} failed".
1799 format(group.name, instance.instance_id))
1800 instance.operational_status = "failed"
1801
1802 elif instance.operational_status == "vnf_terminate_phase":
1803 if all([state == VnfRecordState.TERMINATED for state in instance_vnf_state_list]):
1804 instance.operational_status = "terminated"
1805 rc = yield from self.apply_scaling_group_config(NsdYang.ScalingTrigger.POST_SCALE_IN,
1806 group, instance)
1807 if rc:
1808 self._log.debug("Scale in for group {} and instance {} succeeded".
1809 format(group.name, instance.instance_id))
1810 else:
1811 self._log.error("Post scale in config for scale group {} ({}) failed".
1812 format(group.name, instance.instance_id))
1813
1814 def create_vnffgs(self):
1815 """ This function creates VNFFGs for every VNFFG in the NSD
1816 associated with this NSR"""
1817
1818 for vnffgd in self.nsd_msg.vnffgd:
1819 self._log.debug("Found vnffgd %s in nsr id %s", vnffgd, self.id)
1820 vnffgr = VnffgRecord(self._dts,
1821 self._log,
1822 self._loop,
1823 self._nsm._vnffgmgr,
1824 self,
1825 self.name,
1826 vnffgd,
1827 self._sdn_account_name
1828 )
1829 self._vnffgrs[vnffgr.id] = vnffgr
1830
1831 def resolve_vld_ip_profile(self, nsd_msg, vld):
1832 self._log.debug("Receieved ip profile ref is %s",vld.ip_profile_ref)
1833 if not vld.has_field('ip_profile_ref'):
1834 return None
1835 profile = [profile for profile in nsd_msg.ip_profiles if profile.name == vld.ip_profile_ref]
1836 return profile[0] if profile else None
1837
1838 @asyncio.coroutine
1839 def _create_vls(self, vld, cloud_account,om_datacenter):
1840 """Create a VLR in the cloud account specified using the given VLD
1841
1842 Args:
1843 vld : VLD yang obj
1844 cloud_account : Cloud account name
1845
1846 Returns:
1847 VirtualLinkRecord
1848 """
1849 vlr = yield from VirtualLinkRecord.create_record(
1850 self._dts,
1851 self._log,
1852 self._loop,
1853 self.name,
1854 vld,
1855 cloud_account,
1856 om_datacenter,
1857 self.resolve_vld_ip_profile(self.nsd_msg, vld),
1858 self.id,
1859 restart_mode=self.restart_mode)
1860
1861 return vlr
1862
1863 def _extract_cloud_accounts_for_vl(self, vld):
1864 """
1865 Extracts the list of cloud accounts from the NS Config obj
1866
1867 Rules:
1868 1. Cloud accounts based connection point (vnf_cloud_account_map)
1869 Args:
1870 vld : VLD yang object
1871
1872 Returns:
1873 TYPE: Description
1874 """
1875 cloud_account_list = []
1876
1877 if self._nsr_cfg_msg.vnf_cloud_account_map:
1878 # Handle case where cloud_account is None
1879 vnf_cloud_map = {}
1880 for vnf in self._nsr_cfg_msg.vnf_cloud_account_map:
1881 if vnf.cloud_account is not None or vnf.om_datacenter is not None:
1882 vnf_cloud_map[vnf.member_vnf_index_ref] = (vnf.cloud_account,vnf.om_datacenter)
1883
1884 for vnfc in vld.vnfd_connection_point_ref:
1885 cloud_account = vnf_cloud_map.get(
1886 vnfc.member_vnf_index_ref,
1887 (self.cloud_account_name,self.om_datacenter_name))
1888
1889 cloud_account_list.append(cloud_account)
1890
1891 if self._nsr_cfg_msg.vl_cloud_account_map:
1892 for vld_map in self._nsr_cfg_msg.vl_cloud_account_map:
1893 if vld_map.vld_id_ref == vld.id:
1894 for cloud_account in vld_map.cloud_accounts:
1895 cloud_account_list.extend((cloud_account,None))
1896 for om_datacenter in vld_map.om_datacenters:
1897 cloud_account_list.extend((None,om_datacenter))
1898
1899 # If no config has been provided then fall-back to the default
1900 # account
1901 if not cloud_account_list:
1902 cloud_account_list = [(self.cloud_account_name,self.om_datacenter_name)]
1903
1904 self._log.debug("VL {} cloud accounts: {}".
1905 format(vld.name, cloud_account_list))
1906 return set(cloud_account_list)
1907
1908 @asyncio.coroutine
1909 def create_vls(self):
1910 """ This function creates VLs for every VLD in the NSD
1911 associated with this NSR"""
1912 for vld in self.nsd_msg.vld:
1913
1914 self._log.debug("Found vld %s in nsr id %s", vld, self.id)
1915 cloud_account_list = self._extract_cloud_accounts_for_vl(vld)
1916 for cloud_account,om_datacenter in cloud_account_list:
1917 vlr = yield from self._create_vls(vld, cloud_account,om_datacenter)
1918 self._vlrs.append(vlr)
1919
1920
1921 @asyncio.coroutine
1922 def create_vl_instance(self, vld):
1923 self._log.debug("Create VL for {}: {}".format(self.id, vld.as_dict()))
1924 # Check if the VL is already present
1925 vlr = None
1926 for vl in self._vlrs:
1927 if vl.vld_msg.id == vld.id:
1928 self._log.debug("The VLD %s already in NSR %s as VLR %s with status %s",
1929 vld.id, self.id, vl.id, vl.state)
1930 vlr = vl
1931 if vlr.state != VlRecordState.TERMINATED:
1932 err_msg = "VLR for VL %s in NSR %s already instantiated", \
1933 vld, self.id
1934 self._log.error(err_msg)
1935 raise NsrVlUpdateError(err_msg)
1936 break
1937
1938 if vlr is None:
1939 cloud_account_list = self._extract_cloud_accounts_for_vl(vld)
1940 for account,om_datacenter in cloud_account_list:
1941 vlr = yield from self._create_vls(vld, account,om_datacenter)
1942 self._vlrs.append(vlr)
1943
1944 vlr.state = VlRecordState.INSTANTIATION_PENDING
1945 yield from self.update_state()
1946
1947 try:
1948 yield from self.nsm_plugin.instantiate_vl(self, vlr)
1949 vlr.state = VlRecordState.ACTIVE
1950
1951 except Exception as e:
1952 err_msg = "Error instantiating VL for NSR {} and VLD {}: {}". \
1953 format(self.id, vld.id, e)
1954 self._log.error(err_msg)
1955 self._log.exception(e)
1956 vlr.state = VlRecordState.FAILED
1957
1958 yield from self.update_state()
1959
1960 @asyncio.coroutine
1961 def delete_vl_instance(self, vld):
1962 for vlr in self._vlrs:
1963 if vlr.vld_msg.id == vld.id:
1964 self._log.debug("Found VLR %s for VLD %s in NSR %s",
1965 vlr.id, vld.id, self.id)
1966 vlr.state = VlRecordState.TERMINATE_PENDING
1967 yield from self.update_state()
1968
1969 try:
1970 yield from self.nsm_plugin.terminate_vl(vlr)
1971 vlr.state = VlRecordState.TERMINATED
1972 self._vlrs.remove(vlr)
1973
1974 except Exception as e:
1975 err_msg = "Error terminating VL for NSR {} and VLD {}: {}". \
1976 format(self.id, vld.id, e)
1977 self._log.error(err_msg)
1978 self._log.exception(e)
1979 vlr.state = VlRecordState.FAILED
1980
1981 yield from self.update_state()
1982 break
1983
1984 @asyncio.coroutine
1985 def create_vnfs(self, config_xact):
1986 """
1987 This function creates VNFs for every VNF in the NSD
1988 associated with this NSR
1989 """
1990 self._log.debug("Creating %u VNFs associated with this NS id %s",
1991 len(self.nsd_msg.constituent_vnfd), self.id)
1992
1993 for const_vnfd in self.nsd_msg.constituent_vnfd:
1994 if not const_vnfd.start_by_default:
1995 self._log.debug("start_by_default set to False in constituent VNF (%s). Skipping start.",
1996 const_vnfd.member_vnf_index)
1997 continue
1998
1999 vnfd_msg = self._get_vnfd(const_vnfd.vnfd_id_ref, config_xact)
2000 cloud_account_name,om_datacenter_name = self._get_vnfd_cloud_account(const_vnfd.member_vnf_index)
2001 if cloud_account_name is None:
2002 cloud_account_name = self.cloud_account_name
2003 yield from self.create_vnf_record(vnfd_msg, const_vnfd, cloud_account_name, om_datacenter_name)
2004
2005
2006 def get_placement_groups(self, vnfd_msg, const_vnfd):
2007 placement_groups = []
2008 for group in self.nsd_msg.placement_groups:
2009 for member_vnfd in group.member_vnfd:
2010 if (member_vnfd.vnfd_id_ref == vnfd_msg.id) and \
2011 (member_vnfd.member_vnf_index_ref == const_vnfd.member_vnf_index):
2012 group_info = self.resolve_placement_group_cloud_construct(group)
2013 if group_info is None:
2014 self._log.info("Could not resolve cloud-construct for placement group: %s", group.name)
2015 ### raise PlacementGroupError("Could not resolve cloud-construct for placement group: {}".format(group.name))
2016 else:
2017 self._log.info("Successfully resolved cloud construct for placement group: %s for VNF: %s (Member Index: %s)",
2018 str(group_info),
2019 vnfd_msg.name,
2020 const_vnfd.member_vnf_index)
2021 placement_groups.append(group_info)
2022 return placement_groups
2023
2024 @asyncio.coroutine
2025 def create_vnf_record(self, vnfd_msg, const_vnfd, cloud_account_name, om_datacenter_name, group_name=None, group_instance_id=None):
2026 # Fetch the VNFD associated with this VNF
2027 placement_groups = self.get_placement_groups(vnfd_msg, const_vnfd)
2028 self._log.info("Cloud Account for VNF %d is %s",const_vnfd.member_vnf_index,cloud_account_name)
2029 self._log.info("Launching VNF: %s (Member Index: %s) in NSD plancement Groups: %s",
2030 vnfd_msg.name,
2031 const_vnfd.member_vnf_index,
2032 [ group.name for group in placement_groups])
2033 vnfr = yield from VirtualNetworkFunctionRecord.create_record(self._dts,
2034 self._log,
2035 self._loop,
2036 vnfd_msg,
2037 const_vnfd,
2038 self.nsd_id,
2039 self.name,
2040 cloud_account_name,
2041 om_datacenter_name,
2042 self.id,
2043 group_name,
2044 group_instance_id,
2045 placement_groups,
2046 restart_mode=self.restart_mode,
2047 )
2048 if vnfr.id in self._vnfrs:
2049 err = "VNF with VNFR id %s already in vnf list" % (vnfr.id,)
2050 raise NetworkServiceRecordError(err)
2051
2052 self._vnfrs[vnfr.id] = vnfr
2053 self._nsm.vnfrs[vnfr.id] = vnfr
2054
2055 yield from vnfr.set_config_status(NsrYang.ConfigStates.INIT)
2056
2057 self._log.debug("Added VNFR %s to NSM VNFR list with id %s",
2058 vnfr.name,
2059 vnfr.id)
2060
2061 return vnfr
2062
2063 def create_param_pools(self):
2064 for param_pool in self.nsd_msg.parameter_pool:
2065 self._log.debug("Found parameter pool %s in nsr id %s", param_pool, self.id)
2066
2067 start_value = param_pool.range.start_value
2068 end_value = param_pool.range.end_value
2069 if end_value < start_value:
2070 raise NetworkServiceRecordError(
2071 "Parameter pool %s has invalid range (start: {}, end: {})".format(
2072 start_value, end_value
2073 )
2074 )
2075
2076 self._param_pools[param_pool.name] = config_value_pool.ParameterValuePool(
2077 self._log,
2078 param_pool.name,
2079 range(start_value, end_value)
2080 )
2081
2082 @asyncio.coroutine
2083 def fetch_vnfr(self, vnfr_path):
2084 """ Fetch VNFR record """
2085 vnfr = None
2086 self._log.debug("Fetching VNFR with key %s while instantiating %s",
2087 vnfr_path, self.id)
2088 res_iter = yield from self._dts.query_read(vnfr_path, rwdts.XactFlag.MERGE)
2089
2090 for ent in res_iter:
2091 res = yield from ent
2092 vnfr = res.result
2093
2094 return vnfr
2095
2096 @asyncio.coroutine
2097 def instantiate_vnfs(self, vnfrs, scaleout=False):
2098 """
2099 This function instantiates VNFs for every VNF in this Network Service
2100 """
2101 self._log.debug("Instantiating %u VNFs in NS %s", len(vnfrs), self.id)
2102 for vnf in vnfrs:
2103 self._log.debug("Instantiating VNF: %s in NS %s", vnf, self.id)
2104 yield from self.nsm_plugin.instantiate_vnf(self, vnf,scaleout)
2105
2106 @asyncio.coroutine
2107 def instantiate_vnffgs(self):
2108 """
2109 This function instantiates VNFFGs for every VNFFG in this Network Service
2110 """
2111 self._log.debug("Instantiating %u VNFFGs in NS %s",
2112 len(self.nsd_msg.vnffgd), self.id)
2113 for _, vnfr in self.vnfrs.items():
2114 while vnfr.state in [VnfRecordState.INSTANTIATION_PENDING, VnfRecordState.INIT]:
2115 self._log.debug("Received vnfr state for vnfr %s is %s; retrying",vnfr.name,vnfr.state)
2116 yield from asyncio.sleep(2, loop=self._loop)
2117 if vnfr.state == VnfRecordState.ACTIVE:
2118 self._log.debug("Received vnfr state for vnfr %s is %s ",vnfr.name,vnfr.state)
2119 continue
2120 else:
2121 self._log.debug("Received vnfr state for vnfr %s is %s; failing vnffg creation",vnfr.name,vnfr.state)
2122 self._vnffgr_state = VnffgRecordState.FAILED
2123 return
2124
2125 self._log.info("Waiting for 90 seconds for VMs to come up")
2126 yield from asyncio.sleep(90, loop=self._loop)
2127 self._log.info("Starting VNFFG orchestration")
2128 for vnffg in self._vnffgrs.values():
2129 self._log.debug("Instantiating VNFFG: %s in NS %s", vnffg, self.id)
2130 yield from vnffg.instantiate()
2131
2132 @asyncio.coroutine
2133 def instantiate_scaling_instances(self, config_xact):
2134 """ Instantiate any default scaling instances in this Network Service """
2135 for group in self._scaling_groups.values():
2136 for i in range(group.min_instance_count):
2137 self._log.debug("Instantiating %s default scaling instance %s", group, i)
2138 yield from self.create_scale_group_instance(
2139 group.name, i, config_xact, is_default=True
2140 )
2141
2142 for group_msg in self._nsr_cfg_msg.scaling_group:
2143 if group_msg.scaling_group_name_ref != group.name:
2144 continue
2145
2146 for instance in group_msg.instance:
2147 self._log.debug("Reloading %s scaling instance %s", group_msg, instance.id)
2148 yield from self.create_scale_group_instance(
2149 group.name, instance.id, config_xact, is_default=False
2150 )
2151
2152 def has_scaling_instances(self):
2153 """ Return boolean indicating if the network service has default scaling groups """
2154 for group in self._scaling_groups.values():
2155 if group.min_instance_count > 0:
2156 return True
2157
2158 for group_msg in self._nsr_cfg_msg.scaling_group:
2159 if len(group_msg.instance) > 0:
2160 return True
2161
2162 return False
2163
2164 @asyncio.coroutine
2165 def publish(self):
2166 """ This function publishes this NSR """
2167 self._nsr_msg = self.create_msg()
2168
2169 self._log.debug("Publishing the NSR with xpath %s and nsr %s",
2170 self.nsr_xpath,
2171 self._nsr_msg)
2172
2173 if self._debug_running:
2174 self._log.debug("Publishing NSR in RUNNING state!")
2175 #raise()
2176
2177 with self._dts.transaction() as xact:
2178 yield from self._nsm.nsr_handler.update(xact, self.nsr_xpath, self._nsr_msg)
2179 if self._op_status.state == NetworkServiceRecordState.RUNNING:
2180 self._debug_running = True
2181
2182 @asyncio.coroutine
2183 def unpublish(self, xact):
2184 """ Unpublish this NSR object """
2185 self._log.debug("Unpublishing Network service id %s", self.id)
2186 yield from self._nsm.nsr_handler.delete(xact, self.nsr_xpath)
2187
2188 @property
2189 def nsr_xpath(self):
2190 """ Returns the xpath associated with this NSR """
2191 return(
2192 "D,/nsr:ns-instance-opdata" +
2193 "/nsr:nsr[nsr:ns-instance-config-ref = '{}']"
2194 ).format(self.id)
2195
2196 @staticmethod
2197 def xpath_from_nsr(nsr):
2198 """ Returns the xpath associated with this NSR op data"""
2199 return (NetworkServiceRecord.XPATH +
2200 "[nsr:ns-instance-config-ref = '{}']").format(nsr.id)
2201
2202 @property
2203 def nsd_xpath(self):
2204 """ Return NSD config xpath."""
2205 return(
2206 "C,/nsd:nsd-catalog/nsd:nsd[nsd:id = '{}']"
2207 ).format(self.nsd_id)
2208
2209 @asyncio.coroutine
2210 def instantiate(self, config_xact):
2211 """"Instantiates a NetworkServiceRecord.
2212
2213 This function instantiates a Network service
2214 which involves the following steps,
2215
2216 * Instantiate every VL in NSD by sending create VLR request to DTS.
2217 * Instantiate every VNF in NSD by sending create VNF reuqest to DTS.
2218 * Publish the NSR details to DTS
2219
2220 Arguments:
2221 nsr: The NSR configuration request containing nsr-id and nsd
2222 config_xact: The configuration transaction which initiated the instatiation
2223
2224 Raises:
2225 NetworkServiceRecordError if the NSR creation fails
2226
2227 Returns:
2228 No return value
2229 """
2230
2231 self._log.debug("Instantiating NS - %s xact - %s", self, config_xact)
2232
2233 # Move the state to INIITALIZING
2234 self.set_state(NetworkServiceRecordState.INIT)
2235
2236 event_descr = "Instantiation Request Received NSR Id:%s" % self.id
2237 self.record_event("instantiating", event_descr)
2238
2239 # Find the NSD
2240 self._nsd = self._nsr_cfg_msg.nsd
2241
2242 try:
2243 # Update ref count if nsd present in catalog
2244 self._nsm.get_nsd_ref(self.nsd_id)
2245
2246 except NetworkServiceDescriptorError:
2247 # This could be an NSD not in the nsd-catalog
2248 pass
2249
2250 # Merge any config and initial config primitive values
2251 self.config_store.merge_nsd_config(self.nsd_msg)
2252 self._log.debug("Merged NSD: {}".format(self.nsd_msg.as_dict()))
2253
2254 event_descr = "Fetched NSD with descriptor id %s" % self.nsd_id
2255 self.record_event("nsd-fetched", event_descr)
2256
2257 if self._nsd is None:
2258 msg = "Failed to fetch NSD with nsd-id [%s] for nsr-id %s"
2259 self._log.debug(msg, self.nsd_id, self.id)
2260 raise NetworkServiceRecordError(self)
2261
2262 self._log.debug("Got nsd result %s", self._nsd)
2263
2264 # Substitute any input parameters
2265 self.substitute_input_parameters(self._nsd, self._nsr_cfg_msg)
2266
2267 # Create the record
2268 yield from self.create(config_xact)
2269
2270 # Publish the NSR to DTS
2271 yield from self.publish()
2272
2273 @asyncio.coroutine
2274 def do_instantiate():
2275 """
2276 Instantiate network service
2277 """
2278 self._log.debug("Instantiating VLs nsr id [%s] nsd id [%s]",
2279 self.id, self.nsd_id)
2280
2281 # instantiate the VLs
2282 event_descr = ("Instantiating %s external VLs for NSR id %s" %
2283 (len(self.nsd_msg.vld), self.id))
2284 self.record_event("begin-external-vls-instantiation", event_descr)
2285
2286 self.set_state(NetworkServiceRecordState.VL_INIT_PHASE)
2287
2288 yield from self.instantiate_vls()
2289
2290 # Publish the NSR to DTS
2291 yield from self.publish()
2292
2293 event_descr = ("Finished instantiating %s external VLs for NSR id %s" %
2294 (len(self.nsd_msg.vld), self.id))
2295 self.record_event("end-external-vls-instantiation", event_descr)
2296
2297 self.set_state(NetworkServiceRecordState.VNF_INIT_PHASE)
2298
2299 self._log.debug("Instantiating VNFs ...... nsr[%s], nsd[%s]",
2300 self.id, self.nsd_id)
2301
2302 # instantiate the VNFs
2303 event_descr = ("Instantiating %s VNFS for NSR id %s" %
2304 (len(self.nsd_msg.constituent_vnfd), self.id))
2305
2306 self.record_event("begin-vnf-instantiation", event_descr)
2307
2308 yield from self.instantiate_vnfs(self._vnfrs.values())
2309
2310 self._log.debug(" Finished instantiating %d VNFs for NSR id %s",
2311 len(self.nsd_msg.constituent_vnfd), self.id)
2312
2313 event_descr = ("Finished instantiating %s VNFs for NSR id %s" %
2314 (len(self.nsd_msg.constituent_vnfd), self.id))
2315 self.record_event("end-vnf-instantiation", event_descr)
2316
2317 if len(self.vnffgrs) > 0:
2318 #self.set_state(NetworkServiceRecordState.VNFFG_INIT_PHASE)
2319 event_descr = ("Instantiating %s VNFFGS for NSR id %s" %
2320 (len(self.nsd_msg.vnffgd), self.id))
2321
2322 self.record_event("begin-vnffg-instantiation", event_descr)
2323
2324 yield from self.instantiate_vnffgs()
2325
2326 event_descr = ("Finished instantiating %s VNFFGDs for NSR id %s" %
2327 (len(self.nsd_msg.vnffgd), self.id))
2328 self.record_event("end-vnffg-instantiation", event_descr)
2329
2330 if self.has_scaling_instances():
2331 event_descr = ("Instantiating %s Scaling Groups for NSR id %s" %
2332 (len(self._scaling_groups), self.id))
2333
2334 self.record_event("begin-scaling-group-instantiation", event_descr)
2335 yield from self.instantiate_scaling_instances(config_xact)
2336 self.record_event("end-scaling-group-instantiation", event_descr)
2337
2338 # Give the plugin a chance to deploy the network service now that all
2339 # virtual links and vnfs are instantiated
2340 yield from self.nsm_plugin.deploy(self._nsr_msg)
2341
2342 self._log.debug("Publishing NSR...... nsr[%s], nsd[%s]",
2343 self.id, self.nsd_id)
2344
2345 # Publish the NSR to DTS
2346 yield from self.publish()
2347
2348 self._log.debug("Published NSR...... nsr[%s], nsd[%s]",
2349 self.id, self.nsd_id)
2350
2351 def on_instantiate_done(fut):
2352 # If the do_instantiate fails, then publish NSR with failed result
2353 e = fut.exception()
2354 if e is not None:
2355 import traceback, sys
2356 print(traceback.format_exception(None,e, e.__traceback__), file=sys.stderr, flush=True)
2357 self._log.error("NSR instantiation failed for NSR id %s: %s", self.id, str(e))
2358 self._loop.create_task(self.instantiation_failed(failed_reason=str(e)))
2359
2360 instantiate_task = self._loop.create_task(do_instantiate())
2361 instantiate_task.add_done_callback(on_instantiate_done)
2362
2363 @asyncio.coroutine
2364 def set_config_status(self, status, status_details=None):
2365 if self.config_status != status:
2366 self._log.debug("Updating NSR {} status for {} to {}".
2367 format(self.name, self.config_status, status))
2368 self._config_status = status
2369 self._config_status_details = status_details
2370
2371 if self._config_status == NsrYang.ConfigStates.FAILED:
2372 self.record_event("config-failed", "NS configuration failed",
2373 evt_details=self._config_status_details)
2374
2375 yield from self.publish()
2376
2377 @asyncio.coroutine
2378 def is_active(self):
2379 """ This NS is active """
2380 self.set_state(NetworkServiceRecordState.RUNNING)
2381 if self._is_active:
2382 return
2383
2384 # Publish the NSR to DTS
2385 self._log.debug("Network service %s is active ", self.id)
2386 self._is_active = True
2387
2388 event_descr = "NSR in running state for NSR id %s" % self.id
2389 self.record_event("ns-running", event_descr)
2390
2391 yield from self.publish()
2392
2393 @asyncio.coroutine
2394 def instantiation_failed(self, failed_reason=None):
2395 """ The NS instantiation failed"""
2396 self._log.error("Network service id:%s, name:%s instantiation failed",
2397 self.id, self.name)
2398 self.set_state(NetworkServiceRecordState.FAILED)
2399
2400 event_descr = "Instantiation of NS %s failed" % self.id
2401 self.record_event("ns-failed", event_descr, evt_details=failed_reason)
2402
2403 # Publish the NSR to DTS
2404 yield from self.publish()
2405
2406 @asyncio.coroutine
2407 def terminate_vnfrs(self, vnfrs):
2408 """ Terminate VNFRS in this network service """
2409 self._log.debug("Terminating VNFs in network service %s", self.id)
2410 for vnfr in vnfrs:
2411 yield from self.nsm_plugin.terminate_vnf(vnfr)
2412
2413 @asyncio.coroutine
2414 def terminate(self):
2415 """ Terminate a NetworkServiceRecord."""
2416 def terminate_vnffgrs():
2417 """ Terminate VNFFGRS in this network service """
2418 self._log.debug("Terminating VNFFGRs in network service %s", self.id)
2419 for vnffgr in self.vnffgrs.values():
2420 yield from vnffgr.terminate()
2421
2422 def terminate_vlrs():
2423 """ Terminate VLRs in this netork service """
2424 self._log.debug("Terminating VLs in network service %s", self.id)
2425 for vlr in self.vlrs:
2426 yield from self.nsm_plugin.terminate_vl(vlr)
2427 vlr.state = VlRecordState.TERMINATED
2428 if vlr.id in self.vlr_uptime_tasks:
2429 self.vlr_uptime_tasks[vlr.id].cancel()
2430
2431 self._log.debug("Terminating network service id %s", self.id)
2432
2433 # Move the state to TERMINATE
2434 self.set_state(NetworkServiceRecordState.TERMINATE)
2435 event_descr = "Terminate being processed for NS Id:%s" % self.id
2436 self.record_event("terminate", event_descr)
2437
2438 # Move the state to VNF_TERMINATE_PHASE
2439 self._log.debug("Terminating VNFFGs in NS ID: %s", self.id)
2440 self.set_state(NetworkServiceRecordState.VNFFG_TERMINATE_PHASE)
2441 event_descr = "Terminating VNFFGS in NS Id:%s" % self.id
2442 self.record_event("terminating-vnffgss", event_descr)
2443 yield from terminate_vnffgrs()
2444
2445 # Move the state to VNF_TERMINATE_PHASE
2446 self.set_state(NetworkServiceRecordState.VNF_TERMINATE_PHASE)
2447 event_descr = "Terminating VNFS in NS Id:%s" % self.id
2448 self.record_event("terminating-vnfs", event_descr)
2449 yield from self.terminate_vnfrs(self.vnfrs.values())
2450
2451 # Move the state to VL_TERMINATE_PHASE
2452 self.set_state(NetworkServiceRecordState.VL_TERMINATE_PHASE)
2453 event_descr = "Terminating VLs in NS Id:%s" % self.id
2454 self.record_event("terminating-vls", event_descr)
2455 yield from terminate_vlrs()
2456
2457 yield from self.nsm_plugin.terminate_ns(self)
2458
2459 # Move the state to TERMINATED
2460 self.set_state(NetworkServiceRecordState.TERMINATED)
2461 event_descr = "Terminated NS Id:%s" % self.id
2462 self.record_event("terminated", event_descr)
2463
2464 def enable(self):
2465 """"Enable a NetworkServiceRecord."""
2466 pass
2467
2468 def disable(self):
2469 """"Disable a NetworkServiceRecord."""
2470 pass
2471
2472 def map_config_status(self):
2473 self._log.debug("Config status for ns {} is {}".
2474 format(self.name, self._config_status))
2475 if self._config_status == NsrYang.ConfigStates.CONFIGURING:
2476 return 'configuring'
2477 if self._config_status == NsrYang.ConfigStates.FAILED:
2478 return 'failed'
2479 return 'configured'
2480
2481 def vl_phase_completed(self):
2482 """ Are VLs created in this NS?"""
2483 return self._vl_phase_completed
2484
2485 def vnf_phase_completed(self):
2486 """ Are VLs created in this NS?"""
2487 return self._vnf_phase_completed
2488
2489 def create_msg(self):
2490 """ The network serice record as a message """
2491 nsr_dict = {"ns_instance_config_ref": self.id}
2492 nsr = RwNsrYang.YangData_Nsr_NsInstanceOpdata_Nsr.from_dict(nsr_dict)
2493 #nsr.cloud_account = self.cloud_account_name
2494 nsr.sdn_account = self._sdn_account_name
2495 nsr.name_ref = self.name
2496 nsr.nsd_ref = self.nsd_id
2497 nsr.nsd_name_ref = self.nsd_msg.name
2498 nsr.operational_events = self._op_status.msg
2499 nsr.operational_status = self._op_status.yang_str()
2500 nsr.config_status = self.map_config_status()
2501 nsr.config_status_details = self._config_status_details
2502 nsr.create_time = self._create_time
2503 nsr.uptime = int(time.time()) - self._create_time
2504
2505 for cfg_prim in self.nsd_msg.service_primitive:
2506 cfg_prim = NsrYang.YangData_Nsr_NsInstanceOpdata_Nsr_ServicePrimitive.from_dict(
2507 cfg_prim.as_dict())
2508 nsr.service_primitive.append(cfg_prim)
2509
2510 for init_cfg in self.nsd_msg.initial_config_primitive:
2511 prim = NsrYang.NsrInitialConfigPrimitive.from_dict(
2512 init_cfg.as_dict())
2513 nsr.initial_config_primitive.append(prim)
2514
2515 if self.vl_phase_completed():
2516 for vlr in self.vlrs:
2517 nsr.vlr.append(vlr.create_nsr_vlr_msg(self.vnfrs.values()))
2518
2519 if self.vnf_phase_completed():
2520 for vnfr_id in self.vnfrs:
2521 nsr.constituent_vnfr_ref.append(self.vnfrs[vnfr_id].const_vnfr_msg)
2522 for vnffgr in self.vnffgrs.values():
2523 nsr.vnffgr.append(vnffgr.fetch_vnffgr())
2524 for scaling_group in self._scaling_groups.values():
2525 nsr.scaling_group_record.append(scaling_group.create_record_msg())
2526
2527 return nsr
2528
2529 def all_vnfs_active(self):
2530 """ Are all VNFS in this NS active? """
2531 for _, vnfr in self.vnfrs.items():
2532 if vnfr.active is not True:
2533 return False
2534 return True
2535
2536 @asyncio.coroutine
2537 def update_state(self):
2538 """ Re-evaluate this NS's state """
2539 curr_state = self._op_status.state
2540
2541 if curr_state == NetworkServiceRecordState.TERMINATED:
2542 self._log.debug("NS (%s) in terminated state, not updating state", self.id)
2543 return
2544
2545 new_state = NetworkServiceRecordState.RUNNING
2546 self._log.info("Received update_state for nsr: %s, curr-state: %s",
2547 self.id, curr_state)
2548
2549 # Check all the VNFRs are present
2550 for _, vnfr in self.vnfrs.items():
2551 if vnfr.state in [VnfRecordState.ACTIVE, VnfRecordState.TERMINATED]:
2552 pass
2553 elif vnfr.state == VnfRecordState.FAILED:
2554 if vnfr._prev_state != vnfr.state:
2555 event_descr = "Instantiation of VNF %s failed" % vnfr.id
2556 event_error_details = vnfr.state_failed_reason
2557 self.record_event("vnf-failed", event_descr, evt_details=event_error_details)
2558 vnfr.set_state(VnfRecordState.FAILED)
2559 else:
2560 self._log.info("VNF state did not change, curr=%s, prev=%s",
2561 vnfr.state, vnfr._prev_state)
2562 new_state = NetworkServiceRecordState.FAILED
2563 break
2564 else:
2565 self._log.info("VNF %s in NSR %s is still not active; current state is: %s",
2566 vnfr.id, self.id, vnfr.state)
2567 new_state = curr_state
2568
2569 # If new state is RUNNING; check all VLs
2570 if new_state == NetworkServiceRecordState.RUNNING:
2571 for vl in self.vlrs:
2572
2573 if vl.state in [VlRecordState.ACTIVE, VlRecordState.TERMINATED]:
2574 pass
2575 elif vl.state == VlRecordState.FAILED:
2576 if vl.prev_state != vl.state:
2577 event_descr = "Instantiation of VL %s failed" % vl.id
2578 event_error_details = vl.state_failed_reason
2579 self.record_event("vl-failed", event_descr, evt_details=event_error_details)
2580 vl.prev_state = vl.state
2581 else:
2582 self._log.debug("VL %s already in failed state")
2583 else:
2584 if vl.state in [VlRecordState.INSTANTIATION_PENDING, VlRecordState.INIT]:
2585 new_state = NetworkServiceRecordState.VL_INSTANTIATE
2586 break
2587
2588 if vl.state in [VlRecordState.TERMINATE_PENDING]:
2589 new_state = NetworkServiceRecordState.VL_TERMINATE
2590 break
2591
2592 # If new state is RUNNING; check VNFFGRs are also active
2593 if new_state == NetworkServiceRecordState.RUNNING:
2594 for _, vnffgr in self.vnffgrs.items():
2595 self._log.info("Checking vnffgr state for nsr %s is: %s",
2596 self.id, vnffgr.state)
2597 if vnffgr.state == VnffgRecordState.ACTIVE:
2598 pass
2599 elif vnffgr.state == VnffgRecordState.FAILED:
2600 event_descr = "Instantiation of VNFFGR %s failed" % vnffgr.id
2601 self.record_event("vnffg-failed", event_descr)
2602 new_state = NetworkServiceRecordState.FAILED
2603 break
2604 else:
2605 self._log.info("VNFFGR %s in NSR %s is still not active; current state is: %s",
2606 vnffgr.id, self.id, vnffgr.state)
2607 new_state = curr_state
2608
2609 # Update all the scaling group instance operational status to
2610 # reflect the state of all VNFR within that instance
2611 yield from self._update_scale_group_instances_status()
2612
2613 for _, group in self._scaling_groups.items():
2614 if group.state == scale_group.ScaleGroupState.SCALING_OUT:
2615 new_state = NetworkServiceRecordState.SCALING_OUT
2616 break
2617 elif group.state == scale_group.ScaleGroupState.SCALING_IN:
2618 new_state = NetworkServiceRecordState.SCALING_IN
2619 break
2620
2621 if new_state != curr_state:
2622 self._log.debug("Changing state of Network service %s from %s to %s",
2623 self.id, curr_state, new_state)
2624 if new_state == NetworkServiceRecordState.RUNNING:
2625 yield from self.is_active()
2626 elif new_state == NetworkServiceRecordState.FAILED:
2627 # If the NS is already active and we entered scaling_in, scaling_out,
2628 # do not mark the NS as failing if scaling operation failed.
2629 if curr_state in [NetworkServiceRecordState.SCALING_OUT,
2630 NetworkServiceRecordState.SCALING_IN] and self._is_active:
2631 new_state = NetworkServiceRecordState.RUNNING
2632 self.set_state(new_state)
2633 else:
2634 yield from self.instantiation_failed()
2635 else:
2636 self.set_state(new_state)
2637
2638 yield from self.publish()
2639
2640
2641 class InputParameterSubstitution(object):
2642 """
2643 This class is responsible for substituting input parameters into an NSD.
2644 """
2645
2646 def __init__(self, log):
2647 """Create an instance of InputParameterSubstitution
2648
2649 Arguments:
2650 log - a logger for this object to use
2651
2652 """
2653 self.log = log
2654
2655 def __call__(self, nsd, nsr_config):
2656 """Substitutes input parameters from the NSR config into the NSD
2657
2658 This call modifies the provided NSD with the input parameters that are
2659 contained in the NSR config.
2660
2661 Arguments:
2662 nsd - a GI NSD object
2663 nsr_config - a GI NSR config object
2664
2665 """
2666 if nsd is None or nsr_config is None:
2667 return
2668
2669 # Create a lookup of the xpath elements that this descriptor allows
2670 # to be modified
2671 optional_input_parameters = set()
2672 for input_parameter in nsd.input_parameter_xpath:
2673 optional_input_parameters.add(input_parameter.xpath)
2674
2675 # Apply the input parameters to the descriptor
2676 if nsr_config.input_parameter:
2677 for param in nsr_config.input_parameter:
2678 if param.xpath not in optional_input_parameters:
2679 msg = "tried to set an invalid input parameter ({})"
2680 self.log.error(msg.format(param.xpath))
2681 continue
2682
2683 self.log.debug(
2684 "input-parameter:{} = {}".format(
2685 param.xpath,
2686 param.value,
2687 )
2688 )
2689
2690 try:
2691 xpath.setxattr(nsd, param.xpath, param.value)
2692
2693 except Exception as e:
2694 self.log.exception(e)
2695
2696
2697 class NetworkServiceDescriptor(object):
2698 """
2699 Network service descriptor class
2700 """
2701
2702 def __init__(self, dts, log, loop, nsd, nsm):
2703 self._dts = dts
2704 self._log = log
2705 self._loop = loop
2706
2707 self._nsd = nsd
2708 self._ref_count = 0
2709
2710 self._nsm = nsm
2711
2712 @property
2713 def id(self):
2714 """ Returns nsd id """
2715 return self._nsd.id
2716
2717 @property
2718 def name(self):
2719 """ Returns name of nsd """
2720 return self._nsd.name
2721
2722 @property
2723 def ref_count(self):
2724 """ Returns reference count"""
2725 return self._ref_count
2726
2727 def in_use(self):
2728 """ Returns whether nsd is in use or not """
2729 return True if self.ref_count > 0 else False
2730
2731 def ref(self):
2732 """ Take a reference on this object """
2733 self._ref_count += 1
2734
2735 def unref(self):
2736 """ Release reference on this object """
2737 if self.ref_count < 1:
2738 msg = ("Unref on a NSD object - nsd id %s, ref_count = %s" %
2739 (self.id, self.ref_count))
2740 self._log.critical(msg)
2741 raise NetworkServiceDescriptorError(msg)
2742 self._ref_count -= 1
2743
2744 @property
2745 def msg(self):
2746 """ Return the message associated with this NetworkServiceDescriptor"""
2747 return self._nsd
2748
2749 @staticmethod
2750 def path_for_id(nsd_id):
2751 """ Return path for the passed nsd_id"""
2752 return "C,/nsd:nsd-catalog/nsd:nsd[nsd:id = '{}'".format(nsd_id)
2753
2754 def path(self):
2755 """ Return the message associated with this NetworkServiceDescriptor"""
2756 return NetworkServiceDescriptor.path_for_id(self.id)
2757
2758 def update(self, nsd):
2759 """ Update the NSD descriptor """
2760 self._nsd = nsd
2761
2762
2763 class NsdDtsHandler(object):
2764 """ The network service descriptor DTS handler """
2765 XPATH = "C,/nsd:nsd-catalog/nsd:nsd"
2766
2767 def __init__(self, dts, log, loop, nsm):
2768 self._dts = dts
2769 self._log = log
2770 self._loop = loop
2771 self._nsm = nsm
2772
2773 self._regh = None
2774
2775 @property
2776 def regh(self):
2777 """ Return registration handle """
2778 return self._regh
2779
2780 @asyncio.coroutine
2781 def register(self):
2782 """ Register for Nsd create/update/delete/read requests from dts """
2783
2784 def on_apply(dts, acg, xact, action, scratch):
2785 """Apply the configuration"""
2786 is_recovery = xact.xact is None and action == rwdts.AppconfAction.INSTALL
2787 self._log.debug("Got nsd apply cfg (xact:%s) (action:%s)",
2788 xact, action)
2789 # Create/Update an NSD record
2790 for cfg in self._regh.get_xact_elements(xact):
2791 # Only interested in those NSD cfgs whose ID was received in prepare callback
2792 if cfg.id in scratch.get('nsds', []) or is_recovery:
2793 self._nsm.update_nsd(cfg)
2794
2795 scratch.pop('nsds', None)
2796
2797 return RwTypes.RwStatus.SUCCESS
2798
2799 @asyncio.coroutine
2800 def delete_nsd_libs(nsd_id):
2801 """ Remove any files uploaded with NSD and stored under $RIFT_ARTIFACTS/libs/<id> """
2802 try:
2803 rift_artifacts_dir = os.environ['RIFT_ARTIFACTS']
2804 nsd_dir = os.path.join(rift_artifacts_dir, 'launchpad/libs', nsd_id)
2805
2806 if os.path.exists (nsd_dir):
2807 shutil.rmtree(nsd_dir, ignore_errors=True)
2808 except Exception as e:
2809 self._log.error("Exception in cleaning up NSD libs {}: {}".
2810 format(nsd_id, e))
2811 self._log.excpetion(e)
2812
2813 @asyncio.coroutine
2814 def on_prepare(dts, acg, xact, xact_info, ks_path, msg, scratch):
2815 """ Prepare callback from DTS for NSD config """
2816
2817 self._log.info("Got nsd prepare - config received nsd id %s, msg %s",
2818 msg.id, msg)
2819
2820 fref = ProtobufC.FieldReference.alloc()
2821 fref.goto_whole_message(msg.to_pbcm())
2822
2823 if fref.is_field_deleted():
2824 # Delete an NSD record
2825 self._log.debug("Deleting NSD with id %s", msg.id)
2826 if self._nsm.nsd_in_use(msg.id):
2827 self._log.debug("Cannot delete NSD in use - %s", msg.id)
2828 err = "Cannot delete an NSD in use - %s" % msg.id
2829 raise NetworkServiceDescriptorRefCountExists(err)
2830
2831 yield from delete_nsd_libs(msg.id)
2832 self._nsm.delete_nsd(msg.id)
2833 else:
2834 # Add this NSD to scratch to create/update in apply callback
2835 nsds = scratch.setdefault('nsds', [])
2836 nsds.append(msg.id)
2837 # acg._scratch['nsds'].append(msg.id)
2838
2839 xact_info.respond_xpath(rwdts.XactRspCode.ACK)
2840
2841 self._log.debug(
2842 "Registering for NSD config using xpath: %s",
2843 NsdDtsHandler.XPATH,
2844 )
2845
2846 acg_hdl = rift.tasklets.AppConfGroup.Handler(on_apply=on_apply)
2847 with self._dts.appconf_group_create(handler=acg_hdl) as acg:
2848 # Need a list in scratch to store NSDs to create/update later
2849 # acg._scratch['nsds'] = list()
2850 self._regh = acg.register(
2851 xpath=NsdDtsHandler.XPATH,
2852 flags=rwdts.Flag.SUBSCRIBER | rwdts.Flag.DELTA_READY | rwdts.Flag.CACHE,
2853 on_prepare=on_prepare)
2854
2855
2856 class VnfdDtsHandler(object):
2857 """ DTS handler for VNFD config changes """
2858 XPATH = "C,/vnfd:vnfd-catalog/vnfd:vnfd"
2859
2860 def __init__(self, dts, log, loop, nsm):
2861 self._dts = dts
2862 self._log = log
2863 self._loop = loop
2864 self._nsm = nsm
2865 self._regh = None
2866
2867 @property
2868 def regh(self):
2869 """ DTS registration handle """
2870 return self._regh
2871
2872 @asyncio.coroutine
2873 def register(self):
2874 """ Register for VNFD configuration"""
2875
2876 @asyncio.coroutine
2877 def on_apply(dts, acg, xact, action, scratch):
2878 """Apply the configuration"""
2879 self._log.debug("Got NSM VNFD apply (xact: %s) (action: %s)(scr: %s)",
2880 xact, action, scratch)
2881
2882 # Create/Update a VNFD record
2883 for cfg in self._regh.get_xact_elements(xact):
2884 # Only interested in those VNFD cfgs whose ID was received in prepare callback
2885 if cfg.id in scratch.get('vnfds', []):
2886 self._nsm.update_vnfd(cfg)
2887
2888 for cfg in self._regh.elements:
2889 if cfg.id in scratch.get('deleted_vnfds', []):
2890 yield from self._nsm.delete_vnfd(cfg.id)
2891
2892 scratch.pop('vnfds', None)
2893 scratch.pop('deleted_vnfds', None)
2894
2895 @asyncio.coroutine
2896 def on_prepare(dts, acg, xact, xact_info, ks_path, msg, scratch):
2897 """ on prepare callback """
2898 self._log.debug("Got on prepare for VNFD (path: %s) (action: %s) (msg: %s)",
2899 ks_path.to_xpath(RwNsmYang.get_schema()), xact_info.query_action, msg)
2900
2901 fref = ProtobufC.FieldReference.alloc()
2902 fref.goto_whole_message(msg.to_pbcm())
2903
2904 # Handle deletes in prepare_callback, but adds/updates in apply_callback
2905 if fref.is_field_deleted():
2906 self._log.debug("Adding msg to deleted field")
2907 deleted_vnfds = scratch.setdefault('deleted_vnfds', [])
2908 deleted_vnfds.append(msg.id)
2909 else:
2910 # Add this VNFD to scratch to create/update in apply callback
2911 vnfds = scratch.setdefault('vnfds', [])
2912 vnfds.append(msg.id)
2913
2914 xact_info.respond_xpath(rwdts.XactRspCode.ACK)
2915
2916 self._log.debug(
2917 "Registering for VNFD config using xpath: %s",
2918 VnfdDtsHandler.XPATH,
2919 )
2920 acg_hdl = rift.tasklets.AppConfGroup.Handler(on_apply=on_apply)
2921 with self._dts.appconf_group_create(handler=acg_hdl) as acg:
2922 # Need a list in scratch to store VNFDs to create/update later
2923 # acg._scratch['vnfds'] = list()
2924 # acg._scratch['deleted_vnfds'] = list()
2925 self._regh = acg.register(
2926 xpath=VnfdDtsHandler.XPATH,
2927 flags=rwdts.Flag.SUBSCRIBER | rwdts.Flag.DELTA_READY,
2928 on_prepare=on_prepare)
2929
2930 class NsrRpcDtsHandler(object):
2931 """ The network service instantiation RPC DTS handler """
2932 EXEC_NSR_CONF_XPATH = "I,/nsr:start-network-service"
2933 EXEC_NSR_CONF_O_XPATH = "O,/nsr:start-network-service"
2934 NETCONF_IP_ADDRESS = "127.0.0.1"
2935 NETCONF_PORT = 2022
2936 RESTCONF_PORT = 8888
2937 NETCONF_USER = "admin"
2938 NETCONF_PW = "admin"
2939 REST_BASE_V2_URL = 'https://{}:{}/v2/api/'.format("127.0.0.1",8888)
2940
2941 def __init__(self, dts, log, loop, nsm):
2942 self._dts = dts
2943 self._log = log
2944 self._loop = loop
2945 self._nsm = nsm
2946 self._nsd = None
2947
2948 self._ns_regh = None
2949
2950 self._manager = None
2951 self._nsr_config_url = NsrRpcDtsHandler.REST_BASE_V2_URL + 'config/ns-instance-config'
2952
2953 self._model = RwYang.Model.create_libncx()
2954 self._model.load_schema_ypbc(RwNsrYang.get_schema())
2955
2956 @property
2957 def nsm(self):
2958 """ Return the NS manager instance """
2959 return self._nsm
2960
2961 @staticmethod
2962 def wrap_netconf_config_xml(xml):
2963 xml = '<config xmlns:xc="urn:ietf:params:xml:ns:netconf:base:1.0">{}</config>'.format(xml)
2964 return xml
2965
2966 @asyncio.coroutine
2967 def _connect(self, timeout_secs=240):
2968
2969 start_time = time.time()
2970 while (time.time() - start_time) < timeout_secs:
2971
2972 try:
2973 self._log.debug("Attemping NsmTasklet netconf connection.")
2974
2975 manager = yield from ncclient.asyncio_manager.asyncio_connect(
2976 loop=self._loop,
2977 host=NsrRpcDtsHandler.NETCONF_IP_ADDRESS,
2978 port=NsrRpcDtsHandler.NETCONF_PORT,
2979 username=NsrRpcDtsHandler.NETCONF_USER,
2980 password=NsrRpcDtsHandler.NETCONF_PW,
2981 allow_agent=False,
2982 look_for_keys=False,
2983 hostkey_verify=False,
2984 )
2985
2986 return manager
2987
2988 except ncclient.transport.errors.SSHError as e:
2989 self._log.warning("Netconf connection to launchpad %s failed: %s",
2990 NsrRpcDtsHandler.NETCONF_IP_ADDRESS, str(e))
2991
2992 yield from asyncio.sleep(5, loop=self._loop)
2993
2994 raise NsrInstantiationFailed("Failed to connect to Launchpad within %s seconds" %
2995 timeout_secs)
2996
2997 def _apply_ns_instance_config(self,payload_dict):
2998 #self._log.debug("At apply NS instance config with payload %s",payload_dict)
2999 req_hdr= {'accept':'application/vnd.yang.data+json','content-type':'application/vnd.yang.data+json'}
3000 response=requests.post(self._nsr_config_url, headers=req_hdr, auth=('admin', 'admin'),data=payload_dict,verify=False)
3001 return response
3002
3003 @asyncio.coroutine
3004 def register(self):
3005 """ Register for NS monitoring read from dts """
3006 @asyncio.coroutine
3007 def on_ns_config_prepare(xact_info, action, ks_path, msg):
3008 """ prepare callback from dts start-network-service"""
3009 assert action == rwdts.QueryAction.RPC
3010 rpc_ip = msg
3011 rpc_op = NsrYang.YangOutput_Nsr_StartNetworkService.from_dict({
3012 "nsr_id":str(uuid.uuid4())
3013 })
3014
3015 if not ('name' in rpc_ip and 'nsd_ref' in rpc_ip and ('cloud_account' in rpc_ip or 'om_datacenter' in rpc_ip)):
3016 self._log.error("Mandatory parameters name or nsd_ref or cloud account not found in start-network-service {}".format(rpc_ip))
3017
3018
3019 self._log.debug("start-network-service RPC input: {}".format(rpc_ip))
3020
3021 try:
3022 # Add used value to the pool
3023 self._log.debug("RPC output: {}".format(rpc_op))
3024
3025 nsd_copy = self.nsm.get_nsd(rpc_ip.nsd_ref)
3026
3027 #if not self._manager:
3028 # self._manager = yield from self._connect()
3029
3030 self._log.debug("Configuring ns-instance-config with name %s nsd-ref: %s",
3031 rpc_ip.name, rpc_ip.nsd_ref)
3032
3033 ns_instance_config_dict = {"id":rpc_op.nsr_id, "admin_status":"ENABLED"}
3034 ns_instance_config_copy_dict = {k:v for k, v in rpc_ip.as_dict().items()
3035 if k in RwNsrYang.YangData_Nsr_NsInstanceConfig_Nsr().fields}
3036 ns_instance_config_dict.update(ns_instance_config_copy_dict)
3037
3038 ns_instance_config = RwNsrYang.YangData_Nsr_NsInstanceConfig_Nsr.from_dict(ns_instance_config_dict)
3039 ns_instance_config.nsd = NsrYang.YangData_Nsr_NsInstanceConfig_Nsr_Nsd()
3040 ns_instance_config.nsd.from_dict(nsd_copy.msg.as_dict())
3041
3042 payload_dict = ns_instance_config.to_json(self._model)
3043 #xml = ns_instance_config.to_xml_v2(self._model)
3044 #netconf_xml = self.wrap_netconf_config_xml(xml)
3045
3046 #self._log.debug("Sending configure ns-instance-config xml to %s: %s",
3047 # netconf_xml, NsrRpcDtsHandler.NETCONF_IP_ADDRESS)
3048 self._log.debug("Sending configure ns-instance-config json to %s: %s",
3049 self._nsr_config_url,ns_instance_config)
3050
3051 #response = yield from self._manager.edit_config(
3052 # target="running",
3053 # config=netconf_xml,
3054 # )
3055 response = yield from self._loop.run_in_executor(
3056 None,
3057 self._apply_ns_instance_config,
3058 payload_dict
3059 )
3060 response.raise_for_status()
3061 self._log.debug("Received edit config response: %s", response.json())
3062
3063 xact_info.respond_xpath(rwdts.XactRspCode.ACK,
3064 NsrRpcDtsHandler.EXEC_NSR_CONF_O_XPATH,
3065 rpc_op)
3066 except Exception as e:
3067 self._log.error("Exception processing the "
3068 "start-network-service: {}".format(e))
3069 self._log.exception(e)
3070 xact_info.respond_xpath(rwdts.XactRspCode.NACK,
3071 NsrRpcDtsHandler.EXEC_NSR_CONF_O_XPATH)
3072
3073
3074 hdl_ns = rift.tasklets.DTS.RegistrationHandler(on_prepare=on_ns_config_prepare,)
3075
3076 with self._dts.group_create() as group:
3077 self._ns_regh = group.register(xpath=NsrRpcDtsHandler.EXEC_NSR_CONF_XPATH,
3078 handler=hdl_ns,
3079 flags=rwdts.Flag.PUBLISHER,
3080 )
3081
3082
3083 class NsrDtsHandler(object):
3084 """ The network service DTS handler """
3085 NSR_XPATH = "C,/nsr:ns-instance-config/nsr:nsr"
3086 SCALE_INSTANCE_XPATH = "C,/nsr:ns-instance-config/nsr:nsr/nsr:scaling-group/nsr:instance"
3087 KEY_PAIR_XPATH = "C,/nsr:key-pair"
3088
3089 def __init__(self, dts, log, loop, nsm):
3090 self._dts = dts
3091 self._log = log
3092 self._loop = loop
3093 self._nsm = nsm
3094
3095 self._nsr_regh = None
3096 self._scale_regh = None
3097 self._key_pair_regh = None
3098
3099 @property
3100 def nsm(self):
3101 """ Return the NS manager instance """
3102 return self._nsm
3103
3104 @asyncio.coroutine
3105 def register(self):
3106 """ Register for Nsr create/update/delete/read requests from dts """
3107
3108 def nsr_id_from_keyspec(ks):
3109 nsr_path_entry = NsrYang.YangData_Nsr_NsInstanceConfig_Nsr.schema().keyspec_to_entry(ks)
3110 nsr_id = nsr_path_entry.key00.id
3111 return nsr_id
3112
3113 def group_name_from_keyspec(ks):
3114 group_path_entry = NsrYang.YangData_Nsr_NsInstanceConfig_Nsr_ScalingGroup.schema().keyspec_to_entry(ks)
3115 group_name = group_path_entry.key00.scaling_group_name_ref
3116 return group_name
3117
3118 def is_instance_in_reg_elements(nsr_id, group_name, instance_id):
3119 """ Return boolean indicating if scaling group instance was already commited previously.
3120
3121 By looking at the existing elements in this registration handle (elements not part
3122 of this current xact), we can tell if the instance was configured previously without
3123 keeping any application state.
3124 """
3125 for instance_cfg, keyspec in self._nsr_regh.get_xact_elements(include_keyspec=True):
3126 elem_nsr_id = nsr_id_from_keyspec(keyspec)
3127 elem_group_name = group_name_from_keyspec(keyspec)
3128
3129 if elem_nsr_id != nsr_id or group_name != elem_group_name:
3130 continue
3131
3132 if instance_cfg.id == instance_id:
3133 return True
3134
3135 return False
3136
3137 def get_scale_group_instance_delta(nsr_id, group_name, xact):
3138 delta = {"added": [], "deleted": []}
3139 for instance_cfg, keyspec in self._scale_regh.get_xact_elements(xact, include_keyspec=True):
3140 elem_nsr_id = nsr_id_from_keyspec(keyspec)
3141 if elem_nsr_id != nsr_id:
3142 continue
3143
3144 elem_group_name = group_name_from_keyspec(keyspec)
3145 if elem_group_name != group_name:
3146 continue
3147
3148 delta["added"].append(instance_cfg.id)
3149
3150 for instance_cfg, keyspec in self._scale_regh.get_xact_elements(include_keyspec=True):
3151 elem_nsr_id = nsr_id_from_keyspec(keyspec)
3152 if elem_nsr_id != nsr_id:
3153 continue
3154
3155 elem_group_name = group_name_from_keyspec(keyspec)
3156 if elem_group_name != group_name:
3157 continue
3158
3159 if instance_cfg.id in delta["added"]:
3160 delta["added"].remove(instance_cfg.id)
3161 else:
3162 delta["deleted"].append(instance_cfg.id)
3163
3164 return delta
3165
3166 @asyncio.coroutine
3167 def update_nsr_nsd(nsr_id, xact, scratch):
3168
3169 @asyncio.coroutine
3170 def get_nsr_vl_delta(nsr_id, xact, scratch):
3171 delta = {"added": [], "deleted": []}
3172 for instance_cfg, keyspec in self._nsr_regh.get_xact_elements(xact, include_keyspec=True):
3173 elem_nsr_id = nsr_id_from_keyspec(keyspec)
3174 if elem_nsr_id != nsr_id:
3175 continue
3176
3177 if 'vld' in instance_cfg.nsd:
3178 for vld in instance_cfg.nsd.vld:
3179 delta["added"].append(vld)
3180
3181 for instance_cfg, keyspec in self._nsr_regh.get_xact_elements(include_keyspec=True):
3182 self._log.debug("NSR update: %s", instance_cfg)
3183 elem_nsr_id = nsr_id_from_keyspec(keyspec)
3184 if elem_nsr_id != nsr_id:
3185 continue
3186
3187 if 'vld' in instance_cfg.nsd:
3188 for vld in instance_cfg.nsd.vld:
3189 if vld in delta["added"]:
3190 delta["added"].remove(vld)
3191 else:
3192 delta["deleted"].append(vld)
3193
3194 return delta
3195
3196 vl_delta = yield from get_nsr_vl_delta(nsr_id, xact, scratch)
3197 self._log.debug("Got NSR:%s VL instance delta: %s", nsr_id, vl_delta)
3198
3199 for vld in vl_delta["added"]:
3200 yield from self._nsm.nsr_instantiate_vl(nsr_id, vld)
3201
3202 for vld in vl_delta["deleted"]:
3203 yield from self._nsm.nsr_terminate_vl(nsr_id, vld)
3204
3205 def get_add_delete_update_cfgs(dts_member_reg, xact, key_name, scratch):
3206 # Unfortunately, it is currently difficult to figure out what has exactly
3207 # changed in this xact without Pbdelta support (RIFT-4916)
3208 # As a workaround, we can fetch the pre and post xact elements and
3209 # perform a comparison to figure out adds/deletes/updates
3210 xact_cfgs = list(dts_member_reg.get_xact_elements(xact))
3211 curr_cfgs = list(dts_member_reg.elements)
3212
3213 xact_key_map = {getattr(cfg, key_name): cfg for cfg in xact_cfgs}
3214 curr_key_map = {getattr(cfg, key_name): cfg for cfg in curr_cfgs}
3215
3216 # Find Adds
3217 added_keys = set(xact_key_map) - set(curr_key_map)
3218 added_cfgs = [xact_key_map[key] for key in added_keys]
3219
3220 # Find Deletes
3221 deleted_keys = set(curr_key_map) - set(xact_key_map)
3222 deleted_cfgs = [curr_key_map[key] for key in deleted_keys]
3223
3224 # Find Updates
3225 updated_keys = set(curr_key_map) & set(xact_key_map)
3226 updated_cfgs = [xact_key_map[key] for key in updated_keys
3227 if xact_key_map[key] != curr_key_map[key]]
3228
3229 return added_cfgs, deleted_cfgs, updated_cfgs
3230
3231 def get_nsr_key_pairs(dts_member_reg, xact):
3232 key_pairs = {}
3233 for instance_cfg, keyspec in dts_member_reg.get_xact_elements(xact, include_keyspec=True):
3234 self._log.debug("Key pair received is {} KS: {}".format(instance_cfg, keyspec))
3235 xpath = keyspec.to_xpath(RwNsrYang.get_schema())
3236 key_pairs[instance_cfg.name] = instance_cfg
3237 return key_pairs
3238
3239 def on_apply(dts, acg, xact, action, scratch):
3240 """Apply the configuration"""
3241 self._log.debug("Got nsr apply (xact: %s) (action: %s)(scr: %s)",
3242 xact, action, scratch)
3243
3244 def handle_create_nsr(msg, key_pairs=None, restart_mode=False):
3245 # Handle create nsr requests """
3246 # Do some validations
3247 if not msg.has_field("nsd"):
3248 err = "NSD not provided"
3249 self._log.error(err)
3250 raise NetworkServiceRecordError(err)
3251
3252 self._log.debug("Creating NetworkServiceRecord %s from nsr config %s",
3253 msg.id, msg.as_dict())
3254 nsr = self.nsm.create_nsr(msg, key_pairs=key_pairs, restart_mode=restart_mode)
3255 return nsr
3256
3257 def handle_delete_nsr(msg):
3258 @asyncio.coroutine
3259 def delete_instantiation(ns_id):
3260 """ Delete instantiation """
3261 with self._dts.transaction() as xact:
3262 yield from self._nsm.terminate_ns(ns_id, xact)
3263
3264 # Handle delete NSR requests
3265 self._log.info("Delete req for NSR Id: %s received", msg.id)
3266 # Terminate the NSR instance
3267 nsr = self._nsm.get_ns_by_nsr_id(msg.id)
3268
3269 nsr.set_state(NetworkServiceRecordState.TERMINATE_RCVD)
3270 event_descr = "Terminate rcvd for NS Id:%s" % msg.id
3271 nsr.record_event("terminate-rcvd", event_descr)
3272
3273 self._loop.create_task(delete_instantiation(msg.id))
3274
3275 @asyncio.coroutine
3276 def begin_instantiation(nsr):
3277 # Begin instantiation
3278 self._log.info("Beginning NS instantiation: %s", nsr.id)
3279 yield from self._nsm.instantiate_ns(nsr.id, xact)
3280
3281 self._log.debug("Got nsr apply (xact: %s) (action: %s)(scr: %s)",
3282 xact, action, scratch)
3283
3284 if action == rwdts.AppconfAction.INSTALL and xact.id is None:
3285 key_pairs = []
3286 for element in self._key_pair_regh.elements:
3287 key_pairs.append(element)
3288 for element in self._nsr_regh.elements:
3289 nsr = handle_create_nsr(element, key_pairs, restart_mode=True)
3290 self._loop.create_task(begin_instantiation(nsr))
3291
3292
3293 (added_msgs, deleted_msgs, updated_msgs) = get_add_delete_update_cfgs(self._nsr_regh,
3294 xact,
3295 "id",
3296 scratch)
3297 self._log.debug("Added: %s, Deleted: %s, Updated: %s", added_msgs,
3298 deleted_msgs, updated_msgs)
3299
3300 for msg in added_msgs:
3301 if msg.id not in self._nsm.nsrs:
3302 self._log.info("Create NSR received in on_apply to instantiate NS:%s", msg.id)
3303 key_pairs = get_nsr_key_pairs(self._key_pair_regh, xact)
3304 nsr = handle_create_nsr(msg,key_pairs)
3305 self._loop.create_task(begin_instantiation(nsr))
3306
3307 for msg in deleted_msgs:
3308 self._log.info("Delete NSR received in on_apply to terminate NS:%s", msg.id)
3309 try:
3310 handle_delete_nsr(msg)
3311 except Exception:
3312 self._log.exception("Failed to terminate NS:%s", msg.id)
3313
3314 for msg in updated_msgs:
3315 self._log.info("Update NSR received in on_apply: %s", msg)
3316
3317 self._nsm.nsr_update_cfg(msg.id, msg)
3318
3319 if 'nsd' in msg:
3320 self._loop.create_task(update_nsr_nsd(msg.id, xact, scratch))
3321
3322 for group in msg.scaling_group:
3323 instance_delta = get_scale_group_instance_delta(msg.id, group.scaling_group_name_ref, xact)
3324 self._log.debug("Got NSR:%s scale group instance delta: %s", msg.id, instance_delta)
3325
3326 for instance_id in instance_delta["added"]:
3327 self._nsm.scale_nsr_out(msg.id, group.scaling_group_name_ref, instance_id, xact)
3328
3329 for instance_id in instance_delta["deleted"]:
3330 self._nsm.scale_nsr_in(msg.id, group.scaling_group_name_ref, instance_id)
3331
3332
3333 return RwTypes.RwStatus.SUCCESS
3334
3335 @asyncio.coroutine
3336 def on_prepare(dts, acg, xact, xact_info, ks_path, msg, scratch):
3337 """ Prepare calllback from DTS for NSR """
3338
3339 xpath = ks_path.to_xpath(RwNsrYang.get_schema())
3340 action = xact_info.query_action
3341 self._log.debug(
3342 "Got Nsr prepare callback (xact: %s) (action: %s) (info: %s), %s:%s)",
3343 xact, action, xact_info, xpath, msg
3344 )
3345
3346 @asyncio.coroutine
3347 def delete_instantiation(ns_id):
3348 """ Delete instantiation """
3349 yield from self._nsm.terminate_ns(ns_id, None)
3350
3351 def handle_delete_nsr():
3352 """ Handle delete NSR requests """
3353 self._log.info("Delete req for NSR Id: %s received", msg.id)
3354 # Terminate the NSR instance
3355 nsr = self._nsm.get_ns_by_nsr_id(msg.id)
3356
3357 nsr.set_state(NetworkServiceRecordState.TERMINATE_RCVD)
3358 event_descr = "Terminate rcvd for NS Id:%s" % msg.id
3359 nsr.record_event("terminate-rcvd", event_descr)
3360
3361 self._loop.create_task(delete_instantiation(msg.id))
3362
3363 fref = ProtobufC.FieldReference.alloc()
3364 fref.goto_whole_message(msg.to_pbcm())
3365
3366 if action in [rwdts.QueryAction.CREATE, rwdts.QueryAction.UPDATE, rwdts.QueryAction.DELETE]:
3367 # if this is an NSR create
3368 if action != rwdts.QueryAction.DELETE and msg.id not in self._nsm.nsrs:
3369 # Ensure the Cloud account/datacenter has been specified
3370 if not msg.has_field("cloud_account") and not msg.has_field("om_datacenter"):
3371 raise NsrInstantiationFailed("Cloud account or datacenter not specified in NSR")
3372
3373 # Check if nsd is specified
3374 if not msg.has_field("nsd"):
3375 raise NsrInstantiationFailed("NSD not specified in NSR")
3376
3377 else:
3378 nsr = self._nsm.nsrs[msg.id]
3379
3380 if msg.has_field("nsd"):
3381 if nsr.state != NetworkServiceRecordState.RUNNING:
3382 raise NsrVlUpdateError("Unable to update VL when NSR not in running state")
3383 if 'vld' not in msg.nsd or len(msg.nsd.vld) == 0:
3384 raise NsrVlUpdateError("NS config NSD should have atleast 1 VLD defined")
3385
3386 if msg.has_field("scaling_group"):
3387 self._log.debug("ScaleMsg %s", msg)
3388 self._log.debug("NSSCALINGSTATE %s", nsr.state)
3389 if nsr.state != NetworkServiceRecordState.RUNNING:
3390 raise ScalingOperationError("Unable to perform scaling action when NS is not in running state")
3391
3392 if len(msg.scaling_group) > 1:
3393 raise ScalingOperationError("Only a single scaling group can be configured at a time")
3394
3395 for group_msg in msg.scaling_group:
3396 num_new_group_instances = len(group_msg.instance)
3397 if num_new_group_instances > 1:
3398 raise ScalingOperationError("Only a single scaling instance can be modified at a time")
3399
3400 elif num_new_group_instances == 1:
3401 scale_group = nsr.scaling_groups[group_msg.scaling_group_name_ref]
3402 if action in [rwdts.QueryAction.CREATE, rwdts.QueryAction.UPDATE]:
3403 if len(scale_group.instances) == scale_group.max_instance_count:
3404 raise ScalingOperationError("Max instances for %s reached" % scale_group)
3405
3406 acg.handle.prepare_complete_ok(xact_info.handle)
3407
3408
3409 self._log.debug("Registering for NSR config using xpath: %s",
3410 NsrDtsHandler.NSR_XPATH)
3411
3412 acg_hdl = rift.tasklets.AppConfGroup.Handler(on_apply=on_apply)
3413 with self._dts.appconf_group_create(handler=acg_hdl) as acg:
3414 self._nsr_regh = acg.register(xpath=NsrDtsHandler.NSR_XPATH,
3415 flags=rwdts.Flag.SUBSCRIBER | rwdts.Flag.DELTA_READY | rwdts.Flag.CACHE,
3416 on_prepare=on_prepare)
3417
3418 self._scale_regh = acg.register(
3419 xpath=NsrDtsHandler.SCALE_INSTANCE_XPATH,
3420 flags=rwdts.Flag.SUBSCRIBER | rwdts.Flag.DELTA_READY| rwdts.Flag.CACHE,
3421 )
3422
3423 self._key_pair_regh = acg.register(
3424 xpath=NsrDtsHandler.KEY_PAIR_XPATH,
3425 flags=rwdts.Flag.SUBSCRIBER | rwdts.Flag.DELTA_READY | rwdts.Flag.CACHE,
3426 )
3427
3428
3429 class NsrOpDataDtsHandler(object):
3430 """ The network service op data DTS handler """
3431 XPATH = "D,/nsr:ns-instance-opdata/nsr:nsr"
3432
3433 def __init__(self, dts, log, loop, nsm):
3434 self._dts = dts
3435 self._log = log
3436 self._loop = loop
3437 self._nsm = nsm
3438 self._regh = None
3439
3440 @property
3441 def regh(self):
3442 """ Return the registration handle"""
3443 return self._regh
3444
3445 @property
3446 def nsm(self):
3447 """ Return the NS manager instance """
3448 return self._nsm
3449
3450 @asyncio.coroutine
3451 def register(self):
3452 """ Register for Nsr op data publisher registration"""
3453 self._log.debug("Registering Nsr op data path %s as publisher",
3454 NsrOpDataDtsHandler.XPATH)
3455
3456 hdl = rift.tasklets.DTS.RegistrationHandler()
3457 handlers = rift.tasklets.Group.Handler()
3458 with self._dts.group_create(handler=handlers) as group:
3459 self._regh = group.register(xpath=NsrOpDataDtsHandler.XPATH,
3460 handler=hdl,
3461 flags=rwdts.Flag.PUBLISHER | rwdts.Flag.NO_PREP_READ | rwdts.Flag.DATASTORE)
3462
3463 @asyncio.coroutine
3464 def create(self, path, msg):
3465 """
3466 Create an NS record in DTS with the path and message
3467 """
3468 self._log.debug("Creating NSR %s:%s", path, msg)
3469 self.regh.create_element(path, msg)
3470 self._log.debug("Created NSR, %s:%s", path, msg)
3471
3472 @asyncio.coroutine
3473 def update(self, path, msg, flags=rwdts.XactFlag.REPLACE):
3474 """
3475 Update an NS record in DTS with the path and message
3476 """
3477 self._log.debug("Updating NSR, %s:%s regh = %s", path, msg, self.regh)
3478 self.regh.update_element(path, msg, flags)
3479 self._log.debug("Updated NSR, %s:%s", path, msg)
3480
3481 @asyncio.coroutine
3482 def delete(self, path):
3483 """
3484 Update an NS record in DTS with the path and message
3485 """
3486 self._log.debug("Deleting NSR path:%s", path)
3487 self.regh.delete_element(path)
3488 self._log.debug("Deleted NSR path:%s", path)
3489
3490
3491 class VnfrDtsHandler(object):
3492 """ The virtual network service DTS handler """
3493 XPATH = "D,/vnfr:vnfr-catalog/vnfr:vnfr"
3494
3495 def __init__(self, dts, log, loop, nsm):
3496 self._dts = dts
3497 self._log = log
3498 self._loop = loop
3499 self._nsm = nsm
3500
3501 self._regh = None
3502
3503 @property
3504 def regh(self):
3505 """ Return registration handle """
3506 return self._regh
3507
3508 @property
3509 def nsm(self):
3510 """ Return the NS manager instance """
3511 return self._nsm
3512
3513 @asyncio.coroutine
3514 def register(self):
3515 """ Register for vnfr create/update/delete/ advises from dts """
3516
3517 def on_commit(xact_info):
3518 """ The transaction has been committed """
3519 self._log.debug("Got vnfr commit (xact_info: %s)", xact_info)
3520 return rwdts.MemberRspCode.ACTION_OK
3521
3522 @asyncio.coroutine
3523 def on_prepare(xact_info, action, ks_path, msg):
3524 """ prepare callback from dts """
3525 xpath = ks_path.to_xpath(RwNsrYang.get_schema())
3526 self._log.debug(
3527 "Got vnfr on_prepare cb (xact_info: %s, action: %s): %s:%s",
3528 xact_info, action, ks_path, msg
3529 )
3530
3531 schema = VnfrYang.YangData_Vnfr_VnfrCatalog_Vnfr.schema()
3532 path_entry = schema.keyspec_to_entry(ks_path)
3533 if path_entry.key00.id not in self._nsm._vnfrs:
3534 self._log.error("%s request for non existent record path %s",
3535 action, xpath)
3536 xact_info.respond_xpath(rwdts.XactRspCode.NA, xpath)
3537
3538 return
3539
3540 self._log.debug("Deleting VNFR with id %s", path_entry.key00.id)
3541 if action == rwdts.QueryAction.CREATE or action == rwdts.QueryAction.UPDATE:
3542 yield from self._nsm.update_vnfr(msg)
3543 elif action == rwdts.QueryAction.DELETE:
3544 self._log.debug("Deleting VNFR with id %s", path_entry.key00.id)
3545 self._nsm.delete_vnfr(path_entry.key00.id)
3546
3547 xact_info.respond_xpath(rwdts.XactRspCode.ACK, xpath)
3548
3549 self._log.debug("Registering for VNFR using xpath: %s",
3550 VnfrDtsHandler.XPATH,)
3551
3552 hdl = rift.tasklets.DTS.RegistrationHandler(on_commit=on_commit,
3553 on_prepare=on_prepare,)
3554 with self._dts.group_create() as group:
3555 self._regh = group.register(xpath=VnfrDtsHandler.XPATH,
3556 handler=hdl,
3557 flags=(rwdts.Flag.SUBSCRIBER),)
3558
3559
3560 class NsdRefCountDtsHandler(object):
3561 """ The NSD Ref Count DTS handler """
3562 XPATH = "D,/nsr:ns-instance-opdata/rw-nsr:nsd-ref-count"
3563
3564 def __init__(self, dts, log, loop, nsm):
3565 self._dts = dts
3566 self._log = log
3567 self._loop = loop
3568 self._nsm = nsm
3569
3570 self._regh = None
3571
3572 @property
3573 def regh(self):
3574 """ Return registration handle """
3575 return self._regh
3576
3577 @property
3578 def nsm(self):
3579 """ Return the NS manager instance """
3580 return self._nsm
3581
3582 @asyncio.coroutine
3583 def register(self):
3584 """ Register for NSD ref count read from dts """
3585
3586 @asyncio.coroutine
3587 def on_prepare(xact_info, action, ks_path, msg):
3588 """ prepare callback from dts """
3589 xpath = ks_path.to_xpath(RwNsrYang.get_schema())
3590
3591 if action == rwdts.QueryAction.READ:
3592 schema = RwNsrYang.YangData_Nsr_NsInstanceOpdata_NsdRefCount.schema()
3593 path_entry = schema.keyspec_to_entry(ks_path)
3594 nsd_list = yield from self._nsm.get_nsd_refcount(path_entry.key00.nsd_id_ref)
3595 for xpath, msg in nsd_list:
3596 xact_info.respond_xpath(rsp_code=rwdts.XactRspCode.MORE,
3597 xpath=xpath,
3598 msg=msg)
3599 xact_info.respond_xpath(rwdts.XactRspCode.ACK)
3600 else:
3601 raise NetworkServiceRecordError("Not supported operation %s" % action)
3602
3603 hdl = rift.tasklets.DTS.RegistrationHandler(on_prepare=on_prepare,)
3604 with self._dts.group_create() as group:
3605 self._regh = group.register(xpath=NsdRefCountDtsHandler.XPATH,
3606 handler=hdl,
3607 flags=rwdts.Flag.PUBLISHER,)
3608
3609
3610 class NsManager(object):
3611 """ The Network Service Manager class"""
3612 def __init__(self, dts, log, loop,
3613 nsr_handler, vnfr_handler, vlr_handler, ro_plugin_selector,
3614 vnffgmgr, vnfd_pub_handler, cloud_account_handler):
3615 self._dts = dts
3616 self._log = log
3617 self._loop = loop
3618 self._nsr_handler = nsr_handler
3619 self._vnfr_pub_handler = vnfr_handler
3620 self._vlr_pub_handler = vlr_handler
3621 self._vnffgmgr = vnffgmgr
3622 self._vnfd_pub_handler = vnfd_pub_handler
3623 self._cloud_account_handler = cloud_account_handler
3624
3625 self._ro_plugin_selector = ro_plugin_selector
3626
3627 # Intialize the set of variables for implementing Scaling RPC using REST.
3628 self._headers = {"content-type":"application/json", "accept":"application/json"}
3629 #This will break when we have rbac in the rift code and admin user password is changed or admin it self is removed.
3630 self._user = 'admin'
3631 self._password = 'admin'
3632 self._ip = 'localhost'
3633 self._rport = 8008
3634 self._conf_url = "https://{ip}:{port}/api/config". \
3635 format(ip=self._ip,
3636 port=self._rport)
3637
3638 self._nsrs = {}
3639 self._nsds = {}
3640 self._vnfds = {}
3641 self._vnfrs = {}
3642
3643 self.cfgmgr_obj = conman.ROConfigManager(log, loop, dts, self)
3644
3645 # TODO: All these handlers should move to tasklet level.
3646 # Passing self is often an indication of bad design
3647 self._nsd_dts_handler = NsdDtsHandler(dts, log, loop, self)
3648 self._vnfd_dts_handler = VnfdDtsHandler(dts, log, loop, self)
3649 self._dts_handlers = [self._nsd_dts_handler,
3650 VnfrDtsHandler(dts, log, loop, self),
3651 NsdRefCountDtsHandler(dts, log, loop, self),
3652 NsrDtsHandler(dts, log, loop, self),
3653 ScalingRpcHandler(log, dts, loop, self.scale_rpc_callback),
3654 NsrRpcDtsHandler(dts,log,loop,self),
3655 self._vnfd_dts_handler,
3656 self.cfgmgr_obj,
3657 ]
3658
3659
3660 @property
3661 def log(self):
3662 """ Log handle """
3663 return self._log
3664
3665 @property
3666 def loop(self):
3667 """ Loop """
3668 return self._loop
3669
3670 @property
3671 def dts(self):
3672 """ DTS handle """
3673 return self._dts
3674
3675 @property
3676 def nsr_handler(self):
3677 """" NSR handler """
3678 return self._nsr_handler
3679
3680 @property
3681 def so_obj(self):
3682 """" So Obj handler """
3683 return self._so_obj
3684
3685 @property
3686 def nsrs(self):
3687 """ NSRs in this NSM"""
3688 return self._nsrs
3689
3690 @property
3691 def nsds(self):
3692 """ NSDs in this NSM"""
3693 return self._nsds
3694
3695 @property
3696 def vnfds(self):
3697 """ VNFDs in this NSM"""
3698 return self._vnfds
3699
3700 @property
3701 def vnfrs(self):
3702 """ VNFRs in this NSM"""
3703 return self._vnfrs
3704
3705 @property
3706 def nsr_pub_handler(self):
3707 """ NSR publication handler """
3708 return self._nsr_handler
3709
3710 @property
3711 def vnfr_pub_handler(self):
3712 """ VNFR publication handler """
3713 return self._vnfr_pub_handler
3714
3715 @property
3716 def vlr_pub_handler(self):
3717 """ VLR publication handler """
3718 return self._vlr_pub_handler
3719
3720 @property
3721 def vnfd_pub_handler(self):
3722 return self._vnfd_pub_handler
3723
3724 @asyncio.coroutine
3725 def register(self):
3726 """ Register all static DTS handlers """
3727 for dts_handle in self._dts_handlers:
3728 yield from dts_handle.register()
3729
3730
3731 def get_ns_by_nsr_id(self, nsr_id):
3732 """ get NSR by nsr id """
3733 if nsr_id not in self._nsrs:
3734 raise NetworkServiceRecordError("NSR id %s not found" % nsr_id)
3735
3736 return self._nsrs[nsr_id]
3737
3738 def scale_nsr_out(self, nsr_id, scale_group_name, instance_id, config_xact):
3739 self.log.debug("Scale out NetworkServiceRecord (nsr_id: %s) (scaling group: %s) (instance_id: %s)",
3740 nsr_id,
3741 scale_group_name,
3742 instance_id
3743 )
3744 nsr = self._nsrs[nsr_id]
3745 if nsr.state != NetworkServiceRecordState.RUNNING:
3746 raise ScalingOperationError("Cannot perform scaling operation if NSR is not in running state")
3747
3748 self._loop.create_task(nsr.create_scale_group_instance(scale_group_name, instance_id, config_xact))
3749
3750 def scale_nsr_in(self, nsr_id, scale_group_name, instance_id):
3751 self.log.debug("Scale in NetworkServiceRecord (nsr_id: %s) (scaling group: %s) (instance_id: %s)",
3752 nsr_id,
3753 scale_group_name,
3754 instance_id,
3755 )
3756 nsr = self._nsrs[nsr_id]
3757 if nsr.state != NetworkServiceRecordState.RUNNING:
3758 raise ScalingOperationError("Cannot perform scaling operation if NSR is not in running state")
3759
3760 self._loop.create_task(nsr.delete_scale_group_instance(scale_group_name, instance_id))
3761
3762 def scale_rpc_callback(self, xact, msg, action):
3763 """Callback handler for RPC calls
3764 Args:
3765 xact : Transaction Handler
3766 msg : RPC input
3767 action : Scaling Action
3768 """
3769 def get_scaling_group_information():
3770 scaling_group_url = "{url}/ns-instance-config/nsr/{nsr_id}".format(url=self._conf_url, nsr_id=msg.nsr_id_ref)
3771 output = requests.get(scaling_group_url, headers=self._headers, auth=(self._user, self._password), verify=False)
3772 if output.text == None or len(output.text) == 0:
3773 self.log.error("nsr id %s information not present", self._nsr_id)
3774 return None
3775 scaling_group_info = json.loads(output.text)
3776 return scaling_group_info
3777
3778 def config_scaling_group_information(scaling_group_info):
3779 data_str = json.dumps(scaling_group_info)
3780 self.log.debug("scaling group Info %s", data_str)
3781
3782 scale_out_url = "{url}/ns-instance-config/nsr/{nsr_id}".format(url=self._conf_url, nsr_id=msg.nsr_id_ref)
3783 response = requests.put(scale_out_url, data=data_str, verify=False, auth=(self._user, self._password), headers=self._headers)
3784 response.raise_for_status()
3785
3786 def scale_out():
3787 scaling_group_info = get_scaling_group_information()
3788 if scaling_group_info is None:
3789 return
3790
3791 scaling_group_present = False
3792 if "scaling-group" in scaling_group_info["nsr:nsr"]:
3793 scaling_group_array = scaling_group_info["nsr:nsr"]["scaling-group"]
3794 for scaling_group in scaling_group_array:
3795 if scaling_group["scaling-group-name-ref"] == msg.scaling_group_name_ref:
3796 scaling_group_present = True
3797 if 'instance' not in scaling_group:
3798 scaling_group['instance'] = []
3799 for instance in scaling_group['instance']:
3800 if instance["id"] == int(msg.instance_id):
3801 self.log.error("scaling group with instance id %s exists for scale out", msg.instance_id)
3802 return
3803 scaling_group["instance"].append({"id": int(msg.instance_id)})
3804
3805 if not scaling_group_present:
3806 scaling_group_info["nsr:nsr"]["scaling-group"] = [{"scaling-group-name-ref": msg.scaling_group_name_ref, "instance": [{"id": msg.instance_id}]}]
3807
3808 config_scaling_group_information(scaling_group_info)
3809 return
3810
3811 def scale_in():
3812 scaling_group_info = get_scaling_group_information()
3813 if scaling_group_info is None:
3814 return
3815
3816 scaling_group_array = scaling_group_info["nsr:nsr"]["scaling-group"]
3817 scaling_group_present = False
3818 instance_id_present = False
3819 for scaling_group in scaling_group_array:
3820 if scaling_group["scaling-group-name-ref"] == msg.scaling_group_name_ref:
3821 scaling_group_present = True
3822 if 'instance' in scaling_group:
3823 instance_array = scaling_group["instance"];
3824 for index in range(len(instance_array)):
3825 if instance_array[index]["id"] == int(msg.instance_id):
3826 instance_array.pop(index)
3827 instance_id_present = True
3828 break
3829
3830 if not scaling_group_present:
3831 self.log.error("Scaling group %s doesnot exists for scale in", msg.scaling_group_name_ref)
3832 return
3833
3834 if not instance_id_present:
3835 self.log.error("Instance id %s doesnot exists for scale in", msg.instance_id)
3836 return
3837
3838 config_scaling_group_information(scaling_group_info)
3839 return
3840
3841 if action == ScalingRpcHandler.ACTION.SCALE_OUT:
3842 self._loop.run_in_executor(None, scale_out)
3843 else:
3844 self._loop.run_in_executor(None, scale_in)
3845
3846 def nsr_update_cfg(self, nsr_id, msg):
3847 nsr = self._nsrs[nsr_id]
3848 nsr.nsr_cfg_msg= msg
3849
3850 def nsr_instantiate_vl(self, nsr_id, vld):
3851 self.log.debug("NSR {} create VL {}".format(nsr_id, vld))
3852 nsr = self._nsrs[nsr_id]
3853 if nsr.state != NetworkServiceRecordState.RUNNING:
3854 raise NsrVlUpdateError("Cannot perform VL instantiate if NSR is not in running state")
3855
3856 # Not calling in a separate task as this is called from a separate task
3857 yield from nsr.create_vl_instance(vld)
3858
3859 def nsr_terminate_vl(self, nsr_id, vld):
3860 self.log.debug("NSR {} delete VL {}".format(nsr_id, vld.id))
3861 nsr = self._nsrs[nsr_id]
3862 if nsr.state != NetworkServiceRecordState.RUNNING:
3863 raise NsrVlUpdateError("Cannot perform VL terminate if NSR is not in running state")
3864
3865 # Not calling in a separate task as this is called from a separate task
3866 yield from nsr.delete_vl_instance(vld)
3867
3868 def create_nsr(self, nsr_msg, key_pairs=None,restart_mode=False):
3869 """ Create an NSR instance """
3870 self._log.debug("NSRMSG %s", nsr_msg)
3871 if nsr_msg.id in self._nsrs:
3872 msg = "NSR id %s already exists" % nsr_msg.id
3873 self._log.error(msg)
3874 raise NetworkServiceRecordError(msg)
3875
3876 self._log.info("Create NetworkServiceRecord nsr id %s from nsd_id %s",
3877 nsr_msg.id,
3878 nsr_msg.nsd.id)
3879
3880 nsm_plugin = self._ro_plugin_selector.ro_plugin
3881 sdn_account_name = self._cloud_account_handler.get_cloud_account_sdn_name(nsr_msg.cloud_account)
3882
3883 nsr = NetworkServiceRecord(self._dts,
3884 self._log,
3885 self._loop,
3886 self,
3887 nsm_plugin,
3888 nsr_msg,
3889 sdn_account_name,
3890 key_pairs,
3891 restart_mode=restart_mode,
3892 vlr_handler=self._ro_plugin_selector._records_publisher._vlr_pub_hdlr
3893 )
3894 self._nsrs[nsr_msg.id] = nsr
3895 nsm_plugin.create_nsr(nsr_msg, nsr_msg.nsd, key_pairs)
3896
3897 return nsr
3898
3899 def delete_nsr(self, nsr_id):
3900 """
3901 Delete NSR with the passed nsr id
3902 """
3903 del self._nsrs[nsr_id]
3904
3905 @asyncio.coroutine
3906 def instantiate_ns(self, nsr_id, config_xact):
3907 """ Instantiate an NS instance """
3908 self._log.debug("Instantiating Network service id %s", nsr_id)
3909 if nsr_id not in self._nsrs:
3910 err = "NSR id %s not found " % nsr_id
3911 self._log.error(err)
3912 raise NetworkServiceRecordError(err)
3913
3914 nsr = self._nsrs[nsr_id]
3915 yield from nsr.nsm_plugin.instantiate_ns(nsr, config_xact)
3916
3917 @asyncio.coroutine
3918 def update_vnfr(self, vnfr):
3919 """Create/Update an VNFR """
3920
3921 vnfr_state = self._vnfrs[vnfr.id].state
3922 self._log.debug("Updating VNFR with state %s: vnfr %s", vnfr_state, vnfr)
3923
3924 yield from self._vnfrs[vnfr.id].update_state(vnfr)
3925 nsr = self.find_nsr_for_vnfr(vnfr.id)
3926 yield from nsr.update_state()
3927
3928 def find_nsr_for_vnfr(self, vnfr_id):
3929 """ Find the NSR which )has the passed vnfr id"""
3930 for nsr in list(self.nsrs.values()):
3931 for vnfr in list(nsr.vnfrs.values()):
3932 if vnfr.id == vnfr_id:
3933 return nsr
3934 return None
3935
3936 def delete_vnfr(self, vnfr_id):
3937 """ Delete VNFR with the passed id"""
3938 del self._vnfrs[vnfr_id]
3939
3940 def get_nsd_ref(self, nsd_id):
3941 """ Get network service descriptor for the passed nsd_id
3942 with a reference"""
3943 nsd = self.get_nsd(nsd_id)
3944 nsd.ref()
3945 return nsd
3946
3947 @asyncio.coroutine
3948 def get_nsr_config(self, nsd_id):
3949 xpath = "C,/nsr:ns-instance-config"
3950 results = yield from self._dts.query_read(xpath, rwdts.XactFlag.MERGE)
3951
3952 for result in results:
3953 entry = yield from result
3954 ns_instance_config = entry.result
3955
3956 for nsr in ns_instance_config.nsr:
3957 if nsr.nsd.id == nsd_id:
3958 return nsr
3959
3960 return None
3961
3962 @asyncio.coroutine
3963 def nsd_unref_by_nsr_id(self, nsr_id):
3964 """ Unref the network service descriptor based on NSR id """
3965 self._log.debug("NSR Unref called for Nsr Id:%s", nsr_id)
3966 if nsr_id in self._nsrs:
3967 nsr = self._nsrs[nsr_id]
3968
3969 try:
3970 nsd = self.get_nsd(nsr.nsd_id)
3971 self._log.debug("Releasing ref on NSD %s held by NSR %s - Curr %d",
3972 nsd.id, nsr.id, nsd.ref_count)
3973 nsd.unref()
3974 except NetworkServiceDescriptorError:
3975 # We store a copy of NSD in NSR and the NSD in nsd-catalog
3976 # could be deleted
3977 pass
3978
3979 else:
3980 self._log.error("Cannot find NSR with id %s", nsr_id)
3981 raise NetworkServiceDescriptorUnrefError("No NSR with id" % nsr_id)
3982
3983 @asyncio.coroutine
3984 def nsd_unref(self, nsd_id):
3985 """ Unref the network service descriptor associated with the id """
3986 nsd = self.get_nsd(nsd_id)
3987 nsd.unref()
3988
3989 def get_nsd(self, nsd_id):
3990 """ Get network service descriptor for the passed nsd_id"""
3991 if nsd_id not in self._nsds:
3992 self._log.error("Cannot find NSD id:%s", nsd_id)
3993 raise NetworkServiceDescriptorError("Cannot find NSD id:%s", nsd_id)
3994
3995 return self._nsds[nsd_id]
3996
3997 def create_nsd(self, nsd_msg):
3998 """ Create a network service descriptor """
3999 self._log.debug("Create network service descriptor - %s", nsd_msg)
4000 if nsd_msg.id in self._nsds:
4001 self._log.error("Cannot create NSD %s -NSD ID already exists", nsd_msg)
4002 raise NetworkServiceDescriptorError("NSD already exists-%s", nsd_msg.id)
4003
4004 nsd = NetworkServiceDescriptor(
4005 self._dts,
4006 self._log,
4007 self._loop,
4008 nsd_msg,
4009 self
4010 )
4011 self._nsds[nsd_msg.id] = nsd
4012
4013 return nsd
4014
4015 def update_nsd(self, nsd):
4016 """ update the Network service descriptor """
4017 self._log.debug("Update network service descriptor - %s", nsd)
4018 if nsd.id not in self._nsds:
4019 self._log.debug("No NSD found - creating NSD id = %s", nsd.id)
4020 self.create_nsd(nsd)
4021 else:
4022 self._log.debug("Updating NSD id = %s, nsd = %s", nsd.id, nsd)
4023 self._nsds[nsd.id].update(nsd)
4024
4025 def delete_nsd(self, nsd_id):
4026 """ Delete the Network service descriptor with the passed id """
4027 self._log.debug("Deleting the network service descriptor - %s", nsd_id)
4028 if nsd_id not in self._nsds:
4029 self._log.debug("Delete NSD failed - cannot find nsd-id %s", nsd_id)
4030 raise NetworkServiceDescriptorNotFound("Cannot find %s", nsd_id)
4031
4032 if nsd_id not in self._nsds:
4033 self._log.debug("Cannot delete NSD id %s reference exists %s",
4034 nsd_id,
4035 self._nsds[nsd_id].ref_count)
4036 raise NetworkServiceDescriptorRefCountExists(
4037 "Cannot delete :%s, ref_count:%s",
4038 nsd_id,
4039 self._nsds[nsd_id].ref_count)
4040
4041 del self._nsds[nsd_id]
4042
4043 def get_vnfd_config(self, xact):
4044 vnfd_dts_reg = self._vnfd_dts_handler.regh
4045 for cfg in vnfd_dts_reg.get_xact_elements(xact):
4046 if cfg.id not in self._vnfds:
4047 self.create_vnfd(cfg)
4048
4049 def get_vnfd(self, vnfd_id, xact):
4050 """ Get virtual network function descriptor for the passed vnfd_id"""
4051 if vnfd_id not in self._vnfds:
4052 self._log.error("Cannot find VNFD id:%s", vnfd_id)
4053 self.get_vnfd_config(xact)
4054
4055 if vnfd_id not in self._vnfds:
4056 self._log.error("Cannot find VNFD id:%s", vnfd_id)
4057 raise VnfDescriptorError("Cannot find VNFD id:%s", vnfd_id)
4058
4059 return self._vnfds[vnfd_id]
4060
4061 def create_vnfd(self, vnfd):
4062 """ Create a virtual network function descriptor """
4063 self._log.debug("Create virtual network function descriptor - %s", vnfd)
4064 if vnfd.id in self._vnfds:
4065 self._log.error("Cannot create VNFD %s -VNFD ID already exists", vnfd)
4066 raise VnfDescriptorError("VNFD already exists-%s", vnfd.id)
4067
4068 self._vnfds[vnfd.id] = vnfd
4069 return self._vnfds[vnfd.id]
4070
4071 def update_vnfd(self, vnfd):
4072 """ Update the virtual network function descriptor """
4073 self._log.debug("Update virtual network function descriptor- %s", vnfd)
4074
4075
4076 if vnfd.id not in self._vnfds:
4077 self._log.debug("No VNFD found - creating VNFD id = %s", vnfd.id)
4078 self.create_vnfd(vnfd)
4079 else:
4080 self._log.debug("Updating VNFD id = %s, vnfd = %s", vnfd.id, vnfd)
4081 self._vnfds[vnfd.id] = vnfd
4082
4083 @asyncio.coroutine
4084 def delete_vnfd(self, vnfd_id):
4085 """ Delete the virtual network function descriptor with the passed id """
4086 self._log.debug("Deleting the virtual network function descriptor - %s", vnfd_id)
4087 if vnfd_id not in self._vnfds:
4088 self._log.debug("Delete VNFD failed - cannot find vnfd-id %s", vnfd_id)
4089 raise VnfDescriptorError("Cannot find %s", vnfd_id)
4090
4091 del self._vnfds[vnfd_id]
4092
4093 def nsd_in_use(self, nsd_id):
4094 """ Is the NSD with the passed id in use """
4095 self._log.debug("Is this NSD in use - msg:%s", nsd_id)
4096 if nsd_id in self._nsds:
4097 return self._nsds[nsd_id].in_use()
4098 return False
4099
4100 @asyncio.coroutine
4101 def publish_nsr(self, xact, path, msg):
4102 """ Publish a NSR """
4103 self._log.debug("Publish NSR with path %s, msg %s",
4104 path, msg)
4105 yield from self.nsr_handler.update(xact, path, msg)
4106
4107 @asyncio.coroutine
4108 def unpublish_nsr(self, xact, path):
4109 """ Un Publish an NSR """
4110 self._log.debug("Publishing delete NSR with path %s", path)
4111 yield from self.nsr_handler.delete(path, xact)
4112
4113 def vnfr_is_ready(self, vnfr_id):
4114 """ VNFR with the id is ready """
4115 self._log.debug("VNFR id %s ready", vnfr_id)
4116 if vnfr_id not in self._vnfds:
4117 err = "Did not find VNFR ID with id %s" % vnfr_id
4118 self._log.critical("err")
4119 raise VirtualNetworkFunctionRecordError(err)
4120 self._vnfrs[vnfr_id].is_ready()
4121
4122 @asyncio.coroutine
4123 def get_nsd_refcount(self, nsd_id):
4124 """ Get the nsd_list from this NSM"""
4125
4126 def nsd_refcount_xpath(nsd_id):
4127 """ xpath for ref count entry """
4128 return (NsdRefCountDtsHandler.XPATH +
4129 "[rw-nsr:nsd-id-ref = '{}']").format(nsd_id)
4130
4131 nsd_list = []
4132 if nsd_id is None or nsd_id == "":
4133 for nsd in self._nsds.values():
4134 nsd_msg = RwNsrYang.YangData_Nsr_NsInstanceOpdata_NsdRefCount()
4135 nsd_msg.nsd_id_ref = nsd.id
4136 nsd_msg.instance_ref_count = nsd.ref_count
4137 nsd_list.append((nsd_refcount_xpath(nsd.id), nsd_msg))
4138 elif nsd_id in self._nsds:
4139 nsd_msg = RwNsrYang.YangData_Nsr_NsInstanceOpdata_NsdRefCount()
4140 nsd_msg.nsd_id_ref = self._nsds[nsd_id].id
4141 nsd_msg.instance_ref_count = self._nsds[nsd_id].ref_count
4142 nsd_list.append((nsd_refcount_xpath(nsd_id), nsd_msg))
4143
4144 return nsd_list
4145
4146 @asyncio.coroutine
4147 def terminate_ns(self, nsr_id, xact):
4148 """
4149 Terminate network service for the given NSR Id
4150 """
4151
4152 # Terminate the instances/networks assocaited with this nw service
4153 self._log.debug("Terminating the network service %s", nsr_id)
4154 try :
4155 yield from self._nsrs[nsr_id].terminate()
4156 except Exception as e:
4157 self.log.exception("Failed to terminate NSR[id=%s]", nsr_id)
4158
4159 # Unref the NSD
4160 yield from self.nsd_unref_by_nsr_id(nsr_id)
4161
4162 # Unpublish the NSR record
4163 self._log.debug("Unpublishing the network service %s", nsr_id)
4164 yield from self._nsrs[nsr_id].unpublish(xact)
4165
4166 # Finaly delete the NS instance from this NS Manager
4167 self._log.debug("Deletng the network service %s", nsr_id)
4168 self.delete_nsr(nsr_id)
4169
4170
4171 class NsmRecordsPublisherProxy(object):
4172 """ This class provides a publisher interface that allows plugin objects
4173 to publish NSR/VNFR/VLR"""
4174
4175 def __init__(self, dts, log, loop, nsr_pub_hdlr, vnfr_pub_hdlr, vlr_pub_hdlr):
4176 self._dts = dts
4177 self._log = log
4178 self._loop = loop
4179 self._nsr_pub_hdlr = nsr_pub_hdlr
4180 self._vlr_pub_hdlr = vlr_pub_hdlr
4181 self._vnfr_pub_hdlr = vnfr_pub_hdlr
4182
4183 @asyncio.coroutine
4184 def publish_nsr(self, xact, nsr):
4185 """ Publish an NSR """
4186 path = NetworkServiceRecord.xpath_from_nsr(nsr)
4187 return (yield from self._nsr_pub_hdlr.update(xact, path, nsr))
4188
4189 @asyncio.coroutine
4190 def unpublish_nsr(self, xact, nsr):
4191 """ Unpublish an NSR """
4192 path = NetworkServiceRecord.xpath_from_nsr(nsr)
4193 return (yield from self._nsr_pub_hdlr.delete(xact, path))
4194
4195 @asyncio.coroutine
4196 def publish_vnfr(self, xact, vnfr):
4197 """ Publish an VNFR """
4198 path = VirtualNetworkFunctionRecord.vnfr_xpath(vnfr)
4199 return (yield from self._vnfr_pub_hdlr.update(xact, path, vnfr))
4200
4201 @asyncio.coroutine
4202 def unpublish_vnfr(self, xact, vnfr):
4203 """ Unpublish a VNFR """
4204 path = VirtualNetworkFunctionRecord.vnfr_xpath(vnfr)
4205 return (yield from self._vnfr_pub_hdlr.delete(xact, path))
4206
4207 @asyncio.coroutine
4208 def publish_vlr(self, xact, vlr):
4209 """ Publish a VLR """
4210 path = VirtualLinkRecord.vlr_xpath(vlr)
4211 return (yield from self._vlr_pub_hdlr.update(xact, path, vlr))
4212
4213 @asyncio.coroutine
4214 def unpublish_vlr(self, xact, vlr):
4215 """ Unpublish a VLR """
4216 path = VirtualLinkRecord.vlr_xpath(vlr)
4217 return (yield from self._vlr_pub_hdlr.delete(xact, path))
4218
4219
4220 class ScalingRpcHandler(mano_dts.DtsHandler):
4221 """ The Network service Monitor DTS handler """
4222 SCALE_IN_INPUT_XPATH = "I,/nsr:exec-scale-in"
4223 SCALE_IN_OUTPUT_XPATH = "O,/nsr:exec-scale-in"
4224
4225 SCALE_OUT_INPUT_XPATH = "I,/nsr:exec-scale-out"
4226 SCALE_OUT_OUTPUT_XPATH = "O,/nsr:exec-scale-out"
4227
4228 ACTION = Enum('ACTION', 'SCALE_IN SCALE_OUT')
4229
4230 def __init__(self, log, dts, loop, callback=None):
4231 super().__init__(log, dts, loop)
4232 self.callback = callback
4233 self.last_instance_id = defaultdict(int)
4234
4235 @asyncio.coroutine
4236 def register(self):
4237
4238 @asyncio.coroutine
4239 def on_scale_in_prepare(xact_info, action, ks_path, msg):
4240 assert action == rwdts.QueryAction.RPC
4241
4242 try:
4243 rpc_op = NsrYang.YangOutput_Nsr_ExecScaleIn.from_dict({
4244 "instance_id": msg.instance_id})
4245
4246 xact_info.respond_xpath(
4247 rwdts.XactRspCode.ACK,
4248 self.__class__.SCALE_IN_OUTPUT_XPATH,
4249 rpc_op)
4250
4251 if self.callback:
4252 self.callback(xact_info.xact, msg, self.ACTION.SCALE_IN)
4253 except Exception as e:
4254 self.log.exception(e)
4255 xact_info.respond_xpath(
4256 rwdts.XactRspCode.NACK,
4257 self.__class__.SCALE_IN_OUTPUT_XPATH)
4258
4259 @asyncio.coroutine
4260 def on_scale_out_prepare(xact_info, action, ks_path, msg):
4261 assert action == rwdts.QueryAction.RPC
4262
4263 try:
4264 scaling_group = msg.scaling_group_name_ref
4265 if not msg.instance_id:
4266 last_instance_id = self.last_instance_id[scale_group]
4267 msg.instance_id = last_instance_id + 1
4268 self.last_instance_id[scale_group] += 1
4269
4270 rpc_op = NsrYang.YangOutput_Nsr_ExecScaleOut.from_dict({
4271 "instance_id": msg.instance_id})
4272
4273 xact_info.respond_xpath(
4274 rwdts.XactRspCode.ACK,
4275 self.__class__.SCALE_OUT_OUTPUT_XPATH,
4276 rpc_op)
4277
4278 if self.callback:
4279 self.callback(xact_info.xact, msg, self.ACTION.SCALE_OUT)
4280 except Exception as e:
4281 self.log.exception(e)
4282 xact_info.respond_xpath(
4283 rwdts.XactRspCode.NACK,
4284 self.__class__.SCALE_OUT_OUTPUT_XPATH)
4285
4286 scale_in_hdl = rift.tasklets.DTS.RegistrationHandler(
4287 on_prepare=on_scale_in_prepare)
4288 scale_out_hdl = rift.tasklets.DTS.RegistrationHandler(
4289 on_prepare=on_scale_out_prepare)
4290
4291 with self.dts.group_create() as group:
4292 group.register(
4293 xpath=self.__class__.SCALE_IN_INPUT_XPATH,
4294 handler=scale_in_hdl,
4295 flags=rwdts.Flag.PUBLISHER)
4296 group.register(
4297 xpath=self.__class__.SCALE_OUT_INPUT_XPATH,
4298 handler=scale_out_hdl,
4299 flags=rwdts.Flag.PUBLISHER)
4300
4301
4302 class NsmTasklet(rift.tasklets.Tasklet):
4303 """
4304 The network service manager tasklet
4305 """
4306 def __init__(self, *args, **kwargs):
4307 super(NsmTasklet, self).__init__(*args, **kwargs)
4308 self.rwlog.set_category("rw-mano-log")
4309 self.rwlog.set_subcategory("nsm")
4310
4311 self._dts = None
4312 self._nsm = None
4313
4314 self._ro_plugin_selector = None
4315 self._vnffgmgr = None
4316
4317 self._nsr_handler = None
4318 self._vnfr_pub_handler = None
4319 self._vlr_pub_handler = None
4320 self._vnfd_pub_handler = None
4321 self._scale_cfg_handler = None
4322
4323 self._records_publisher_proxy = None
4324
4325 def start(self):
4326 """ The task start callback """
4327 super(NsmTasklet, self).start()
4328 self.log.info("Starting NsmTasklet")
4329
4330 self.log.debug("Registering with dts")
4331 self._dts = rift.tasklets.DTS(self.tasklet_info,
4332 RwNsmYang.get_schema(),
4333 self.loop,
4334 self.on_dts_state_change)
4335
4336 self.log.debug("Created DTS Api GI Object: %s", self._dts)
4337
4338 def stop(self):
4339 try:
4340 self._dts.deinit()
4341 except Exception:
4342 print("Caught Exception in NSM stop:", sys.exc_info()[0])
4343 raise
4344
4345 def on_instance_started(self):
4346 """ Task instance started callback """
4347 self.log.debug("Got instance started callback")
4348
4349 @asyncio.coroutine
4350 def init(self):
4351 """ Task init callback """
4352 self.log.debug("Got instance started callback")
4353
4354 self.log.debug("creating config account handler")
4355
4356 self._nsr_pub_handler = publisher.NsrOpDataDtsHandler(self._dts, self.log, self.loop)
4357 yield from self._nsr_pub_handler.register()
4358
4359 self._vnfr_pub_handler = publisher.VnfrPublisherDtsHandler(self._dts, self.log, self.loop)
4360 yield from self._vnfr_pub_handler.register()
4361
4362 self._vlr_pub_handler = publisher.VlrPublisherDtsHandler(self._dts, self.log, self.loop)
4363 yield from self._vlr_pub_handler.register()
4364
4365 manifest = self.tasklet_info.get_pb_manifest()
4366 use_ssl = manifest.bootstrap_phase.rwsecurity.use_ssl
4367 ssl_cert = manifest.bootstrap_phase.rwsecurity.cert
4368 ssl_key = manifest.bootstrap_phase.rwsecurity.key
4369
4370 self._vnfd_pub_handler = publisher.VnfdPublisher(use_ssl, ssl_cert, ssl_key, self.loop)
4371
4372 self._records_publisher_proxy = NsmRecordsPublisherProxy(
4373 self._dts,
4374 self.log,
4375 self.loop,
4376 self._nsr_pub_handler,
4377 self._vnfr_pub_handler,
4378 self._vlr_pub_handler,
4379 )
4380
4381 # Register the NSM to receive the nsm plugin
4382 # when cloud account is configured
4383 self._ro_plugin_selector = cloud.ROAccountPluginSelector(
4384 self._dts,
4385 self.log,
4386 self.loop,
4387 self._records_publisher_proxy,
4388 )
4389 yield from self._ro_plugin_selector.register()
4390
4391 self._cloud_account_handler = cloud.CloudAccountConfigSubscriber(
4392 self._log,
4393 self._dts,
4394 self.log_hdl)
4395
4396 yield from self._cloud_account_handler.register()
4397
4398 self._vnffgmgr = rwvnffgmgr.VnffgMgr(self._dts, self.log, self.log_hdl, self.loop)
4399 yield from self._vnffgmgr.register()
4400
4401 self._nsm = NsManager(
4402 self._dts,
4403 self.log,
4404 self.loop,
4405 self._nsr_pub_handler,
4406 self._vnfr_pub_handler,
4407 self._vlr_pub_handler,
4408 self._ro_plugin_selector,
4409 self._vnffgmgr,
4410 self._vnfd_pub_handler,
4411 self._cloud_account_handler
4412 )
4413
4414 yield from self._nsm.register()
4415
4416 @asyncio.coroutine
4417 def run(self):
4418 """ Task run callback """
4419 pass
4420
4421 @asyncio.coroutine
4422 def on_dts_state_change(self, state):
4423 """Take action according to current dts state to transition
4424 application into the corresponding application state
4425
4426 Arguments
4427 state - current dts state
4428 """
4429 switch = {
4430 rwdts.State.INIT: rwdts.State.REGN_COMPLETE,
4431 rwdts.State.CONFIG: rwdts.State.RUN,
4432 }
4433
4434 handlers = {
4435 rwdts.State.INIT: self.init,
4436 rwdts.State.RUN: self.run,
4437 }
4438
4439 # Transition application to next state
4440 handler = handlers.get(state, None)
4441 if handler is not None:
4442 yield from handler()
4443
4444 # Transition dts to next state
4445 next_state = switch.get(state, None)
4446 if next_state is not None:
4447 self.log.debug("Changing state to %s", next_state)
4448 self._dts.handle.set_state(next_state)