register / remove metrics of dumpling

pingcap · Dec 31, 2020 · 1f24a99 · 1f24a99
1 parent 5f55bf3
commit 1f24a99
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/dumpling/metrics.go b/dumpling/metrics.go
@@ -14,6 +14,7 @@
 package dumpling
 
 import (
+	"github.com/pingcap/dumpling/v4/export"
 	"github.com/prometheus/client_golang/prometheus"
 
 	"github.com/pingcap/dm/pkg/metricsproxy"
@@ -33,8 +34,10 @@ var (
 // RegisterMetrics registers metrics.
 func RegisterMetrics(registry *prometheus.Registry) {
 	registry.MustRegister(dumplingExitWithErrorCounter)
+	export.RegisterMetrics(registry)
 }
 
 func (m *Dumpling) removeLabelValuesWithTaskInMetrics(task string) {
 	dumplingExitWithErrorCounter.DeleteAllAboutLabels(prometheus.Labels{"task": task})
+	export.RemoveLabelValuesWithTaskInMetrics(prometheus.Labels{"task": task})
 }