feat: add customizable probes (#111)

databendlabs · Oct 30, 2024 · ebc335e · ebc335e
1 parent d7dd928
commit ebc335e
Show file tree

Hide file tree

Showing 6 changed files with 32 additions and 8 deletions.
diff --git a/charts/databend-meta/Chart.yaml b/charts/databend-meta/Chart.yaml
@@ -26,7 +26,7 @@ type: application
 # This is the chart version. This version number should be incremented each time you make changes
 # to the chart and its templates, including the app version.
 # Versions are expected to follow Semantic Versioning (https://semver.org/)
-version: 0.9.5
+version: 0.9.6
 
 # This is the version number of the application being deployed. This version number should be
 # incremented each time you make changes to the application. Versions are not expected to

diff --git a/charts/databend-meta/templates/statefulset.yaml b/charts/databend-meta/templates/statefulset.yaml
@@ -94,17 +94,24 @@ spec:
             {{- end }}
             - name: raft
               containerPort: 28004
+          {{- if .Values.livenessProbe.enabled }}
           livenessProbe:
             httpGet:
               path: /v1/health
               port: admin
+            initialDelaySeconds: {{ default 5 .Values.livenessProbe.initialDelaySeconds }}
+            periodSeconds: {{ default 15 .Values.livenessProbe.initialDelaySeconds }}
+            failureThreshold: {{ default 3 .Values.livenessProbe.initialDelaySeconds }}
+          {{- end }}
+          {{- if .Values.readinessProbe.enabled }}
           readinessProbe:
             httpGet:
               path: /v1/health
               port: admin
-            initialDelaySeconds: 5
-            periodSeconds: 15
-            failureThreshold: 3
+            initialDelaySeconds: {{ default 5 .Values.readinessProbe.initialDelaySeconds }}
+            periodSeconds: {{ default 15 .Values.readinessProbe.initialDelaySeconds }}
+            failureThreshold: {{ default 3 .Values.readinessProbe.initialDelaySeconds }}
+          {{- end }}
           resources:
             {{- toYaml .Values.resources | nindent 12 }}
           env:

diff --git a/charts/databend-meta/values.yaml b/charts/databend-meta/values.yaml
@@ -33,6 +33,11 @@ serviceAccount:
   # If not set and create is true, a name is generated using the fullname template
   name: ""
 
+livenessProbe:
+  enabled: true
+readinessProbe:
+  enabled: true
+
 service:
   type: ClusterIP
   ports:

diff --git a/charts/databend-query/Chart.yaml b/charts/databend-query/Chart.yaml
@@ -25,7 +25,7 @@ type: application
 # This is the chart version. This version number should be incremented each time you make changes
 # to the chart and its templates, including the app version.
 # Versions are expected to follow Semantic Versioning (https://semver.org/)
-version: 0.10.7
+version: 0.10.8
 
 # This is the version number of the application being deployed. This version number should be
 # incremented each time you make changes to the application. Versions are not expected to

diff --git a/charts/databend-query/templates/statefulset.yaml b/charts/databend-query/templates/statefulset.yaml
@@ -52,17 +52,24 @@ spec:
             - name: {{ $key }}
               containerPort: {{ $val}}
             {{- end }}
+          {{- if .Values.livenessProbe.enabled }}
           livenessProbe:
             httpGet:
               path: /v1/health
               port: admin
+            initialDelaySeconds: {{ default 5 .Values.livenessProbe.initialDelaySeconds }}
+            periodSeconds: {{ default 15 .Values.livenessProbe.initialDelaySeconds }}
+            failureThreshold: {{ default 3 .Values.livenessProbe.initialDelaySeconds }}
+          {{- end }}
+          {{- if .Values.readinessProbe.enabled }}
           readinessProbe:
             httpGet:
               path: /v1/health
               port: admin
-            initialDelaySeconds: 5
-            periodSeconds: 15
-            failureThreshold: 3
+            initialDelaySeconds: {{ default 5 .Values.readinessProbe.initialDelaySeconds }}
+            periodSeconds: {{ default 15 .Values.readinessProbe.initialDelaySeconds }}
+            failureThreshold: {{ default 3 .Values.readinessProbe.initialDelaySeconds }}
+          {{- end }}
           resources:
             {{- toYaml .Values.resources | nindent 12 }}
           env:

diff --git a/charts/databend-query/values.yaml b/charts/databend-query/values.yaml
@@ -31,6 +31,11 @@ serviceAccount:
   # If not set and create is true, a name is generated using the fullname template
   name: ""
 
+livenessProbe:
+  enabled: true
+readinessProbe:
+  enabled: true
+
 service:
   type: ClusterIP
   ports: