blob: e7b85cd881743b6b89b32f960135fc59551e4d1c [file] [log] [blame]
tierno59d22d22018-09-25 18:10:19 +02001# -*- coding: utf-8 -*-
2
tierno2e215512018-11-28 09:37:52 +00003##
4# Copyright 2018 Telefonica S.A.
5#
6# Licensed under the Apache License, Version 2.0 (the "License"); you may
7# not use this file except in compliance with the License. You may obtain
8# a copy of the License at
9#
10# http://www.apache.org/licenses/LICENSE-2.0
11#
12# Unless required by applicable law or agreed to in writing, software
13# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
14# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
15# License for the specific language governing permissions and limitations
16# under the License.
17##
tierno59d22d22018-09-25 18:10:19 +020018
kuused124bfe2019-06-18 12:09:24 +020019import asyncio
aticigdffa6212022-04-12 15:27:53 +030020import checksumdir
tierno59d22d22018-09-25 18:10:19 +020021from collections import OrderedDict
aticig1dda84c2022-09-10 01:56:58 +030022import hashlib
aticigdffa6212022-04-12 15:27:53 +030023import os
aticig9bc63ac2022-07-27 09:32:06 +030024import shutil
25import traceback
tierno79cd8ad2019-10-18 13:03:10 +000026from time import time
aticig9bc63ac2022-07-27 09:32:06 +030027
28from osm_common.fsbase import FsException
bravof922c4172020-11-24 21:21:43 -030029from osm_lcm.data_utils.database.database import Database
30from osm_lcm.data_utils.filesystem.filesystem import Filesystem
aticig9bc63ac2022-07-27 09:32:06 +030031import yaml
32from zipfile import ZipFile, BadZipfile
bravof922c4172020-11-24 21:21:43 -030033
tiernobaa51102018-12-14 13:16:18 +000034# from osm_common.dbbase import DbException
tierno59d22d22018-09-25 18:10:19 +020035
36__author__ = "Alfonso Tierno"
37
38
39class LcmException(Exception):
40 pass
41
42
gcalvinoed7f6d42018-12-14 14:44:56 +010043class LcmExceptionExit(LcmException):
44 pass
45
46
tierno59d22d22018-09-25 18:10:19 +020047def versiontuple(v):
tierno27246d82018-09-27 15:59:09 +020048 """utility for compare dot separate versions. Fills with zeros to proper number comparison
49 package version will be something like 4.0.1.post11+gb3f024d.dirty-1. Where 4.0.1 is the git tag, postXX is the
50 number of commits from this tag, and +XXXXXXX is the git commit short id. Total length is 16 with until 999 commits
51 """
tierno59d22d22018-09-25 18:10:19 +020052 filled = []
53 for point in v.split("."):
tiernoe64f7fb2019-09-11 08:55:52 +000054 point, _, _ = point.partition("+")
55 point, _, _ = point.partition("-")
56 filled.append(point.zfill(20))
tierno59d22d22018-09-25 18:10:19 +020057 return tuple(filled)
58
59
tierno744303e2020-01-13 16:46:31 +000060def deep_get(target_dict, key_list, default_value=None):
tierno626e0152019-11-29 14:16:16 +000061 """
62 Get a value from target_dict entering in the nested keys. If keys does not exist, it returns None
63 Example target_dict={a: {b: 5}}; key_list=[a,b] returns 5; both key_list=[a,b,c] and key_list=[f,h] return None
64 :param target_dict: dictionary to be read
65 :param key_list: list of keys to read from target_dict
tierno744303e2020-01-13 16:46:31 +000066 :param default_value: value to return if key is not present in the nested dictionary
tierno626e0152019-11-29 14:16:16 +000067 :return: The wanted value if exist, None otherwise
68 """
69 for key in key_list:
70 if not isinstance(target_dict, dict) or key not in target_dict:
tierno744303e2020-01-13 16:46:31 +000071 return default_value
tierno626e0152019-11-29 14:16:16 +000072 target_dict = target_dict[key]
73 return target_dict
74
75
tierno744303e2020-01-13 16:46:31 +000076def get_iterable(in_dict, in_key):
77 """
78 Similar to <dict>.get(), but if value is None, False, ..., An empty tuple is returned instead
79 :param in_dict: a dictionary
80 :param in_key: the key to look for at in_dict
81 :return: in_dict[in_var] or () if it is None or not present
82 """
83 if not in_dict.get(in_key):
84 return ()
85 return in_dict[in_key]
86
87
aticigdffa6212022-04-12 15:27:53 +030088def check_juju_bundle_existence(vnfd: dict) -> str:
89 """Checks the existence of juju-bundle in the descriptor
90
91 Args:
92 vnfd: Descriptor as a dictionary
93
94 Returns:
95 Juju bundle if dictionary has juju-bundle else None
96
97 """
98 if vnfd.get("vnfd"):
99 vnfd = vnfd["vnfd"]
100
101 for kdu in vnfd.get("kdu", []):
102 return kdu.get("juju-bundle", None)
103
104
105def get_charm_artifact_path(base_folder, charm_name, charm_type, revision=str()) -> str:
106 """Finds the charm artifact paths
107
108 Args:
109 base_folder: Main folder which will be looked up for charm
110 charm_name: Charm name
111 charm_type: Type of charm native_charm, lxc_proxy_charm or k8s_proxy_charm
112 revision: vnf package revision number if there is
113
114 Returns:
115 artifact_path: (str)
116
117 """
118 extension = ""
119 if revision:
120 extension = ":" + str(revision)
121
122 if base_folder.get("pkg-dir"):
123 artifact_path = "{}/{}/{}/{}".format(
aticigd7083542022-05-30 20:45:55 +0300124 base_folder["folder"].split(":")[0] + extension,
aticigdffa6212022-04-12 15:27:53 +0300125 base_folder["pkg-dir"],
126 "charms"
127 if charm_type in ("native_charm", "lxc_proxy_charm", "k8s_proxy_charm")
128 else "helm-charts",
129 charm_name,
130 )
131
132 else:
133 # For SOL004 packages
134 artifact_path = "{}/Scripts/{}/{}".format(
aticigd7083542022-05-30 20:45:55 +0300135 base_folder["folder"].split(":")[0] + extension,
aticigdffa6212022-04-12 15:27:53 +0300136 "charms"
137 if charm_type in ("native_charm", "lxc_proxy_charm", "k8s_proxy_charm")
138 else "helm-charts",
139 charm_name,
140 )
141
142 return artifact_path
143
144
tierno744303e2020-01-13 16:46:31 +0000145def populate_dict(target_dict, key_list, value):
146 """
147 Update target_dict creating nested dictionaries with the key_list. Last key_list item is asigned the value.
148 Example target_dict={K: J}; key_list=[a,b,c]; target_dict will be {K: J, a: {b: {c: value}}}
149 :param target_dict: dictionary to be changed
150 :param key_list: list of keys to insert at target_dict
151 :param value:
152 :return: None
153 """
154 for key in key_list[0:-1]:
155 if key not in target_dict:
156 target_dict[key] = {}
157 target_dict = target_dict[key]
158 target_dict[key_list[-1]] = value
159
160
Gabriel Cubae539a8d2022-10-10 11:34:51 -0500161def get_ee_id_parts(ee_id):
162 """
163 Parses ee_id stored at database that can be either 'version:namespace.helm_id' or only
164 namespace.helm_id for backward compatibility
165 If exists helm version can be helm-v3 or helm (helm-v2 old version)
166 """
167 version, _, part_id = ee_id.rpartition(":")
168 namespace, _, helm_id = part_id.rpartition(".")
169 return version, namespace, helm_id
170
171
Gabriel Cubac7737442023-02-14 13:09:18 -0500172def vld_to_ro_ip_profile(source_data):
173 if source_data:
174 return {
175 "ip_version": "IPv4"
176 if "v4" in source_data.get("ip-version", "ipv4")
177 else "IPv6",
178 "subnet_address": source_data.get("cidr")
179 or source_data.get("subnet-address"),
180 "gateway_address": source_data.get("gateway-ip")
181 or source_data.get("gateway-address"),
182 "dns_address": ";".join(
183 [v["address"] for v in source_data["dns-server"] if v.get("address")]
184 )
185 if source_data.get("dns-server")
186 else None,
187 "dhcp_enabled": source_data.get("dhcp-params", {}).get("enabled", False)
188 or source_data.get("dhcp-enabled", False),
189 "dhcp_start_address": source_data["dhcp-params"].get("start-address")
190 if source_data.get("dhcp-params")
191 else None,
192 "dhcp_count": source_data["dhcp-params"].get("count")
193 if source_data.get("dhcp-params")
194 else None,
Gabriel Cubaf0af5e62023-03-14 00:27:49 -0500195 "ipv6_address_mode": source_data["ipv6-address-mode"]
196 if "ipv6-address-mode" in source_data
197 else None,
Gabriel Cubac7737442023-02-14 13:09:18 -0500198 }
199
200
kuused124bfe2019-06-18 12:09:24 +0200201class LcmBase:
bravof922c4172020-11-24 21:21:43 -0300202 def __init__(self, msg, logger):
kuused124bfe2019-06-18 12:09:24 +0200203 """
204
205 :param db: database connection
206 """
bravof922c4172020-11-24 21:21:43 -0300207 self.db = Database().instance.db
kuused124bfe2019-06-18 12:09:24 +0200208 self.msg = msg
bravof922c4172020-11-24 21:21:43 -0300209 self.fs = Filesystem().instance.fs
kuused124bfe2019-06-18 12:09:24 +0200210 self.logger = logger
211
212 def update_db_2(self, item, _id, _desc):
213 """
214 Updates database with _desc information. If success _desc is cleared
Pedro Escaleirada21d262022-04-21 16:31:06 +0100215 :param item: collection
216 :param _id: the _id to use in the query filter
kuused124bfe2019-06-18 12:09:24 +0200217 :param _desc: dictionary with the content to update. Keys are dot separated keys for
218 :return: None. Exception is raised on error
219 """
220 if not _desc:
221 return
tierno79cd8ad2019-10-18 13:03:10 +0000222 now = time()
223 _desc["_admin.modified"] = now
rshri932105f2024-07-05 15:11:55 +0000224 self.logger.info("Desc: {} Item: {} _id: {}".format(_desc, item, _id))
kuused124bfe2019-06-18 12:09:24 +0200225 self.db.set_one(item, {"_id": _id}, _desc)
226 _desc.clear()
227 # except DbException as e:
228 # self.logger.error("Updating {} _id={} with '{}'. Error: {}".format(item, _id, _desc, e))
229
rshri932105f2024-07-05 15:11:55 +0000230 def update_operation_history(
231 self, content, workflow_status=None, resource_status=None
232 ):
233 self.logger.info("Update Operation History in LcmBase")
234 self.logger.info(
235 "Content: {} Workflow Status: {} Resource Status: {}".format(
236 content, workflow_status, resource_status
237 )
238 )
239
240 op_id = content["current_operation"]
241 self.logger.info("OP_id: {}".format(op_id))
242 length = 0
garciadeblas96b94f52024-07-08 16:18:21 +0200243 for operation in content["operationHistory"]:
244 self.logger.info("Operations: {}".format(operation))
245 if operation["op_id"] == op_id:
rshri932105f2024-07-05 15:11:55 +0000246 self.logger.info("Length: {}".format(length))
247 now = time()
248 if workflow_status:
249 content["operationHistory"][length]["workflowState"] = "COMPLETED"
250 else:
251 content["operationHistory"][length]["workflowState"] = "ERROR"
252
253 if resource_status:
254 content["operationHistory"][length]["resourceState"] = "READY"
255 else:
256 content["operationHistory"][length]["resourceState"] = "NOT_READY"
257
258 content["operationHistory"][length]["endDate"] = now
259 break
260 length += 1
261 self.logger.info("content: {}".format(content))
262
263 return content
264
aticig1dda84c2022-09-10 01:56:58 +0300265 @staticmethod
266 def calculate_charm_hash(zipped_file):
267 """Calculate the hash of charm files which ends with .charm
268
269 Args:
270 zipped_file (str): Existing charm package full path
271
272 Returns:
273 hex digest (str): The hash of the charm file
274 """
Gabriel Cuba4c0e6802023-10-09 13:22:38 -0500275 filehash = hashlib.sha256()
aticig1dda84c2022-09-10 01:56:58 +0300276 with open(zipped_file, mode="rb") as file:
277 contents = file.read()
278 filehash.update(contents)
279 return filehash.hexdigest()
280
281 @staticmethod
282 def compare_charm_hash(current_charm, target_charm):
283 """Compare the existing charm and the target charm if the charms
284 are given as zip files ends with .charm
285
286 Args:
287 current_charm (str): Existing charm package full path
288 target_charm (str): Target charm package full path
289
290 Returns:
291 True/False (bool): if charm has changed it returns True
292 """
293 return LcmBase.calculate_charm_hash(
294 current_charm
295 ) != LcmBase.calculate_charm_hash(target_charm)
296
297 @staticmethod
298 def compare_charmdir_hash(current_charm_dir, target_charm_dir):
299 """Compare the existing charm and the target charm if the charms
300 are given as directories
301
302 Args:
303 current_charm_dir (str): Existing charm package directory path
304 target_charm_dir (str): Target charm package directory path
305
306 Returns:
307 True/False (bool): if charm has changed it returns True
308 """
309 return checksumdir.dirhash(current_charm_dir) != checksumdir.dirhash(
310 target_charm_dir
311 )
312
aticigdffa6212022-04-12 15:27:53 +0300313 def check_charm_hash_changed(
314 self, current_charm_path: str, target_charm_path: str
315 ) -> bool:
316 """Find the target charm has changed or not by checking the hash of
317 old and new charm packages
318
319 Args:
320 current_charm_path (str): Existing charm package artifact path
321 target_charm_path (str): Target charm package artifact path
322
323 Returns:
324 True/False (bool): if charm has changed it returns True
325
326 """
aticig1dda84c2022-09-10 01:56:58 +0300327 try:
328 # Check if the charm artifacts are available
329 current_charm = self.fs.path + current_charm_path
330 target_charm = self.fs.path + target_charm_path
aticigdffa6212022-04-12 15:27:53 +0300331
aticig1dda84c2022-09-10 01:56:58 +0300332 if os.path.exists(current_charm) and os.path.exists(target_charm):
aticig1dda84c2022-09-10 01:56:58 +0300333 # Compare the hash of .charm files
334 if current_charm.endswith(".charm"):
335 return LcmBase.compare_charm_hash(current_charm, target_charm)
aticigdffa6212022-04-12 15:27:53 +0300336
aticig1dda84c2022-09-10 01:56:58 +0300337 # Compare the hash of charm folders
338 return LcmBase.compare_charmdir_hash(current_charm, target_charm)
339
340 else:
341 raise LcmException(
342 "Charm artifact {} does not exist in the VNF Package".format(
343 self.fs.path + target_charm_path
344 )
aticigdffa6212022-04-12 15:27:53 +0300345 )
aticig1dda84c2022-09-10 01:56:58 +0300346 except (IOError, OSError, TypeError) as error:
347 self.logger.debug(traceback.format_exc())
348 self.logger.error(f"{error} occured while checking the charm hashes")
349 raise LcmException(error)
aticigdffa6212022-04-12 15:27:53 +0300350
aticig9bc63ac2022-07-27 09:32:06 +0300351 @staticmethod
352 def get_charm_name(charm_metadata_file: str) -> str:
353 """Get the charm name from metadata file.
354
355 Args:
356 charm_metadata_file (str): charm metadata file full path
357
358 Returns:
359 charm_name (str): charm name
360
361 """
362 # Read charm metadata.yaml to get the charm name
363 with open(charm_metadata_file, "r") as metadata_file:
364 content = yaml.safe_load(metadata_file)
365 charm_name = content["name"]
366 return str(charm_name)
367
368 def _get_charm_path(
369 self, nsd_package_path: str, nsd_package_name: str, charm_folder_name: str
370 ) -> str:
371 """Get the full path of charm folder.
372
373 Args:
374 nsd_package_path (str): NSD package full path
375 nsd_package_name (str): NSD package name
376 charm_folder_name (str): folder name
377
378 Returns:
379 charm_path (str): charm folder full path
380 """
381 charm_path = (
382 self.fs.path
383 + nsd_package_path
384 + "/"
385 + nsd_package_name
386 + "/charms/"
387 + charm_folder_name
388 )
389 return charm_path
390
391 def _get_charm_metadata_file(
392 self,
393 charm_folder_name: str,
394 nsd_package_path: str,
395 nsd_package_name: str,
396 charm_path: str = None,
397 ) -> str:
398 """Get the path of charm metadata file.
399
400 Args:
401 charm_folder_name (str): folder name
402 nsd_package_path (str): NSD package full path
403 nsd_package_name (str): NSD package name
404 charm_path (str): Charm full path
405
406 Returns:
407 charm_metadata_file_path (str): charm metadata file full path
408
409 """
410 # Locate the charm metadata.yaml
411 if charm_folder_name.endswith(".charm"):
412 extract_path = (
413 self.fs.path
414 + nsd_package_path
415 + "/"
416 + nsd_package_name
417 + "/charms/"
aticiga37c6ff2022-08-20 20:56:19 +0300418 + charm_folder_name.replace(".charm", "")
aticig9bc63ac2022-07-27 09:32:06 +0300419 )
420 # Extract .charm to extract path
421 with ZipFile(charm_path, "r") as zipfile:
422 zipfile.extractall(extract_path)
423 return extract_path + "/metadata.yaml"
424 else:
425 return charm_path + "/metadata.yaml"
426
427 def find_charm_name(self, db_nsr: dict, charm_folder_name: str) -> str:
428 """Get the charm name from metadata.yaml of charm package.
429
430 Args:
431 db_nsr (dict): NS record as a dictionary
432 charm_folder_name (str): charm folder name
433
434 Returns:
435 charm_name (str): charm name
436 """
437 try:
438 if not charm_folder_name:
439 raise LcmException("charm_folder_name should be provided.")
440
441 # Find nsd_package details: path, name
442 revision = db_nsr.get("revision", "")
aticiga37c6ff2022-08-20 20:56:19 +0300443
444 # Get the NSD package path
445 if revision:
preethika.p28b0bf82022-09-23 07:36:28 +0000446 nsd_package_path = db_nsr["nsd-id"] + ":" + str(revision)
aticiga37c6ff2022-08-20 20:56:19 +0300447 db_nsd = self.db.get_one("nsds_revisions", {"_id": nsd_package_path})
448
449 else:
450 nsd_package_path = db_nsr["nsd-id"]
451
452 db_nsd = self.db.get_one("nsds", {"_id": nsd_package_path})
453
454 # Get the NSD package name
455 nsd_package_name = db_nsd["_admin"]["storage"]["pkg-dir"]
aticig9bc63ac2022-07-27 09:32:06 +0300456
457 # Remove the existing nsd package and sync from FsMongo
458 shutil.rmtree(self.fs.path + nsd_package_path, ignore_errors=True)
459 self.fs.sync(from_path=nsd_package_path)
460
461 # Get the charm path
462 charm_path = self._get_charm_path(
463 nsd_package_path, nsd_package_name, charm_folder_name
464 )
465
466 # Find charm metadata file full path
467 charm_metadata_file = self._get_charm_metadata_file(
468 charm_folder_name, nsd_package_path, nsd_package_name, charm_path
469 )
470
471 # Return charm name
472 return self.get_charm_name(charm_metadata_file)
473
474 except (
475 yaml.YAMLError,
476 IOError,
477 FsException,
478 KeyError,
479 TypeError,
480 FileNotFoundError,
481 BadZipfile,
482 ) as error:
483 self.logger.debug(traceback.format_exc())
484 self.logger.error(f"{error} occured while getting the charm name")
485 raise LcmException(error)
486
Gabriel Cuba879483e2024-03-19 18:01:13 -0500487 def get_vca_info(self, ee_item, db_nsr, get_charm_name: bool):
488 vca_name = charm_name = vca_type = None
489 if ee_item.get("juju"):
490 vca_name = ee_item["juju"].get("charm")
491 if get_charm_name:
492 charm_name = self.find_charm_name(db_nsr, str(vca_name))
493 vca_type = (
494 "lxc_proxy_charm"
495 if ee_item["juju"].get("charm") is not None
496 else "native_charm"
497 )
498 if ee_item["juju"].get("cloud") == "k8s":
499 vca_type = "k8s_proxy_charm"
500 elif ee_item["juju"].get("proxy") is False:
501 vca_type = "native_charm"
502 elif ee_item.get("helm-chart"):
503 vca_name = ee_item["helm-chart"]
504 vca_type = "helm-v3"
505 return vca_name, charm_name, vca_type
506
kuused124bfe2019-06-18 12:09:24 +0200507
508class TaskRegistry(LcmBase):
tierno59d22d22018-09-25 18:10:19 +0200509 """
510 Implements a registry of task needed for later cancelation, look for related tasks that must be completed before
511 etc. It stores a four level dict
512 First level is the topic, ns, vim_account, sdn
513 Second level is the _id
514 Third level is the operation id
515 Fourth level is a descriptive name, the value is the task class
kuused124bfe2019-06-18 12:09:24 +0200516
517 The HA (High-Availability) methods are used when more than one LCM instance is running.
518 To register the current task in the external DB, use LcmBase as base class, to be able
519 to reuse LcmBase.update_db_2()
520 The DB registry uses the following fields to distinguish a task:
521 - op_type: operation type ("nslcmops" or "nsilcmops")
522 - op_id: operation ID
523 - worker: the worker ID for this process
tierno59d22d22018-09-25 18:10:19 +0200524 """
525
kuuse6a470c62019-07-10 13:52:45 +0200526 # NS/NSI: "services" VIM/WIM/SDN: "accounts"
garciadeblas5697b8b2021-03-24 09:17:02 +0100527 topic_service_list = ["ns", "nsi"]
rshri932105f2024-07-05 15:11:55 +0000528 topic_account_list = [
529 "vim",
530 "wim",
531 "sdn",
532 "k8scluster",
533 "vca",
534 "k8srepo",
535 "cluster",
536 "k8s_app",
537 "k8s_resource",
538 "k8s_infra_controller",
539 "k8s_infra_config",
yshah771dea82024-07-05 15:11:49 +0000540 "oka",
541 "ksu",
rshri932105f2024-07-05 15:11:55 +0000542 ]
kuuse6a470c62019-07-10 13:52:45 +0200543
544 # Map topic to InstanceID
garciadeblas5697b8b2021-03-24 09:17:02 +0100545 topic2instid_dict = {"ns": "nsInstanceId", "nsi": "netsliceInstanceId"}
kuuse6a470c62019-07-10 13:52:45 +0200546
547 # Map topic to DB table name
548 topic2dbtable_dict = {
garciadeblas5697b8b2021-03-24 09:17:02 +0100549 "ns": "nslcmops",
550 "nsi": "nsilcmops",
551 "vim": "vim_accounts",
552 "wim": "wim_accounts",
553 "sdn": "sdns",
554 "k8scluster": "k8sclusters",
555 "vca": "vca",
556 "k8srepo": "k8srepos",
rshri932105f2024-07-05 15:11:55 +0000557 "cluster": "k8sclusters",
558 "k8s_app": "k8sapp",
559 "k8s_resource": "k8sresource",
560 "k8s_infra_controller": "k8sinfra_controller",
561 "k8s_infra_config": "k8sinfra_config",
yshah771dea82024-07-05 15:11:49 +0000562 "oka": "oka",
563 "ksu": "ksus",
garciadeblas5697b8b2021-03-24 09:17:02 +0100564 }
kuused124bfe2019-06-18 12:09:24 +0200565
bravof922c4172020-11-24 21:21:43 -0300566 def __init__(self, worker_id=None, logger=None):
tierno59d22d22018-09-25 18:10:19 +0200567 self.task_registry = {
568 "ns": {},
Felipe Vicensc2033f22018-11-15 15:09:58 +0100569 "nsi": {},
tierno59d22d22018-09-25 18:10:19 +0200570 "vim_account": {},
tiernoe37b57d2018-12-11 17:22:51 +0000571 "wim_account": {},
tierno59d22d22018-09-25 18:10:19 +0200572 "sdn": {},
calvinosanch9f9c6f22019-11-04 13:37:39 +0100573 "k8scluster": {},
David Garciac1fe90a2021-03-31 19:12:02 +0200574 "vca": {},
calvinosanch9f9c6f22019-11-04 13:37:39 +0100575 "k8srepo": {},
rshri932105f2024-07-05 15:11:55 +0000576 "cluster": {},
577 "k8s_app": {},
578 "k8s_resource": {},
579 "k8s_infra_controller": {},
580 "k8s_infra_config": {},
yshah771dea82024-07-05 15:11:49 +0000581 "oka": {},
582 "ksu": {},
rshri932105f2024-07-05 15:11:55 +0000583 "odu": {},
tierno59d22d22018-09-25 18:10:19 +0200584 }
kuused124bfe2019-06-18 12:09:24 +0200585 self.worker_id = worker_id
bravof922c4172020-11-24 21:21:43 -0300586 self.db = Database().instance.db
kuused124bfe2019-06-18 12:09:24 +0200587 self.logger = logger
rshri932105f2024-07-05 15:11:55 +0000588 # self.logger.info("Task registry: {}".format(self.task_registry))
tierno59d22d22018-09-25 18:10:19 +0200589
590 def register(self, topic, _id, op_id, task_name, task):
591 """
592 Register a new task
Felipe Vicensc2033f22018-11-15 15:09:58 +0100593 :param topic: Can be "ns", "nsi", "vim_account", "sdn"
tierno59d22d22018-09-25 18:10:19 +0200594 :param _id: _id of the related item
595 :param op_id: id of the operation of the related item
596 :param task_name: Task descriptive name, as create, instantiate, terminate. Must be unique in this op_id
597 :param task: Task class
598 :return: none
599 """
rshri932105f2024-07-05 15:11:55 +0000600 self.logger.info(
601 "topic : {}, _id:{}, op_id:{}, taskname:{}, task:{}".format(
602 topic, _id, op_id, task_name, task
603 )
604 )
tierno59d22d22018-09-25 18:10:19 +0200605 if _id not in self.task_registry[topic]:
606 self.task_registry[topic][_id] = OrderedDict()
607 if op_id not in self.task_registry[topic][_id]:
608 self.task_registry[topic][_id][op_id] = {task_name: task}
609 else:
610 self.task_registry[topic][_id][op_id][task_name] = task
rshri932105f2024-07-05 15:11:55 +0000611 self.logger.info("Task resgistry: {}".format(self.task_registry))
tierno59d22d22018-09-25 18:10:19 +0200612 # print("registering task", topic, _id, op_id, task_name, task)
613
614 def remove(self, topic, _id, op_id, task_name=None):
615 """
tiernobaa51102018-12-14 13:16:18 +0000616 When task is ended, it should be removed. It ignores missing tasks. It also removes tasks done with this _id
Felipe Vicensc2033f22018-11-15 15:09:58 +0100617 :param topic: Can be "ns", "nsi", "vim_account", "sdn"
tierno59d22d22018-09-25 18:10:19 +0200618 :param _id: _id of the related item
619 :param op_id: id of the operation of the related item
tiernobaa51102018-12-14 13:16:18 +0000620 :param task_name: Task descriptive name. If none it deletes all tasks with same _id and op_id
621 :return: None
tierno59d22d22018-09-25 18:10:19 +0200622 """
tiernobaa51102018-12-14 13:16:18 +0000623 if not self.task_registry[topic].get(_id):
tierno59d22d22018-09-25 18:10:19 +0200624 return
625 if not task_name:
tiernobaa51102018-12-14 13:16:18 +0000626 self.task_registry[topic][_id].pop(op_id, None)
627 elif self.task_registry[topic][_id].get(op_id):
628 self.task_registry[topic][_id][op_id].pop(task_name, None)
629
630 # delete done tasks
631 for op_id_ in list(self.task_registry[topic][_id]):
632 for name, task in self.task_registry[topic][_id][op_id_].items():
633 if not task.done():
634 break
635 else:
636 del self.task_registry[topic][_id][op_id_]
tierno59d22d22018-09-25 18:10:19 +0200637 if not self.task_registry[topic][_id]:
638 del self.task_registry[topic][_id]
639
640 def lookfor_related(self, topic, _id, my_op_id=None):
641 task_list = []
642 task_name_list = []
643 if _id not in self.task_registry[topic]:
644 return "", task_name_list
645 for op_id in reversed(self.task_registry[topic][_id]):
646 if my_op_id:
647 if my_op_id == op_id:
648 my_op_id = None # so that the next task is taken
649 continue
650
651 for task_name, task in self.task_registry[topic][_id][op_id].items():
tiernobaa51102018-12-14 13:16:18 +0000652 if not task.done():
653 task_list.append(task)
654 task_name_list.append(task_name)
tierno59d22d22018-09-25 18:10:19 +0200655 break
656 return ", ".join(task_name_list), task_list
657
658 def cancel(self, topic, _id, target_op_id=None, target_task_name=None):
659 """
kuused124bfe2019-06-18 12:09:24 +0200660 Cancel all active tasks of a concrete ns, nsi, vim_account, sdn identified for _id. If op_id is supplied only
Felipe Vicensc2033f22018-11-15 15:09:58 +0100661 this is cancelled, and the same with task_name
Gabriel Cubab6049d32023-10-30 13:44:49 -0500662 :return: cancelled task to be awaited if needed
tierno59d22d22018-09-25 18:10:19 +0200663 """
664 if not self.task_registry[topic].get(_id):
665 return
666 for op_id in reversed(self.task_registry[topic][_id]):
667 if target_op_id and target_op_id != op_id:
668 continue
Gabriel Cubab6049d32023-10-30 13:44:49 -0500669 for task_name, task in list(self.task_registry[topic][_id][op_id].items()):
tierno59d22d22018-09-25 18:10:19 +0200670 if target_task_name and target_task_name != task_name:
671 continue
672 # result =
673 task.cancel()
Gabriel Cubab6049d32023-10-30 13:44:49 -0500674 yield task
tierno59d22d22018-09-25 18:10:19 +0200675 # if result:
676 # self.logger.debug("{} _id={} order_id={} task={} cancelled".format(topic, _id, op_id, task_name))
677
kuuse6a470c62019-07-10 13:52:45 +0200678 # Is topic NS/NSI?
679 def _is_service_type_HA(self, topic):
680 return topic in self.topic_service_list
681
682 # Is topic VIM/WIM/SDN?
683 def _is_account_type_HA(self, topic):
684 return topic in self.topic_account_list
685
686 # Input: op_id, example: 'abc123def:3' Output: account_id='abc123def', op_index=3
687 def _get_account_and_op_HA(self, op_id):
688 if not op_id:
tiernofa076c32020-08-13 14:25:47 +0000689 return None, None
garciadeblas5697b8b2021-03-24 09:17:02 +0100690 account_id, _, op_index = op_id.rpartition(":")
tiernofa076c32020-08-13 14:25:47 +0000691 if not account_id or not op_index.isdigit():
692 return None, None
kuuse6a470c62019-07-10 13:52:45 +0200693 return account_id, op_index
694
695 # Get '_id' for any topic and operation
696 def _get_instance_id_HA(self, topic, op_type, op_id):
697 _id = None
698 # Special operation 'ANY', for SDN account associated to a VIM account: op_id as '_id'
garciadeblas5697b8b2021-03-24 09:17:02 +0100699 if op_type == "ANY":
kuuse6a470c62019-07-10 13:52:45 +0200700 _id = op_id
701 # NS/NSI: Use op_id as '_id'
702 elif self._is_service_type_HA(topic):
703 _id = op_id
calvinosanch9f9c6f22019-11-04 13:37:39 +0100704 # VIM/SDN/WIM/K8SCLUSTER: Split op_id to get Account ID and Operation Index, use Account ID as '_id'
kuuse6a470c62019-07-10 13:52:45 +0200705 elif self._is_account_type_HA(topic):
706 _id, _ = self._get_account_and_op_HA(op_id)
707 return _id
708
709 # Set DB _filter for querying any related process state
710 def _get_waitfor_filter_HA(self, db_lcmop, topic, op_type, op_id):
711 _filter = {}
712 # Special operation 'ANY', for SDN account associated to a VIM account: op_id as '_id'
713 # In this special case, the timestamp is ignored
garciadeblas5697b8b2021-03-24 09:17:02 +0100714 if op_type == "ANY":
715 _filter = {"operationState": "PROCESSING"}
kuuse6a470c62019-07-10 13:52:45 +0200716 # Otherwise, get 'startTime' timestamp for this operation
717 else:
718 # NS/NSI
719 if self._is_service_type_HA(topic):
tierno79cd8ad2019-10-18 13:03:10 +0000720 now = time()
kuuse6a470c62019-07-10 13:52:45 +0200721 starttime_this_op = db_lcmop.get("startTime")
722 instance_id_label = self.topic2instid_dict.get(topic)
723 instance_id = db_lcmop.get(instance_id_label)
garciadeblas5697b8b2021-03-24 09:17:02 +0100724 _filter = {
725 instance_id_label: instance_id,
726 "operationState": "PROCESSING",
727 "startTime.lt": starttime_this_op,
728 "_admin.modified.gt": now
729 - 2 * 3600, # ignore if tow hours of inactivity
730 }
calvinosanch9f9c6f22019-11-04 13:37:39 +0100731 # VIM/WIM/SDN/K8scluster
kuuse6a470c62019-07-10 13:52:45 +0200732 elif self._is_account_type_HA(topic):
733 _, op_index = self._get_account_and_op_HA(op_id)
garciadeblas5697b8b2021-03-24 09:17:02 +0100734 _ops = db_lcmop["_admin"]["operations"]
kuuse6a470c62019-07-10 13:52:45 +0200735 _this_op = _ops[int(op_index)]
garciadeblas5697b8b2021-03-24 09:17:02 +0100736 starttime_this_op = _this_op.get("startTime", None)
737 _filter = {
738 "operationState": "PROCESSING",
739 "startTime.lt": starttime_this_op,
740 }
kuuse6a470c62019-07-10 13:52:45 +0200741 return _filter
742
743 # Get DB params for any topic and operation
744 def _get_dbparams_for_lock_HA(self, topic, op_type, op_id):
745 q_filter = {}
746 update_dict = {}
747 # NS/NSI
748 if self._is_service_type_HA(topic):
garciadeblas5697b8b2021-03-24 09:17:02 +0100749 q_filter = {"_id": op_id, "_admin.worker": None}
750 update_dict = {"_admin.worker": self.worker_id}
kuuse6a470c62019-07-10 13:52:45 +0200751 # VIM/WIM/SDN
752 elif self._is_account_type_HA(topic):
753 account_id, op_index = self._get_account_and_op_HA(op_id)
754 if not account_id:
755 return None, None
garciadeblas5697b8b2021-03-24 09:17:02 +0100756 if op_type == "create":
kuuse6a470c62019-07-10 13:52:45 +0200757 # Creating a VIM/WIM/SDN account implies setting '_admin.current_operation' = 0
758 op_index = 0
garciadeblas5697b8b2021-03-24 09:17:02 +0100759 q_filter = {
760 "_id": account_id,
761 "_admin.operations.{}.worker".format(op_index): None,
762 }
763 update_dict = {
764 "_admin.operations.{}.worker".format(op_index): self.worker_id,
765 "_admin.current_operation": op_index,
766 }
kuuse6a470c62019-07-10 13:52:45 +0200767 return q_filter, update_dict
768
kuused124bfe2019-06-18 12:09:24 +0200769 def lock_HA(self, topic, op_type, op_id):
770 """
kuuse6a470c62019-07-10 13:52:45 +0200771 Lock a task, if possible, to indicate to the HA system that
kuused124bfe2019-06-18 12:09:24 +0200772 the task will be executed in this LCM instance.
kuuse6a470c62019-07-10 13:52:45 +0200773 :param topic: Can be "ns", "nsi", "vim", "wim", or "sdn"
774 :param op_type: Operation type, can be "nslcmops", "nsilcmops", "create", "edit", "delete"
775 :param op_id: NS, NSI: Operation ID VIM,WIM,SDN: Account ID + ':' + Operation Index
kuused124bfe2019-06-18 12:09:24 +0200776 :return:
kuuse6a470c62019-07-10 13:52:45 +0200777 True=lock was successful => execute the task (not registered by any other LCM instance)
kuused124bfe2019-06-18 12:09:24 +0200778 False=lock failed => do NOT execute the task (already registered by another LCM instance)
kuuse6a470c62019-07-10 13:52:45 +0200779
780 HA tasks and backward compatibility:
781 If topic is "account type" (VIM/WIM/SDN) and op_id is None, 'op_id' was not provided by NBI.
782 This means that the running NBI instance does not support HA.
783 In such a case this method should always return True, to always execute
784 the task in this instance of LCM, without querying the DB.
tierno59d22d22018-09-25 18:10:19 +0200785 """
786
calvinosanch9f9c6f22019-11-04 13:37:39 +0100787 # Backward compatibility for VIM/WIM/SDN/k8scluster without op_id
rshri932105f2024-07-05 15:11:55 +0000788 self.logger.info("Lock_HA")
kuuse6a470c62019-07-10 13:52:45 +0200789 if self._is_account_type_HA(topic) and op_id is None:
790 return True
tierno59d22d22018-09-25 18:10:19 +0200791
kuuse6a470c62019-07-10 13:52:45 +0200792 # Try to lock this task
tiernofa076c32020-08-13 14:25:47 +0000793 db_table_name = self.topic2dbtable_dict[topic]
kuuse6a470c62019-07-10 13:52:45 +0200794 q_filter, update_dict = self._get_dbparams_for_lock_HA(topic, op_type, op_id)
rshri932105f2024-07-05 15:11:55 +0000795 self.logger.info(
796 "db table name: {} update dict: {}".format(db_table_name, update_dict)
797 )
garciadeblas5697b8b2021-03-24 09:17:02 +0100798 db_lock_task = self.db.set_one(
799 db_table_name,
800 q_filter=q_filter,
801 update_dict=update_dict,
802 fail_on_empty=False,
803 )
kuused124bfe2019-06-18 12:09:24 +0200804 if db_lock_task is None:
garciadeblas5697b8b2021-03-24 09:17:02 +0100805 self.logger.debug(
806 "Task {} operation={} already locked by another worker".format(
807 topic, op_id
808 )
809 )
kuused124bfe2019-06-18 12:09:24 +0200810 return False
811 else:
kuuse6a470c62019-07-10 13:52:45 +0200812 # Set 'detailed-status' to 'In progress' for VIM/WIM/SDN operations
813 if self._is_account_type_HA(topic):
garciadeblas5697b8b2021-03-24 09:17:02 +0100814 detailed_status = "In progress"
kuuse6a470c62019-07-10 13:52:45 +0200815 account_id, op_index = self._get_account_and_op_HA(op_id)
garciadeblas5697b8b2021-03-24 09:17:02 +0100816 q_filter = {"_id": account_id}
817 update_dict = {
818 "_admin.operations.{}.detailed-status".format(
819 op_index
820 ): detailed_status
821 }
822 self.db.set_one(
823 db_table_name,
824 q_filter=q_filter,
825 update_dict=update_dict,
826 fail_on_empty=False,
827 )
kuused124bfe2019-06-18 12:09:24 +0200828 return True
829
tiernofa076c32020-08-13 14:25:47 +0000830 def unlock_HA(self, topic, op_type, op_id, operationState, detailed_status):
kuuse6a470c62019-07-10 13:52:45 +0200831 """
832 Register a task, done when finished a VIM/WIM/SDN 'create' operation.
833 :param topic: Can be "vim", "wim", or "sdn"
834 :param op_type: Operation type, can be "create", "edit", "delete"
835 :param op_id: Account ID + ':' + Operation Index
836 :return: nothing
837 """
rshri932105f2024-07-05 15:11:55 +0000838 self.logger.info("Unlock HA")
kuuse6a470c62019-07-10 13:52:45 +0200839 # Backward compatibility
tiernofa076c32020-08-13 14:25:47 +0000840 if not self._is_account_type_HA(topic) or not op_id:
kuuse6a470c62019-07-10 13:52:45 +0200841 return
842
843 # Get Account ID and Operation Index
844 account_id, op_index = self._get_account_and_op_HA(op_id)
tiernofa076c32020-08-13 14:25:47 +0000845 db_table_name = self.topic2dbtable_dict[topic]
rshri932105f2024-07-05 15:11:55 +0000846 self.logger.info("db_table_name: {}".format(db_table_name))
kuuse6a470c62019-07-10 13:52:45 +0200847 # If this is a 'delete' operation, the account may have been deleted (SUCCESS) or may still exist (FAILED)
848 # If the account exist, register the HA task.
849 # Update DB for HA tasks
garciadeblas5697b8b2021-03-24 09:17:02 +0100850 q_filter = {"_id": account_id}
851 update_dict = {
852 "_admin.operations.{}.operationState".format(op_index): operationState,
853 "_admin.operations.{}.detailed-status".format(op_index): detailed_status,
854 "_admin.operations.{}.worker".format(op_index): None,
855 "_admin.current_operation": None,
856 }
rshri932105f2024-07-05 15:11:55 +0000857 self.logger.info("Update dict: {}".format(update_dict))
garciadeblas5697b8b2021-03-24 09:17:02 +0100858 self.db.set_one(
859 db_table_name,
860 q_filter=q_filter,
861 update_dict=update_dict,
862 fail_on_empty=False,
863 )
kuuse6a470c62019-07-10 13:52:45 +0200864 return
865
kuused124bfe2019-06-18 12:09:24 +0200866 async def waitfor_related_HA(self, topic, op_type, op_id=None):
tierno59d22d22018-09-25 18:10:19 +0200867 """
kuused124bfe2019-06-18 12:09:24 +0200868 Wait for any pending related HA tasks
tierno59d22d22018-09-25 18:10:19 +0200869 """
kuused124bfe2019-06-18 12:09:24 +0200870
kuuse6a470c62019-07-10 13:52:45 +0200871 # Backward compatibility
garciadeblas5697b8b2021-03-24 09:17:02 +0100872 if not (
873 self._is_service_type_HA(topic) or self._is_account_type_HA(topic)
874 ) and (op_id is None):
kuuse6a470c62019-07-10 13:52:45 +0200875 return
kuused124bfe2019-06-18 12:09:24 +0200876
kuuse6a470c62019-07-10 13:52:45 +0200877 # Get DB table name
878 db_table_name = self.topic2dbtable_dict.get(topic)
879
880 # Get instance ID
881 _id = self._get_instance_id_HA(topic, op_type, op_id)
882 _filter = {"_id": _id}
garciadeblas5697b8b2021-03-24 09:17:02 +0100883 db_lcmop = self.db.get_one(db_table_name, _filter, fail_on_empty=False)
kuused124bfe2019-06-18 12:09:24 +0200884 if not db_lcmop:
tierno59d22d22018-09-25 18:10:19 +0200885 return
kuuse6a470c62019-07-10 13:52:45 +0200886
887 # Set DB _filter for querying any related process state
888 _filter = self._get_waitfor_filter_HA(db_lcmop, topic, op_type, op_id)
kuused124bfe2019-06-18 12:09:24 +0200889
890 # For HA, get list of tasks from DB instead of from dictionary (in-memory) variable.
garciadeblas5697b8b2021-03-24 09:17:02 +0100891 timeout_wait_for_task = (
892 3600 # Max time (seconds) to wait for a related task to finish
893 )
kuused124bfe2019-06-18 12:09:24 +0200894 # interval_wait_for_task = 30 # A too long polling interval slows things down considerably
garciadeblas5697b8b2021-03-24 09:17:02 +0100895 interval_wait_for_task = 10 # Interval in seconds for polling related tasks
kuused124bfe2019-06-18 12:09:24 +0200896 time_left = timeout_wait_for_task
897 old_num_related_tasks = 0
898 while True:
kuuse6a470c62019-07-10 13:52:45 +0200899 # Get related tasks (operations within the same instance as this) which are
kuused124bfe2019-06-18 12:09:24 +0200900 # still running (operationState='PROCESSING') and which were started before this task.
kuuse6a470c62019-07-10 13:52:45 +0200901 # In the case of op_type='ANY', get any related tasks with operationState='PROCESSING', ignore timestamps.
garciadeblas5697b8b2021-03-24 09:17:02 +0100902 db_waitfor_related_task = self.db.get_list(db_table_name, q_filter=_filter)
kuused124bfe2019-06-18 12:09:24 +0200903 new_num_related_tasks = len(db_waitfor_related_task)
kuuse6a470c62019-07-10 13:52:45 +0200904 # If there are no related tasks, there is nothing to wait for, so return.
kuused124bfe2019-06-18 12:09:24 +0200905 if not new_num_related_tasks:
kuused124bfe2019-06-18 12:09:24 +0200906 return
907 # If number of pending related tasks have changed,
908 # update the 'detailed-status' field and log the change.
kuuse6a470c62019-07-10 13:52:45 +0200909 # Do NOT update the 'detailed-status' for SDNC-associated-to-VIM operations ('ANY').
garciadeblas5697b8b2021-03-24 09:17:02 +0100910 if (op_type != "ANY") and (new_num_related_tasks != old_num_related_tasks):
911 step = "Waiting for {} related tasks to be completed.".format(
912 new_num_related_tasks
913 )
rshri932105f2024-07-05 15:11:55 +0000914 self.logger.info("{}".format(step))
kuuse6a470c62019-07-10 13:52:45 +0200915 update_dict = {}
garciadeblas5697b8b2021-03-24 09:17:02 +0100916 q_filter = {"_id": _id}
kuuse6a470c62019-07-10 13:52:45 +0200917 # NS/NSI
918 if self._is_service_type_HA(topic):
garciadeblas5697b8b2021-03-24 09:17:02 +0100919 update_dict = {
920 "detailed-status": step,
921 "queuePosition": new_num_related_tasks,
922 }
kuuse6a470c62019-07-10 13:52:45 +0200923 # VIM/WIM/SDN
924 elif self._is_account_type_HA(topic):
925 _, op_index = self._get_account_and_op_HA(op_id)
garciadeblas5697b8b2021-03-24 09:17:02 +0100926 update_dict = {
927 "_admin.operations.{}.detailed-status".format(op_index): step
928 }
kuuse6a470c62019-07-10 13:52:45 +0200929 self.logger.debug("Task {} operation={} {}".format(topic, _id, step))
garciadeblas5697b8b2021-03-24 09:17:02 +0100930 self.db.set_one(
931 db_table_name,
932 q_filter=q_filter,
933 update_dict=update_dict,
934 fail_on_empty=False,
935 )
kuused124bfe2019-06-18 12:09:24 +0200936 old_num_related_tasks = new_num_related_tasks
937 time_left -= interval_wait_for_task
938 if time_left < 0:
939 raise LcmException(
940 "Timeout ({}) when waiting for related tasks to be completed".format(
garciadeblas5697b8b2021-03-24 09:17:02 +0100941 timeout_wait_for_task
942 )
943 )
kuused124bfe2019-06-18 12:09:24 +0200944 await asyncio.sleep(interval_wait_for_task)
945
946 return