blob: 5817b16a701d8d8cedbb4dcce69f65d87701bfdb [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
kuused124bfe2019-06-18 12:09:24 +0200224 self.db.set_one(item, {"_id": _id}, _desc)
225 _desc.clear()
226 # except DbException as e:
227 # self.logger.error("Updating {} _id={} with '{}'. Error: {}".format(item, _id, _desc, e))
228
aticig1dda84c2022-09-10 01:56:58 +0300229 @staticmethod
230 def calculate_charm_hash(zipped_file):
231 """Calculate the hash of charm files which ends with .charm
232
233 Args:
234 zipped_file (str): Existing charm package full path
235
236 Returns:
237 hex digest (str): The hash of the charm file
238 """
Gabriel Cuba4c0e6802023-10-09 13:22:38 -0500239 filehash = hashlib.sha256()
aticig1dda84c2022-09-10 01:56:58 +0300240 with open(zipped_file, mode="rb") as file:
241 contents = file.read()
242 filehash.update(contents)
243 return filehash.hexdigest()
244
245 @staticmethod
246 def compare_charm_hash(current_charm, target_charm):
247 """Compare the existing charm and the target charm if the charms
248 are given as zip files ends with .charm
249
250 Args:
251 current_charm (str): Existing charm package full path
252 target_charm (str): Target charm package full path
253
254 Returns:
255 True/False (bool): if charm has changed it returns True
256 """
257 return LcmBase.calculate_charm_hash(
258 current_charm
259 ) != LcmBase.calculate_charm_hash(target_charm)
260
261 @staticmethod
262 def compare_charmdir_hash(current_charm_dir, target_charm_dir):
263 """Compare the existing charm and the target charm if the charms
264 are given as directories
265
266 Args:
267 current_charm_dir (str): Existing charm package directory path
268 target_charm_dir (str): Target charm package directory path
269
270 Returns:
271 True/False (bool): if charm has changed it returns True
272 """
273 return checksumdir.dirhash(current_charm_dir) != checksumdir.dirhash(
274 target_charm_dir
275 )
276
aticigdffa6212022-04-12 15:27:53 +0300277 def check_charm_hash_changed(
278 self, current_charm_path: str, target_charm_path: str
279 ) -> bool:
280 """Find the target charm has changed or not by checking the hash of
281 old and new charm packages
282
283 Args:
284 current_charm_path (str): Existing charm package artifact path
285 target_charm_path (str): Target charm package artifact path
286
287 Returns:
288 True/False (bool): if charm has changed it returns True
289
290 """
aticig1dda84c2022-09-10 01:56:58 +0300291 try:
292 # Check if the charm artifacts are available
293 current_charm = self.fs.path + current_charm_path
294 target_charm = self.fs.path + target_charm_path
aticigdffa6212022-04-12 15:27:53 +0300295
aticig1dda84c2022-09-10 01:56:58 +0300296 if os.path.exists(current_charm) and os.path.exists(target_charm):
aticig1dda84c2022-09-10 01:56:58 +0300297 # Compare the hash of .charm files
298 if current_charm.endswith(".charm"):
299 return LcmBase.compare_charm_hash(current_charm, target_charm)
aticigdffa6212022-04-12 15:27:53 +0300300
aticig1dda84c2022-09-10 01:56:58 +0300301 # Compare the hash of charm folders
302 return LcmBase.compare_charmdir_hash(current_charm, target_charm)
303
304 else:
305 raise LcmException(
306 "Charm artifact {} does not exist in the VNF Package".format(
307 self.fs.path + target_charm_path
308 )
aticigdffa6212022-04-12 15:27:53 +0300309 )
aticig1dda84c2022-09-10 01:56:58 +0300310 except (IOError, OSError, TypeError) as error:
311 self.logger.debug(traceback.format_exc())
312 self.logger.error(f"{error} occured while checking the charm hashes")
313 raise LcmException(error)
aticigdffa6212022-04-12 15:27:53 +0300314
aticig9bc63ac2022-07-27 09:32:06 +0300315 @staticmethod
316 def get_charm_name(charm_metadata_file: str) -> str:
317 """Get the charm name from metadata file.
318
319 Args:
320 charm_metadata_file (str): charm metadata file full path
321
322 Returns:
323 charm_name (str): charm name
324
325 """
326 # Read charm metadata.yaml to get the charm name
327 with open(charm_metadata_file, "r") as metadata_file:
328 content = yaml.safe_load(metadata_file)
329 charm_name = content["name"]
330 return str(charm_name)
331
332 def _get_charm_path(
333 self, nsd_package_path: str, nsd_package_name: str, charm_folder_name: str
334 ) -> str:
335 """Get the full path of charm folder.
336
337 Args:
338 nsd_package_path (str): NSD package full path
339 nsd_package_name (str): NSD package name
340 charm_folder_name (str): folder name
341
342 Returns:
343 charm_path (str): charm folder full path
344 """
345 charm_path = (
346 self.fs.path
347 + nsd_package_path
348 + "/"
349 + nsd_package_name
350 + "/charms/"
351 + charm_folder_name
352 )
353 return charm_path
354
355 def _get_charm_metadata_file(
356 self,
357 charm_folder_name: str,
358 nsd_package_path: str,
359 nsd_package_name: str,
360 charm_path: str = None,
361 ) -> str:
362 """Get the path of charm metadata file.
363
364 Args:
365 charm_folder_name (str): folder name
366 nsd_package_path (str): NSD package full path
367 nsd_package_name (str): NSD package name
368 charm_path (str): Charm full path
369
370 Returns:
371 charm_metadata_file_path (str): charm metadata file full path
372
373 """
374 # Locate the charm metadata.yaml
375 if charm_folder_name.endswith(".charm"):
376 extract_path = (
377 self.fs.path
378 + nsd_package_path
379 + "/"
380 + nsd_package_name
381 + "/charms/"
aticiga37c6ff2022-08-20 20:56:19 +0300382 + charm_folder_name.replace(".charm", "")
aticig9bc63ac2022-07-27 09:32:06 +0300383 )
384 # Extract .charm to extract path
385 with ZipFile(charm_path, "r") as zipfile:
386 zipfile.extractall(extract_path)
387 return extract_path + "/metadata.yaml"
388 else:
389 return charm_path + "/metadata.yaml"
390
391 def find_charm_name(self, db_nsr: dict, charm_folder_name: str) -> str:
392 """Get the charm name from metadata.yaml of charm package.
393
394 Args:
395 db_nsr (dict): NS record as a dictionary
396 charm_folder_name (str): charm folder name
397
398 Returns:
399 charm_name (str): charm name
400 """
401 try:
402 if not charm_folder_name:
403 raise LcmException("charm_folder_name should be provided.")
404
405 # Find nsd_package details: path, name
406 revision = db_nsr.get("revision", "")
aticiga37c6ff2022-08-20 20:56:19 +0300407
408 # Get the NSD package path
409 if revision:
preethika.p28b0bf82022-09-23 07:36:28 +0000410 nsd_package_path = db_nsr["nsd-id"] + ":" + str(revision)
aticiga37c6ff2022-08-20 20:56:19 +0300411 db_nsd = self.db.get_one("nsds_revisions", {"_id": nsd_package_path})
412
413 else:
414 nsd_package_path = db_nsr["nsd-id"]
415
416 db_nsd = self.db.get_one("nsds", {"_id": nsd_package_path})
417
418 # Get the NSD package name
419 nsd_package_name = db_nsd["_admin"]["storage"]["pkg-dir"]
aticig9bc63ac2022-07-27 09:32:06 +0300420
421 # Remove the existing nsd package and sync from FsMongo
422 shutil.rmtree(self.fs.path + nsd_package_path, ignore_errors=True)
423 self.fs.sync(from_path=nsd_package_path)
424
425 # Get the charm path
426 charm_path = self._get_charm_path(
427 nsd_package_path, nsd_package_name, charm_folder_name
428 )
429
430 # Find charm metadata file full path
431 charm_metadata_file = self._get_charm_metadata_file(
432 charm_folder_name, nsd_package_path, nsd_package_name, charm_path
433 )
434
435 # Return charm name
436 return self.get_charm_name(charm_metadata_file)
437
438 except (
439 yaml.YAMLError,
440 IOError,
441 FsException,
442 KeyError,
443 TypeError,
444 FileNotFoundError,
445 BadZipfile,
446 ) as error:
447 self.logger.debug(traceback.format_exc())
448 self.logger.error(f"{error} occured while getting the charm name")
449 raise LcmException(error)
450
kuused124bfe2019-06-18 12:09:24 +0200451
452class TaskRegistry(LcmBase):
tierno59d22d22018-09-25 18:10:19 +0200453 """
454 Implements a registry of task needed for later cancelation, look for related tasks that must be completed before
455 etc. It stores a four level dict
456 First level is the topic, ns, vim_account, sdn
457 Second level is the _id
458 Third level is the operation id
459 Fourth level is a descriptive name, the value is the task class
kuused124bfe2019-06-18 12:09:24 +0200460
461 The HA (High-Availability) methods are used when more than one LCM instance is running.
462 To register the current task in the external DB, use LcmBase as base class, to be able
463 to reuse LcmBase.update_db_2()
464 The DB registry uses the following fields to distinguish a task:
465 - op_type: operation type ("nslcmops" or "nsilcmops")
466 - op_id: operation ID
467 - worker: the worker ID for this process
tierno59d22d22018-09-25 18:10:19 +0200468 """
469
kuuse6a470c62019-07-10 13:52:45 +0200470 # NS/NSI: "services" VIM/WIM/SDN: "accounts"
garciadeblas5697b8b2021-03-24 09:17:02 +0100471 topic_service_list = ["ns", "nsi"]
472 topic_account_list = ["vim", "wim", "sdn", "k8scluster", "vca", "k8srepo"]
kuuse6a470c62019-07-10 13:52:45 +0200473
474 # Map topic to InstanceID
garciadeblas5697b8b2021-03-24 09:17:02 +0100475 topic2instid_dict = {"ns": "nsInstanceId", "nsi": "netsliceInstanceId"}
kuuse6a470c62019-07-10 13:52:45 +0200476
477 # Map topic to DB table name
478 topic2dbtable_dict = {
garciadeblas5697b8b2021-03-24 09:17:02 +0100479 "ns": "nslcmops",
480 "nsi": "nsilcmops",
481 "vim": "vim_accounts",
482 "wim": "wim_accounts",
483 "sdn": "sdns",
484 "k8scluster": "k8sclusters",
485 "vca": "vca",
486 "k8srepo": "k8srepos",
487 }
kuused124bfe2019-06-18 12:09:24 +0200488
bravof922c4172020-11-24 21:21:43 -0300489 def __init__(self, worker_id=None, logger=None):
tierno59d22d22018-09-25 18:10:19 +0200490 self.task_registry = {
491 "ns": {},
Felipe Vicensc2033f22018-11-15 15:09:58 +0100492 "nsi": {},
tierno59d22d22018-09-25 18:10:19 +0200493 "vim_account": {},
tiernoe37b57d2018-12-11 17:22:51 +0000494 "wim_account": {},
tierno59d22d22018-09-25 18:10:19 +0200495 "sdn": {},
calvinosanch9f9c6f22019-11-04 13:37:39 +0100496 "k8scluster": {},
David Garciac1fe90a2021-03-31 19:12:02 +0200497 "vca": {},
calvinosanch9f9c6f22019-11-04 13:37:39 +0100498 "k8srepo": {},
tierno59d22d22018-09-25 18:10:19 +0200499 }
kuused124bfe2019-06-18 12:09:24 +0200500 self.worker_id = worker_id
bravof922c4172020-11-24 21:21:43 -0300501 self.db = Database().instance.db
kuused124bfe2019-06-18 12:09:24 +0200502 self.logger = logger
tierno59d22d22018-09-25 18:10:19 +0200503
504 def register(self, topic, _id, op_id, task_name, task):
505 """
506 Register a new task
Felipe Vicensc2033f22018-11-15 15:09:58 +0100507 :param topic: Can be "ns", "nsi", "vim_account", "sdn"
tierno59d22d22018-09-25 18:10:19 +0200508 :param _id: _id of the related item
509 :param op_id: id of the operation of the related item
510 :param task_name: Task descriptive name, as create, instantiate, terminate. Must be unique in this op_id
511 :param task: Task class
512 :return: none
513 """
514 if _id not in self.task_registry[topic]:
515 self.task_registry[topic][_id] = OrderedDict()
516 if op_id not in self.task_registry[topic][_id]:
517 self.task_registry[topic][_id][op_id] = {task_name: task}
518 else:
519 self.task_registry[topic][_id][op_id][task_name] = task
520 # print("registering task", topic, _id, op_id, task_name, task)
521
522 def remove(self, topic, _id, op_id, task_name=None):
523 """
tiernobaa51102018-12-14 13:16:18 +0000524 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 +0100525 :param topic: Can be "ns", "nsi", "vim_account", "sdn"
tierno59d22d22018-09-25 18:10:19 +0200526 :param _id: _id of the related item
527 :param op_id: id of the operation of the related item
tiernobaa51102018-12-14 13:16:18 +0000528 :param task_name: Task descriptive name. If none it deletes all tasks with same _id and op_id
529 :return: None
tierno59d22d22018-09-25 18:10:19 +0200530 """
tiernobaa51102018-12-14 13:16:18 +0000531 if not self.task_registry[topic].get(_id):
tierno59d22d22018-09-25 18:10:19 +0200532 return
533 if not task_name:
tiernobaa51102018-12-14 13:16:18 +0000534 self.task_registry[topic][_id].pop(op_id, None)
535 elif self.task_registry[topic][_id].get(op_id):
536 self.task_registry[topic][_id][op_id].pop(task_name, None)
537
538 # delete done tasks
539 for op_id_ in list(self.task_registry[topic][_id]):
540 for name, task in self.task_registry[topic][_id][op_id_].items():
541 if not task.done():
542 break
543 else:
544 del self.task_registry[topic][_id][op_id_]
tierno59d22d22018-09-25 18:10:19 +0200545 if not self.task_registry[topic][_id]:
546 del self.task_registry[topic][_id]
547
548 def lookfor_related(self, topic, _id, my_op_id=None):
549 task_list = []
550 task_name_list = []
551 if _id not in self.task_registry[topic]:
552 return "", task_name_list
553 for op_id in reversed(self.task_registry[topic][_id]):
554 if my_op_id:
555 if my_op_id == op_id:
556 my_op_id = None # so that the next task is taken
557 continue
558
559 for task_name, task in self.task_registry[topic][_id][op_id].items():
tiernobaa51102018-12-14 13:16:18 +0000560 if not task.done():
561 task_list.append(task)
562 task_name_list.append(task_name)
tierno59d22d22018-09-25 18:10:19 +0200563 break
564 return ", ".join(task_name_list), task_list
565
566 def cancel(self, topic, _id, target_op_id=None, target_task_name=None):
567 """
kuused124bfe2019-06-18 12:09:24 +0200568 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 +0100569 this is cancelled, and the same with task_name
tierno59d22d22018-09-25 18:10:19 +0200570 """
571 if not self.task_registry[topic].get(_id):
572 return
573 for op_id in reversed(self.task_registry[topic][_id]):
574 if target_op_id and target_op_id != op_id:
575 continue
576 for task_name, task in self.task_registry[topic][_id][op_id].items():
577 if target_task_name and target_task_name != task_name:
578 continue
579 # result =
580 task.cancel()
581 # if result:
582 # self.logger.debug("{} _id={} order_id={} task={} cancelled".format(topic, _id, op_id, task_name))
583
kuuse6a470c62019-07-10 13:52:45 +0200584 # Is topic NS/NSI?
585 def _is_service_type_HA(self, topic):
586 return topic in self.topic_service_list
587
588 # Is topic VIM/WIM/SDN?
589 def _is_account_type_HA(self, topic):
590 return topic in self.topic_account_list
591
592 # Input: op_id, example: 'abc123def:3' Output: account_id='abc123def', op_index=3
593 def _get_account_and_op_HA(self, op_id):
594 if not op_id:
tiernofa076c32020-08-13 14:25:47 +0000595 return None, None
garciadeblas5697b8b2021-03-24 09:17:02 +0100596 account_id, _, op_index = op_id.rpartition(":")
tiernofa076c32020-08-13 14:25:47 +0000597 if not account_id or not op_index.isdigit():
598 return None, None
kuuse6a470c62019-07-10 13:52:45 +0200599 return account_id, op_index
600
601 # Get '_id' for any topic and operation
602 def _get_instance_id_HA(self, topic, op_type, op_id):
603 _id = None
604 # Special operation 'ANY', for SDN account associated to a VIM account: op_id as '_id'
garciadeblas5697b8b2021-03-24 09:17:02 +0100605 if op_type == "ANY":
kuuse6a470c62019-07-10 13:52:45 +0200606 _id = op_id
607 # NS/NSI: Use op_id as '_id'
608 elif self._is_service_type_HA(topic):
609 _id = op_id
calvinosanch9f9c6f22019-11-04 13:37:39 +0100610 # 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 +0200611 elif self._is_account_type_HA(topic):
612 _id, _ = self._get_account_and_op_HA(op_id)
613 return _id
614
615 # Set DB _filter for querying any related process state
616 def _get_waitfor_filter_HA(self, db_lcmop, topic, op_type, op_id):
617 _filter = {}
618 # Special operation 'ANY', for SDN account associated to a VIM account: op_id as '_id'
619 # In this special case, the timestamp is ignored
garciadeblas5697b8b2021-03-24 09:17:02 +0100620 if op_type == "ANY":
621 _filter = {"operationState": "PROCESSING"}
kuuse6a470c62019-07-10 13:52:45 +0200622 # Otherwise, get 'startTime' timestamp for this operation
623 else:
624 # NS/NSI
625 if self._is_service_type_HA(topic):
tierno79cd8ad2019-10-18 13:03:10 +0000626 now = time()
kuuse6a470c62019-07-10 13:52:45 +0200627 starttime_this_op = db_lcmop.get("startTime")
628 instance_id_label = self.topic2instid_dict.get(topic)
629 instance_id = db_lcmop.get(instance_id_label)
garciadeblas5697b8b2021-03-24 09:17:02 +0100630 _filter = {
631 instance_id_label: instance_id,
632 "operationState": "PROCESSING",
633 "startTime.lt": starttime_this_op,
634 "_admin.modified.gt": now
635 - 2 * 3600, # ignore if tow hours of inactivity
636 }
calvinosanch9f9c6f22019-11-04 13:37:39 +0100637 # VIM/WIM/SDN/K8scluster
kuuse6a470c62019-07-10 13:52:45 +0200638 elif self._is_account_type_HA(topic):
639 _, op_index = self._get_account_and_op_HA(op_id)
garciadeblas5697b8b2021-03-24 09:17:02 +0100640 _ops = db_lcmop["_admin"]["operations"]
kuuse6a470c62019-07-10 13:52:45 +0200641 _this_op = _ops[int(op_index)]
garciadeblas5697b8b2021-03-24 09:17:02 +0100642 starttime_this_op = _this_op.get("startTime", None)
643 _filter = {
644 "operationState": "PROCESSING",
645 "startTime.lt": starttime_this_op,
646 }
kuuse6a470c62019-07-10 13:52:45 +0200647 return _filter
648
649 # Get DB params for any topic and operation
650 def _get_dbparams_for_lock_HA(self, topic, op_type, op_id):
651 q_filter = {}
652 update_dict = {}
653 # NS/NSI
654 if self._is_service_type_HA(topic):
garciadeblas5697b8b2021-03-24 09:17:02 +0100655 q_filter = {"_id": op_id, "_admin.worker": None}
656 update_dict = {"_admin.worker": self.worker_id}
kuuse6a470c62019-07-10 13:52:45 +0200657 # VIM/WIM/SDN
658 elif self._is_account_type_HA(topic):
659 account_id, op_index = self._get_account_and_op_HA(op_id)
660 if not account_id:
661 return None, None
garciadeblas5697b8b2021-03-24 09:17:02 +0100662 if op_type == "create":
kuuse6a470c62019-07-10 13:52:45 +0200663 # Creating a VIM/WIM/SDN account implies setting '_admin.current_operation' = 0
664 op_index = 0
garciadeblas5697b8b2021-03-24 09:17:02 +0100665 q_filter = {
666 "_id": account_id,
667 "_admin.operations.{}.worker".format(op_index): None,
668 }
669 update_dict = {
670 "_admin.operations.{}.worker".format(op_index): self.worker_id,
671 "_admin.current_operation": op_index,
672 }
kuuse6a470c62019-07-10 13:52:45 +0200673 return q_filter, update_dict
674
kuused124bfe2019-06-18 12:09:24 +0200675 def lock_HA(self, topic, op_type, op_id):
676 """
kuuse6a470c62019-07-10 13:52:45 +0200677 Lock a task, if possible, to indicate to the HA system that
kuused124bfe2019-06-18 12:09:24 +0200678 the task will be executed in this LCM instance.
kuuse6a470c62019-07-10 13:52:45 +0200679 :param topic: Can be "ns", "nsi", "vim", "wim", or "sdn"
680 :param op_type: Operation type, can be "nslcmops", "nsilcmops", "create", "edit", "delete"
681 :param op_id: NS, NSI: Operation ID VIM,WIM,SDN: Account ID + ':' + Operation Index
kuused124bfe2019-06-18 12:09:24 +0200682 :return:
kuuse6a470c62019-07-10 13:52:45 +0200683 True=lock was successful => execute the task (not registered by any other LCM instance)
kuused124bfe2019-06-18 12:09:24 +0200684 False=lock failed => do NOT execute the task (already registered by another LCM instance)
kuuse6a470c62019-07-10 13:52:45 +0200685
686 HA tasks and backward compatibility:
687 If topic is "account type" (VIM/WIM/SDN) and op_id is None, 'op_id' was not provided by NBI.
688 This means that the running NBI instance does not support HA.
689 In such a case this method should always return True, to always execute
690 the task in this instance of LCM, without querying the DB.
tierno59d22d22018-09-25 18:10:19 +0200691 """
692
calvinosanch9f9c6f22019-11-04 13:37:39 +0100693 # Backward compatibility for VIM/WIM/SDN/k8scluster without op_id
kuuse6a470c62019-07-10 13:52:45 +0200694 if self._is_account_type_HA(topic) and op_id is None:
695 return True
tierno59d22d22018-09-25 18:10:19 +0200696
kuuse6a470c62019-07-10 13:52:45 +0200697 # Try to lock this task
tiernofa076c32020-08-13 14:25:47 +0000698 db_table_name = self.topic2dbtable_dict[topic]
kuuse6a470c62019-07-10 13:52:45 +0200699 q_filter, update_dict = self._get_dbparams_for_lock_HA(topic, op_type, op_id)
garciadeblas5697b8b2021-03-24 09:17:02 +0100700 db_lock_task = self.db.set_one(
701 db_table_name,
702 q_filter=q_filter,
703 update_dict=update_dict,
704 fail_on_empty=False,
705 )
kuused124bfe2019-06-18 12:09:24 +0200706 if db_lock_task is None:
garciadeblas5697b8b2021-03-24 09:17:02 +0100707 self.logger.debug(
708 "Task {} operation={} already locked by another worker".format(
709 topic, op_id
710 )
711 )
kuused124bfe2019-06-18 12:09:24 +0200712 return False
713 else:
kuuse6a470c62019-07-10 13:52:45 +0200714 # Set 'detailed-status' to 'In progress' for VIM/WIM/SDN operations
715 if self._is_account_type_HA(topic):
garciadeblas5697b8b2021-03-24 09:17:02 +0100716 detailed_status = "In progress"
kuuse6a470c62019-07-10 13:52:45 +0200717 account_id, op_index = self._get_account_and_op_HA(op_id)
garciadeblas5697b8b2021-03-24 09:17:02 +0100718 q_filter = {"_id": account_id}
719 update_dict = {
720 "_admin.operations.{}.detailed-status".format(
721 op_index
722 ): detailed_status
723 }
724 self.db.set_one(
725 db_table_name,
726 q_filter=q_filter,
727 update_dict=update_dict,
728 fail_on_empty=False,
729 )
kuused124bfe2019-06-18 12:09:24 +0200730 return True
731
tiernofa076c32020-08-13 14:25:47 +0000732 def unlock_HA(self, topic, op_type, op_id, operationState, detailed_status):
kuuse6a470c62019-07-10 13:52:45 +0200733 """
734 Register a task, done when finished a VIM/WIM/SDN 'create' operation.
735 :param topic: Can be "vim", "wim", or "sdn"
736 :param op_type: Operation type, can be "create", "edit", "delete"
737 :param op_id: Account ID + ':' + Operation Index
738 :return: nothing
739 """
740
741 # Backward compatibility
tiernofa076c32020-08-13 14:25:47 +0000742 if not self._is_account_type_HA(topic) or not op_id:
kuuse6a470c62019-07-10 13:52:45 +0200743 return
744
745 # Get Account ID and Operation Index
746 account_id, op_index = self._get_account_and_op_HA(op_id)
tiernofa076c32020-08-13 14:25:47 +0000747 db_table_name = self.topic2dbtable_dict[topic]
kuuse6a470c62019-07-10 13:52:45 +0200748
749 # If this is a 'delete' operation, the account may have been deleted (SUCCESS) or may still exist (FAILED)
750 # If the account exist, register the HA task.
751 # Update DB for HA tasks
garciadeblas5697b8b2021-03-24 09:17:02 +0100752 q_filter = {"_id": account_id}
753 update_dict = {
754 "_admin.operations.{}.operationState".format(op_index): operationState,
755 "_admin.operations.{}.detailed-status".format(op_index): detailed_status,
756 "_admin.operations.{}.worker".format(op_index): None,
757 "_admin.current_operation": None,
758 }
759 self.db.set_one(
760 db_table_name,
761 q_filter=q_filter,
762 update_dict=update_dict,
763 fail_on_empty=False,
764 )
kuuse6a470c62019-07-10 13:52:45 +0200765 return
766
kuused124bfe2019-06-18 12:09:24 +0200767 async def waitfor_related_HA(self, topic, op_type, op_id=None):
tierno59d22d22018-09-25 18:10:19 +0200768 """
kuused124bfe2019-06-18 12:09:24 +0200769 Wait for any pending related HA tasks
tierno59d22d22018-09-25 18:10:19 +0200770 """
kuused124bfe2019-06-18 12:09:24 +0200771
kuuse6a470c62019-07-10 13:52:45 +0200772 # Backward compatibility
garciadeblas5697b8b2021-03-24 09:17:02 +0100773 if not (
774 self._is_service_type_HA(topic) or self._is_account_type_HA(topic)
775 ) and (op_id is None):
kuuse6a470c62019-07-10 13:52:45 +0200776 return
kuused124bfe2019-06-18 12:09:24 +0200777
kuuse6a470c62019-07-10 13:52:45 +0200778 # Get DB table name
779 db_table_name = self.topic2dbtable_dict.get(topic)
780
781 # Get instance ID
782 _id = self._get_instance_id_HA(topic, op_type, op_id)
783 _filter = {"_id": _id}
garciadeblas5697b8b2021-03-24 09:17:02 +0100784 db_lcmop = self.db.get_one(db_table_name, _filter, fail_on_empty=False)
kuused124bfe2019-06-18 12:09:24 +0200785 if not db_lcmop:
tierno59d22d22018-09-25 18:10:19 +0200786 return
kuuse6a470c62019-07-10 13:52:45 +0200787
788 # Set DB _filter for querying any related process state
789 _filter = self._get_waitfor_filter_HA(db_lcmop, topic, op_type, op_id)
kuused124bfe2019-06-18 12:09:24 +0200790
791 # For HA, get list of tasks from DB instead of from dictionary (in-memory) variable.
garciadeblas5697b8b2021-03-24 09:17:02 +0100792 timeout_wait_for_task = (
793 3600 # Max time (seconds) to wait for a related task to finish
794 )
kuused124bfe2019-06-18 12:09:24 +0200795 # interval_wait_for_task = 30 # A too long polling interval slows things down considerably
garciadeblas5697b8b2021-03-24 09:17:02 +0100796 interval_wait_for_task = 10 # Interval in seconds for polling related tasks
kuused124bfe2019-06-18 12:09:24 +0200797 time_left = timeout_wait_for_task
798 old_num_related_tasks = 0
799 while True:
kuuse6a470c62019-07-10 13:52:45 +0200800 # Get related tasks (operations within the same instance as this) which are
kuused124bfe2019-06-18 12:09:24 +0200801 # still running (operationState='PROCESSING') and which were started before this task.
kuuse6a470c62019-07-10 13:52:45 +0200802 # In the case of op_type='ANY', get any related tasks with operationState='PROCESSING', ignore timestamps.
garciadeblas5697b8b2021-03-24 09:17:02 +0100803 db_waitfor_related_task = self.db.get_list(db_table_name, q_filter=_filter)
kuused124bfe2019-06-18 12:09:24 +0200804 new_num_related_tasks = len(db_waitfor_related_task)
kuuse6a470c62019-07-10 13:52:45 +0200805 # If there are no related tasks, there is nothing to wait for, so return.
kuused124bfe2019-06-18 12:09:24 +0200806 if not new_num_related_tasks:
kuused124bfe2019-06-18 12:09:24 +0200807 return
808 # If number of pending related tasks have changed,
809 # update the 'detailed-status' field and log the change.
kuuse6a470c62019-07-10 13:52:45 +0200810 # Do NOT update the 'detailed-status' for SDNC-associated-to-VIM operations ('ANY').
garciadeblas5697b8b2021-03-24 09:17:02 +0100811 if (op_type != "ANY") and (new_num_related_tasks != old_num_related_tasks):
812 step = "Waiting for {} related tasks to be completed.".format(
813 new_num_related_tasks
814 )
kuuse6a470c62019-07-10 13:52:45 +0200815 update_dict = {}
garciadeblas5697b8b2021-03-24 09:17:02 +0100816 q_filter = {"_id": _id}
kuuse6a470c62019-07-10 13:52:45 +0200817 # NS/NSI
818 if self._is_service_type_HA(topic):
garciadeblas5697b8b2021-03-24 09:17:02 +0100819 update_dict = {
820 "detailed-status": step,
821 "queuePosition": new_num_related_tasks,
822 }
kuuse6a470c62019-07-10 13:52:45 +0200823 # VIM/WIM/SDN
824 elif self._is_account_type_HA(topic):
825 _, op_index = self._get_account_and_op_HA(op_id)
garciadeblas5697b8b2021-03-24 09:17:02 +0100826 update_dict = {
827 "_admin.operations.{}.detailed-status".format(op_index): step
828 }
kuuse6a470c62019-07-10 13:52:45 +0200829 self.logger.debug("Task {} operation={} {}".format(topic, _id, step))
garciadeblas5697b8b2021-03-24 09:17:02 +0100830 self.db.set_one(
831 db_table_name,
832 q_filter=q_filter,
833 update_dict=update_dict,
834 fail_on_empty=False,
835 )
kuused124bfe2019-06-18 12:09:24 +0200836 old_num_related_tasks = new_num_related_tasks
837 time_left -= interval_wait_for_task
838 if time_left < 0:
839 raise LcmException(
840 "Timeout ({}) when waiting for related tasks to be completed".format(
garciadeblas5697b8b2021-03-24 09:17:02 +0100841 timeout_wait_for_task
842 )
843 )
kuused124bfe2019-06-18 12:09:24 +0200844 await asyncio.sleep(interval_wait_for_task)
845
846 return