Merge pull request #3371 from askeipx/issues/3341

delete disk_size metrics when collection deleted

1 files changed, 19 insertions, 0 deletions

diff --git a/weed/stats/metrics.go b/weed/stats/metrics.go
index f0b810608..a2fd373af 100644
--- a/weed/stats/metrics.go
+++ b/weed/stats/metrics.go
@@ -17,6 +17,16 @@ import (
 	"github.com/prometheus/client_golang/prometheus/push"
 )
 
+// Readonly volume types
+const (
+	IsReadOnly       = "IsReadOnly"
+	NoWriteOrDelete  = "noWriteOrDelete"
+	NoWriteCanDelete = "noWriteCanDelete"
+	IsDiskSpaceLow   = "isDiskSpaceLow"
+)
+
+var readOnlyVolumeTypes = [4]string{IsReadOnly, NoWriteOrDelete, NoWriteCanDelete, IsDiskSpaceLow}
+
 var (
 	Gather = prometheus.NewRegistry()
 
@@ -249,3 +259,12 @@ func SourceName(port uint32) string {
 	}
 	return net.JoinHostPort(hostname, strconv.Itoa(int(port)))
 }
+
+// todo - can be changed to DeletePartialMatch when https://github.com/prometheus/client_golang/pull/1013 gets released
+func DeleteCollectionMetrics(collection string) {
+	VolumeServerDiskSizeGauge.DeleteLabelValues(collection, "normal")
+	for _, volume_type := range readOnlyVolumeTypes {
+		VolumeServerReadOnlyVolumeGauge.DeleteLabelValues(collection, volume_type)
+	}
+	VolumeServerVolumeCounter.DeleteLabelValues(collection, "volume")
+}