+
+
+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"]
+ 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 = []
+ init_target = deep_get(db_k8scluster, ("_admin", "init"))
+ # helm-chart
+ if not init_target or "helm-chart" in init_target:
+ k8s_hc_id = None
+ try:
+ k8s_hc_id, uninstall_sw = await self.helm_k8scluster.init_env(k8s_credentials,
+ reuse_cluster_uuid=k8scluster_id)
+ 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)
+
+ if not init_target or "juju-bundle" in init_target:
+ # 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 + "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"]
+ 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_jb_id: # delete in reverse order of creation
+ step = "Removing juju-bundle '{}'".format(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)
+ db_k8scluster_update["_admin.juju-bundle.id"] = None
+
+ if k8s_hc_id:
+ step = "Removing helm-chart '{}'".format(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)
+ db_k8scluster_update["_admin.helm-chart.id"] = None
+
+ # 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})
+ db_k8scluster_update = None
+ self.logger.debug(logging_text + "Done")
+
+ except Exception as e:
+ if isinstance(e, (LcmException, DbException, K8sException, N2VCException)):
+ 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)