added health check to mock-service container

elastic · Jul 17, 2024 · 38628e5 · 38628e5
1 parent 69fddcd
commit 38628e5
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 7 deletions.
diff --git a/packages/websocket/_dev/deploy/docker/docker-compose.yml b/packages/websocket/_dev/deploy/docker/docker-compose.yml
@@ -8,4 +8,8 @@ services:
       - ./websocket-mock-service:/app
     ports:
       - "3000:3000"
+    healthcheck:
+      test: "wget --no-verbose --tries=1 --spider http://localhost:3000/health || exit 1"
+      interval: 10s
+      timeout: 5s
     command: ["go", "run", "main.go"]
diff --git a/packages/websocket/_dev/deploy/docker/websocket-mock-service/main.go b/packages/websocket/_dev/deploy/docker/websocket-mock-service/main.go
@@ -17,15 +17,10 @@ func main() {
 }
 
 func handleWebSocket(w http.ResponseWriter, r *http.Request) {
-	upgrader := websocket.Upgrader{
-		CheckOrigin: func(r *http.Request) bool { return true },
-	}
-	conn, err := upgrader.Upgrade(w, r, nil)
-	if err != nil {
-		log.Println(err)
+
+	if r.URL.Path == "/health" {
 		return
 	}
-	defer conn.Close()
 
 	if r.URL.Path == "/testbasicauth" {
 		// Check if the 'Authorization' header is set for basic authentication
@@ -38,6 +33,16 @@ func handleWebSocket(w http.ResponseWriter, r *http.Request) {
 		}
 	}
 
+	upgrader := websocket.Upgrader{
+		CheckOrigin: func(r *http.Request) bool { return true },
+	}
+	conn, err := upgrader.Upgrade(w, r, nil)
+	if err != nil {
+		log.Println(err)
+		return
+	}
+	defer conn.Close()
+
 	var responseMessage []map[string]string
 
 	if r.URL.Path == "/testbasicauth" {