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

Add trace log for search (#8302) 

Signed-off-by: cai.zhang <cai.zhang@zilliz.com>
pull/8334/head
cai.zhang 2021-09-22 19:36:00 +08:00 committed by GitHub
parent 79b7dcc75a
commit 729d409a32
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
internal/proxy/impl.go
View File

@ -1418,6 +1418,8 @@ func (node *Proxy) Search(ctx context.Context, request *milvuspb.SearchRequest)
Status: unhealthyStatus(),
}, nil
}
sp, ctx := trace.StartSpanFromContextWithOperationName(ctx, "Proxy-Search")
defer sp.Finish()
qt := &searchTask{
ctx: ctx,
Condition: NewTaskCondition(ctx),

8
internal/proxy/task.go
View File

@ -27,6 +27,8 @@ import (
"time"
"unsafe"
"github.com/milvus-io/milvus/internal/util/trace"
"github.com/milvus-io/milvus/internal/common"
"go.uber.org/zap"
@ -1439,6 +1441,8 @@ func (st *searchTask) getVChannels() ([]vChan, error) {
}
func (st *searchTask) PreExecute(ctx context.Context) error {
sp, ctx := trace.StartSpanFromContextWithOperationName(st.TraceCtx(), "Proxy-Search-PreExecute")
defer sp.Finish()
st.Base.MsgType = commonpb.MsgType_Search
st.Base.SourceID = Params.ProxyID
@ -1630,6 +1634,8 @@ func (st *searchTask) PreExecute(ctx context.Context) error {
}
func (st *searchTask) Execute(ctx context.Context) error {
sp, ctx := trace.StartSpanFromContextWithOperationName(st.TraceCtx(), "Proxy-Search-Execute")
defer sp.Finish()
var tsMsg msgstream.TsMsg = &msgstream.SearchMsg{
SearchRequest: *st.SearchRequest,
BaseMsg: msgstream.BaseMsg{
@ -1957,6 +1963,8 @@ func reduceSearchResultData(searchResultData []*schemapb.SearchResultData, avail
//}
func (st *searchTask) PostExecute(ctx context.Context) error {
sp, ctx := trace.StartSpanFromContextWithOperationName(st.TraceCtx(), "Proxy-Search-PostExecute")
defer sp.Finish()
t0 := time.Now()
defer func() {
log.Debug("WaitAndPostExecute", zap.Any("time cost", time.Since(t0)))