From 3c99bd4fb2f2505e210063519538310fae1da6d4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Andreas=20Lindh=C3=A9?=
 <7773090+lindhe@users.noreply.github.com>
Date: Thu, 12 Dec 2024 14:35:07 +0100
Subject: [PATCH] feat(helm): Add health probes to write
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Adds configurable `livenessProbe` and `startupProbe` to the loki
container in the write pods.

Signed-off-by: Andreas Lindhé <7773090+lindhe@users.noreply.github.com>
---
 production/helm/loki/CHANGELOG.md                           | 1 +
 production/helm/loki/templates/write/statefulset-write.yaml | 4 ++++
 2 files changed, 5 insertions(+)

diff --git a/production/helm/loki/CHANGELOG.md b/production/helm/loki/CHANGELOG.md
index ffa8a364ecf51..4de92730786ec 100644
--- a/production/helm/loki/CHANGELOG.md
+++ b/production/helm/loki/CHANGELOG.md
@@ -39,6 +39,7 @@ Entries should include a reference to the pull request that introduced the chang
 - [ENHANCEMENT] Add configurable `livenessProbe` and `startupProbe` to ruler.
 - [ENHANCEMENT] Add configurable `startupProbe` to the loki-sc-rules sidecar container in the single-binary pods.
 - [ENHANCEMENT] Add configurable `startupProbe` to table-manager.
+- [ENHANCEMENT] Add configurable `livenessProbe` and `startupProbe` to write.
 
 ## 6.23.0
 
diff --git a/production/helm/loki/templates/write/statefulset-write.yaml b/production/helm/loki/templates/write/statefulset-write.yaml
index fc3b301354db8..484de46d4e246 100644
--- a/production/helm/loki/templates/write/statefulset-write.yaml
+++ b/production/helm/loki/templates/write/statefulset-write.yaml
@@ -114,8 +114,12 @@ spec:
           {{- end }}
           securityContext:
             {{- toYaml .Values.loki.containerSecurityContext | nindent 12 }}
+          livenessProbe:
+            {{- toYaml .Values.loki.livenessProbe | nindent 12 }}
           readinessProbe:
             {{- toYaml .Values.loki.readinessProbe | nindent 12 }}
+          startupProbe:
+            {{- toYaml .Values.loki.startupProbe | nindent 12 }}
           {{- if .Values.write.lifecycle }}
           lifecycle:
             {{- toYaml .Values.write.lifecycle | nindent 12 }}