X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=osm_lcm%2Fvim_sdn.py;h=6593b4688cd1cf6d6b87f502518918cd0f65b587;hb=refs%2Fheads%2Ffeature7184;hp=7c0bc0b81a20ffb058f7b6ca31129028538448d5;hpb=8069ce54d91918c53ec3bf9f8fd4eee8b6aacb93;p=osm%2FLCM.git diff --git a/osm_lcm/vim_sdn.py b/osm_lcm/vim_sdn.py index 7c0bc0b..6593b46 100644 --- a/osm_lcm/vim_sdn.py +++ b/osm_lcm/vim_sdn.py @@ -16,10 +16,14 @@ # under the License. ## +import yaml import logging import logging.handlers from osm_lcm import ROclient -from osm_lcm.lcm_utils import LcmException, LcmBase +from osm_lcm.lcm_utils import LcmException, LcmBase, deep_get +from n2vc.k8s_helm_conn import K8sHelmConnector +from n2vc.k8s_juju_conn import K8sJujuConnector +from n2vc.exceptions import K8sException, N2VCException from osm_common.dbbase import DbException from copy import deepcopy @@ -31,7 +35,7 @@ class VimLcm(LcmBase): vim_config_encrypted = {"1.1": ("admin_password", "nsx_password", "vcenter_password"), "default": ("admin_password", "nsx_password", "vcenter_password", "vrops_password")} - def __init__(self, db, msg, fs, lcm_tasks, ro_config, loop): + def __init__(self, db, msg, fs, lcm_tasks, config, loop): """ Init, Connect to database, filesystem storage, and messaging :param config: two level dictionary with configuration. Top level should contain 'database', 'storage', @@ -41,7 +45,7 @@ class VimLcm(LcmBase): self.logger = logging.getLogger('lcm.vim') self.loop = loop self.lcm_tasks = lcm_tasks - self.ro_config = ro_config + self.ro_config = config["ro_config"] super().__init__(db, msg, fs, self.logger) @@ -381,7 +385,7 @@ class WimLcm(LcmBase): # values that are encrypted at wim config because they are passwords wim_config_encrypted = () - def __init__(self, db, msg, fs, lcm_tasks, ro_config, loop): + def __init__(self, db, msg, fs, lcm_tasks, config, loop): """ Init, Connect to database, filesystem storage, and messaging :param config: two level dictionary with configuration. Top level should contain 'database', 'storage', @@ -391,7 +395,7 @@ class WimLcm(LcmBase): self.logger = logging.getLogger('lcm.vim') self.loop = loop self.lcm_tasks = lcm_tasks - self.ro_config = ro_config + self.ro_config = config["ro_config"] super().__init__(db, msg, fs, self.logger) @@ -684,7 +688,7 @@ class WimLcm(LcmBase): class SdnLcm(LcmBase): - def __init__(self, db, msg, fs, lcm_tasks, ro_config, loop): + def __init__(self, db, msg, fs, lcm_tasks, config, loop): """ Init, Connect to database, filesystem storage, and messaging :param config: two level dictionary with configuration. Top level should contain 'database', 'storage', @@ -694,7 +698,7 @@ class SdnLcm(LcmBase): self.logger = logging.getLogger('lcm.sdn') self.loop = loop self.lcm_tasks = lcm_tasks - self.ro_config = ro_config + self.ro_config = config["ro_config"] super().__init__(db, msg, fs, self.logger) @@ -912,3 +916,333 @@ class SdnLcm(LcmBase): except DbException as e: self.logger.error(logging_text + "Cannot update database: {}".format(e)) self.lcm_tasks.remove("sdn", sdn_id, order_id) + + +class K8sClusterLcm(LcmBase): + + def __init__(self, db, msg, fs, lcm_tasks, config, loop): + """ + Init, Connect to database, filesystem storage, and messaging + :param config: two level dictionary with configuration. Top level should contain 'database', 'storage', + :return: None + """ + + self.logger = logging.getLogger('lcm.k8scluster') + self.loop = loop + self.lcm_tasks = lcm_tasks + self.vca_config = config["VCA"] + self.fs = fs + self.db = db + + self.helm_k8scluster = K8sHelmConnector( + kubectl_command=self.vca_config.get("kubectlpath"), + helm_command=self.vca_config.get("helmpath"), + fs=self.fs, + log=self.logger, + db=self.db, + on_update_db=None + ) + + self.juju_k8scluster = K8sJujuConnector( + kubectl_command=self.vca_config.get("kubectlpath"), + juju_command=self.vca_config.get("jujupath"), + fs=self.fs, + log=self.logger, + db=self.db, + on_update_db=None + ) + + super().__init__(db, msg, fs, self.logger) + + async def create(self, k8scluster_content, order_id): + + # HA tasks and backward compatibility: + # If 'vim_content' does not include 'op_id', we a running a legacy NBI version. + # In such a case, HA is not supported by NBI, 'op_id' is None, and lock_HA() will do nothing. + # Register 'create' task here for related future HA operations + op_id = k8scluster_content.pop('op_id', None) + if not self.lcm_tasks.lock_HA('k8scluster', 'create', op_id): + return + + k8scluster_id = k8scluster_content["_id"] + k8scluster_content.pop("op_id", None) + logging_text = "Task k8scluster_create={} ".format(k8scluster_id) + self.logger.debug(logging_text + "Enter") + + db_k8scluster = None + db_k8scluster_update = {} + + exc = None + operationState_HA = '' + detailed_status_HA = '' + try: + step = "Getting k8scluster-id='{}' from db".format(k8scluster_id) + self.logger.debug(logging_text + step) + db_k8scluster = self.db.get_one("k8sclusters", {"_id": k8scluster_id}) + self.db.encrypt_decrypt_fields(db_k8scluster.get("credentials"), 'decrypt', ['password', 'secret'], + schema_version=db_k8scluster["schema_version"], salt=db_k8scluster["_id"]) + k8s_credentials = yaml.safe_dump(db_k8scluster.get("credentials")) + error_text_list = [] + # helm-chart + k8s_hc_id = None + try: + k8s_hc_id, uninstall_sw = await self.helm_k8scluster.init_env(k8s_credentials) + db_k8scluster_update["_admin.helm-chart.id"] = k8s_hc_id + db_k8scluster_update["_admin.helm-chart.created"] = uninstall_sw + except Exception as e: + error_text_list.append("Failing init helm-chart: {}".format(e)) + db_k8scluster_update["_admin.helm-chart.error_msg"] = str(e) + if isinstance(e, K8sException): + self.logger.error(logging_text + "Failing init helm-chart: {}".format(e)) + else: + self.logger.error(logging_text + "Failing init helm-chart: {}".format(e), exc_info=True) + + # Juju/k8s cluster + k8s_jb_id = None + try: + k8s_jb_id, uninstall_sw = await self.juju_k8scluster.init_env(k8s_credentials) + db_k8scluster_update["_admin.juju-bundle.id"] = k8s_jb_id + db_k8scluster_update["_admin.juju-bundle.created"] = uninstall_sw + except Exception as e: + error_text_list.append("Failing init juju-bundle: {}".format(e)) + db_k8scluster_update["_admin.juju-bundle.error_msg"] = str(e) + if isinstance(e, N2VCException): + self.logger.error(logging_text + "Failing init juju-bundle: {}".format(e)) + else: + self.logger.error(logging_text + "Failing init juju-bundle: {}".format(e), exc_info=True) + + # mark as an error if both helm-chart and juju-bundle have been failed + if k8s_hc_id or k8s_jb_id: + self.logger.debug(logging_text + " successfully created") + db_k8scluster_update["_admin.operationalState"] = "ENABLED" + else: + self.logger.debug(logging_text + " successfully created with errors") + db_k8scluster_update["_admin.operationalState"] = "ERROR" + db_k8scluster_update["_admin.detailed-status"] = ";".join(error_text_list) + + except Exception as e: + self.logger.critical(logging_text + "Exit Exception {}".format(e), exc_info=True) + exc = e + finally: + if exc and db_k8scluster: + db_k8scluster_update["_admin.operationalState"] = "ERROR" + db_k8scluster_update["_admin.detailed-status"] = "ERROR {}: {}".format(step, exc) + + # Mark the k8scluster 'create' HA task as erroneous + operationState_HA = 'FAILED' + detailed_status_HA = "ERROR {}: {}".format(step, exc) + try: + if db_k8scluster_update: + self.update_db_2("k8sclusters", k8scluster_id, db_k8scluster_update) + + # Register the K8scluster 'create' HA task either + # succesful or erroneous, or do nothing (if legacy NBI) + self.lcm_tasks.register_HA('k8scluster', 'create', op_id, + operationState=operationState_HA, + detailed_status=detailed_status_HA) + except DbException as e: + self.logger.error(logging_text + "Cannot update database: {}".format(e)) + self.lcm_tasks.remove("k8scluster", k8scluster_id, order_id) + + async def delete(self, k8scluster_content, order_id): + + # HA tasks and backward compatibility: + # If 'vim_content' does not include 'op_id', we a running a legacy NBI version. + # In such a case, HA is not supported by NBI, 'op_id' is None, and lock_HA() will do nothing. + # Register 'delete' task here for related future HA operations + op_id = k8scluster_content.pop('op_id', None) + if not self.lcm_tasks.lock_HA('k8scluster', 'delete', op_id): + return + + k8scluster_id = k8scluster_content["_id"] + k8scluster_content.pop("op_id", None) + logging_text = "Task k8scluster_delete={} ".format(k8scluster_id) + self.logger.debug(logging_text + "Enter") + + db_k8scluster = None + db_k8scluster_update = {} + exc = None + operationState_HA = '' + detailed_status_HA = '' + try: + step = "Getting k8scluster='{}' from db".format(k8scluster_id) + self.logger.debug(logging_text + step) + db_k8scluster = self.db.get_one("k8sclusters", {"_id": k8scluster_id}) + k8s_hc_id = deep_get(db_k8scluster, ("_admin", "helm-chart", "id")) + k8s_jb_id = deep_get(db_k8scluster, ("_admin", "juju-bundle", "id")) + + uninstall_sw = deep_get(db_k8scluster, ("_admin", "helm-chart", "created")) + cluster_removed = True + if k8s_hc_id: + uninstall_sw = uninstall_sw or False + cluster_removed = await self.helm_k8scluster.reset(cluster_uuid=k8s_hc_id, uninstall_sw=uninstall_sw) + + if k8s_jb_id: + uninstall_sw = uninstall_sw or False + cluster_removed = await self.juju_k8scluster.reset(cluster_uuid=k8s_jb_id, uninstall_sw=uninstall_sw) + + # Try to remove from cluster_inserted to clean old versions + if k8s_hc_id and cluster_removed: + step = "Removing k8scluster='{}' from k8srepos".format(k8scluster_id) + self.logger.debug(logging_text + step) + db_k8srepo_list = self.db.get_list("k8srepos", {"_admin.cluster-inserted": k8s_hc_id}) + for k8srepo in db_k8srepo_list: + try: + cluster_list = k8srepo["_admin"]["cluster-inserted"] + cluster_list.remove(k8s_hc_id) + self.update_db_2("k8srepos", k8srepo["_id"], {"_admin.cluster-inserted": cluster_list}) + except Exception as e: + self.logger.error("{}: {}".format(step, e)) + self.db.del_one("k8sclusters", {"_id": k8scluster_id}) + else: + raise LcmException("An error happened during the reset of the k8s cluster '{}'".format(k8scluster_id)) + # if not cluster_removed: + # raise Exception("K8scluster was not properly removed") + + except Exception as e: + if isinstance(e, (LcmException, DbException)): + self.logger.error(logging_text + "Exit Exception {}".format(e)) + else: + self.logger.critical(logging_text + "Exit Exception {}".format(e), exc_info=True) + exc = e + finally: + if exc and db_k8scluster: + db_k8scluster_update["_admin.operationalState"] = "ERROR" + db_k8scluster_update["_admin.detailed-status"] = "ERROR {}: {}".format(step, exc) + # Mark the WIM 'create' HA task as erroneous + operationState_HA = 'FAILED' + detailed_status_HA = "ERROR {}: {}".format(step, exc) + try: + if db_k8scluster_update: + self.update_db_2("k8sclusters", k8scluster_id, db_k8scluster_update) + # Register the K8scluster 'delete' HA task either + # succesful or erroneous, or do nothing (if legacy NBI) + self.lcm_tasks.register_HA('k8scluster', 'delete', op_id, + operationState=operationState_HA, + detailed_status=detailed_status_HA) + except DbException as e: + self.logger.error(logging_text + "Cannot update database: {}".format(e)) + self.lcm_tasks.remove("k8scluster", k8scluster_id, order_id) + + +class K8sRepoLcm(LcmBase): + + def __init__(self, db, msg, fs, lcm_tasks, config, loop): + """ + Init, Connect to database, filesystem storage, and messaging + :param config: two level dictionary with configuration. Top level should contain 'database', 'storage', + :return: None + """ + + self.logger = logging.getLogger('lcm.k8srepo') + self.loop = loop + self.lcm_tasks = lcm_tasks + self.vca_config = config["VCA"] + self.fs = fs + self.db = db + + self.k8srepo = K8sHelmConnector( + kubectl_command=self.vca_config.get("kubectlpath"), + helm_command=self.vca_config.get("helmpath"), + fs=self.fs, + log=self.logger, + db=self.db, + on_update_db=None + ) + + super().__init__(db, msg, fs, self.logger) + + async def create(self, k8srepo_content, order_id): + + # HA tasks and backward compatibility: + # If 'vim_content' does not include 'op_id', we a running a legacy NBI version. + # In such a case, HA is not supported by NBI, 'op_id' is None, and lock_HA() will do nothing. + # Register 'create' task here for related future HA operations + + op_id = k8srepo_content.pop('op_id', None) + if not self.lcm_tasks.lock_HA('k8srepo', 'create', op_id): + return + + k8srepo_id = k8srepo_content.get("_id") + logging_text = "Task k8srepo_create={} ".format(k8srepo_id) + self.logger.debug(logging_text + "Enter") + + db_k8srepo = None + db_k8srepo_update = {} + exc = None + operationState_HA = '' + detailed_status_HA = '' + try: + step = "Getting k8srepo-id='{}' from db".format(k8srepo_id) + self.logger.debug(logging_text + step) + db_k8srepo = self.db.get_one("k8srepos", {"_id": k8srepo_id}) + db_k8srepo_update["_admin.operationalState"] = "ENABLED" + except Exception as e: + self.logger.critical(logging_text + "Exit Exception {}".format(e), exc_info=True) + exc = e + finally: + if exc and db_k8srepo: + db_k8srepo_update["_admin.operationalState"] = "ERROR" + db_k8srepo_update["_admin.detailed-status"] = "ERROR {}: {}".format(step, exc) + # Mark the WIM 'create' HA task as erroneous + operationState_HA = 'FAILED' + detailed_status_HA = "ERROR {}: {}".format(step, exc) + try: + if db_k8srepo_update: + self.update_db_2("k8srepos", k8srepo_id, db_k8srepo_update) + # Register the K8srepo 'create' HA task either + # succesful or erroneous, or do nothing (if legacy NBI) + self.lcm_tasks.register_HA('k8srepo', 'create', op_id, + operationState=operationState_HA, + detailed_status=detailed_status_HA) + except DbException as e: + self.logger.error(logging_text + "Cannot update database: {}".format(e)) + self.lcm_tasks.remove("k8srepo", k8srepo_id, order_id) + + async def delete(self, k8srepo_content, order_id): + + # HA tasks and backward compatibility: + # If 'vim_content' does not include 'op_id', we a running a legacy NBI version. + # In such a case, HA is not supported by NBI, 'op_id' is None, and lock_HA() will do nothing. + # Register 'delete' task here for related future HA operations + op_id = k8srepo_content.pop('op_id', None) + if not self.lcm_tasks.lock_HA('k8srepo', 'delete', op_id): + return + + k8srepo_id = k8srepo_content.get("_id") + logging_text = "Task k8srepo_delete={} ".format(k8srepo_id) + self.logger.debug(logging_text + "Enter") + + db_k8srepo = None + db_k8srepo_update = {} + + exc = None + operationState_HA = '' + detailed_status_HA = '' + try: + step = "Getting k8srepo-id='{}' from db".format(k8srepo_id) + self.logger.debug(logging_text + step) + db_k8srepo = self.db.get_one("k8srepos", {"_id": k8srepo_id}) + + except Exception as e: + self.logger.critical(logging_text + "Exit Exception {}".format(e), exc_info=True) + exc = e + finally: + if exc and db_k8srepo: + db_k8srepo_update["_admin.operationalState"] = "ERROR" + db_k8srepo_update["_admin.detailed-status"] = "ERROR {}: {}".format(step, exc) + # Mark the WIM 'create' HA task as erroneous + operationState_HA = 'FAILED' + detailed_status_HA = "ERROR {}: {}".format(step, exc) + try: + if db_k8srepo_update: + self.update_db_2("k8srepos", k8srepo_id, db_k8srepo_update) + # Register the K8srepo 'delete' HA task either + # succesful or erroneous, or do nothing (if legacy NBI) + self.lcm_tasks.register_HA('k8srepo', 'delete', op_id, + operationState=operationState_HA, + detailed_status=detailed_status_HA) + self.db.del_one("k8srepos", {"_id": k8srepo_id}) + except DbException as e: + self.logger.error(logging_text + "Cannot update database: {}".format(e)) + self.lcm_tasks.remove("k8srepo", k8srepo_id, order_id)