feat: Let user specify multiple replica count for pgbouncer

Signed-off-by: Benoit Tigeot <[email protected]>
airflow-helm · Jan 8, 2025 · b03efda · b03efda
1 parent 732a0f0
commit b03efda
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/charts/airflow/README.md b/charts/airflow/README.md
@@ -475,6 +475,7 @@ Parameter | Description | Default
 `pgbouncer.podAnnotations` | Pod annotations for the pgbouncer Deployment | `{}`
 `pgbouncer.safeToEvict` | if we add the annotation: "cluster-autoscaler.kubernetes.io/safe-to-evict" = "true" | `true`
 `pgbouncer.podDisruptionBudget.*` | configs for the PodDisruptionBudget of the pgbouncer | `<see values.yaml>`
+`pgbouncer.replicaCount` | Pod replication capability | `<see values.yaml>`
 `pgbouncer.livenessProbe.*` | configs for the pgbouncer Pods' liveness probe | `<see values.yaml>`
 `pgbouncer.startupProbe.*` | configs for the pgbouncer Pods' startup probe | `<see values.yaml>`
 `pgbouncer.terminationGracePeriodSeconds` | the maximum number of seconds to wait for queries upon pod termination, before force killing | `120`

diff --git a/charts/airflow/templates/pgbouncer/pgbouncer-deployment.yaml b/charts/airflow/templates/pgbouncer/pgbouncer-deployment.yaml
@@ -74,7 +74,7 @@ metadata:
   {{- toYaml .Values.pgbouncer.annotations | nindent 4 }}
   {{- end }}
 spec:
-  replicas: 1
+  replicas: {{ .Values.pgbouncer.replicaCount }}
   strategy:
     rollingUpdate:
       ## multiple pgbouncer pods can safely run concurrently
@@ -223,4 +223,4 @@ spec:
             sources:
               {{- include "airflow.pgbouncer.certs_volume_sources" . | indent 14 }}
         {{- end }}
-{{- end }}
+{{- end }}
diff --git a/charts/airflow/values.yaml b/charts/airflow/values.yaml
@@ -1951,6 +1951,10 @@ pgbouncer:
     ##
     minAvailable:
 
+  ## the number of replica count. Careful, each pgbouncer instance maintains its own connection pool
+  ##
+  replicaCount: 1
+
   ## configs for the pgbouncer Pods' liveness probe
   ##
   livenessProbe: