mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-29 15:05:31 +08:00
84319ed311
* [skip ci] Update query test cases Signed-off-by: wangting0128 <ting.wang@zilliz.com> * [skip ci] Conflict resolution Signed-off-by: wangting0128 <ting.wang@zilliz.com> * [skip ci] Update teardown Signed-off-by: wangting0128 <ting.wang@zilliz.com>
45 lines
1.5 KiB
Python
45 lines
1.5 KiB
Python
import traceback
|
|
from utils.util_log import test_log as log
|
|
|
|
|
|
class Error:
|
|
def __init__(self, error):
|
|
self.code = getattr(error, 'code', -1)
|
|
self.message = getattr(error, 'message', str(error))
|
|
|
|
|
|
log_row_length = 3000
|
|
|
|
|
|
def api_request_catch():
|
|
def wrapper(func):
|
|
def inner_wrapper(*args, **kwargs):
|
|
try:
|
|
res = func(*args, **kwargs)
|
|
log_res = str(res)[0:log_row_length] + '......' if len(str(res)) > log_row_length else str(res)
|
|
log.debug("(api_response) Response : %s " % log_res)
|
|
return res, True
|
|
except Exception as e:
|
|
log_e = str(e)[0:log_row_length] + '......' if len(str(e)) > log_row_length else str(e)
|
|
log.error(traceback.format_exc())
|
|
log.error("(api_response) [Milvus API Exception]%s: %s" % (str(func), log_e))
|
|
return Error(e), False
|
|
return inner_wrapper
|
|
return wrapper
|
|
|
|
|
|
@api_request_catch()
|
|
def api_request(_list, **kwargs):
|
|
if isinstance(_list, list):
|
|
func = _list[0]
|
|
if callable(func):
|
|
arg = []
|
|
if len(_list) > 1:
|
|
for a in _list[1:]:
|
|
arg.append(a)
|
|
log_arg = str(arg)[0:log_row_length] + '......' if len(str(arg)) > log_row_length else str(arg)
|
|
log.debug("(api_request) Request: [%s] args: %s, kwargs: %s" % (str(func), log_arg, str(kwargs)))
|
|
return func(*arg, **kwargs)
|
|
return False, False
|
|
|