fix(helm/loki): Allow prefixing of rollout-group value and ingester name

grafana · Dec 9, 2024 · cd3ffbd · cd3ffbd
1 parent e784c93
commit cd3ffbd
Show file tree

Hide file tree

Showing 10 changed files with 57 additions and 33 deletions.
diff --git a/production/helm/loki/templates/ingester/_helpers-ingester.tpl b/production/helm/loki/templates/ingester/_helpers-ingester.tpl
@@ -71,4 +71,22 @@ expects a dict
 */}}
 {{- define "loki.ingester.maxUnavailable" -}}
 {{- ceil (mulf .replicas (divf (int .ctx.Values.ingester.zoneAwareReplication.maxUnavailablePct) 100)) -}}
-{{- end -}}
+{{- end -}}
+
+{{/*
+Return rollout-group prefix if it is set
+*/}}
+{{- define "loki.prefixRolloutGroup" -}}
+{{- if .Values.ingester.rolloutGroupPrefix -}}
+{{- .Values.ingester.rolloutGroupPrefix -}}-
+{{- end -}}
+{{- end -}}
+
+{{/*
+Return ingester name prefix if required
+*/}}
+{{- define "loki.prefixIngesterName" -}}
+{{- if .Values.ingester.addIngesterNamePrefix -}}
+loki-
+{{- end -}}
+{{- end -}}
diff --git a/production/helm/loki/templates/ingester/poddisruptionbudget-ingester-rollout.yaml b/production/helm/loki/templates/ingester/poddisruptionbudget-ingester-rollout.yaml
@@ -13,7 +13,7 @@ metadata:
 spec:
   selector:
     matchLabels:
-      rollout-group: ingester
+      rollout-group: {{ include "loki.prefixRolloutGroup" . }}ingester
   {{- with .Values.ingester.maxUnavailable }}
   maxUnavailable: {{ . }}
   {{- end }}

diff --git a/production/helm/loki/templates/ingester/poddisruptionbudget-ingester.yaml b/production/helm/loki/templates/ingester/poddisruptionbudget-ingester.yaml
@@ -19,7 +19,7 @@ spec:
       - key: rollout-group
         operator: NotIn
         values:
-          - "ingester"
+          - '{{ include "loki.prefixRolloutGroup" . }}ingester'
   {{- with .Values.ingester.maxUnavailable }}
   maxUnavailable: {{ . }}
   {{- end }}

diff --git a/production/helm/loki/templates/ingester/service-ingester-zone-a-headless.yaml b/production/helm/loki/templates/ingester/service-ingester-zone-a-headless.yaml
@@ -33,6 +33,6 @@ spec:
       {{- end }}
   selector:
     {{- include "loki.ingesterSelectorLabels" . | nindent 4 }}
-    name: ingester-zone-a
-    rollout-group: ingester
+    name: {{ include "loki.prefixIngesterName" . }}ingester-zone-a
+    rollout-group: {{ include "loki.prefixRolloutGroup" . }}ingester
 {{- end -}}
diff --git a/production/helm/loki/templates/ingester/service-ingester-zone-b-headless.yaml b/production/helm/loki/templates/ingester/service-ingester-zone-b-headless.yaml
@@ -33,6 +33,6 @@ spec:
       {{- end }}
   selector:
     {{- include "loki.ingesterSelectorLabels" . | nindent 4 }}
-    name: ingester-zone-b
-    rollout-group: ingester
+    name: {{ include "loki.prefixIngesterName" . }}ingester-zone-b
+    rollout-group: {{ include "loki.prefixRolloutGroup" . }}ingester
 {{- end -}}
diff --git a/production/helm/loki/templates/ingester/service-ingester-zone-c-headless.yaml b/production/helm/loki/templates/ingester/service-ingester-zone-c-headless.yaml
@@ -33,6 +33,6 @@ spec:
       {{- end }}
   selector:
     {{- include "loki.ingesterSelectorLabels" . | nindent 4 }}
