add /health endpoint

README.md

@@ -27,8 +27,6 @@ Usage of ./build/cspr-collector:
         enable http output
   -output-http-host string
         http host to send the csp violations to (default "http://localhost:80/")
-  -output-http-headers string
-        additional headers for HTTP output
   -output-http-headers string
         header header key with value to send with the http request. Example "Authorization: ApiKey <APIKEY>"
   -output-stdout
@@ -102,6 +100,12 @@ curl -X POST \
 go fmt ./...
 ```
 
+## Health check endpoint
+endpoint with health response "OK"
+```bash
+curl http://localhost:8080/health
+```
+
 ## License
 
 The MIT License (MIT). Please see the [license file](LICENSE.md) for more information.

collector.go

@@ -19,6 +19,11 @@ type Collector struct {
 }
 
 func (c *Collector) ServeHTTP(w http.ResponseWriter, r *http.Request) {
+	if r.RequestURI == "/health" {
+		c.handleHealth(w)
+		return
+	}
+
 	if r.RequestURI != "/" {
 		message := fmt.Sprintf("Path \"%s\" not found.", r.RequestURI)
 		c.response(w, http.StatusNotFound, message)
@@ -58,7 +63,6 @@ func (c *Collector) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 	c.WorkQueue <- data
 
 	c.response(w, http.StatusCreated, "")
-	return
 }
 
 func (c *Collector) response(w http.ResponseWriter, status int, message string) {
@@ -82,3 +86,8 @@ func (c *Collector) response(w http.ResponseWriter, status int, message string)
 		log.Fatal(err)
 	}
 }
+
+func (c *Collector) handleHealth(w http.ResponseWriter) {
+	w.WriteHeader(http.StatusOK)
+	w.Write([]byte("OK"))
+}

dispatcher.go

@@ -31,15 +31,20 @@ func (d *Dispatcher) Run() {
 }
 
 func (d *Dispatcher) start() {
-	for {
-		select {
-		case work := <-d.WorkQueue:
-			log.Print("Received work request.")
-			go func() {
-				worker := <-d.WorkerQueue
-				log.Print("Dispatching work request.")
-				worker <- work
-			}()
-		}
+	for work := range d.WorkQueue {
+		log.Print("Received work request.")
+		go func(w interface{}) {
+			worker := <-d.WorkerQueue
+			log.Print("Dispatching work request.")
+
+			// Perform a type assertion
+			typedWork, ok := w.(CSPRequest)
+			if !ok {
+				log.Print("Invalid work request type")
+				return
+			}
+
+			worker <- typedWork
+		}(work)
 	}
 }