blob: 7ce1841faac51dfa9fccf0efa7402bed2cecaba4 [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
tiernof578e552018-11-08 19:07:20 +010043class LcmExceptionNoMgmtIP(LcmException):
44 pass
45
46
gcalvinoed7f6d42018-12-14 14:44:56 +010047class LcmExceptionExit(LcmException):
48 pass
49
50
tierno59d22d22018-09-25 18:10:19 +020051def versiontuple(v):
tierno27246d82018-09-27 15:59:09 +020052 """utility for compare dot separate versions. Fills with zeros to proper number comparison
53 package version will be something like 4.0.1.post11+gb3f024d.dirty-1. Where 4.0.1 is the git tag, postXX is the
54 number of commits from this tag, and +XXXXXXX is the git commit short id. Total length is 16 with until 999 commits
55 """
tierno59d22d22018-09-25 18:10:19 +020056 filled = []
57 for point in v.split("."):
tiernoe64f7fb2019-09-11 08:55:52 +000058 point, _, _ = point.partition("+")
59 point, _, _ = point.partition("-")
60 filled.append(point.zfill(20))
tierno59d22d22018-09-25 18:10:19 +020061 return tuple(filled)
62
63
tierno744303e2020-01-13 16:46:31 +000064def deep_get(target_dict, key_list, default_value=None):
tierno626e0152019-11-29 14:16:16 +000065 """
66 Get a value from target_dict entering in the nested keys. If keys does not exist, it returns None
67 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
68 :param target_dict: dictionary to be read
69 :param key_list: list of keys to read from target_dict
tierno744303e2020-01-13 16:46:31 +000070 :param default_value: value to return if key is not present in the nested dictionary
tierno626e0152019-11-29 14:16:16 +000071 :return: The wanted value if exist, None otherwise
72 """
73 for key in key_list:
74 if not isinstance(target_dict, dict) or key not in target_dict:
tierno744303e2020-01-13 16:46:31 +000075 return default_value
tierno626e0152019-11-29 14:16:16 +000076 target_dict = target_dict[key]
77 return target_dict
78
79
tierno744303e2020-01-13 16:46:31 +000080def get_iterable(in_dict, in_key):
81 """
82 Similar to <dict>.get(), but if value is None, False, ..., An empty tuple is returned instead
83 :param in_dict: a dictionary
84 :param in_key: the key to look for at in_dict
85 :return: in_dict[in_var] or () if it is None or not present
86 """
87 if not in_dict.get(in_key):
88 return ()
89 return in_dict[in_key]
90
91
aticigdffa6212022-04-12 15:27:53 +030092def check_juju_bundle_existence(vnfd: dict) -> str:
93 """Checks the existence of juju-bundle in the descriptor
94
95 Args:
96 vnfd: Descriptor as a dictionary
97
98 Returns:
99 Juju bundle if dictionary has juju-bundle else None
100
101 """
102 if vnfd.get("vnfd"):
103 vnfd = vnfd["vnfd"]
104
105 for kdu in vnfd.get("kdu", []):
106 return kdu.get("juju-bundle", None)
107
108
109def get_charm_artifact_path(base_folder, charm_name, charm_type, revision=str()) -> str:
110 """Finds the charm artifact paths
111
112 Args:
113 base_folder: Main folder which will be looked up for charm
114 charm_name: Charm name
115 charm_type: Type of charm native_charm, lxc_proxy_charm or k8s_proxy_charm
116 revision: vnf package revision number if there is
117
118 Returns:
119 artifact_path: (str)
120
121 """
122 extension = ""
123 if revision:
124 extension = ":" + str(revision)
125
126 if base_folder.get("pkg-dir"):
127 artifact_path = "{}/{}/{}/{}".format(
aticigd7083542022-05-30 20:45:55 +0300128 base_folder["folder"].split(":")[0] + extension,
aticigdffa6212022-04-12 15:27:53 +0300129 base_folder["pkg-dir"],
130 "charms"
131 if charm_type in ("native_charm", "lxc_proxy_charm", "k8s_proxy_charm")
132 else "helm-charts",
133 charm_name,
134 )
135
136 else:
137 # For SOL004 packages
138 artifact_path = "{}/Scripts/{}/{}".format(
aticigd7083542022-05-30 20:45:55 +0300139 base_folder["folder"].split(":")[0] + extension,
aticigdffa6212022-04-12 15:27:53 +0300140 "charms"
141 if charm_type in ("native_charm", "lxc_proxy_charm", "k8s_proxy_charm")
142 else "helm-charts",
143 charm_name,
144 )
145
146 return artifact_path
147
148
tierno744303e2020-01-13 16:46:31 +0000149def populate_dict(target_dict, key_list, value):
150 """
151 Update target_dict creating nested dictionaries with the key_list. Last key_list item is asigned the value.
152 Example target_dict={K: J}; key_list=[a,b,c]; target_dict will be {K: J, a: {b: {c: value}}}
153 :param target_dict: dictionary to be changed
154 :param key_list: list of keys to insert at target_dict
155 :param value:
156 :return: None
157 """
158 for key in key_list[0:-1]:
159 if key not in target_dict:
160 target_dict[key] = {}
161 target_dict = target_dict[key]
162 target_dict[key_list[-1]] = value
163
164
Gabriel Cubae539a8d2022-10-10 11:34:51 -0500165def get_ee_id_parts(ee_id):
166 """
167 Parses ee_id stored at database that can be either 'version:namespace.helm_id' or only
168 namespace.helm_id for backward compatibility
169 If exists helm version can be helm-v3 or helm (helm-v2 old version)
170 """
171 version, _, part_id = ee_id.rpartition(":")
172 namespace, _, helm_id = part_id.rpartition(".")
173 return version, namespace, helm_id
174
175
kuused124bfe2019-06-18 12:09:24 +0200176class LcmBase:
bravof922c4172020-11-24 21:21:43 -0300177 def __init__(self, msg, logger):
kuused124bfe2019-06-18 12:09:24 +0200178 """
179
180 :param db: database connection
181 """
bravof922c4172020-11-24 21:21:43 -0300182 self.db = Database().instance.db
kuused124bfe2019-06-18 12:09:24 +0200183 self.msg = msg
bravof922c4172020-11-24 21:21:43 -0300184 self.fs = Filesystem().instance.fs
kuused124bfe2019-06-18 12:09:24 +0200185 self.logger = logger
186
187 def update_db_2(self, item, _id, _desc):
188 """
189 Updates database with _desc information. If success _desc is cleared
Pedro Escaleirada21d262022-04-21 16:31:06 +0100190 :param item: collection
191 :param _id: the _id to use in the query filter
kuused124bfe2019-06-18 12:09:24 +0200192 :param _desc: dictionary with the content to update. Keys are dot separated keys for
193 :return: None. Exception is raised on error
194 """
195 if not _desc:
196 return
tierno79cd8ad2019-10-18 13:03:10 +0000197 now = time()
198 _desc["_admin.modified"] = now
kuused124bfe2019-06-18 12:09:24 +0200199 self.db.set_one(item, {"_id": _id}, _desc)
200 _desc.clear()
201 # except DbException as e:
202 # self.logger.error("Updating {} _id={} with '{}'. Error: {}".format(item, _id, _desc, e))
203
aticig1dda84c2022-09-10 01:56:58 +0300204 @staticmethod
205 def calculate_charm_hash(zipped_file):
206 """Calculate the hash of charm files which ends with .charm
207
208 Args:
209 zipped_file (str): Existing charm package full path
210
211 Returns:
212 hex digest (str): The hash of the charm file
213 """
214 filehash = hashlib.md5()
215 with open(zipped_file, mode="rb") as file:
216 contents = file.read()
217 filehash.update(contents)
218 return filehash.hexdigest()
219
220 @staticmethod
221 def compare_charm_hash(current_charm, target_charm):
222 """Compare the existing charm and the target charm if the charms
223 are given as zip files ends with .charm
224
225 Args:
226 current_charm (str): Existing charm package full path
227 target_charm (str): Target charm package full path
228
229 Returns:
230 True/False (bool): if charm has changed it returns True
231 """
232 return LcmBase.calculate_charm_hash(
233 current_charm
234 ) != LcmBase.calculate_charm_hash(target_charm)
235
236 @staticmethod
237 def compare_charmdir_hash(current_charm_dir, target_charm_dir):
238 """Compare the existing charm and the target charm if the charms
239 are given as directories
240
241 Args:
242 current_charm_dir (str): Existing charm package directory path
243 target_charm_dir (str): Target charm package directory path
244
245 Returns:
246 True/False (bool): if charm has changed it returns True
247 """
248 return checksumdir.dirhash(current_charm_dir) != checksumdir.dirhash(
249 target_charm_dir
250 )
251
aticigdffa6212022-04-12 15:27:53 +0300252 def check_charm_hash_changed(
253 self, current_charm_path: str, target_charm_path: str
254 ) -> bool:
255 """Find the target charm has changed or not by checking the hash of
256 old and new charm packages
257
258 Args:
259 current_charm_path (str): Existing charm package artifact path
260 target_charm_path (str): Target charm package artifact path
261
262 Returns:
263 True/False (bool): if charm has changed it returns True
264
265 """
aticig1dda84c2022-09-10 01:56:58 +0300266 try:
267 # Check if the charm artifacts are available
268 current_charm = self.fs.path + current_charm_path
269 target_charm = self.fs.path + target_charm_path
aticigdffa6212022-04-12 15:27:53 +0300270
aticig1dda84c2022-09-10 01:56:58 +0300271 if os.path.exists(current_charm) and os.path.exists(target_charm):
aticig1dda84c2022-09-10 01:56:58 +0300272 # Compare the hash of .charm files
273 if current_charm.endswith(".charm"):
274 return LcmBase.compare_charm_hash(current_charm, target_charm)
aticigdffa6212022-04-12 15:27:53 +0300275
aticig1dda84c2022-09-10 01:56:58 +0300276 # Compare the hash of charm folders
277 return LcmBase.compare_charmdir_hash(current_charm, target_charm)
278
279 else:
280 raise LcmException(
281 "Charm artifact {} does not exist in the VNF Package".format(
282 self.fs.path + target_charm_path
283 )
aticigdffa6212022-04-12 15:27:53 +0300284 )
aticig1dda84c2022-09-10 01:56:58 +0300285 except (IOError, OSError, TypeError) as error:
286 self.logger.debug(traceback.format_exc())
287 self.logger.error(f"{error} occured while checking the charm hashes")
288 raise LcmException(error)
aticigdffa6212022-04-12 15:27:53 +0300289
aticig9bc63ac2022-07-27 09:32:06 +0300290 @staticmethod
291 def get_charm_name(charm_metadata_file: str) -> str:
292 """Get the charm name from metadata file.
293
294 Args:
295 charm_metadata_file (str): charm metadata file full path
296
297 Returns:
298 charm_name (str): charm name
299
300 """
301 # Read charm metadata.yaml to get the charm name
302 with open(charm_metadata_file, "r") as metadata_file:
303 content = yaml.safe_load(metadata_file)
304 charm_name = content["name"]
305 return str(charm_name)
306
307 def _get_charm_path(
308 self, nsd_package_path: str, nsd_package_name: str, charm_folder_name: str
309 ) -> str:
310 """Get the full path of charm folder.
311
312 Args:
313 nsd_package_path (str): NSD package full path
314 nsd_package_name (str): NSD package name
315 charm_folder_name (str): folder name
316
317 Returns:
318 charm_path (str): charm folder full path
319 """
320 charm_path = (
321 self.fs.path
322 + nsd_package_path
323 + "/"
324 + nsd_package_name
325 + "/charms/"
326 + charm_folder_name
327 )
328 return charm_path
329
330 def _get_charm_metadata_file(
331 self,
332 charm_folder_name: str,
333 nsd_package_path: str,
334 nsd_package_name: str,
335 charm_path: str = None,
336 ) -> str:
337 """Get the path of charm metadata file.
338
339 Args:
340 charm_folder_name (str): folder name
341 nsd_package_path (str): NSD package full path
342 nsd_package_name (str): NSD package name
343 charm_path (str): Charm full path
344
345 Returns:
346 charm_metadata_file_path (str): charm metadata file full path
347
348 """
349 # Locate the charm metadata.yaml
350 if charm_folder_name.endswith(".charm"):
351 extract_path = (
352 self.fs.path
353 + nsd_package_path
354 + "/"
355 + nsd_package_name
356 + "/charms/"
aticiga37c6ff2022-08-20 20:56:19 +0300357 + charm_folder_name.replace(".charm", "")
aticig9bc63ac2022-07-27 09:32:06 +0300358 )
359 # Extract .charm to extract path
360 with ZipFile(charm_path, "r") as zipfile:
361 zipfile.extractall(extract_path)
362 return extract_path + "/metadata.yaml"
363 else:
364 return charm_path + "/metadata.yaml"
365
366 def find_charm_name(self, db_nsr: dict, charm_folder_name: str) -> str:
367 """Get the charm name from metadata.yaml of charm package.
368
369 Args:
370 db_nsr (dict): NS record as a dictionary
371 charm_folder_name (str): charm folder name
372
373 Returns:
374 charm_name (str): charm name
375 """
376 try:
377 if not charm_folder_name:
378 raise LcmException("charm_folder_name should be provided.")
379
380 # Find nsd_package details: path, name
381 revision = db_nsr.get("revision", "")
aticiga37c6ff2022-08-20 20:56:19 +0300382
383 # Get the NSD package path
384 if revision:
preethika.p28b0bf82022-09-23 07:36:28 +0000385 nsd_package_path = db_nsr["nsd-id"] + ":" + str(revision)
aticiga37c6ff2022-08-20 20:56:19 +0300386 db_nsd = self.db.get_one("nsds_revisions", {"_id": nsd_package_path})
387
388 else:
389 nsd_package_path = db_nsr["nsd-id"]
390
391 db_nsd = self.db.get_one("nsds", {"_id": nsd_package_path})
392
393 # Get the NSD package name
394 nsd_package_name = db_nsd["_admin"]["storage"]["pkg-dir"]
aticig9bc63ac2022-07-27 09:32:06 +0300395
396 # Remove the existing nsd package and sync from FsMongo
397 shutil.rmtree(self.fs.path + nsd_package_path, ignore_errors=True)
398 self.fs.sync(from_path=nsd_package_path)
399
400 # Get the charm path
401 charm_path = self._get_charm_path(
402 nsd_package_path, nsd_package_name, charm_folder_name
403 )
404
405 # Find charm metadata file full path
406 charm_metadata_file = self._get_charm_metadata_file(
407 charm_folder_name, nsd_package_path, nsd_package_name, charm_path
408 )
409
410 # Return charm name
411 return self.get_charm_name(charm_metadata_file)
412
413 except (
414 yaml.YAMLError,
415 IOError,
416 FsException,
417 KeyError,
418 TypeError,
419 FileNotFoundError,
420 BadZipfile,
421 ) as error:
422 self.logger.debug(traceback.format_exc())
423 self.logger.error(f"{error} occured while getting the charm name")
424 raise LcmException(error)
425
kuused124bfe2019-06-18 12:09:24 +0200426
427class TaskRegistry(LcmBase):
tierno59d22d22018-09-25 18:10:19 +0200428 """
429 Implements a registry of task needed for later cancelation, look for related tasks that must be completed before
430 etc. It stores a four level dict
431 First level is the topic, ns, vim_account, sdn
432 Second level is the _id
433 Third level is the operation id
434 Fourth level is a descriptive name, the value is the task class
kuused124bfe2019-06-18 12:09:24 +0200435
436 The HA (High-Availability) methods are used when more than one LCM instance is running.
437 To register the current task in the external DB, use LcmBase as base class, to be able
438 to reuse LcmBase.update_db_2()
439 The DB registry uses the following fields to distinguish a task:
440 - op_type: operation type ("nslcmops" or "nsilcmops")
441 - op_id: operation ID
442 - worker: the worker ID for this process
tierno59d22d22018-09-25 18:10:19 +0200443 """
444
kuuse6a470c62019-07-10 13:52:45 +0200445 # NS/NSI: "services" VIM/WIM/SDN: "accounts"
garciadeblas5697b8b2021-03-24 09:17:02 +0100446 topic_service_list = ["ns", "nsi"]
447 topic_account_list = ["vim", "wim", "sdn", "k8scluster", "vca", "k8srepo"]
kuuse6a470c62019-07-10 13:52:45 +0200448
449 # Map topic to InstanceID
garciadeblas5697b8b2021-03-24 09:17:02 +0100450 topic2instid_dict = {"ns": "nsInstanceId", "nsi": "netsliceInstanceId"}
kuuse6a470c62019-07-10 13:52:45 +0200451
452 # Map topic to DB table name
453 topic2dbtable_dict = {
garciadeblas5697b8b2021-03-24 09:17:02 +0100454 "ns": "nslcmops",
455 "nsi": "nsilcmops",
456 "vim": "vim_accounts",
457 "wim": "wim_accounts",
458 "sdn": "sdns",
459 "k8scluster": "k8sclusters",
460 "vca": "vca",
461 "k8srepo": "k8srepos",
462 }
kuused124bfe2019-06-18 12:09:24 +0200463
bravof922c4172020-11-24 21:21:43 -0300464 def __init__(self, worker_id=None, logger=None):
tierno59d22d22018-09-25 18:10:19 +0200465 self.task_registry = {
466 "ns": {},
Felipe Vicensc2033f22018-11-15 15:09:58 +0100467 "nsi": {},
tierno59d22d22018-09-25 18:10:19 +0200468 "vim_account": {},
tiernoe37b57d2018-12-11 17:22:51 +0000469 "wim_account": {},
tierno59d22d22018-09-25 18:10:19 +0200470 "sdn": {},
calvinosanch9f9c6f22019-11-04 13:37:39 +0100471 "k8scluster": {},
David Garciac1fe90a2021-03-31 19:12:02 +0200472 "vca": {},
calvinosanch9f9c6f22019-11-04 13:37:39 +0100473 "k8srepo": {},
tierno59d22d22018-09-25 18:10:19 +0200474 }
kuused124bfe2019-06-18 12:09:24 +0200475 self.worker_id = worker_id
bravof922c4172020-11-24 21:21:43 -0300476 self.db = Database().instance.db
kuused124bfe2019-06-18 12:09:24 +0200477 self.logger = logger
tierno59d22d22018-09-25 18:10:19 +0200478
479 def register(self, topic, _id, op_id, task_name, task):
480 """
481 Register a new task
Felipe Vicensc2033f22018-11-15 15:09:58 +0100482 :param topic: Can be "ns", "nsi", "vim_account", "sdn"
tierno59d22d22018-09-25 18:10:19 +0200483 :param _id: _id of the related item
484 :param op_id: id of the operation of the related item
485 :param task_name: Task descriptive name, as create, instantiate, terminate. Must be unique in this op_id
486 :param task: Task class
487 :return: none
488 """
489 if _id not in self.task_registry[topic]:
490 self.task_registry[topic][_id] = OrderedDict()
491 if op_id not in self.task_registry[topic][_id]:
492 self.task_registry[topic][_id][op_id] = {task_name: task}
493 else:
494 self.task_registry[topic][_id][op_id][task_name] = task
495 # print("registering task", topic, _id, op_id, task_name, task)
496
497 def remove(self, topic, _id, op_id, task_name=None):
498 """
tiernobaa51102018-12-14 13:16:18 +0000499 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 +0100500 :param topic: Can be "ns", "nsi", "vim_account", "sdn"
tierno59d22d22018-09-25 18:10:19 +0200501 :param _id: _id of the related item
502 :param op_id: id of the operation of the related item
tiernobaa51102018-12-14 13:16:18 +0000503 :param task_name: Task descriptive name. If none it deletes all tasks with same _id and op_id
504 :return: None
tierno59d22d22018-09-25 18:10:19 +0200505 """
tiernobaa51102018-12-14 13:16:18 +0000506 if not self.task_registry[topic].get(_id):
tierno59d22d22018-09-25 18:10:19 +0200507 return
508 if not task_name:
tiernobaa51102018-12-14 13:16:18 +0000509 self.task_registry[topic][_id].pop(op_id, None)
510 elif self.task_registry[topic][_id].get(op_id):
511 self.task_registry[topic][_id][op_id].pop(task_name, None)
512
513 # delete done tasks
514 for op_id_ in list(self.task_registry[topic][_id]):
515 for name, task in self.task_registry[topic][_id][op_id_].items():
516 if not task.done():
517 break
518 else:
519 del self.task_registry[topic][_id][op_id_]
tierno59d22d22018-09-25 18:10:19 +0200520 if not self.task_registry[topic][_id]:
521 del self.task_registry[topic][_id]
522
523 def lookfor_related(self, topic, _id, my_op_id=None):
524 task_list = []
525 task_name_list = []
526 if _id not in self.task_registry[topic]:
527 return "", task_name_list
528 for op_id in reversed(self.task_registry[topic][_id]):
529 if my_op_id:
530 if my_op_id == op_id:
531 my_op_id = None # so that the next task is taken
532 continue
533
534 for task_name, task in self.task_registry[topic][_id][op_id].items():
tiernobaa51102018-12-14 13:16:18 +0000535 if not task.done():
536 task_list.append(task)
537 task_name_list.append(task_name)
tierno59d22d22018-09-25 18:10:19 +0200538 break
539 return ", ".join(task_name_list), task_list
540
541 def cancel(self, topic, _id, target_op_id=None, target_task_name=None):
542 """
kuused124bfe2019-06-18 12:09:24 +0200543 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 +0100544 this is cancelled, and the same with task_name
tierno59d22d22018-09-25 18:10:19 +0200545 """
546 if not self.task_registry[topic].get(_id):
547 return
548 for op_id in reversed(self.task_registry[topic][_id]):
549 if target_op_id and target_op_id != op_id:
550 continue
551 for task_name, task in self.task_registry[topic][_id][op_id].items():
552 if target_task_name and target_task_name != task_name:
553 continue
554 # result =
555 task.cancel()
556 # if result:
557 # self.logger.debug("{} _id={} order_id={} task={} cancelled".format(topic, _id, op_id, task_name))
558
kuuse6a470c62019-07-10 13:52:45 +0200559 # Is topic NS/NSI?
560 def _is_service_type_HA(self, topic):
561 return topic in self.topic_service_list
562
563 # Is topic VIM/WIM/SDN?
564 def _is_account_type_HA(self, topic):
565 return topic in self.topic_account_list
566
567 # Input: op_id, example: 'abc123def:3' Output: account_id='abc123def', op_index=3
568 def _get_account_and_op_HA(self, op_id):
569 if not op_id:
tiernofa076c32020-08-13 14:25:47 +0000570 return None, None
garciadeblas5697b8b2021-03-24 09:17:02 +0100571 account_id, _, op_index = op_id.rpartition(":")
tiernofa076c32020-08-13 14:25:47 +0000572 if not account_id or not op_index.isdigit():
573 return None, None
kuuse6a470c62019-07-10 13:52:45 +0200574 return account_id, op_index
575
576 # Get '_id' for any topic and operation
577 def _get_instance_id_HA(self, topic, op_type, op_id):
578 _id = None
579 # Special operation 'ANY', for SDN account associated to a VIM account: op_id as '_id'
garciadeblas5697b8b2021-03-24 09:17:02 +0100580 if op_type == "ANY":
kuuse6a470c62019-07-10 13:52:45 +0200581 _id = op_id
582 # NS/NSI: Use op_id as '_id'
583 elif self._is_service_type_HA(topic):
584 _id = op_id
calvinosanch9f9c6f22019-11-04 13:37:39 +0100585 # 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 +0200586 elif self._is_account_type_HA(topic):
587 _id, _ = self._get_account_and_op_HA(op_id)
588 return _id
589
590 # Set DB _filter for querying any related process state
591 def _get_waitfor_filter_HA(self, db_lcmop, topic, op_type, op_id):
592 _filter = {}
593 # Special operation 'ANY', for SDN account associated to a VIM account: op_id as '_id'
594 # In this special case, the timestamp is ignored
garciadeblas5697b8b2021-03-24 09:17:02 +0100595 if op_type == "ANY":
596 _filter = {"operationState": "PROCESSING"}
kuuse6a470c62019-07-10 13:52:45 +0200597 # Otherwise, get 'startTime' timestamp for this operation
598 else:
599 # NS/NSI
600 if self._is_service_type_HA(topic):
tierno79cd8ad2019-10-18 13:03:10 +0000601 now = time()
kuuse6a470c62019-07-10 13:52:45 +0200602 starttime_this_op = db_lcmop.get("startTime")
603 instance_id_label = self.topic2instid_dict.get(topic)
604 instance_id = db_lcmop.get(instance_id_label)
garciadeblas5697b8b2021-03-24 09:17:02 +0100605 _filter = {
606 instance_id_label: instance_id,
607 "operationState": "PROCESSING",
608 "startTime.lt": starttime_this_op,
609 "_admin.modified.gt": now
610 - 2 * 3600, # ignore if tow hours of inactivity
611 }
calvinosanch9f9c6f22019-11-04 13:37:39 +0100612 # VIM/WIM/SDN/K8scluster
kuuse6a470c62019-07-10 13:52:45 +0200613 elif self._is_account_type_HA(topic):
614 _, op_index = self._get_account_and_op_HA(op_id)
garciadeblas5697b8b2021-03-24 09:17:02 +0100615 _ops = db_lcmop["_admin"]["operations"]
kuuse6a470c62019-07-10 13:52:45 +0200616 _this_op = _ops[int(op_index)]
garciadeblas5697b8b2021-03-24 09:17:02 +0100617 starttime_this_op = _this_op.get("startTime", None)
618 _filter = {
619 "operationState": "PROCESSING",
620 "startTime.lt": starttime_this_op,
621 }
kuuse6a470c62019-07-10 13:52:45 +0200622 return _filter
623
624 # Get DB params for any topic and operation
625 def _get_dbparams_for_lock_HA(self, topic, op_type, op_id):
626 q_filter = {}
627 update_dict = {}
628 # NS/NSI
629 if self._is_service_type_HA(topic):
garciadeblas5697b8b2021-03-24 09:17:02 +0100630 q_filter = {"_id": op_id, "_admin.worker": None}
631 update_dict = {"_admin.worker": self.worker_id}
kuuse6a470c62019-07-10 13:52:45 +0200632 # VIM/WIM/SDN
633 elif self._is_account_type_HA(topic):
634 account_id, op_index = self._get_account_and_op_HA(op_id)
635 if not account_id:
636 return None, None
garciadeblas5697b8b2021-03-24 09:17:02 +0100637 if op_type == "create":
kuuse6a470c62019-07-10 13:52:45 +0200638 # Creating a VIM/WIM/SDN account implies setting '_admin.current_operation' = 0
639 op_index = 0
garciadeblas5697b8b2021-03-24 09:17:02 +0100640 q_filter = {
641 "_id": account_id,
642 "_admin.operations.{}.worker".format(op_index): None,
643 }
644 update_dict = {
645 "_admin.operations.{}.worker".format(op_index): self.worker_id,
646 "_admin.current_operation": op_index,
647 }
kuuse6a470c62019-07-10 13:52:45 +0200648 return q_filter, update_dict
649
kuused124bfe2019-06-18 12:09:24 +0200650 def lock_HA(self, topic, op_type, op_id):
651 """
kuuse6a470c62019-07-10 13:52:45 +0200652 Lock a task, if possible, to indicate to the HA system that
kuused124bfe2019-06-18 12:09:24 +0200653 the task will be executed in this LCM instance.
kuuse6a470c62019-07-10 13:52:45 +0200654 :param topic: Can be "ns", "nsi", "vim", "wim", or "sdn"
655 :param op_type: Operation type, can be "nslcmops", "nsilcmops", "create", "edit", "delete"
656 :param op_id: NS, NSI: Operation ID VIM,WIM,SDN: Account ID + ':' + Operation Index
kuused124bfe2019-06-18 12:09:24 +0200657 :return:
kuuse6a470c62019-07-10 13:52:45 +0200658 True=lock was successful => execute the task (not registered by any other LCM instance)
kuused124bfe2019-06-18 12:09:24 +0200659 False=lock failed => do NOT execute the task (already registered by another LCM instance)
kuuse6a470c62019-07-10 13:52:45 +0200660
661 HA tasks and backward compatibility:
662 If topic is "account type" (VIM/WIM/SDN) and op_id is None, 'op_id' was not provided by NBI.
663 This means that the running NBI instance does not support HA.
664 In such a case this method should always return True, to always execute
665 the task in this instance of LCM, without querying the DB.
tierno59d22d22018-09-25 18:10:19 +0200666 """
667
calvinosanch9f9c6f22019-11-04 13:37:39 +0100668 # Backward compatibility for VIM/WIM/SDN/k8scluster without op_id
kuuse6a470c62019-07-10 13:52:45 +0200669 if self._is_account_type_HA(topic) and op_id is None:
670 return True
tierno59d22d22018-09-25 18:10:19 +0200671
kuuse6a470c62019-07-10 13:52:45 +0200672 # Try to lock this task
tiernofa076c32020-08-13 14:25:47 +0000673 db_table_name = self.topic2dbtable_dict[topic]
kuuse6a470c62019-07-10 13:52:45 +0200674 q_filter, update_dict = self._get_dbparams_for_lock_HA(topic, op_type, op_id)
garciadeblas5697b8b2021-03-24 09:17:02 +0100675 db_lock_task = self.db.set_one(
676 db_table_name,
677 q_filter=q_filter,
678 update_dict=update_dict,
679 fail_on_empty=False,
680 )
kuused124bfe2019-06-18 12:09:24 +0200681 if db_lock_task is None:
garciadeblas5697b8b2021-03-24 09:17:02 +0100682 self.logger.debug(
683 "Task {} operation={} already locked by another worker".format(
684 topic, op_id
685 )
686 )
kuused124bfe2019-06-18 12:09:24 +0200687 return False
688 else:
kuuse6a470c62019-07-10 13:52:45 +0200689 # Set 'detailed-status' to 'In progress' for VIM/WIM/SDN operations
690 if self._is_account_type_HA(topic):
garciadeblas5697b8b2021-03-24 09:17:02 +0100691 detailed_status = "In progress"
kuuse6a470c62019-07-10 13:52:45 +0200692 account_id, op_index = self._get_account_and_op_HA(op_id)
garciadeblas5697b8b2021-03-24 09:17:02 +0100693 q_filter = {"_id": account_id}
694 update_dict = {
695 "_admin.operations.{}.detailed-status".format(
696 op_index
697 ): detailed_status
698 }
699 self.db.set_one(
700 db_table_name,
701 q_filter=q_filter,
702 update_dict=update_dict,
703 fail_on_empty=False,
704 )
kuused124bfe2019-06-18 12:09:24 +0200705 return True
706
tiernofa076c32020-08-13 14:25:47 +0000707 def unlock_HA(self, topic, op_type, op_id, operationState, detailed_status):
kuuse6a470c62019-07-10 13:52:45 +0200708 """
709 Register a task, done when finished a VIM/WIM/SDN 'create' operation.
710 :param topic: Can be "vim", "wim", or "sdn"
711 :param op_type: Operation type, can be "create", "edit", "delete"
712 :param op_id: Account ID + ':' + Operation Index
713 :return: nothing
714 """
715
716 # Backward compatibility
tiernofa076c32020-08-13 14:25:47 +0000717 if not self._is_account_type_HA(topic) or not op_id:
kuuse6a470c62019-07-10 13:52:45 +0200718 return
719
720 # Get Account ID and Operation Index
721 account_id, op_index = self._get_account_and_op_HA(op_id)
tiernofa076c32020-08-13 14:25:47 +0000722 db_table_name = self.topic2dbtable_dict[topic]
kuuse6a470c62019-07-10 13:52:45 +0200723
724 # If this is a 'delete' operation, the account may have been deleted (SUCCESS) or may still exist (FAILED)
725 # If the account exist, register the HA task.
726 # Update DB for HA tasks
garciadeblas5697b8b2021-03-24 09:17:02 +0100727 q_filter = {"_id": account_id}
728 update_dict = {
729 "_admin.operations.{}.operationState".format(op_index): operationState,
730 "_admin.operations.{}.detailed-status".format(op_index): detailed_status,
731 "_admin.operations.{}.worker".format(op_index): None,
732 "_admin.current_operation": None,
733 }
734 self.db.set_one(
735 db_table_name,
736 q_filter=q_filter,
737 update_dict=update_dict,
738 fail_on_empty=False,
739 )
kuuse6a470c62019-07-10 13:52:45 +0200740 return
741
kuused124bfe2019-06-18 12:09:24 +0200742 async def waitfor_related_HA(self, topic, op_type, op_id=None):
tierno59d22d22018-09-25 18:10:19 +0200743 """
kuused124bfe2019-06-18 12:09:24 +0200744 Wait for any pending related HA tasks
tierno59d22d22018-09-25 18:10:19 +0200745 """
kuused124bfe2019-06-18 12:09:24 +0200746
kuuse6a470c62019-07-10 13:52:45 +0200747 # Backward compatibility
garciadeblas5697b8b2021-03-24 09:17:02 +0100748 if not (
749 self._is_service_type_HA(topic) or self._is_account_type_HA(topic)
750 ) and (op_id is None):
kuuse6a470c62019-07-10 13:52:45 +0200751 return
kuused124bfe2019-06-18 12:09:24 +0200752
kuuse6a470c62019-07-10 13:52:45 +0200753 # Get DB table name
754 db_table_name = self.topic2dbtable_dict.get(topic)
755
756 # Get instance ID
757 _id = self._get_instance_id_HA(topic, op_type, op_id)
758 _filter = {"_id": _id}
garciadeblas5697b8b2021-03-24 09:17:02 +0100759 db_lcmop = self.db.get_one(db_table_name, _filter, fail_on_empty=False)
kuused124bfe2019-06-18 12:09:24 +0200760 if not db_lcmop:
tierno59d22d22018-09-25 18:10:19 +0200761 return
kuuse6a470c62019-07-10 13:52:45 +0200762
763 # Set DB _filter for querying any related process state
764 _filter = self._get_waitfor_filter_HA(db_lcmop, topic, op_type, op_id)
kuused124bfe2019-06-18 12:09:24 +0200765
766 # For HA, get list of tasks from DB instead of from dictionary (in-memory) variable.
garciadeblas5697b8b2021-03-24 09:17:02 +0100767 timeout_wait_for_task = (
768 3600 # Max time (seconds) to wait for a related task to finish
769 )
kuused124bfe2019-06-18 12:09:24 +0200770 # interval_wait_for_task = 30 # A too long polling interval slows things down considerably
garciadeblas5697b8b2021-03-24 09:17:02 +0100771 interval_wait_for_task = 10 # Interval in seconds for polling related tasks
kuused124bfe2019-06-18 12:09:24 +0200772 time_left = timeout_wait_for_task
773 old_num_related_tasks = 0
774 while True:
kuuse6a470c62019-07-10 13:52:45 +0200775 # Get related tasks (operations within the same instance as this) which are
kuused124bfe2019-06-18 12:09:24 +0200776 # still running (operationState='PROCESSING') and which were started before this task.
kuuse6a470c62019-07-10 13:52:45 +0200777 # In the case of op_type='ANY', get any related tasks with operationState='PROCESSING', ignore timestamps.
garciadeblas5697b8b2021-03-24 09:17:02 +0100778 db_waitfor_related_task = self.db.get_list(db_table_name, q_filter=_filter)
kuused124bfe2019-06-18 12:09:24 +0200779 new_num_related_tasks = len(db_waitfor_related_task)
kuuse6a470c62019-07-10 13:52:45 +0200780 # If there are no related tasks, there is nothing to wait for, so return.
kuused124bfe2019-06-18 12:09:24 +0200781 if not new_num_related_tasks:
kuused124bfe2019-06-18 12:09:24 +0200782 return
783 # If number of pending related tasks have changed,
784 # update the 'detailed-status' field and log the change.
kuuse6a470c62019-07-10 13:52:45 +0200785 # Do NOT update the 'detailed-status' for SDNC-associated-to-VIM operations ('ANY').
garciadeblas5697b8b2021-03-24 09:17:02 +0100786 if (op_type != "ANY") and (new_num_related_tasks != old_num_related_tasks):
787 step = "Waiting for {} related tasks to be completed.".format(
788 new_num_related_tasks
789 )
kuuse6a470c62019-07-10 13:52:45 +0200790 update_dict = {}
garciadeblas5697b8b2021-03-24 09:17:02 +0100791 q_filter = {"_id": _id}
kuuse6a470c62019-07-10 13:52:45 +0200792 # NS/NSI
793 if self._is_service_type_HA(topic):
garciadeblas5697b8b2021-03-24 09:17:02 +0100794 update_dict = {
795 "detailed-status": step,
796 "queuePosition": new_num_related_tasks,
797 }
kuuse6a470c62019-07-10 13:52:45 +0200798 # VIM/WIM/SDN
799 elif self._is_account_type_HA(topic):
800 _, op_index = self._get_account_and_op_HA(op_id)
garciadeblas5697b8b2021-03-24 09:17:02 +0100801 update_dict = {
802 "_admin.operations.{}.detailed-status".format(op_index): step
803 }
kuuse6a470c62019-07-10 13:52:45 +0200804 self.logger.debug("Task {} operation={} {}".format(topic, _id, step))
garciadeblas5697b8b2021-03-24 09:17:02 +0100805 self.db.set_one(
806 db_table_name,
807 q_filter=q_filter,
808 update_dict=update_dict,
809 fail_on_empty=False,
810 )
kuused124bfe2019-06-18 12:09:24 +0200811 old_num_related_tasks = new_num_related_tasks
812 time_left -= interval_wait_for_task
813 if time_left < 0:
814 raise LcmException(
815 "Timeout ({}) when waiting for related tasks to be completed".format(
garciadeblas5697b8b2021-03-24 09:17:02 +0100816 timeout_wait_for_task
817 )
818 )
kuused124bfe2019-06-18 12:09:24 +0200819 await asyncio.sleep(interval_wait_for_task)
820
821 return