-    name: ingester-zone-c
-    rollout-group: ingester
+    name: {{ include "loki.prefixIngesterName" . }}ingester-zone-c
+    rollout-group: {{ include "loki.prefixRolloutGroup" . }}ingester
 {{- end -}}
diff --git a/production/helm/loki/templates/ingester/statefulset-ingester-zone-a.yaml b/production/helm/loki/templates/ingester/statefulset-ingester-zone-a.yaml
@@ -9,8 +9,8 @@ metadata:
   labels:
     {{- include "loki.ingesterLabels" . | nindent 4 }}
     app.kubernetes.io/part-of: memberlist
-    rollout-group: ingester
-    name: ingester-zone-a
+    name: {{ include "loki.prefixIngesterName" . }}ingester-zone-a
+    rollout-group: {{ include "loki.prefixRolloutGroup" . }}ingester
   annotations:
     rollout-max-unavailable: "{{ include "loki.ingester.maxUnavailable" (dict "ctx" . "replicas" $replicas)}}"
     {{- with .Values.loki.annotations }}
@@ -38,8 +38,8 @@ spec:
   selector:
     matchLabels:
       {{- include "loki.ingesterSelectorLabels" . | nindent 6 }}
-      name: ingester-zone-a
-      rollout-group: ingester
+      name: {{ include "loki.prefixIngesterName" . }}ingester-zone-a
+      rollout-group: {{ include "loki.prefixRolloutGroup" . }}ingester
 {{- with .Values.ingester.updateStrategy }}
   updateStrategy:
      {{- tpl (. | toYaml) $ | nindent 4 }}
@@ -60,8 +60,8 @@ spec:
       labels:
         {{- include "loki.ingesterSelectorLabels" . | nindent 8 }}
         app.kubernetes.io/part-of: memberlist
-        name: ingester-zone-a
-        rollout-group: ingester
+        name: {{ include "loki.prefixIngesterName" . }}ingester-zone-a
+        rollout-group: {{ include "loki.prefixRolloutGroup" . }}ingester
         {{- with .Values.loki.podLabels }}
         {{- toYaml . | nindent 8 }}
         {{- end }}
@@ -164,11 +164,11 @@ spec:
                   - key: rollout-group
                     operator: In
                     values:
-                      - ingester
+                      - {{ include "loki.prefixRolloutGroup" . }}ingester
                   - key: name
                     operator: NotIn
                     values:
-                      - ingester-zone-a
+                      - {{ include "loki.prefixIngesterName" . }}ingester-zone-a
               topologyKey: kubernetes.io/hostname
         {{- with .Values.ingester.zoneAwareReplication.zoneA.extraAffinity }}
         {{- toYaml . | nindent 8 }}

diff --git a/production/helm/loki/templates/ingester/statefulset-ingester-zone-b.yaml b/production/helm/loki/templates/ingester/statefulset-ingester-zone-b.yaml
@@ -9,8 +9,8 @@ metadata:
   labels:
     {{- include "loki.ingesterLabels" . | nindent 4 }}
     app.kubernetes.io/part-of: memberlist
-    rollout-group: ingester
-    name: ingester-zone-b
+    name: {{ include "loki.prefixIngesterName" . }}ingester-zone-b
+    rollout-group: {{ include "loki.prefixRolloutGroup" . }}ingester
   annotations:
     rollout-max-unavailable: "{{ include "loki.ingester.maxUnavailable" (dict "ctx" . "replicas" $replicas)}}"
     {{- with .Values.loki.annotations }}
@@ -38,8 +38,8 @@ spec:
   selector:
     matchLabels:
       {{- include "loki.ingesterSelectorLabels" . | nindent 6 }}
-      name: ingester-zone-b
-      rollout-group: ingester
+      name: {{ include "loki.prefixIngesterName" . }}ingester-zone-b
+      rollout-group: {{ include "loki.prefixRolloutGroup" . }}ingester
 {{- with .Values.ingester.updateStrategy }}
   updateStrategy:
      {{- tpl (. | toYaml) $ | nindent 4 }}
