From 27c7cbbc72e38a9ddf500c6f4b11a09fb6eaac24 Mon Sep 17 00:00:00 2001 From: "yihao.dai" Date: Wed, 12 Mar 2025 19:08:05 +0800 Subject: [PATCH] fix: Fix QueryNodeNumEntities metric (#40602) fix QueryNodeNumEntities metric introduced by pr https://github.com/milvus-io/milvus/pull/39536 issue: https://github.com/milvus-io/milvus/issues/38162 Signed-off-by: bigsheeper --- internal/querynodev2/metrics_info.go | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/internal/querynodev2/metrics_info.go b/internal/querynodev2/metrics_info.go index 57f98606a6..36514a9967 100644 --- a/internal/querynodev2/metrics_info.go +++ b/internal/querynodev2/metrics_info.go @@ -92,11 +92,8 @@ func getQuotaMetrics(node *QueryNode) (*metricsinfo.QueryNodeQuotaMetrics, error metrics.QueryNodeEntitiesSize.WithLabelValues(nodeID, fmt.Sprint(collection), segments.SegmentTypeGrowing.String()).Set(float64(size)) } - growingGroupByPartition := lo.GroupBy(growingSegments, func(seg segments.Segment) int64 { - return seg.Partition() - }) - for _, segs := range growingGroupByPartition { + for _, segs := range growingGroupByCollection { numEntities := lo.SumBy(segs, func(seg segments.Segment) int64 { return seg.RowNum() })