Skip to content

Commit 370562b

Browse files
authored
fix: fix partition loaded num metric (milvus-io#33316)
issue: milvus-io#32108 Signed-off-by: sunby <sunbingyi1992@gmail.com>
1 parent 3c7d020 commit 370562b

1 file changed

Lines changed: 7 additions & 2 deletions

File tree

internal/querynodev2/segments/manager.go

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,7 @@ import (
3636
"github.com/milvus-io/milvus/internal/proto/datapb"
3737
"github.com/milvus-io/milvus/internal/proto/querypb"
3838
"github.com/milvus-io/milvus/internal/querynodev2/segments/metricsutil"
39+
"github.com/milvus-io/milvus/pkg/common"
3940
"github.com/milvus-io/milvus/pkg/eventlog"
4041
"github.com/milvus-io/milvus/pkg/log"
4142
"github.com/milvus-io/milvus/pkg/metrics"
@@ -723,11 +724,15 @@ func (mgr *segmentManager) updateMetric() {
723724
collections, partiations := make(typeutil.Set[int64]), make(typeutil.Set[int64])
724725
for _, seg := range mgr.growingSegments {
725726
collections.Insert(seg.Collection())
726-
partiations.Insert(seg.Partition())
727+
if seg.Partition() != common.AllPartitionsID {
728+
partiations.Insert(seg.Partition())
729+
}
727730
}
728731
for _, seg := range mgr.sealedSegments {
729732
collections.Insert(seg.Collection())
730-
partiations.Insert(seg.Partition())
733+
if seg.Partition() != common.AllPartitionsID {
734+
partiations.Insert(seg.Partition())
735+
}
731736
}
732737
metrics.QueryNodeNumCollections.WithLabelValues(fmt.Sprint(paramtable.GetNodeID())).Set(float64(collections.Len()))
733738
metrics.QueryNodeNumPartitions.WithLabelValues(fmt.Sprint(paramtable.GetNodeID())).Set(float64(partiations.Len()))

0 commit comments

Comments
 (0)