fix: readiness check for ingesters and frontend (#3435)

grafana · Jul 22, 2024 · feefa81 · feefa81
1 parent 336b7dd
commit feefa81
Show file tree

Hide file tree

Showing 4 changed files with 23 additions and 8 deletions.
diff --git a/Makefile b/Makefile
@@ -77,13 +77,10 @@ build: frontend/build go/bin ## Do a production build (requiring the frontend bu
 build-dev: ## Do a dev build (without requiring the frontend)
 	$(MAKE) EMBEDASSETS="" go/bin
 
-.PHONY: frontend/build
-frontend/build: frontend/deps ## Do a production build for the frontend
-	yarn build
 
-.PHONY: frontend/deps
-frontend/deps:
-	yarn --frozen-lockfile
+.PHONY: frontend/build
+frontend/build:
+	docker build  -f cmd/pyroscope/frontend.Dockerfile --output=public/build .
 
 .PHONY: release
 release/prereq: $(BIN)/goreleaser ## Ensure release pre requesites are met

diff --git a/cmd/pyroscope/frontend.Dockerfile b/cmd/pyroscope/frontend.Dockerfile
@@ -1,4 +1,4 @@
-FROM node:18 as builder
+FROM node:18 AS builder
 RUN apt-get update && apt-get install -y libpango1.0-dev libcairo2-dev
 WORKDIR /pyroscope
 COPY yarn.lock package.json tsconfig.json ./

diff --git a/pkg/phlare/modules.go b/pkg/phlare/modules.go
@@ -109,6 +109,7 @@ func (f *Phlare) initQueryFrontend() (services.Service, error) {
 	f.API.RegisterPyroscopeHandlers(frontendSvc)
 	f.API.RegisterQueryFrontend(frontendSvc)
 	f.API.RegisterQuerier(frontendSvc)
+	f.frontend = frontendSvc
 
 	return frontendSvc, nil
 }
@@ -408,6 +409,7 @@ func (f *Phlare) initIngester() (_ services.Service, err error) {
 	}
 
 	f.API.RegisterIngester(svc)
+	f.ingester = svc
 
 	return svc, nil
 }

diff --git a/pkg/phlare/phlare.go b/pkg/phlare/phlare.go
@@ -239,7 +239,9 @@ type Phlare struct {
 
 	grpcGatewayMux *grpcgw.ServeMux
 
-	auth connect.Option
+	auth     connect.Option
+	ingester *ingester.Ingester
+	frontend *frontend.Frontend
 }
 
 func New(cfg Config) (*Phlare, error) {
@@ -502,6 +504,20 @@ func (f *Phlare) readyHandler(sm *services.Manager) http.HandlerFunc {
 			return
 		}
 
+		if f.ingester != nil {
+			if err := f.ingester.CheckReady(r.Context()); err != nil {
+				http.Error(w, "Ingester not ready: "+err.Error(), http.StatusServiceUnavailable)
+				return
+			}
+		}
+
+		if f.frontend != nil {
+			if err := f.frontend.CheckReady(r.Context()); err != nil {
+				http.Error(w, "Query Frontend not ready: "+err.Error(), http.StatusServiceUnavailable)
+				return
+			}
+		}
+
 		util.WriteTextResponse(w, "ready")
 	}
 }