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

more detail tracing in search

Browse Source
This commit is contained in:
peng.xu 2019-09-27 10:21:17 +08:00
parent a6a1ff2f13
commit 81a78a40cb
1 changed files with 11 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
mishards/service_handler.py
View File

@ -145,7 +145,9 @@ class ServiceHandler(milvus_pb2_grpc.MilvusServiceServicer):
conn = self.query_conn(addr, metadata=metadata)
start = time.time()
with self.tracer.start_span('search_{}_span'.format(addr),
span = kwargs.get('span', None)
span = span if span else context.get_active_span().context
with self.tracer.start_span('search_{}'.format(addr),
child_of=context.get_active_span().context):
ret = conn.search_vectors_in_files(table_name=query_params['table_id'],
file_ids=query_params['file_ids'],
@ -158,13 +160,15 @@ class ServiceHandler(milvus_pb2_grpc.MilvusServiceServicer):
all_topk_results.append(ret)
with ThreadPoolExecutor(max_workers=workers) as pool:
for addr, params in routing.items():
res = pool.submit(search, addr, params, vectors, topk, nprobe)
rs.append(res)
with self.tracer.start_span('do_search',
child_of=context.get_active_span().context) as span:
with ThreadPoolExecutor(max_workers=workers) as pool:
for addr, params in routing.items():
res = pool.submit(search, addr, params, vectors, topk, nprobe, span=span)
rs.append(res)
for res in rs:
res.result()
for res in rs:
res.result()
reverse = table_meta.metric_type == types.MetricType.IP
with self.tracer.start_span('do_merge',