diff --git a/charts/rabbitmq/Chart.yaml b/charts/rabbitmq/Chart.yaml
index f2c6afb6dae2483fa3c98a8a23dfd583c0a0028e..e070b78d08982f9fa72dcc98e7ec63299bbdb5c1 100644
--- a/charts/rabbitmq/Chart.yaml
+++ b/charts/rabbitmq/Chart.yaml
@@ -7,6 +7,6 @@ type: application
maintainers:
- name: groundhog2k
-version: 0.4.12
+version: 0.4.13
appVersion: "3.9.13"
diff --git a/charts/rabbitmq/README.md b/charts/rabbitmq/README.md
index 97f51102d1ff5ee6c7e956b68ec8159e31c4ec88..83170b433e373590f393930f5f2104540da50e0b 100644
--- a/charts/rabbitmq/README.md
+++ b/charts/rabbitmq/README.md
@@ -1,6 +1,6 @@
# RabbitMQ
-  
+  
## Changelog
@@ -76,6 +76,7 @@ helm uninstall my-release
| securityContext | object | `see values.yaml` | Container security context |
| env | list | `[]` | Additional container environmment variables |
| args | list | `[]` | Additional container command arguments |
+| terminationGracePeriodSeconds | int | `60` | Container termination grace period in seconds |
| rbac.create | bool | `true` | Enable creation of RBAC |
| serviceAccount.annotations | object | `{}` | Additional service account annotations |
| serviceAccount.create | bool | `true` | Enable service account creation |
@@ -151,7 +152,7 @@ Section to define custom services
| authentication.user | string | `"guest"` | Initial user name |
| authentication.password | string | `"guest"` | Initial password |
| authentication.erlangCookie | string | `nil` | Erlang cookie (MANDATORY) (Alternative: Set the environment variable ERLANG_COOKIE) |
-| clustering.rebalance | bool | `true` | Enable rebalance queues with master when new replica is created |
+| clustering.rebalance | bool | `false` | Enable rebalance queues with master when new replica is created |
| clustering.useLongName | bool | `true` | Use FQDN for RabbitMQ node names |
## RabbitMQ memory parameters
diff --git a/charts/rabbitmq/RELEASENOTES.md b/charts/rabbitmq/RELEASENOTES.md
index c4d24a8333ee98aa5ed99f867cb3a1a270a4ee0a..86437b08d8e3272f12cb3f35b223e70964b492f3 100644
--- a/charts/rabbitmq/RELEASENOTES.md
+++ b/charts/rabbitmq/RELEASENOTES.md
@@ -9,4 +9,5 @@
| 0.4.10 | 3.9.13 | Implemented support for extra secrets and advanced configuration capabilites |
| 0.4.11 | 3.9.13 | Corrected chart documentation |
| 0.4.12 | 3.9.13 | Fixed headless service and enabled publishNotReadyAddresses |
+| 0.4.13 | 3.9.13 | Added graceful shutdown hook |
| | | |
diff --git a/charts/rabbitmq/templates/scripts.yaml b/charts/rabbitmq/templates/scripts.yaml
index aa3be2d4c3ec4bb379081456255b1f91682b6d19..42f2984ca5d11d5e4b61ec12c4f1a7b9de5465a4 100644
--- a/charts/rabbitmq/templates/scripts.yaml
+++ b/charts/rabbitmq/templates/scripts.yaml
@@ -33,3 +33,42 @@ data:
fi
chmod 600 /var/lib/rabbitmq/.erlang.cookie
echo "Finished."
+ startup.sh: |
+ until rabbitmqctl cluster_status >/dev/null; do
+ echo "PostStart: Waiting for cluster readiness..." >/proc/1/fd/1
+ sleep 5
+ done
+ echo "PostStart: Rebalancing all queues.." >/proc/1/fd/1
+ rabbitmq-queues rebalance "all" >/proc/1/fd/1
+ shutdown.sh: |
+ {{- if and (.Values.terminationGracePeriodSeconds) (gt (int .Values.terminationGracePeriodSeconds) 10) }}
+ if rabbitmqctl cluster_status; then
+ echo "PreStop: Will wait up to {{ sub .Values.terminationGracePeriodSeconds 10 }} seconds for node to make sure cluster is healthy after node shutdown" >/proc/1/fd/1
+ timeout {{ sub .Values.terminationGracePeriodSeconds 10 }} /scripts/safeshutdown.sh
+ fi
+ {{- end }}
+ echo "PreStop: Stopping RabbitMQ" >/proc/1/fd/1
+ rabbitmqctl stop_app
+ echo "PreStop: RabbitMQ stopped" >/proc/1/fd/1
+ sleep 5
+ safeshutdown.sh: |
+ while true; do
+ waiting="false"
+ if ! rabbitmq-diagnostics -q check_if_node_is_mirror_sync_critical >/proc/1/fd/1; then
+ echo "PreStop: check_if_node_is_mirror_sync_critical returns error. Continuing to wait" >/proc/1/fd/1
+ waiting="true"
+ else
+ echo "PreStop: check_if_node_is_mirror_sync_critical returns o.k." >/proc/1/fd/1
+ fi
+ if ! rabbitmq-diagnostics -q check_if_node_is_quorum_critical >/proc/1/fd/1; then
+ echo "PreStop: check_if_node_is_quorum_critical returns error. Continuing to wait" >/proc/1/fd/1
+ waiting="true"
+ else
+ echo "PreStop: check_if_node_is_quorum_critical returns o.k." >/proc/1/fd/1
+ fi
+ if [ $waiting = "true" ]; then
+ sleep 1
+ else
+ break
+ fi
+ done
diff --git a/charts/rabbitmq/templates/statefulset.yaml b/charts/rabbitmq/templates/statefulset.yaml
index 24d4486cfa756ca8a59ccbbca7e77f1749770646..d65e7725b6b4a5071eecfd317c6378013dd20dd6 100644
--- a/charts/rabbitmq/templates/statefulset.yaml
+++ b/charts/rabbitmq/templates/statefulset.yaml
@@ -74,6 +74,9 @@ spec:
{{- with .Values.extraInitContainers }}
{{- toYaml . | nindent 8 }}
{{- end }}
+ {{- if .Values.terminationGracePeriodSeconds }}
+ terminationGracePeriodSeconds: {{ .Values.terminationGracePeriodSeconds }}
+ {{- end }}
containers:
- name: {{ .Chart.Name }}
{{- with .Values.securityContext }}
@@ -177,19 +180,14 @@ spec:
command:
- /bin/bash
- -ec
- - |
- until rabbitmqctl cluster_status >/dev/null; do
- echo "Waiting for cluster readiness..."
- sleep 5
- done
- rabbitmq-queues rebalance "all"
+ - /scripts/startup.sh
{{- end }}
preStop:
exec:
command:
- bash
- -ec
- - rabbitmqctl stop_app
+ - /scripts/shutdown.sh
env:
- name: RABBITMQ_USE_LONGNAME
value: {{ .Values.clustering.useLongName | quote }}
@@ -227,6 +225,8 @@ spec:
mountPath: /var/lib/rabbitmq
- name: config
mountPath: /etc/rabbitmq
+ - name: scripts
+ mountPath: /scripts
{{- if .Values.certificates.enabled }}
- name: certs
mountPath: /ssl
diff --git a/charts/rabbitmq/values.yaml b/charts/rabbitmq/values.yaml
index 555af3d581d657bb14cfaa848d69f8048ae759ff..3916bbbd9d25fef129c5982018681a62d0845870 100644
--- a/charts/rabbitmq/values.yaml
+++ b/charts/rabbitmq/values.yaml
@@ -202,6 +202,9 @@ extraContainers: []
## Default Kubernetes cluster domain
clusterDomain: cluster.local
+## Container termination grace period
+terminationGracePeriodSeconds: 60
+
# A list of plugins (without whitespace!) - f.e.: rabbitmq_federation_management,rabbitmq_mqtt,rabbitmq_stomp
plugins: []
@@ -326,7 +329,7 @@ extraSecretAdvancedConfigs:
## Common (recommended) cluster settings
clustering:
## Rebalance queues with master when new replica is created
- rebalance: true
+ rebalance: false
## Use FQDN for RabbitMQ node names
useLongName: true