@@ -60,8 +60,8 @@ spec:
       labels:
         {{- include "loki.ingesterSelectorLabels" . | nindent 8 }}
         app.kubernetes.io/part-of: memberlist
-        name: ingester-zone-b
-        rollout-group: ingester
+        name: {{ include "loki.prefixIngesterName" . }}ingester-zone-b
+        rollout-group: {{ include "loki.prefixRolloutGroup" . }}ingester
         {{- with .Values.loki.podLabels }}
         {{- toYaml . | nindent 8 }}
         {{- end }}
@@ -164,11 +164,11 @@ spec:
                   - key: rollout-group
                     operator: In
                     values:
-                      - ingester
+                      - {{ include "loki.prefixRolloutGroup" . }}ingester
                   - key: name
                     operator: NotIn
                     values:
-                      - ingester-zone-b
+                      - {{ include "loki.prefixIngesterName" . }}ingester-zone-b
               topologyKey: kubernetes.io/hostname
         {{- with .Values.ingester.zoneAwareReplication.zoneB.extraAffinity }}
         {{- toYaml . | nindent 8 }}

diff --git a/production/helm/loki/templates/ingester/statefulset-ingester-zone-c.yaml b/production/helm/loki/templates/ingester/statefulset-ingester-zone-c.yaml
@@ -9,8 +9,8 @@ metadata:
   labels:
     {{- include "loki.ingesterLabels" . | nindent 4 }}
     app.kubernetes.io/part-of: memberlist
-    rollout-group: ingester
-    name: ingester-zone-c
+    name: {{ include "loki.prefixIngesterName" . }}ingester-zone-c
+    rollout-group: {{ include "loki.prefixRolloutGroup" . }}ingester
   annotations:
     rollout-max-unavailable: "{{ include "loki.ingester.maxUnavailable" (dict "ctx" . "replicas" $replicas)}}"
     {{- with .Values.loki.annotations }}
@@ -38,8 +38,8 @@ spec:
   selector:
     matchLabels:
       {{- include "loki.ingesterSelectorLabels" . | nindent 6 }}
-      name: ingester-zone-c
-      rollout-group: ingester
+      name: {{ include "loki.prefixIngesterName" . }}ingester-zone-c
+      rollout-group: {{ include "loki.prefixRolloutGroup" . }}ingester
 {{- with .Values.ingester.updateStrategy }}
   updateStrategy:
      {{- tpl (. | toYaml) $ | nindent 4 }}
@@ -60,8 +60,8 @@ spec:
       labels:
         {{- include "loki.ingesterSelectorLabels" . | nindent 8 }}
         app.kubernetes.io/part-of: memberlist
-        name: ingester-zone-c
-        rollout-group: ingester
+        name: {{ include "loki.prefixIngesterName" . }}ingester-zone-c
+        rollout-group: {{ include "loki.prefixRolloutGroup" . }}ingester
         {{- with .Values.loki.podLabels }}
         {{- toYaml . | nindent 8 }}
         {{- end }}
@@ -164,11 +164,11 @@ spec:
                   - key: rollout-group
                     operator: In
                     values:
-                      - ingester
+                      - {{ include "loki.prefixIngesterName" . }}ingester
                   - key: name
                     operator: NotIn
                     values:
-                      - ingester-zone-c
+                      - {{ include "loki.prefixIngesterName" . }}ingester-zone-c
               topologyKey: kubernetes.io/hostname
         {{- with .Values.ingester.zoneAwareReplication.zoneC.extraAffinity }}
         {{- toYaml . | nindent 8 }}

diff --git a/production/helm/loki/values.yaml b/production/helm/loki/values.yaml
@@ -1893,6 +1893,12 @@ ingester:
       excludeDefaultZone: false
       readPath: false
       writePath: false
+
+  # optionally allow adding arbitrary prefix to the ingester rollout-group label
+  rolloutGroupPrefix: null
+  # optionally allow adding 'loki-' prefix to ingester name label
+  addIngesterNamePrefix: false
+
 # --  Configuration for the distributor
 distributor:
   # -- Number of replicas for the distributor