From 83e84865be44bf57ce28e548ed6b531800a5c699 Mon Sep 17 00:00:00 2001
From: zhuhao <37029601+hwzhuhao@users.noreply.github.com>
Date: Wed, 4 Sep 2024 14:25:46 +0800
Subject: [PATCH] feat: add health check for pg and redis in
 docker-compose.middleware.yaml (#7961) (#7962)

---
 docker/docker-compose.middleware.yaml | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/docker/docker-compose.middleware.yaml b/docker/docker-compose.middleware.yaml
index 9a4c40448bf112..adaecc69ec3f1d 100644
--- a/docker/docker-compose.middleware.yaml
+++ b/docker/docker-compose.middleware.yaml
@@ -19,6 +19,11 @@ services:
       - ./volumes/db/data:/var/lib/postgresql/data
     ports:
       - "${EXPOSE_POSTGRES_PORT:-5432}:5432"
+    healthcheck:
+      test: [ "CMD", "pg_isready" ]
+      interval: 1s
+      timeout: 3s
+      retries: 30
 
   # The redis cache.
   redis:
@@ -31,6 +36,8 @@ services:
     command: redis-server --requirepass difyai123456
     ports:
       - "${EXPOSE_REDIS_PORT:-6379}:6379"
+    healthcheck:
+      test: [ "CMD", "redis-cli", "ping" ]
 
   # The DifySandbox
   sandbox: