test/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2024-12-02 11:59:00 +08:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

[skip e2e]Update MilvusSys to adapt to new version pymilvus (#16408)

Browse Source 
Signed-off-by: zhuwenxing <wenxing.zhu@zilliz.com>
This commit is contained in:
zhuwenxing 2022-04-07 10:37:31 +08:00 committed by GitHub
parent 909dd73741
commit dc47acba29
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
tests/python_client/common/milvus_sys.py
View File

@ -11,18 +11,17 @@ sys_logs_req = ujson.dumps({"metric_type": "system_logs"})
class MilvusSys: class MilvusSys:
def __init__(self, alias='default'): def __init__(self, alias='default'):
self.alias = alias self.alias = alias
self.client = connections.get_connection(alias=self.alias) self.handler = connections._fetch_handler(alias=self.alias)
if self.client is None: if self.handler is None:
raise Exception(f"Connection {alias} is disconnected or nonexistent") raise Exception(f"Connection {alias} is disconnected or nonexistent")
# TODO: for now it only supports non_orm style API for getMetricsRequest # TODO: for now it only supports non_orm style API for getMetricsRequest
with self.client._connection() as handler:
req = milvus_types.GetMetricsRequest(request=sys_info_req) req = milvus_types.GetMetricsRequest(request=sys_info_req)
self.sys_info = handler._stub.GetMetrics(req, wait_for_ready=True, timeout=None) self.sys_info = self.handler._stub.GetMetrics(req, wait_for_ready=True, timeout=None)
req = milvus_types.GetMetricsRequest(request=sys_statistics_req) req = milvus_types.GetMetricsRequest(request=sys_statistics_req)
self.sys_statistics = handler._stub.GetMetrics(req, wait_for_ready=True, timeout=None) self.sys_statistics = self.handler._stub.GetMetrics(req, wait_for_ready=True, timeout=None)
req = milvus_types.GetMetricsRequest(request=sys_logs_req) req = milvus_types.GetMetricsRequest(request=sys_logs_req)
self.sys_logs = handler._stub.GetMetrics(req, wait_for_ready=True, timeout=None) self.sys_logs = self.handler._stub.GetMetrics(req, wait_for_ready=True, timeout=None)
@property @property
def build_version(self): def build_version(self):
@ -50,7 +49,7 @@ class MilvusSys:
"""get all query nodes in Milvus deployment""" """get all query nodes in Milvus deployment"""
query_nodes = [] query_nodes = []
for node in self.nodes: for node in self.nodes:
if 'QueryNode' == node.get('infos').get('type'): if 'querynode' == node.get('infos').get('type'):
query_nodes.append(node) query_nodes.append(node)
return query_nodes return query_nodes
@ -59,7 +58,7 @@ class MilvusSys:
"""get all data nodes in Milvus deployment""" """get all data nodes in Milvus deployment"""
data_nodes = [] data_nodes = []
for node in self.nodes: for node in self.nodes:
if 'DataNode' == node.get('infos').get('type'): if 'datanode' == node.get('infos').get('type'):
data_nodes.append(node) data_nodes.append(node)
return data_nodes return data_nodes
@ -68,7 +67,7 @@ class MilvusSys:
"""get all index nodes in Milvus deployment""" """get all index nodes in Milvus deployment"""
index_nodes = [] index_nodes = []
for node in self.nodes: for node in self.nodes:
if 'IndexNode' == node.get('infos').get('type'): if 'indexnode' == node.get('infos').get('type'):
index_nodes.append(node) index_nodes.append(node)
return index_nodes return index_nodes
@ -77,7 +76,7 @@ class MilvusSys:
"""get all proxy nodes in Milvus deployment""" """get all proxy nodes in Milvus deployment"""
proxy_nodes = [] proxy_nodes = []
for node in self.nodes: for node in self.nodes:
if 'Proxy' == node.get('infos').get('type'): if 'proxy' == node.get('infos').get('type'):
proxy_nodes.append(node) proxy_nodes.append(node)
return proxy_nodes return proxy_nodes