fix: lost metrics of flowgraph and write buffer (#42549)

issue: #42548

Signed-off-by: chyezh <chyezh@outlook.com>
This commit is contained in:
Zhen Ye 2025-06-06 19:08:31 +08:00 committed by GitHub
parent 8511881d3f
commit f81652ebbc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 14 additions and 0 deletions

View File

@ -18,6 +18,7 @@ package metrics
import ( import (
"fmt" "fmt"
"sync"
"github.com/prometheus/client_golang/prometheus" "github.com/prometheus/client_golang/prometheus"
@ -336,8 +337,17 @@ var (
}, []string{nodeIDLabelName, "type"}) }, []string{nodeIDLabelName, "type"})
) )
var registerDNOnce sync.Once
// RegisterDataNode registers DataNode metrics // RegisterDataNode registers DataNode metrics
func RegisterDataNode(registry *prometheus.Registry) { func RegisterDataNode(registry *prometheus.Registry) {
registerDNOnce.Do(func() {
registerDataNodeOnce(registry)
})
}
// registerDataNodeOnce registers DataNode metrics
func registerDataNodeOnce(registry *prometheus.Registry) {
registry.MustRegister(DataNodeNumFlowGraphs) registry.MustRegister(DataNodeNumFlowGraphs)
// input related // input related
registry.MustRegister(DataNodeConsumeMsgRowsCount) registry.MustRegister(DataNodeConsumeMsgRowsCount)

View File

@ -496,6 +496,10 @@ func RegisterStreamingNode(registry *prometheus.Registry) {
registry.MustRegister(StreamingNodeConsumeBytes) registry.MustRegister(StreamingNodeConsumeBytes)
registerWAL(registry) registerWAL(registry)
// TODO: after remove the implementation of old data node
// Such as flowgraph and writebuffer, we can remove these metrics from streaming node.
RegisterDataNode(registry)
} }
// registerWAL registers wal metrics // registerWAL registers wal metrics