Set up rabbitmqadmin automatically (#426)

rabbitmq · Oct 28, 2020 · 3cba654 · 3cba654
1 parent 6fc47e6
commit 3cba654
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 2 deletions.
diff --git a/internal/resource/statefulset.go b/internal/resource/statefulset.go
@@ -534,7 +534,11 @@ func (builder *StatefulSetBuilder) podTemplateSpec(annotations, labels map[strin
 							"&& chmod 600 /var/lib/rabbitmq/.erlang.cookie ; " +
 							"cp /tmp/rabbitmq-plugins/enabled_plugins /operator/enabled_plugins " +
 							"&& chown 999:999 /operator/enabled_plugins ; " +
-							"chgrp 999 /var/lib/rabbitmq/mnesia/",
+							"chgrp 999 /var/lib/rabbitmq/mnesia/ ; " +
+							"echo '[default]' > /var/lib/rabbitmq/.rabbitmqadmin.conf " +
+							"&& sed -e 's/default_user/username/' -e 's/default_pass/password/' /tmp/default_user.conf >> /var/lib/rabbitmq/.rabbitmqadmin.conf " +
+							"&& chown 999:999 /var/lib/rabbitmq/.rabbitmqadmin.conf " +
+							"&& chmod 600 /var/lib/rabbitmq/.rabbitmqadmin.conf",
 					},
 					Resources: corev1.ResourceRequirements{
 						Limits: map[corev1.ResourceName]k8sresource.Quantity{
@@ -567,6 +571,11 @@ func (builder *StatefulSetBuilder) podTemplateSpec(annotations, labels map[strin
 							Name:      "persistence",
 							MountPath: "/var/lib/rabbitmq/mnesia/",
 						},
+						{
+							Name:      "rabbitmq-confd",
+							MountPath: "/tmp/default_user.conf",
+							SubPath:   "default_user.conf",
+						},
 					},
 				},
 			},

diff --git a/internal/resource/statefulset_test.go b/internal/resource/statefulset_test.go
@@ -986,7 +986,11 @@ var _ = Describe("StatefulSet", func() {
 						"&& chmod 600 /var/lib/rabbitmq/.erlang.cookie ; "+
 						"cp /tmp/rabbitmq-plugins/enabled_plugins /operator/enabled_plugins "+
 						"&& chown 999:999 /operator/enabled_plugins ; "+
-						"chgrp 999 /var/lib/rabbitmq/mnesia/",
+						"chgrp 999 /var/lib/rabbitmq/mnesia/ ; "+
+						"echo '[default]' > /var/lib/rabbitmq/.rabbitmqadmin.conf "+
+						"&& sed -e 's/default_user/username/' -e 's/default_pass/password/' /tmp/default_user.conf >> /var/lib/rabbitmq/.rabbitmqadmin.conf "+
+						"&& chown 999:999 /var/lib/rabbitmq/.rabbitmqadmin.conf "+
+						"&& chmod 600 /var/lib/rabbitmq/.rabbitmqadmin.conf",
 				),
 				"VolumeMounts": ConsistOf(
 					corev1.VolumeMount{
@@ -1009,6 +1013,11 @@ var _ = Describe("StatefulSet", func() {
 						Name:      "persistence",
 						MountPath: "/var/lib/rabbitmq/mnesia/",
 					},
+					corev1.VolumeMount{
+						Name:      "rabbitmq-confd",
+						MountPath: "/tmp/default_user.conf",
+						SubPath:   "default_user.conf",
+					},
 				),
 			}))
 		})