diff --git a/internal/querynodev2/handlers.go b/internal/querynodev2/handlers.go index 6707081573..61b2c91e5d 100644 --- a/internal/querynodev2/handlers.go +++ b/internal/querynodev2/handlers.go @@ -435,6 +435,12 @@ func (node *QueryNode) searchChannel(ctx context.Context, req *querypb.SearchReq reduce.NewReduceSearchResultInfo(req.GetReq().GetNq(), req.GetReq().GetTopk()).WithMetricType(req.GetReq().GetMetricType()).WithGroupByField(req.GetReq().GetGroupByFieldId()). WithGroupSize(req.GetReq().GetGroupSize()).WithAdvance(req.GetReq().GetIsAdvanced())) + + reduceLatency := tr.RecordSpan() + metrics.QueryNodeReduceLatency. + WithLabelValues(fmt.Sprint(node.GetNodeID()), metrics.SearchLabel, metrics.ReduceShards, metrics.BatchReduce). + Observe(float64(reduceLatency.Milliseconds())) + if err != nil { return nil, err } diff --git a/internal/querynodev2/services.go b/internal/querynodev2/services.go index 459fdf13d7..b48e7117c3 100644 --- a/internal/querynodev2/services.go +++ b/internal/querynodev2/services.go @@ -878,13 +878,8 @@ func (node *QueryNode) Search(ctx context.Context, req *querypb.SearchRequest) ( return resp, nil } - tr.RecordSpan() ret.Status = merr.Success() - reduceLatency := tr.RecordSpan() - metrics.QueryNodeReduceLatency. - WithLabelValues(fmt.Sprint(node.GetNodeID()), metrics.SearchLabel, metrics.ReduceShards, metrics.BatchReduce). - Observe(float64(reduceLatency.Milliseconds())) metrics.QueryNodeExecuteCounter.WithLabelValues(strconv.FormatInt(node.GetNodeID(), 10), metrics.SearchLabel). Add(float64(proto.Size(req)))