mirror of
https://gitee.com/milvus-io/milvus.git
synced 2026-01-07 03:13:22 +08:00
a86b6f2a54
issue: #42416 - Rename the InsertMetric into ModifiedMetric. - Add L0 control configuration. - Add some L0 current state collect. Signed-off-by: chyezh <chyezh@outlook.com>
51 lines
2.2 KiB
Go
51 lines
2.2 KiB
Go
package stats
|
|
|
|
import (
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
|
|
"github.com/milvus-io/milvus/pkg/v2/metrics"
|
|
"github.com/milvus-io/milvus/pkg/v2/proto/datapb"
|
|
"github.com/milvus-io/milvus/pkg/v2/util/paramtable"
|
|
)
|
|
|
|
// newMetricsHelper creates a new metrics helper for the WAL segment.
|
|
func newMetricsHelper() *metricsHelper {
|
|
return &metricsHelper{
|
|
growingBytesHWM: metrics.WALGrowingSegmentHWMBytes.With(prometheus.Labels{metrics.NodeIDLabelName: paramtable.GetStringNodeID()}),
|
|
growingBytesLWM: metrics.WALGrowingSegmentLWMBytes.With(prometheus.Labels{metrics.NodeIDLabelName: paramtable.GetStringNodeID()}),
|
|
growingBytes: metrics.WALGrowingSegmentBytes.MustCurryWith(prometheus.Labels{metrics.NodeIDLabelName: paramtable.GetStringNodeID()}),
|
|
growingRowsTotal: metrics.WALGrowingSegmentRowsTotal.MustCurryWith(prometheus.Labels{metrics.NodeIDLabelName: paramtable.GetStringNodeID()}),
|
|
}
|
|
}
|
|
|
|
// metricsHelper is a helper struct for managing metrics related to WAL segments.
|
|
type metricsHelper struct {
|
|
growingBytesHWM prometheus.Gauge
|
|
growingBytesLWM prometheus.Gauge
|
|
growingBytes *prometheus.GaugeVec
|
|
growingRowsTotal *prometheus.GaugeVec
|
|
}
|
|
|
|
// ObservePChannelBytesUpdate updates the bytes of a pchannel.
|
|
func (m *metricsHelper) ObservePChannelBytesUpdate(pchannel string, am *aggregatedMetrics) {
|
|
for _, lv := range []datapb.SegmentLevel{datapb.SegmentLevel_L0, datapb.SegmentLevel_L1} {
|
|
metric := am.Get(lv)
|
|
if metric.BinarySize <= 0 {
|
|
metrics.WALGrowingSegmentBytes.DeletePartialMatch(prometheus.Labels{metrics.WALChannelLabelName: pchannel, metrics.WALSegmentLevelLabelName: lv.String()})
|
|
} else {
|
|
m.growingBytes.WithLabelValues(pchannel, lv.String()).Set(float64(metric.BinarySize))
|
|
}
|
|
if metric.Rows <= 0 {
|
|
metrics.WALGrowingSegmentRowsTotal.DeletePartialMatch(prometheus.Labels{metrics.WALChannelLabelName: pchannel, metrics.WALSegmentLevelLabelName: lv.String()})
|
|
} else {
|
|
m.growingRowsTotal.WithLabelValues(pchannel, lv.String()).Set(float64(metric.Rows))
|
|
}
|
|
}
|
|
}
|
|
|
|
// ObserveConfigUpdate is a update method for configuration changes.
|
|
func (m *metricsHelper) ObserveConfigUpdate(cfg statsConfig) {
|
|
m.growingBytesHWM.Set(float64(cfg.growingBytesHWM))
|
|
m.growingBytesLWM.Set(float64(cfg.growingBytesLWM))
|
|
}
|