add checks

hatchet-dev · Feb 29, 2024 · 21ff722 · 21ff722
1 parent 207196b
commit 21ff722
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/api/v1/server/handlers/metadata/health.go b/api/v1/server/handlers/metadata/health.go
@@ -1,6 +1,8 @@
 package metadata
 
 import (
+	"fmt"
+
 	"github.com/labstack/echo/v4"
 
 	"github.com/hatchet-dev/hatchet/api/v1/server/oas/gen"
@@ -11,6 +13,13 @@ func (u *MetadataService) LivenessGet(ctx echo.Context, request gen.LivenessGetR
 }
 
 func (u *MetadataService) ReadinessGet(ctx echo.Context, request gen.ReadinessGetRequestObject) (gen.ReadinessGetResponseObject, error) {
-	// TODO check if db and queue are ready
+	if !u.config.Repository.Health().IsHealthy() {
+		return nil, fmt.Errorf("repository is not healthy")
+	}
+
+	if !u.config.TaskQueue.IsReady() {
+		return nil, fmt.Errorf("repository is not healthy")
+	}
+
 	return gen.ReadinessGet200Response{}, nil
 }