milvus-io
/
milvus
mirror of https://github.com/milvus-io/milvus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[skip ci] Add guarantee_timestamp of search_performance templet (#12774) 

Signed-off-by: wangting0128 <ting.wang@zilliz.com>
pull/12784/merge
wt 2021-12-06 18:14:52 +08:00 committed by GitHub
parent 7504dd5df2
commit 473ebd215b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/benchmark/milvus_benchmark/runners/search.py
View File

@ -23,6 +23,7 @@ class SearchRunner(BaseRunner):
top_ks = collection["top_ks"]
nqs = collection["nqs"]
filters = collection["filters"] if "filters" in collection else []
guarantee_timestamp = collection["guarantee_timestamp"] if "guarantee_timestamp" in collection else None
search_params = collection["search_params"]
# TODO: get fields by describe_index
@ -73,7 +74,8 @@ class SearchRunner(BaseRunner):
"nq": nq,
"topk": top_k,
"search_param": search_param,
"filter": filter_param
"filter": filter_param,
"guarantee_timestamp": guarantee_timestamp
}
vector_query = {"vector": {index_field_name: search_info}}
case = {
@ -82,6 +84,7 @@ class SearchRunner(BaseRunner):
"run_count": run_count,
"filter_query": filter_query,
"vector_query": vector_query,
"guarantee_timestamp": guarantee_timestamp
}
cases.append(case)
case_metrics.append(case_metric)
@ -108,7 +111,8 @@ class SearchRunner(BaseRunner):
for i in range(run_count):
logger.debug("Start run query, run %d of %s" % (i+1, run_count))
start_time = time.time()
_query_res = self.milvus.query(case_param["vector_query"], filter_query=case_param["filter_query"])
_query_res = self.milvus.query(case_param["vector_query"], filter_query=case_param["filter_query"],
guarantee_timestamp=case_param["guarantee_timestamp"])
interval_time = time.time() - start_time
total_query_time += interval_time
if (i == 0) or (min_query_time > interval_time):