From 08967ff536db13b6d7c46810bb61a2a6c36d46c0 Mon Sep 17 00:00:00 2001
From: Ramit Goolry <93002515+ramit-wandb@users.noreply.github.com>
Date: Wed, 20 Mar 2024 23:51:00 +0530
Subject: [PATCH] chore: readiness and liveness probes for parquet (#95)

---
 charts/operator-wandb/Chart.yaml              |  2 +-
 .../charts/parquet/templates/deployment.yaml  | 20 +++++++++++++++++++
 2 files changed, 21 insertions(+), 1 deletion(-)

diff --git a/charts/operator-wandb/Chart.yaml b/charts/operator-wandb/Chart.yaml
index f6271601..243db21a 100644
--- a/charts/operator-wandb/Chart.yaml
+++ b/charts/operator-wandb/Chart.yaml
@@ -2,7 +2,7 @@ apiVersion: v2
 name: operator-wandb
 description: A Helm chart for deploying W&B to Kubernetes
 type: application
-version: 0.11.1
+version: 0.11.2
 appVersion: 1.0.0
 icon: https://wandb.ai/logo.svg
 
diff --git a/charts/operator-wandb/charts/parquet/templates/deployment.yaml b/charts/operator-wandb/charts/parquet/templates/deployment.yaml
index c1f9f170..494760eb 100644
--- a/charts/operator-wandb/charts/parquet/templates/deployment.yaml
+++ b/charts/operator-wandb/charts/parquet/templates/deployment.yaml
@@ -113,6 +113,26 @@ spec:
             {{- include "parquet.extraEnv" (dict "global" .Values.global "local" .Values) | nindent 12 }}
             {{- include "wandb.extraEnvFrom" (dict "root" $ "local" .) | nindent 12 }}
 
+          livenessProbe:
+            httpGet:
+              path: /ready 
+              port: 8087
+            initialDelaySeconds: 30
+            periodSeconds: 10
+            timeoutSeconds: 5
+            failureThreshold: 3
+            successThreshold: 1
+
+          readinessProbe:
+            httpGet:
+              path: /ready 
+              port: 8087
+            initialDelaySeconds: 30
+            periodSeconds: 10
+            timeoutSeconds: 5
+            failureThreshold: 3
+            successThreshold: 1
+          
           resources:
             {{- toYaml .Values.resources | nindent 12 }}
       volumes: