Fix not exist milvus chart (#7271)

Signed-off-by: ThreadDao <yufen.zong@zilliz.com>
This commit is contained in:
ThreadDao 2021-08-24 19:35:51 +08:00 committed by GitHub
parent bd250f28ac
commit ecfbcd4760
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -6,16 +6,16 @@ from utils.util_log import test_log as log
from common import common_func as cf from common import common_func as cf
from scale import scale_common as sc from scale import scale_common as sc
milvus_chart_path = sc.get_milvus_chart_env_var()
class HelmEnv: class HelmEnv:
def __init__(self, release_name=None, **kwargs): def __init__(self, release_name=None, **kwargs):
self.release_name = release_name if release_name else cf.gen_unique_str("scale") self.release_name = release_name if release_name else cf.gen_unique_str("scale")
self.proxy = kwargs.get(constants.PROXY, 1) self.proxy = kwargs.get(constants.PROXY, 1)
self.data_node = kwargs.get(constants.DATA_NODE, 1) self.data_node = kwargs.get(constants.DATA_NODE, 1)
self.index_node = kwargs.get(constants.INDEX_NODE, 1) self.index_node = kwargs.get(constants.INDEX_NODE, 1)
self.query_node = kwargs.get(constants.QUERY_NODE, 1) self.query_node = kwargs.get(constants.QUERY_NODE, 1)
self.milvus_chart_path = sc.get_milvus_chart_env_var()
def helm_install_cluster_milvus(self, image_pull_policy=constants.IF_NOT_PRESENT): def helm_install_cluster_milvus(self, image_pull_policy=constants.IF_NOT_PRESENT):
""" """
@ -40,7 +40,7 @@ class HelmEnv:
f'{self.release_name} . ' f'{self.release_name} . '
log.debug(f'install_cmd: {install_cmd}') log.debug(f'install_cmd: {install_cmd}')
try: try:
os.system(f'cd {milvus_chart_path} && {install_cmd}') os.system(f'cd {self.milvus_chart_path} && {install_cmd}')
except Exception: except Exception:
raise Exception("Failed to deploy cluster milvus") raise Exception("Failed to deploy cluster milvus")
return self.get_service_ip() return self.get_service_ip()
@ -63,7 +63,7 @@ class HelmEnv:
f'--set queryNode.replicas={query_node} ' \ f'--set queryNode.replicas={query_node} ' \
f'--reuse-values' f'--reuse-values'
log.debug(f'upgrade_cmd: {upgrade_cmd}') log.debug(f'upgrade_cmd: {upgrade_cmd}')
if os.system(f'cd {milvus_chart_path} && {upgrade_cmd}'): if os.system(f'cd {self.milvus_chart_path} && {upgrade_cmd}'):
raise Exception(f'Failed to upgrade cluster milvus with {kwargs}') raise Exception(f'Failed to upgrade cluster milvus with {kwargs}')
def helm_uninstall_cluster_milvus(self): def helm_uninstall_cluster_milvus(self):