From 6972efe6baaee02fbf9162aa05d235b7987d0908 Mon Sep 17 00:00:00 2001
From: wgcn <1026688210@qq.com>
Date: Tue, 27 Sep 2022 19:40:54 +0800
Subject: [PATCH] fix reduce latency metric (#19479)

Signed-off-by: wgcn <1026688210@qq.com>

Signed-off-by: wgcn <1026688210@qq.com>
---
 internal/querynode/task_search.go | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/internal/querynode/task_search.go b/internal/querynode/task_search.go
index 9fd93bbfe6..6c6b8e5f67 100644
--- a/internal/querynode/task_search.go
+++ b/internal/querynode/task_search.go
@@ -231,7 +231,11 @@ func (s *searchTask) reduceResults(ctx context.Context, searchReq *searchRequest
 			log.Ctx(ctx).Warn("marshal for historical results error", zap.Int64("msgID", s.ID()), zap.Error(err))
 			return err
 		}
-		defer deleteSearchResultDataBlobs(blobs)
+
+		defer func() {
+			deleteSearchResultDataBlobs(blobs)
+			s.reduceDur = s.tr.RecordSpan()
+		}()
 
 		for i := 0; i < cnt; i++ {
 			blob, err := getSearchResultDataBlob(blobs, i)
@@ -274,8 +278,9 @@ func (s *searchTask) reduceResults(ctx context.Context, searchReq *searchRequest
 				SlicedNumCount: 1,
 			}
 		}
+
+		s.reduceDur = s.tr.RecordSpan()
 	}
-	s.reduceDur = s.tr.RecordSpan()
 	return nil
 }