Export SRT statistics as Prometheus metrics.

[nageru] / shared / metrics.cpp

diff --git a/shared/metrics.cpp b/shared/metrics.cpp
index 248da38a277992289e9ae703bd1442f16ce077e5..91b03e83483e1335d85d747dca730d607ff10438 100644 (file)
--- a/shared/metrics.cpp
+++ b/shared/metrics.cpp
@@ -106,6 +106,23 @@ void Metrics::remove(const string &name, const vector<pair<string, string>> &lab
        metrics.erase(it);
 }
 
+void Metrics::remove_if_exists(const string &name, const vector<pair<string, string>> &labels)
+{
+       lock_guard<mutex> lock(mu);
+       auto it = metrics.find(MetricKey(name, labels));
+       if (it == metrics.end()) {
+               return;
+       }
+
+       // If this is the last metric with this name, remove the type as well.
+       if (!((it != metrics.begin() && prev(it)->first.name == name) ||
+             (it != metrics.end() && next(it) != metrics.end() && next(it)->first.name == name))) {
+               types.erase(name);
+       }
+
+       metrics.erase(it);
+}
+
 string Metrics::serialize() const
 {
        stringstream ss;