delete old metrics in puma collector

didww · Jul 9, 2019 · 809b382 · 809b382
1 parent 3095996
commit 809b382
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/lib/prometheus_exporter/server/puma_collector.rb b/lib/prometheus_exporter/server/puma_collector.rb
@@ -2,6 +2,9 @@
 
 module PrometheusExporter::Server
   class PumaCollector < TypeCollector
+
+    MAX_PUMA_METRIC_AGE = 30
+
     PUMA_GAUGES = {
       workers_total: "Number of puma workers.",
       booted_workers_total: "Number of puma workers booted.",
@@ -47,7 +50,16 @@ def metrics
     end
 
     def collect(obj)
-      @puma_metrics << obj
+      now = ::Process.clock_gettime(::Process::CLOCK_MONOTONIC)
+
+      obj["created_at"] = now
+
+      @puma_metrics .delete_if do |current|
+        (obj["pid"] == current["pid"] && obj["hostname"] == current["hostname"]) ||
+            (current["created_at"] + MAX_PUMA_METRIC_AGE < now)
+      end
+
+      @puma_metrics  << obj
     end
   end
 end