Return http 500 on collection failure

corelight · Jul 30, 2024 · 412050b · 412050b
1 parent 998e7d9
commit 412050b
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/main.go b/main.go
@@ -320,6 +320,14 @@ var (
 		newCounterMetric("napatech", "dispatch_drop_packets_total", "", "pkts"),
 		newCounterMetric("napatech", "dispatch_drop_bytes_total", "", "byte"),
 	}
+
+	// Metric desc used for reporting collection failures
+	FailedCollectionDesc = prometheus.NewDesc(
+		prometheus.BuildFQName("suricata", "collection", "failure"),
+		"invalid metric for reporting collection failures",
+		[]string{},
+		nil,
+	)
 )
 
 func NewSuricataClient(socketPath string) *SuricataClient {
@@ -792,6 +800,7 @@ func (sc *suricataCollector) Collect(ch chan<- prometheus.Metric) {
 
 	if err := sc.client.EnsureConnection(); err != nil {
 		log.Printf("ERROR: Failed to connect to %v", err)
+		ch <- prometheus.NewInvalidMetric(FailedCollectionDesc, fmt.Errorf("Failed to connect"))
 		return
 	}
 
@@ -828,7 +837,9 @@ func main() {
 	r := prometheus.NewRegistry()
 	r.MustRegister(&suricataCollector{NewSuricataClient(*socketPath), sync.Mutex{}})
 
-	http.Handle(*path, promhttp.HandlerFor(r, promhttp.HandlerOpts{}))
+	http.Handle(*path, promhttp.HandlerFor(r, promhttp.HandlerOpts{
+		ErrorHandling: promhttp.HTTPErrorOnError,
+	}))
 	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
 		_, err := w.Write([]byte(`<html>
 			<head><title>Suricata Exporter</title></head>