feat(dvcx-worker): Expose configuration for livenessProbe

iterative · Jun 16, 2024 · 876cabb · 876cabb
1 parent 5263742
commit 876cabb
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 6 deletions.
diff --git a/charts/studio/Chart.yaml b/charts/studio/Chart.yaml
@@ -2,7 +2,7 @@ apiVersion: v2
 name: studio
 description: A Helm chart for Kubernetes
 type: application
-version: 0.14.13
+version: 0.14.14
 appVersion: "v2.123.2"
 maintainers:
   - name: iterative

diff --git a/charts/studio/README.md b/charts/studio/README.md
@@ -1,6 +1,6 @@
 # studio
 
-![Version: 0.14.13](https://img.shields.io/badge/Version-0.14.13-informational?style=flat-square) ![Type: application](https://img.shields.io/badge/Type-application-informational?style=flat-square) ![AppVersion: v2.123.2](https://img.shields.io/badge/AppVersion-v2.123.2-informational?style=flat-square)
+![Version: 0.14.14](https://img.shields.io/badge/Version-0.14.14-informational?style=flat-square) ![Type: application](https://img.shields.io/badge/Type-application-informational?style=flat-square) ![AppVersion: v2.123.2](https://img.shields.io/badge/AppVersion-v2.123.2-informational?style=flat-square)
 
 A Helm chart for Kubernetes
 
@@ -138,7 +138,7 @@ A Helm chart for Kubernetes
 | studioBlobvault.image | object | `{"repository":"nginx","tag":"1.27.0-alpine"}` | Image to use for the blobvault service |
 | studioBlobvault.image.repository | string | `"nginx"` | Image repository |
 | studioBlobvault.image.tag | string | `"1.27.0-alpine"` | Image tag |
-| studioDvcxWorker | object | `{"affinity":{},"autoscaling":{"annotations":{},"behavior":{},"enabled":false,"maxReplicas":5,"minReplicas":1,"targetCPUUtilizationPercentage":80,"template":[]},"envFromSecret":"","envVars":{},"ephemeralStorage":{"persistentVolumeClaim":{"claimName":"dvcx-worker","storageClass":""},"size":"20Gi","type":"ephemeral"},"image":{"pullPolicy":"IfNotPresent","repository":"docker.iterative.ai/studio-dvcx-worker"},"logLevel":"info","nodeSelector":{},"podAnnotations":{},"podSecurityContext":{},"replicaCount":1,"resources":{"limits":{"ephemeral-storage":"20Gi","memory":"16Gi"},"requests":{"cpu":"1000m","ephemeral-storage":"10Gi","memory":"3Gi"}},"securityContext":{},"serviceAccount":{"annotations":{},"create":false,"name":""},"strategy":{"rollingUpdate":{"maxSurge":1,"maxUnavailable":0}},"tolerations":[]}` | Studio DVCx Worker settings group |
+| studioDvcxWorker | object | `{"affinity":{},"autoscaling":{"annotations":{},"behavior":{},"enabled":false,"maxReplicas":5,"minReplicas":1,"targetCPUUtilizationPercentage":80,"template":[]},"envFromSecret":"","envVars":{},"ephemeralStorage":{"persistentVolumeClaim":{"claimName":"dvcx-worker","storageClass":""},"size":"20Gi","type":"ephemeral"},"image":{"pullPolicy":"IfNotPresent","repository":"docker.iterative.ai/studio-dvcx-worker"},"livenessProbe":{"initialDelaySeconds":10,"periodSeconds":10,"timeoutSeconds":10},"logLevel":"info","nodeSelector":{},"podAnnotations":{},"podSecurityContext":{},"replicaCount":1,"resources":{"limits":{"ephemeral-storage":"20Gi","memory":"16Gi"},"requests":{"cpu":"1000m","ephemeral-storage":"10Gi","memory":"3Gi"}},"securityContext":{},"serviceAccount":{"annotations":{},"create":false,"name":""},"strategy":{"rollingUpdate":{"maxSurge":1,"maxUnavailable":0}},"tolerations":[]}` | Studio DVCx Worker settings group |
 | studioDvcxWorker.affinity | object | `{}` | DVCx worker pod affinity configuration |
 | studioDvcxWorker.autoscaling | object | `{"annotations":{},"behavior":{},"enabled":false,"maxReplicas":5,"minReplicas":1,"targetCPUUtilizationPercentage":80,"template":[]}` | DVCx worker autoscaling configuration |
 | studioDvcxWorker.autoscaling.annotations | object | `{}` | Worker autoscaling annotation |
@@ -159,6 +159,9 @@ A Helm chart for Kubernetes
 | studioDvcxWorker.image | object | `{"pullPolicy":"IfNotPresent","repository":"docker.iterative.ai/studio-dvcx-worker"}` | DVCx worker image settings |
 | studioDvcxWorker.image.pullPolicy | string | `"IfNotPresent"` | DVCx worker image pull policy |
 | studioDvcxWorker.image.repository | string | `"docker.iterative.ai/studio-dvcx-worker"` | DVCx worker image repository |
+| studioDvcxWorker.livenessProbe.initialDelaySeconds | int | `10` | DVCx worker liveness probe `initialDelaySeconds` |
+| studioDvcxWorker.livenessProbe.periodSeconds | int | `10` | DVCx worker liveness probe `periodSeconds` |
+| studioDvcxWorker.livenessProbe.timeoutSeconds | int | `10` | DVCx worker liveness probe `timeoutSeconds` |
 | studioDvcxWorker.logLevel | string | `"info"` | DVCx worker log level |
 | studioDvcxWorker.nodeSelector | object | `{}` | DVCx worker pod node selector configuration |
 | studioDvcxWorker.podAnnotations | object | `{}` | Additional DVCx worker pod annotations |

diff --git a/charts/studio/templates/deployment-studio-dvcx-worker.yaml b/charts/studio/templates/deployment-studio-dvcx-worker.yaml
@@ -77,9 +77,9 @@ spec:
                 - python
                 - -c
                 - 'from viewer.utils import sqlalchemy_liveness_probe; sqlalchemy_liveness_probe()'
-            initialDelaySeconds: 10
-            periodSeconds: 10
-            timeoutSeconds: 10
+            initialDelaySeconds: {{.Values.studioDvcxWorker.livenessProbe.initialDelaySeconds | default 10}}
+            periodSeconds: {{.Values.studioDvcxWorker.livenessProbe.periodSeconds | default 10}}
+            timeoutSeconds: {{.Values.studioDvcxWorker.livenessProbe.timeoutSeconds | default 10}}
           volumeMounts:
             {{- if not .Values.global.blobvault.bucket }}
             - name: blobvault

diff --git a/charts/studio/values.yaml b/charts/studio/values.yaml
@@ -708,6 +708,14 @@ studioDvcxWorker:
       maxUnavailable: 0
       maxSurge: 1
 
+  livenessProbe:
+    # -- DVCx worker liveness probe `initialDelaySeconds`
+    initialDelaySeconds: 10
+    # -- DVCx worker liveness probe `periodSeconds`
+    periodSeconds: 10
+    # -- DVCx worker liveness probe `timeoutSeconds`
+    timeoutSeconds: 10
+
   # -- DVCx worker autoscaling configuration
   autoscaling:
     # -- DVCx worker autoscaling enabled flag