Add readiness probe

netgroup-polito · Jan 8, 2025 · 224f768 · 224f768
1 parent 55d9bf2
commit 224f768
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 0 deletions.
diff --git a/operators/deploy/cloudimg-registry/templates/deployment.yaml b/operators/deploy/cloudimg-registry/templates/deployment.yaml
@@ -38,6 +38,12 @@ spec:
             - name: http
               containerPort: 8080
               protocol: TCP
+          readinessProbe:
+            httpGet:
+              path: /healthz
+              port: http
+            initialDelaySeconds: 3
+            periodSeconds: 3
           resources:
             {{- toYaml .Values.resources | nindent 12 }}
           volumeMounts:

diff --git a/operators/pkg/ciregistry/handlers.go b/operators/pkg/ciregistry/handlers.go
@@ -219,3 +219,15 @@ func HandleDeleteTag(log klog.Logger) usecase.Interactor {
 
 	return u
 }
+
+// HealthzHandler is used for performing readiness probes.
+func HealthzHandler() usecase.Interactor {
+	u := usecase.NewInteractor(func(_ context.Context, _ struct{}, out *BasicJSONReply) error {
+		out.Success = true
+		return nil
+	})
+
+	u.SetExpectedErrors(status.Unavailable)
+
+	return u
+}
diff --git a/operators/pkg/ciregistry/router.go b/operators/pkg/ciregistry/router.go
@@ -36,6 +36,7 @@ func NewRouter() http.Handler {
 	s.OpenAPISchema().SetDescription("API for managing cloudimage repositories and metadata.")
 	s.OpenAPISchema().SetVersion("1.0.0")
 
+	s.Get("/healthz", HealthzHandler())
 	s.Get("/{repo}", HandleGetImages(klog.LoggerWithName(log, "imagelist")))
 	s.Get("/{repo}/{image}", HandleGetImageTags(klog.LoggerWithName(log, "taglist")))
 	s.Get("/{repo}/{image}/{tag}", HandleGetImage(klog.LoggerWithName(log, "imagebin")), nethttp.SuccessfulResponseContentType("application/octet-stream"))