feat(helm): add liveness probe to the controller, proxy and registry

enix · May 24, 2024 · bb2ee9b · bb2ee9b
1 parent 3879969
commit bb2ee9b
Show file tree

Hide file tree

Showing 5 changed files with 35 additions and 0 deletions.
diff --git a/helm/kube-image-keeper/templates/controller-deployment.yaml b/helm/kube-image-keeper/templates/controller-deployment.yaml
@@ -91,6 +91,10 @@ spec:
           readinessProbe:
             {{- toYaml . | nindent 12 }}
           {{- end }}
+          {{- with .Values.controllers.livenessProbe }}
+          livenessProbe:
+            {{- toYaml . | nindent 12 }}
+          {{- end }}
           {{- with .Values.controllers.resources }}
           resources:
             {{- toYaml . | nindent 12 }}

diff --git a/helm/kube-image-keeper/templates/proxy-daemonset.yaml b/helm/kube-image-keeper/templates/proxy-daemonset.yaml
@@ -95,6 +95,14 @@ spec:
           readinessProbe:
             {{- $readinessProbe | toYaml | nindent 12 }}
           {{- end }}
+          {{- $livenessProbe := deepCopy .Values.proxy.livenessProbe }}
+          {{- if .Values.proxy.hostNetwork }}
+            {{- $livenessProbe := merge $livenessProbe.httpGet (dict "host" "localhost") }}
+          {{- end }}
+          {{- with .Values.proxy.livenessProbe }}
+          livenessProbe:
+            {{- $livenessProbe | toYaml | nindent 12 }}
+          {{- end }}
           {{- with .Values.proxy.resources }}
           resources:
             {{- toYaml . | nindent 12 }}

diff --git a/helm/kube-image-keeper/templates/registry-deployment.yaml b/helm/kube-image-keeper/templates/registry-deployment.yaml
@@ -94,6 +94,10 @@ spec:
           readinessProbe:
             {{- toYaml . | nindent 12 }}
           {{- end }}
+          {{- with .Values.registry.livenessProbe }}
+          livenessProbe:
+            {{- toYaml . | nindent 12 }}
+          {{- end }}
       {{- with .Values.registry.nodeSelector }}
       nodeSelector:
         {{- toYaml . | nindent 8 }}

diff --git a/helm/kube-image-keeper/templates/registry-statefulset.yaml b/helm/kube-image-keeper/templates/registry-statefulset.yaml
@@ -97,6 +97,10 @@ spec:
           readinessProbe:
             {{- toYaml . | nindent 12 }}
           {{- end }}
+          {{- with .Values.registry.livenessProbe }}
+          livenessProbe:
+            {{- toYaml . | nindent 12 }}
+          {{- end }}
       {{- with .Values.registry.nodeSelector }}
       nodeSelector:
         {{- toYaml . | nindent 8 }}

diff --git a/helm/kube-image-keeper/values.yaml b/helm/kube-image-keeper/values.yaml
@@ -57,6 +57,11 @@ controllers:
     httpGet:
       path: /readyz
       port: 8081
+  # -- Liveness probe definition for the controllers pod
+  livenessProbe:
+    httpGet:
+      path: /healthz
+      port: 8081
   resources:
     requests:
       # -- Cpu requests for the controller pod
@@ -164,6 +169,11 @@ proxy:
     httpGet:
       path: /v2/
       port: 7439
+  # -- Liveness probe definition for the proxy pod
+  livenessProbe:
+    httpGet:
+      path: /v2/
+      port: 7439
   resources:
     requests:
       # -- Cpu requests for the proxy pod
@@ -239,6 +249,11 @@ registry:
     httpGet:
       path: /v2/
       port: 5000
+    # -- Liveness probe definition for the proxy pod
+  livenessProbe:
+    httpGet:
+      path: /v2/
+      port: 5000
   resources:
     requests:
       # -- Cpu requests for the registry pod