Health check

charts/clamav/templates/clamav-deployment.yaml

@@ -36,15 +36,15 @@ spec:
           protocol: TCP
         livenessProbe:
           httpGet:
-            path: /healthz
+            path: /v1alpha/healthz
             port: 8080
           initialDelaySeconds: 10
           timeoutSeconds: 5
           periodSeconds: 5
           failureThreshold: 10
         readinessProbe:
           httpGet:
-            path: /healthz
+            path: /v1alpha/healthz
             port: 8080
           initialDelaySeconds: 10
           timeoutSeconds: 5

clamav-http/server/server.go

@@ -32,6 +32,10 @@ func RunHTTPListener(clamd_address string, port int, max_file_mem int64, logger
 		Max_file_mem: max_file_mem,
 		Logger:       logger,
 	})
+	m.Handle("/v1alpha/healthz", &v1alpha.HealthHandler{
+		Address: clamd_address,
+		Logger:  logger,
+	})
 	m.Handle("/v1alpha/scan", &v1alpha.ScanHandler{
 		Address:      clamd_address,
 		Max_file_mem: max_file_mem,

clamav-http/server/v1alpha/health_handler.go

@@ -0,0 +1,39 @@
+package v1alpha
+
+import (
+	"net/http"
+	"strings"
+
+	"github.com/dutchcoders/go-clamd"
+	"github.com/sirupsen/logrus"
+)
+
+const eicar = "X5O!P%@AP[4\\PZX54(P^)7CC)7}$EICAR-STANDARD-ANTIVIRUS-TEST-FILE!$H+H*"
+
+type HealthHandler struct {
+	Address string
+	Logger  *logrus.Logger
+}
+
+func (hh *HealthHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
+	c := clamd.NewClamd(hh.Address)
+	response, err := c.ScanStream(strings.NewReader(eicar), make(chan bool))
+
+	if err != nil {
+		w.WriteHeader(http.StatusInternalServerError)
+		w.Write([]byte("not okay"))
+	}
+
+	result := <-response
+	if result.Status == "FOUND" {
+		w.WriteHeader(http.StatusOK)
+		hh.Logger.Info("healthcheck: OK")
+		w.Write([]byte("healthcheck: OK\n"))
+	} else {
+		w.WriteHeader(http.StatusInternalServerError)
+		hh.Logger.Info("healthcheck: ERROR")
+		w.Write([]byte("healthcheck: ERROR\n"))
+	}
+
+	return
+}