admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2026-01-04 18:02:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

[skip ci](shards): handle connection empty cases

Browse Source 
Signed-off-by: peng.xu <peng.xu@zilliz.com>
This commit is contained in:
peng.xu 2020-04-30 23:26:19 +08:00
parent 0032c13445
commit 29a41ef645
No known key found for this signature in database
GPG Key ID: 3DF8F92DC43F2CEC
2 changed files with 19 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
shards/mishards/exception_handlers.py
View File

@ -83,3 +83,9 @@ def DBErrorHandler(err):
def InvalidArgumentErrorHandler(err):
logger.error(err)
return resp_handler(err, status_pb2.ILLEGAL_RANGE)
@server.errorhandler(exceptions.ConnectionNotFoundError)
def InvalidArgumentErrorHandler(err):
logger.error(err)
return resp_handler(err, status_pb2.UNEXPECTED_ERROR)

15
shards/mishards/router/__init__.py
View File

@ -17,8 +17,19 @@ class RouterMixin:
return conn.conn
def query_conn(self, name, metadata=None):
conn = self.readonly_topo.get_group(name).get(name).fetch()
if not name:
raise exceptions.ConnectionNotFoundError(
message=f'Conn Group is Empty. Please Check your configurations',
metadata=metadata)
group = self.readonly_topo.get_group(name)
if not group:
raise exceptions.ConnectionNotFoundError(
message=f'Conn Group {name} is Empty. Please Check your configurations',
metadata=metadata)
conn = group.get(name).fetch()
if not conn:
raise exceptions.ConnectionNotFoundError(name, metadata=metadata)
raise exceptions.ConnectionNotFoundError(
message=f'Conn {name} Not Found', metadata=metadata)
conn.on_connect(metadata=metadata)
return conn