Fix Bug 2041: Backup VDU interfaces for healing features
[osm/RO.git] / NG-RO / osm_ng_ro / ns.py
index 68d2a72..bc8e628 100644 (file)
@@ -19,8 +19,6 @@
 from http import HTTPStatus
 import logging
 from random import choice as random_choice
-
-import yaml
 from threading import Lock
 from time import time
 from traceback import format_exc as traceback_format_exc
@@ -51,6 +49,7 @@ from osm_common.fsbase import FsBase, FsException
 from osm_common.msgbase import MsgException
 from osm_ng_ro.ns_thread import deep_get, NsWorker, NsWorkerException
 from osm_ng_ro.validation import deploy_schema, validate_input
+import yaml
 
 __author__ = "Alfonso Tierno <alfonso.tiernosepulveda@telefonica.com>"
 min_common_version = "0.1.16"
@@ -509,6 +508,7 @@ class Ns(object):
                 "vim_name": None,
                 "vim_status": None,
                 "vim_details": None,
+                "vim_message": None,
                 "refresh_at": None,
             },
             "modified_at": now,
@@ -1195,7 +1195,7 @@ class Ns(object):
         """
         vnfr = kwargs.get("vnfr")
         vdu2cloud_init = kwargs.get("vdu2cloud_init")
-        #logger = kwargs.get("logger")
+        # logger = kwargs.get("logger")
         db = kwargs.get("db")
         fs = kwargs.get("fs")
         ro_nsr_public_key = kwargs.get("ro_nsr_public_key")
@@ -1232,7 +1232,11 @@ class Ns(object):
             }
             existing_ifaces = existing_vdu["vim_info"][target_id].get("interfaces", [])
             net_id = next(
-                (i["vim_net_id"] for i in existing_ifaces if i["ip_address"] == interface["ip-address"]),
+                (
+                    i["vim_net_id"]
+                    for i in existing_ifaces
+                    if i["ip_address"] == interface["ip-address"]
+                ),
                 None,
             )
 
@@ -1311,8 +1315,13 @@ class Ns(object):
             affinity_group = {}
             for affinity_group_id in existing_vdu["affinity-or-anti-affinity-group-id"]:
                 for group in db_nsr.get("affinity-or-anti-affinity-group"):
-                    if group["id"] == affinity_group_id and group["vim_info"][target_id].get("vim_id", None) is not None:
-                        affinity_group["affinity_group_id"] = group["vim_info"][target_id].get("vim_id", None)
+                    if (
+                        group["id"] == affinity_group_id
+                        and group["vim_info"][target_id].get("vim_id", None) is not None
+                    ):
+                        affinity_group["affinity_group_id"] = group["vim_info"][
+                            target_id
+                        ].get("vim_id", None)
                         affinity_group_list.append(affinity_group)
 
         extra_dict["params"] = {
@@ -1658,9 +1667,7 @@ class Ns(object):
                 extra_dict=change.get("extra_dict", None),
             )
 
-            self.logger.warning(
-                "ns.define_all_tasks task={}".format(task)
-            )
+            self.logger.warning("ns.define_all_tasks task={}".format(task))
             tasks_by_target_record_id[change["target_record_id"]] = task
             db_new_tasks.append(task)
 
@@ -1746,11 +1753,13 @@ class Ns(object):
             self.logger.debug("Updating database, Creating ro_tasks")
             db_ro_task = Ns._create_ro_task(target_id, db_task)
 
-            # If DELETE task: the associated created items shoud be removed 
+            # If DELETE task: the associated created items should be removed
             # (except persistent volumes):
             if action == "DELETE":
                 db_ro_task["vim_info"]["created"] = True
-                db_ro_task["vim_info"]["created_items"] = db_task.get("created_items", {})
+                db_ro_task["vim_info"]["created_items"] = db_task.get(
+                    "created_items", {}
+                )
                 db_ro_task["vim_info"]["vim_id"] = db_task.get("vim_id", None)
 
             nb_ro_tasks += 1
@@ -1837,7 +1846,10 @@ class Ns(object):
                 item_index = 0
                 existing_instance = None
                 for instance in existing_vnf.get("vdur", None):
-                    if (instance["vdu-name"] == vdu_name and instance["count-index"] == count_index):
+                    if (
+                        instance["vdu-name"] == vdu_name
+                        and instance["count-index"] == count_index
+                    ):
                         existing_instance = instance
                         break
                     else:
@@ -1992,7 +2004,7 @@ class Ns(object):
                 )
 
                 # Delete all ro_tasks registered for the targets vdurs (target_record)
-                # If task of type CREATE exist then vim will try to get info form deleted VMs. 
+                # If task of type CREATE exist then vim will try to get info form deleted VMs.
                 # So remove all task related to target record.
                 ro_tasks = self.db.get_list("ro_tasks", {"tasks.nsr_id": nsr_id})
                 for change in changes_list:
@@ -2007,7 +2019,7 @@ class Ns(object):
                                     },
                                     fail_on_empty=False,
                                 )
-                
+
                 step = "Updating database, Appending tasks to ro_tasks"
                 self.upload_recreate_tasks(
                     db_new_tasks=db_new_tasks,
@@ -2269,7 +2281,7 @@ class Ns(object):
                         error_text = "Error at {} {}: {}".format(
                             task["action"].lower(),
                             task["item"],
-                            ro_task["vim_info"].get("vim_details") or "unknown",
+                            ro_task["vim_info"].get("vim_message") or "unknown",
                         )
                         details.append(error_text)
                     elif task["status"] in ("SCHEDULED", "BUILD"):
@@ -2323,3 +2335,118 @@ class Ns(object):
                     return_data.append(task["action_id"])
 
         return return_data, None, True
+
+    def migrate_task(
+        self, vdu, vnf, vdu_index, action_id, nsr_id, task_index, extra_dict
+    ):
+        target_vim, vim_info = next(k_v for k_v in vdu["vim_info"].items())
+        self._assign_vim(target_vim)
+        target_record = "vnfrs:{}:vdur.{}".format(vnf["_id"], vdu_index)
+        target_record_id = "vnfrs:{}:vdur.{}".format(vnf["_id"], vdu["id"])
+        deployment_info = {
+            "action_id": action_id,
+            "nsr_id": nsr_id,
+            "task_index": task_index,
+        }
+
+        task = Ns._create_task(
+            deployment_info=deployment_info,
+            target_id=target_vim,
+            item="migrate",
+            action="EXEC",
+            target_record=target_record,
+            target_record_id=target_record_id,
+            extra_dict=extra_dict,
+        )
+
+        return task
+
+    def migrate(self, session, indata, version, nsr_id, *args, **kwargs):
+        task_index = 0
+        extra_dict = {}
+        now = time()
+        action_id = indata.get("action_id", str(uuid4()))
+        step = ""
+        logging_text = "Task deploy nsr_id={} action_id={} ".format(nsr_id, action_id)
+        self.logger.debug(logging_text + "Enter")
+        try:
+            vnf_instance_id = indata["vnfInstanceId"]
+            step = "Getting vnfrs from db"
+            db_vnfr = self.db.get_one("vnfrs", {"_id": vnf_instance_id})
+            vdu = indata.get("vdu")
+            migrateToHost = indata.get("migrateToHost")
+            db_new_tasks = []
+
+            with self.write_lock:
+                if vdu is not None:
+                    vdu_id = indata["vdu"]["vduId"]
+                    vdu_count_index = indata["vdu"].get("vduCountIndex", 0)
+                    for vdu_index, vdu in enumerate(db_vnfr["vdur"]):
+                        if (
+                            vdu["vdu-id-ref"] == vdu_id
+                            and vdu["count-index"] == vdu_count_index
+                        ):
+                            extra_dict["params"] = {
+                                "vim_vm_id": vdu["vim-id"],
+                                "migrate_host": migrateToHost,
+                                "vdu_vim_info": vdu["vim_info"],
+                            }
+                            step = "Creating migration task for vdu:{}".format(vdu)
+                            task = self.migrate_task(
+                                vdu,
+                                db_vnfr,
+                                vdu_index,
+                                action_id,
+                                nsr_id,
+                                task_index,
+                                extra_dict,
+                            )
+                            db_new_tasks.append(task)
+                            task_index += 1
+                            break
+                else:
+
+                    for vdu_index, vdu in enumerate(db_vnfr["vdur"]):
+                        extra_dict["params"] = {
+                            "vim_vm_id": vdu["vim-id"],
+                            "migrate_host": migrateToHost,
+                            "vdu_vim_info": vdu["vim_info"],
+                        }
+                        step = "Creating migration task for vdu:{}".format(vdu)
+                        task = self.migrate_task(
+                            vdu,
+                            db_vnfr,
+                            vdu_index,
+                            action_id,
+                            nsr_id,
+                            task_index,
+                            extra_dict,
+                        )
+                        db_new_tasks.append(task)
+                        task_index += 1
+
+                self.upload_all_tasks(
+                    db_new_tasks=db_new_tasks,
+                    now=now,
+                )
+
+            self.logger.debug(
+                logging_text + "Exit. Created {} tasks".format(len(db_new_tasks))
+            )
+            return (
+                {"status": "ok", "nsr_id": nsr_id, "action_id": action_id},
+                action_id,
+                True,
+            )
+        except Exception as e:
+            if isinstance(e, (DbException, NsException)):
+                self.logger.error(
+                    logging_text + "Exit Exception while '{}': {}".format(step, e)
+                )
+            else:
+                e = traceback_format_exc()
+                self.logger.critical(
+                    logging_text + "Exit Exception while '{}': {}".format(step, e),
+                    exc_info=True,
+                )
+            raise NsException(e)