osm_lcm/lcm_utils.py

   1 # -*- coding: utf-8 -*-
   2
   3 ##
   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 ##
  18
  19 import asyncio
  20 import checksumdir
  21 from collections import OrderedDict
  22 import hashlib
  23 import os
  24 import shutil
  25 import traceback
  26 from time import time
  27
  28 from osm_common.fsbase import FsException
  29 from osm_lcm.data_utils.database.database import Database
  30 from osm_lcm.data_utils.filesystem.filesystem import Filesystem
  31 import yaml
  32 from zipfile import ZipFile, BadZipfile
  33
  34 # from osm_common.dbbase import DbException
  35
  36 __author__ = "Alfonso Tierno"
  37
  38
  39 class LcmException(Exception):
  40     pass
  41
  42
  43 class LcmExceptionExit(LcmException):
  44     pass
  45
  46
  47 def versiontuple(v):
  48     """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     """
  52     filled = []
  53     for point in v.split("."):
  54         point, _, _ = point.partition("+")
  55         point, _, _ = point.partition("-")
  56         filled.append(point.zfill(20))
  57     return tuple(filled)
  58
  59
  60 def deep_get(target_dict, key_list, default_value=None):
  61     """
  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
  66     :param default_value: value to return if key is not present in the nested dictionary
  67     :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:
  71             return default_value
  72         target_dict = target_dict[key]
  73     return target_dict
  74
  75
  76 def 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
  88 def 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
 105 def 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(
 124             base_folder["folder"].split(":")[0] + extension,
 125             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(
 135             base_folder["folder"].split(":")[0] + extension,
 136             "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
 145 def 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
 161 def 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
 172 def 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,
 195             "ipv6_address_mode": source_data["ipv6-address-mode"]
 196             if "ipv6-address-mode" in source_data
 197             else None,
 198         }
 199
 200
 201 class LcmBase:
 202     def __init__(self, msg, logger):
 203         """
 204
 205         :param db: database connection
 206         """
 207         self.db = Database().instance.db
 208         self.msg = msg
 209         self.fs = Filesystem().instance.fs
 210         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
 215         :param item: collection
 216         :param _id: the _id to use in the query filter
 217         :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
 222         now = time()
 223         _desc["_admin.modified"] = now
 224         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
 229     @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         """
 239         filehash = hashlib.sha256()
 240         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
 277     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         """
 291         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
 295
 296             if os.path.exists(current_charm) and os.path.exists(target_charm):
 297                 # Compare the hash of .charm files
 298                 if current_charm.endswith(".charm"):
 299                     return LcmBase.compare_charm_hash(current_charm, target_charm)
 300
 301                 # 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                     )
 309                 )
 310         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)
 314
 315     @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/"
 382                 + charm_folder_name.replace(".charm", "")
 383             )
 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", "")
 407
 408             # Get the NSD package path
 409             if revision:
 410                 nsd_package_path = db_nsr["nsd-id"] + ":" + str(revision)
 411                 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"]
 420
 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
 451
 452 class TaskRegistry(LcmBase):
 453     """
 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
 460
 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
 468     """
 469
 470     # NS/NSI: "services" VIM/WIM/SDN: "accounts"
 471     topic_service_list = ["ns", "nsi"]
 472     topic_account_list = ["vim", "wim", "sdn", "k8scluster", "vca", "k8srepo"]
 473
 474     # Map topic to InstanceID
 475     topic2instid_dict = {"ns": "nsInstanceId", "nsi": "netsliceInstanceId"}
 476
 477     # Map topic to DB table name
 478     topic2dbtable_dict = {
 479         "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     }
 488
 489     def __init__(self, worker_id=None, logger=None):
 490         self.task_registry = {
 491             "ns": {},
 492             "nsi": {},
 493             "vim_account": {},
 494             "wim_account": {},
 495             "sdn": {},
 496             "k8scluster": {},
 497             "vca": {},
 498             "k8srepo": {},
 499         }
 500         self.worker_id = worker_id
 501         self.db = Database().instance.db
 502         self.logger = logger
 503
 504     def register(self, topic, _id, op_id, task_name, task):
 505         """
 506         Register a new task
 507         :param topic: Can be "ns", "nsi", "vim_account", "sdn"
 508         :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         """
 524         When task is ended, it should be removed. It ignores missing tasks. It also removes tasks done with this _id
 525         :param topic: Can be "ns", "nsi", "vim_account", "sdn"
 526         :param _id: _id of the related item
 527         :param op_id: id of the operation of the related item
 528         :param task_name: Task descriptive name. If none it deletes all tasks with same _id and op_id
 529         :return: None
 530         """
 531         if not self.task_registry[topic].get(_id):
 532             return
 533         if not task_name:
 534             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_]
 545         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():
 560                 if not task.done():
 561                     task_list.append(task)
 562                     task_name_list.append(task_name)
 563             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         """
 568         Cancel all active tasks of a concrete ns, nsi, vim_account, sdn identified for _id. If op_id is supplied only
 569         this is cancelled, and the same with task_name
 570         """
 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
 584     # 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:
 595             return None, None
 596         account_id, _, op_index = op_id.rpartition(":")
 597         if not account_id or not op_index.isdigit():
 598             return None, None
 599         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'
 605         if op_type == "ANY":
 606             _id = op_id
 607         # NS/NSI: Use op_id as '_id'
 608         elif self._is_service_type_HA(topic):
 609             _id = op_id
 610         # VIM/SDN/WIM/K8SCLUSTER: Split op_id to get Account ID and Operation Index, use Account ID as '_id'
 611         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
 620         if op_type == "ANY":
 621             _filter = {"operationState": "PROCESSING"}
 622         # Otherwise, get 'startTime' timestamp for this operation
 623         else:
 624             # NS/NSI
 625             if self._is_service_type_HA(topic):
 626                 now = time()
 627                 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)
 630                 _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                 }
 637             # VIM/WIM/SDN/K8scluster
 638             elif self._is_account_type_HA(topic):
 639                 _, op_index = self._get_account_and_op_HA(op_id)
 640                 _ops = db_lcmop["_admin"]["operations"]
 641                 _this_op = _ops[int(op_index)]
 642                 starttime_this_op = _this_op.get("startTime", None)
 643                 _filter = {
 644                     "operationState": "PROCESSING",
 645                     "startTime.lt": starttime_this_op,
 646                 }
 647         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):
 655             q_filter = {"_id": op_id, "_admin.worker": None}
 656             update_dict = {"_admin.worker": self.worker_id}
 657         # 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
 662             if op_type == "create":
 663                 # Creating a VIM/WIM/SDN account implies setting '_admin.current_operation' = 0
 664                 op_index = 0
 665             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             }
 673         return q_filter, update_dict
 674
 675     def lock_HA(self, topic, op_type, op_id):
 676         """
 677         Lock a task, if possible, to indicate to the HA system that
 678         the task will be executed in this LCM instance.
 679         :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
 682         :return:
 683         True=lock was successful => execute the task (not registered by any other LCM instance)
 684         False=lock failed => do NOT execute the task (already registered by another LCM instance)
 685
 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.
 691         """
 692
 693         # Backward compatibility for VIM/WIM/SDN/k8scluster without op_id
 694         if self._is_account_type_HA(topic) and op_id is None:
 695             return True
 696
 697         # Try to lock this task
 698         db_table_name = self.topic2dbtable_dict[topic]
 699         q_filter, update_dict = self._get_dbparams_for_lock_HA(topic, op_type, op_id)
 700         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         )
 706         if db_lock_task is None:
 707             self.logger.debug(
 708                 "Task {} operation={} already locked by another worker".format(
 709                     topic, op_id
 710                 )
 711             )
 712             return False
 713         else:
 714             # Set 'detailed-status' to 'In progress' for VIM/WIM/SDN operations
 715             if self._is_account_type_HA(topic):
 716                 detailed_status = "In progress"
 717                 account_id, op_index = self._get_account_and_op_HA(op_id)
 718                 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                 )
 730             return True
 731
 732     def unlock_HA(self, topic, op_type, op_id, operationState, detailed_status):
 733         """
 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
 742         if not self._is_account_type_HA(topic) or not op_id:
 743             return
 744
 745         # Get Account ID and Operation Index
 746         account_id, op_index = self._get_account_and_op_HA(op_id)
 747         db_table_name = self.topic2dbtable_dict[topic]
 748
 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
 752         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         )
 765         return
 766
 767     async def waitfor_related_HA(self, topic, op_type, op_id=None):
 768         """
 769         Wait for any pending related HA tasks
 770         """
 771
 772         # Backward compatibility
 773         if not (
 774             self._is_service_type_HA(topic) or self._is_account_type_HA(topic)
 775         ) and (op_id is None):
 776             return
 777
 778         # 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}
 784         db_lcmop = self.db.get_one(db_table_name, _filter, fail_on_empty=False)
 785         if not db_lcmop:
 786             return
 787
 788         # Set DB _filter for querying any related process state
 789         _filter = self._get_waitfor_filter_HA(db_lcmop, topic, op_type, op_id)
 790
 791         # For HA, get list of tasks from DB instead of from dictionary (in-memory) variable.
 792         timeout_wait_for_task = (
 793             3600  # Max time (seconds) to wait for a related task to finish
 794         )
 795         # interval_wait_for_task = 30    #  A too long polling interval slows things down considerably
 796         interval_wait_for_task = 10  # Interval in seconds for polling related tasks
 797         time_left = timeout_wait_for_task
 798         old_num_related_tasks = 0
 799         while True:
 800             # Get related tasks (operations within the same instance as this) which are
 801             # still running (operationState='PROCESSING') and which were started before this task.
 802             # In the case of op_type='ANY', get any related tasks with operationState='PROCESSING', ignore timestamps.
 803             db_waitfor_related_task = self.db.get_list(db_table_name, q_filter=_filter)
 804             new_num_related_tasks = len(db_waitfor_related_task)
 805             # If there are no related tasks, there is nothing to wait for, so return.
 806             if not new_num_related_tasks:
 807                 return
 808             # If number of pending related tasks have changed,
 809             # update the 'detailed-status' field and log the change.
 810             # Do NOT update the 'detailed-status' for SDNC-associated-to-VIM operations ('ANY').
 811             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                 )
 815                 update_dict = {}
 816                 q_filter = {"_id": _id}
 817                 # NS/NSI
 818                 if self._is_service_type_HA(topic):
 819                     update_dict = {
 820                         "detailed-status": step,
 821                         "queuePosition": new_num_related_tasks,
 822                     }
 823                 # VIM/WIM/SDN
 824                 elif self._is_account_type_HA(topic):
 825                     _, op_index = self._get_account_and_op_HA(op_id)
 826                     update_dict = {
 827                         "_admin.operations.{}.detailed-status".format(op_index): step
 828                     }
 829                 self.logger.debug("Task {} operation={} {}".format(topic, _id, step))
 830                 self.db.set_one(
 831                     db_table_name,
 832                     q_filter=q_filter,
 833                     update_dict=update_dict,
 834                     fail_on_empty=False,
 835                 )
 836                 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(
 841                         timeout_wait_for_task
 842                     )
 843                 )
 844             await asyncio.sleep(interval_wait_for_task)
 845
 846         return