Add health check

priobike · Jun 18, 2024 · 39ae948 · 39ae948
1 parent 6e033cf
commit 39ae948
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -1,4 +1,5 @@
 FROM golang
 COPY main.go /go/src/main.go
 RUN go build -o /go/bin/main /go/src/main.go
+HEALTHCHECK --interval=30s --timeout=30s --start-period=5s --retries=3 CMD curl -f http://localhost:8000/health || exit 1
 CMD ["/go/bin/main"]
diff --git a/main.go b/main.go
@@ -248,6 +248,12 @@ func checkAndProxy(w http.ResponseWriter, r *http.Request) {
 		log.Printf("Request: %s %s\n", r.Method, r.URL.Path)
 	}
 
+	// Health check endpoint
+	if r.URL.Path == "/health" {
+		w.WriteHeader(http.StatusOK)
+		return
+	}
+
 	if !checkRequest(w, r) {
 		http.Error(w, "Invalid", http.StatusBadRequest)
 		return