Files
wehub-resource-sync 498b235461
Build and test / Build and test AMD64 Ubuntu 22.04 (push) Failing after 0s
Publish Builder / amazonlinux2023 (push) Failing after 1s
Build and test / UT for Go (push) Has been skipped
Publish KRTE Images / KRTE (push) Failing after 1s
Build and test / Integration Test (push) Has been skipped
Build and test / Upload Code Coverage (push) Has been skipped
Publish Builder / rockylinux9 (push) Failing after 1s
Publish Builder / ubuntu22.04 (push) Failing after 0s
Publish Builder / ubuntu24.04 (push) Failing after 0s
Publish Gpu Builder / publish-gpu-builder (push) Failing after 1s
Publish Test Images / PyTest (push) Failing after 0s
Build and test / UT for Cpp (push) Has been cancelled
chore: import upstream snapshot with attribution
2026-07-13 12:31:17 +08:00

59 lines
1.7 KiB
Go

package flusherimpl
import (
"strconv"
"github.com/prometheus/client_golang/prometheus"
"github.com/milvus-io/milvus/pkg/v3/metrics"
"github.com/milvus-io/milvus/pkg/v3/streaming/util/types"
"github.com/milvus-io/milvus/pkg/v3/util/paramtable"
"github.com/milvus-io/milvus/pkg/v3/util/tsoutil"
)
const (
flusherStateInRecovering flusherState = "in_recovery"
flusherStateInWorking flusherState = "working"
flusherStateOnClosing flusherState = "closing"
)
type flusherState = string
func newFlusherMetrics(pchannel types.PChannelInfo) *flusherMetrics {
constLabels := prometheus.Labels{
metrics.NodeIDLabelName: paramtable.GetStringNodeID(),
metrics.WALChannelLabelName: pchannel.Name,
metrics.WALChannelTermLabelName: strconv.FormatInt(pchannel.Term, 10),
}
m := &flusherMetrics{
constLabels: constLabels,
info: metrics.WALFlusherInfo.MustCurryWith(constLabels),
timetick: metrics.WALFlusherTimeTick.With(constLabels),
state: flusherStateInRecovering,
}
m.info.WithLabelValues(flusherStateInRecovering).Set(1)
return m
}
type flusherMetrics struct {
constLabels prometheus.Labels
info *prometheus.GaugeVec
timetick prometheus.Gauge
state flusherState
}
func (m *flusherMetrics) IntoState(state flusherState) {
metrics.WALFlusherInfo.DeletePartialMatch(m.constLabels)
m.state = state
m.info.WithLabelValues(m.state).Set(1)
}
func (m *flusherMetrics) ObserveMetrics(tickTime uint64) {
m.timetick.Set(tsoutil.PhysicalTimeSeconds(tickTime))
}
func (m *flusherMetrics) Close() {
metrics.WALFlusherInfo.DeletePartialMatch(m.constLabels)
metrics.WALFlusherTimeTick.DeletePartialMatch(m.constLabels)
}