feat: add ayd_incident_count to Prometheus metrics

macrat · Apr 23, 2021 · 7c046a9 · 7c046a9
1 parent 5770c55
commit 7c046a9
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/exporter/metrics.go b/exporter/metrics.go
@@ -32,5 +32,9 @@ func MetricsExporter(s *store.Store) http.HandlerFunc {
 				fmt.Fprintln(w)
 			}
 		}
+		fmt.Fprintln(w)
+		fmt.Fprintln(w, "# HELP ayd_incident_count The number of incident happend since server started")
+		fmt.Fprintln(w, "# TYPE ayd_incident_count Counter")
+		fmt.Fprintf(w, "ayd_incident_count %d\n", s.IncidentCount)
 	}
 }
diff --git a/store/store.go b/store/store.go
@@ -65,7 +65,8 @@ type Store struct {
 	CurrentIncidents []*Incident
 	IncidentHistory  []*Incident
 
-	OnIncident []IncidentHandler
+	OnIncident    []IncidentHandler
+	IncidentCount int
 
 	file      *os.File
 	lastError error
@@ -115,6 +116,7 @@ func (s *Store) setIncidentIfNeed(r Record, needCallback bool) {
 		s.CurrentIncidents = append(s.CurrentIncidents, incident)
 
 		if needCallback {
+			s.IncidentCount++
 			for _, cb := range s.OnIncident {
 				s.appendWithoutLock(cb(incident))
 			}