From e5a16050cecabbaad9fbcdca94bd39d7d3685682 Mon Sep 17 00:00:00 2001 From: congqixia Date: Tue, 20 Feb 2024 22:14:53 +0800 Subject: [PATCH] fix: Update disk usage metrics after segment released (#30702) See also #30701 Signed-off-by: Congqi Xia --- internal/querynodev2/segments/segment.go | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/internal/querynodev2/segments/segment.go b/internal/querynodev2/segments/segment.go index 3e386ec58e..e8564b9ec8 100644 --- a/internal/querynodev2/segments/segment.go +++ b/internal/querynodev2/segments/segment.go @@ -1264,6 +1264,12 @@ func (s *LocalSegment) Release() { strconv.FormatInt(int64(len(s.Indexes())), 10), ).Sub(float64(s.InsertCount())) + localDiskUsage, err := GetLocalUsedSize(context.Background(), paramtable.Get().LocalStorageCfg.Path.GetValue()) + // ignore error here, shall not block releasing + if err == nil { + metrics.QueryNodeDiskUsedSize.WithLabelValues(fmt.Sprint(paramtable.GetNodeID())).Set(float64(localDiskUsage) / 1024 / 1024) // in MB + } + log.Info("delete segment from memory", zap.Int64("collectionID", s.collectionID), zap.Int64("partitionID", s.partitionID),