[alertmanager] support to auto set replicas by node num

Signed-off-by: junot <[email protected]>
WizTelemetry · Apr 2, 2024 · b29d6d0 · b29d6d0
1 parent 2f5a122
commit b29d6d0
Show file tree

Hide file tree

Showing 5 changed files with 29 additions and 2 deletions.
diff --git a/charts/alertmanager/Chart.yaml b/charts/alertmanager/Chart.yaml
@@ -6,7 +6,7 @@ icon: https://raw.githubusercontent.com/prometheus/prometheus.github.io/master/a
 sources:
   - https://github.com/prometheus/alertmanager
 type: application
-version: 1.8.3
+version: 1.8.4
 appVersion: v0.26.0
 kubeVersion: ">=1.19.0-0"
 keywords:

diff --git a/charts/alertmanager/templates/_helpers.tpl b/charts/alertmanager/templates/_helpers.tpl
@@ -109,3 +109,22 @@ labelNameLengthLimit: {{ . }}
 labelValueLengthLimit: {{ . }}
 {{- end }}
 {{- end }}
+
+{{- define "alertmanager.replicas" -}}
+  {{- $replicas := 0 -}}
+  {{ if .Values.replicaAutoSet }}
+    {{- $nodeList := lookup "v1" "Node" "" "" -}}
+    {{- if $nodeList }}
+        {{- if lt (len $nodeList.items) 3 -}}
+            {{- $replicas = 1 -}}
+        {{- else -}}
+            {{- $replicas = 3 -}}
+        {{- end -}}
+    {{- else }}
+        {{- $replicas = .Values.replicaCount -}}
+    {{- end -}}
+  {{- else -}}
+    {{- $replicas = .Values.replicaCount -}}
+  {{- end -}}
+  {{- $replicas -}}
+{{- end -}}
diff --git a/charts/alertmanager/templates/statefulset.yaml b/charts/alertmanager/templates/statefulset.yaml
@@ -11,7 +11,7 @@ metadata:
   {{- end }}
   namespace: {{ include "alertmanager.namespace" . }}
 spec:
-  replicas: {{ .Values.replicaCount }}
+  replicas: {{ include "alertmanager.replicas" . }}
   minReadySeconds: {{ .Values.minReadySeconds }}
   revisionHistoryLimit: {{ .Values.revisionHistoryLimit }}
   selector:

diff --git a/charts/alertmanager/values.schema.json b/charts/alertmanager/values.schema.json
@@ -457,6 +457,9 @@
             },
             "type": "object"
         },
+        "replicaAutoSet": {
+            "type": "boolean"
+        },
         "replicaCount": {
             "type": "integer"
         },

diff --git a/charts/alertmanager/values.yaml b/charts/alertmanager/values.yaml
@@ -6,7 +6,12 @@
 global:
   imagePullSecrets: []
 
+# The alertmanager replica count.
+# If you want to control it by yourself, please firstly configure the replicaAutoSet item as false.
 replicaCount: 1
+# If true, will ignore the upper replicaCount iterm and auto set alertmanager replica count by the node num:
+#   set the alertmanager replica count as 1 when the node num < 3, or set as 3.
+replicaAutoSet: true
 
 # Number of old history to retain to allow rollback
 # Default Kubernetes value is set to 10