From 4757fe817df5633c77697d30793abadbc1c0a8bc Mon Sep 17 00:00:00 2001
From: Ilya Nikokoshev <ilya.nikokoshev@gmail.com>
Date: Tue, 9 Jan 2024 22:00:17 +0100
Subject: [PATCH] fix: add an `initContainer`

---
 langfuse/templates/deployment.yaml | 34 +++++++++++++++++++++++++++++-
 1 file changed, 33 insertions(+), 1 deletion(-)

diff --git a/langfuse/templates/deployment.yaml b/langfuse/templates/deployment.yaml
index 7d200aa..f34f85f 100644
--- a/langfuse/templates/deployment.yaml
+++ b/langfuse/templates/deployment.yaml
@@ -27,6 +27,32 @@ spec:
       serviceAccountName: {{ include "langfuse.serviceAccountName" . }}
       securityContext:
         {{- toYaml .Values.podSecurityContext | nindent 8 }}
+      initContainers:
+        - name: check-postgres
+          image: busybox:latest
+          env:
+            {{- if .Values.postgresql.enabled }}
+            - name: DATABASE_HOST
+              value: {{ include "langfuse.databaseHost" . | quote }}
+            - name: DATABASE_PORT
+              value: "5432"
+            {{- else }}
+            - name: DATABASE_URL
+              valueFrom:
+                secretKeyRef:
+                  name: {{ include "langfuse.connectionSecret" . | quote }}
+                  key: database-url
+            {{- end }}
+          command:
+            - sh
+            - -c
+            - >
+              until nc -z -v -w30 
+              ${DATABASE_HOST:-$(echo $DATABASE_URL | awk -F[/:] '{print $4}')} 
+              ${DATABASE_PORT:-$(echo $DATABASE_URL | awk -F[/:] '{print $5}'):-"5432"} ; do
+                echo 'Waiting for Postgres...' ;
+                sleep 5 ;
+              done
       containers:
         - name: {{ .Chart.Name }}
           securityContext:
@@ -80,10 +106,16 @@ spec:
             - configMapRef:
                 name: {{ include "langfuse.configMap" . }}
             {{- include "langfuse.additionalSecrets" . }}
-          readinessProbe:
+          startupProbe:
+            tcpSocket:
+              port: http
+            failureThreshold: 30
+            periodSeconds: 10
+          livenessProbe:
             httpGet:
               path: /api/public/health
               port: http
+            periodSeconds: 20
           resources:
             {{- toYaml .Values.resources | nindent 12 }}
       {{- with .Values.nodeSelector }}