Add operate api to GVNFM
[vfc/gvnfm/vnflcm.git] / lcm / lcm / pub / vimapi / adaptor.py
index 6eeaf73..8ebddab 100644 (file)
 
 import logging
 import time
+import ast
 
 from lcm.pub.utils.values import ignore_case_get, set_opt_val
 from lcm.pub.msapi.aai import get_flavor_info
 from . import api
 from .exceptions import VimException
+from lcm.nf.const import ACTION_TYPE
 
 logger = logging.getLogger(__name__)
 
@@ -64,6 +66,50 @@ def get_res_id(res_cache, res_type, key):
     return res_cache[res_type][key]
 
 
+def action_vm(action_type, server, vimId, tenantId):
+    param = {}
+    if action_type == ACTION_TYPE.START:
+        param = {
+            "os-start": None,
+        }
+    elif action_type == ACTION_TYPE.STOP:
+        param = {
+            "os-stop": None,
+        }
+    elif action_type == ACTION_TYPE.REBOOT:
+        param = {
+            "reboot": {}
+        }
+        if server["status"] == "ACTIVE":
+            param["reboot"]["type"] = "SOFT"
+        else:
+            param["reboot"]["type"] = "HARD"
+    res_id = server["id"]
+    api.action_vm(vimId, tenantId, res_id, param)
+
+
+# TODO Have to check if the resources should be started and stopped in some order.
+def operate_vim_res(data, changeStateTo, stopType, gracefulStopTimeout, do_notify_op):
+    for res in ignore_case_get(data, "vm"):
+        try:
+            if changeStateTo == "STARTED":
+                action_vm(ACTION_TYPE.START, res, res["vim_id"], res["tenant_id"])
+                do_notify_op("ACTIVE", res["id"])
+            elif changeStateTo == "STOPPED":
+                if stopType == "GRACEFUL":
+                    if gracefulStopTimeout > 60:
+                        gracefulStopTimeout = 60
+                    time.sleep(gracefulStopTimeout)
+                action_vm(ACTION_TYPE.STOP, res, res["vim_id"], res["tenant_id"])
+                # TODO check if the we should poll getvm to get the status or the action_vm api
+                # successful return should suffice to mark vm as Active/Inactive
+                do_notify_op("INACTIVE", res["id"])
+        except VimException as e:
+            # TODO Have to update database appropriately on failure
+            logger.error("Failed to Heal %s(%s)", RES_VM, res["res_id"])
+            logger.error("%s:%s", e.http_code, e.message)
+
+
 def create_vim_res(data, do_notify):
     vim_cache, res_cache = {}, {}
     for vol in ignore_case_get(data, "volume_storages"):
@@ -181,7 +227,7 @@ def create_port(vim_cache, res_cache, data, port, do_notify, res_type):
                 vdu["cps"].append(port["cp_id"])
             break
     if not location_info:
-        err_msg = "vdu_id(%s) for cp(%s) is not defined"
+        err_msg = "vdu_id(%s) for cp(%s) is not defined."
         raise VimException(err_msg % (port_ref_vdu_id, port["cp_id"]), ERR_CODE)
     network_id = ignore_case_get(port, "networkId")
     subnet_id = ignore_case_get(port, "subnetId")
@@ -196,7 +242,7 @@ def create_port(vim_cache, res_cache, data, port, do_notify, res_type):
     set_opt_val(param, "macAddress", ignore_case_get(port["properties"], "mac_address"))
     ip_address = []
     for one_protocol_data in port["properties"]["protocol_data"]:
-        l3_address_data = one_protocol_data["address_data"]["l3_address_data"]
+        l3_address_data = one_protocol_data["address_data"]["l3_address_data"]  # l3 is not 13
         fixed_ip_address = ignore_case_get(l3_address_data, "fixed_ip_address")
         ip_address.extend(fixed_ip_address)
     set_opt_val(param, "ip", ",".join(ip_address))
@@ -210,20 +256,39 @@ def create_port(vim_cache, res_cache, data, port, do_notify, res_type):
     set_res_cache(res_cache, res_type, port["cp_id"], ret["id"])
 
 
-def search_flavor_aai(vim_id, memory_page_size):
+def parse_unit(val, base_unit):
+    recognized_units = ["B", "kB", "KiB", "MB", "MiB", "GB", "GiB", "TB", "TiB"]
+    units_rate = [1, 1000, 1024, 1000000, 1048576, 1000000000, 1073741824, 1000000000000, 1099511627776]
+    unit_rate_map = {unit.upper(): rate for unit, rate in zip(recognized_units, units_rate)}
+    num_unit = val.strip().split(" ")
+    if len(num_unit) != 2:
+        return val.strip
+    num, unit = num_unit[0], num_unit[1]
+    return int(num) * unit_rate_map[unit.upper()] / unit_rate_map[base_unit.upper()]
+
+
+def search_flavor_aai(vim_id, memory_page_size, memory_page_unit):
     aai_flavors = get_flavor_info(vim_id)
     if not aai_flavors:
         return None
     logger.debug("aai_flavors:%s" % aai_flavors)
