fix: Collect entites num group by collection instead of partition (#41788)

issue: #41787

Signed-off-by: Cai Zhang <cai.zhang@zilliz.com>
This commit is contained in:
cai.zhang 2025-05-15 12:04:22 +08:00 committed by GitHub
parent a6d09ff4cd
commit 9eebb9b464
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -116,10 +116,8 @@ func getQuotaMetrics(node *QueryNode) (*metricsinfo.QueryNodeQuotaMetrics, error
metrics.QueryNodeEntitiesSize.WithLabelValues(fmt.Sprint(node.GetNodeID()),
fmt.Sprint(collection), segments.SegmentTypeSealed.String()).Set(float64(size))
}
sealedGroupByPartition := lo.GroupBy(sealedSegments, func(seg segments.Segment) int64 {
return seg.Partition()
})
for _, segs := range sealedGroupByPartition {
for _, segs := range sealedGroupByCollection {
numEntities := lo.SumBy(segs, func(seg segments.Segment) int64 {
return seg.RowNum()
})