From ea40540f5cdf880fd08facf0c7de42c9900d2a3e Mon Sep 17 00:00:00 2001
From: RainbowMango <renhongcai@huawei.com>
Date: Wed, 13 Nov 2019 19:52:40 +0800
Subject: [PATCH] Add Reset() API to stability framework

---
 staging/src/k8s.io/component-base/metrics/counter.go   | 9 +++++++++
 staging/src/k8s.io/component-base/metrics/gauge.go     | 9 +++++++++
 staging/src/k8s.io/component-base/metrics/histogram.go | 9 +++++++++
 staging/src/k8s.io/component-base/metrics/summary.go   | 9 +++++++++
 4 files changed, 36 insertions(+)

diff --git a/staging/src/k8s.io/component-base/metrics/counter.go b/staging/src/k8s.io/component-base/metrics/counter.go
index ccc1f084b6b1..d641b0782993 100644
--- a/staging/src/k8s.io/component-base/metrics/counter.go
+++ b/staging/src/k8s.io/component-base/metrics/counter.go
@@ -159,3 +159,12 @@ func (v *CounterVec) Delete(labels map[string]string) bool {
 	}
 	return v.CounterVec.Delete(labels)
 }
+
+// Reset deletes all metrics in this vector.
+func (v *CounterVec) Reset() {
+	if !v.IsCreated() {
+		return
+	}
+
+	v.CounterVec.Reset()
+}
diff --git a/staging/src/k8s.io/component-base/metrics/gauge.go b/staging/src/k8s.io/component-base/metrics/gauge.go
index 5c997443b73a..c9c187b7b004 100644
--- a/staging/src/k8s.io/component-base/metrics/gauge.go
+++ b/staging/src/k8s.io/component-base/metrics/gauge.go
@@ -161,6 +161,15 @@ func (v *GaugeVec) Delete(labels map[string]string) bool {
 	return v.GaugeVec.Delete(labels)
 }
 
+// Reset deletes all metrics in this vector.
+func (v *GaugeVec) Reset() {
+	if !v.IsCreated() {
+		return
+	}
+
+	v.GaugeVec.Reset()
+}
+
 func newGaugeFunc(opts GaugeOpts, function func() float64, v semver.Version) GaugeFunc {
 	g := NewGauge(&opts)
 
diff --git a/staging/src/k8s.io/component-base/metrics/histogram.go b/staging/src/k8s.io/component-base/metrics/histogram.go
index 1b2de88fc227..1a4bbda813ef 100644
--- a/staging/src/k8s.io/component-base/metrics/histogram.go
+++ b/staging/src/k8s.io/component-base/metrics/histogram.go
@@ -166,3 +166,12 @@ func (v *HistogramVec) Delete(labels map[string]string) bool {
 	}
 	return v.HistogramVec.Delete(labels)
 }
+
+// Reset deletes all metrics in this vector.
+func (v *HistogramVec) Reset() {
+	if !v.IsCreated() {
+		return
+	}
+
+	v.HistogramVec.Reset()
+}
diff --git a/staging/src/k8s.io/component-base/metrics/summary.go b/staging/src/k8s.io/component-base/metrics/summary.go
index cad4ce633767..3d8da0064597 100644
--- a/staging/src/k8s.io/component-base/metrics/summary.go
+++ b/staging/src/k8s.io/component-base/metrics/summary.go
@@ -160,3 +160,12 @@ func (v *SummaryVec) Delete(labels map[string]string) bool {
 	}
 	return v.SummaryVec.Delete(labels)
 }
+
+// Reset deletes all metrics in this vector.
+func (v *SummaryVec) Reset() {
+	if !v.IsCreated() {
+		return
+	}
+
+	v.SummaryVec.Reset()
+}