-    aai_flavor = aai_flavors[0]["flavors"]["flavor"]
+    aai_flavor = aai_flavors["flavor"]
     for one_aai_flavor in aai_flavor:
-        hpa_capabilities = one_aai_flavor["hpa-capabilities"]
+        if one_aai_flavor["flavor-name"].find("onap.") == -1:
+            continue
+        hpa_capabilities = one_aai_flavor["hpa-capabilities"]["hpa-capability"]
+        logger.debug("hpa_capabilities=%s", hpa_capabilities)
         for one_hpa_capa in hpa_capabilities:
+            logger.debug("one_hpa_capa=%s", one_hpa_capa)
             hpa_feature_attr = one_hpa_capa["hpa-feature-attributes"]
             for one_hpa_attr in hpa_feature_attr:
                 hpa_key = one_hpa_attr["hpa-attribute-key"]
-                hpa_value = one_hpa_attr["hpa-attribute-value"]["value"]
-                if hpa_key == "memoryPageSize" and int(hpa_value) == memory_page_size:
+                hpa_attr_value = ast.literal_eval(one_hpa_attr["hpa-attribute-value"])
+                mem_size = ignore_case_get(hpa_attr_value, 'value')
+                mem_unit = ignore_case_get(hpa_attr_value, 'unit')
+                value = mem_size + " " + mem_unit
+                hpa_mem_size = parse_unit(value, memory_page_unit)
+                if hpa_key == "memoryPageSize" and hpa_mem_size == memory_page_size:
                     return one_aai_flavor
 
 
@@ -231,23 +296,26 @@ def create_flavor(vim_cache, res_cache, data, flavor, do_notify, res_type):
     location_info = flavor["properties"]["location_info"]
     vim_id, tenant_name = location_info["vimid"], location_info["tenant"]
     virtual_compute = flavor["virtual_compute"]
+    virtual_storage = flavor["virtual_storage"]
+    virtual_cpu = ignore_case_get(virtual_compute, "virtual_cpu")
+    virtual_memory = ignore_case_get(virtual_compute, "virtual_memory")
     param = {
         "name": "Flavor_%s" % flavor["vdu_id"],
-        "vcpu": int(virtual_compute["virtual_cpu"]["num_virtual_cpu"]),
-        "memory": int(virtual_compute["virtual_memory"]["virtual_mem_size"].replace('MB', '').strip()),
+        "vcpu": int(ignore_case_get(virtual_cpu, "num_virtual_cpu")),
+        "memory": int(ignore_case_get(virtual_memory, "virtual_mem_size").replace('MB', '').strip()),
         "isPublic": True
     }
 
     # just do memory huge page
     flavor_extra_specs = ""
-    vdu_memory_requirements = virtual_compute["virtual_memory"]["vdu_memory_requirements"]
+    vdu_memory_requirements = ignore_case_get(virtual_memory, "vdu_memory_requirements")
     if "memoryPageSize" in vdu_memory_requirements:
         memory_page_size = int(vdu_memory_requirements["memoryPageSize"].replace('MB', '').strip())
-        flavor_extra_specs = ("hw:mem_page_size=%sMB" % memory_page_size)
+        flavor_extra_specs = {"hw": memory_page_size, }  # TODO
         logger.debug("flavor_extra_specs:%s" % flavor_extra_specs)
 
-    # search aai flavor
-    aai_flavor = search_flavor_aai(vim_id, memory_page_size)
+    # FIXME: search aai flavor
+    aai_flavor = search_flavor_aai(vim_id, memory_page_size, "MB")
 
     # add aai flavor
     if aai_flavor:
@@ -256,8 +324,8 @@ def create_flavor(vim_cache, res_cache, data, flavor, do_notify, res_type):
         set_res_cache(res_cache, res_type, flavor["vdu_id"], ret["flavor-id"])
     else:
         extra_specs = []
-        disk_type = virtual_compute["virtual_storage"]["type_of_storage"]
-        disk_size = int(virtual_compute["virtual_storage"]["size_of_storage"].replace('GB', '').strip())
+        disk_type = ignore_case_get(virtual_storage, "type_of_storage")
+        disk_size = int(ignore_case_get(virtual_storage, "size_of_storage").replace('GB', '').strip())
         if disk_type == "root":
             param["disk"] = disk_size
         elif disk_type == "ephemeral":