From b3602d36464b23d760fbd7e0b731b6bd2b48a1f3 Mon Sep 17 00:00:00 2001
From: marsal wang <707783088@qq.com>
Date: Mon, 10 Jan 2022 14:52:43 +0800
Subject: [PATCH] stomp

---
 rabbitmq/templates/statefulset.yaml |  2 ++
 rabbitmq/templates/svc.yaml         | 11 +++++++++++
 rabbitmq/values.yaml                | 18 ++++++++++++++++++
 3 files changed, 31 insertions(+)

diff --git a/rabbitmq/templates/statefulset.yaml b/rabbitmq/templates/statefulset.yaml
index b283266..94b8b05 100644
--- a/rabbitmq/templates/statefulset.yaml
+++ b/rabbitmq/templates/statefulset.yaml
@@ -224,6 +224,8 @@ spec:
               containerPort: 25672
             - name: stats
               containerPort: 15672
+            - name: stomp
+              containerPort: 15674
             - name: epmd
               containerPort: 4369
             {{- if .Values.metrics.enabled }}
diff --git a/rabbitmq/templates/svc.yaml b/rabbitmq/templates/svc.yaml
index ad193fd..1c122ed 100644
--- a/rabbitmq/templates/svc.yaml
+++ b/rabbitmq/templates/svc.yaml
@@ -75,6 +75,17 @@ spec:
       targetPort: stats
       {{- if eq .Values.service.type "ClusterIP" }}
       nodePort: null
+      {{- else if and (or (eq .Values.service.type "NodePort") (eq .Values.service.type "LoadBalancer")) (not (empty .Values.service.stompNodePort)) }}
+      nodePort: {{ .Values.service.stompNodePort }}
+      {{- end }}
+    {{- end }}
+    
+    {{- if .Values.service.stompPortEnabled }}
+    - name: {{ .Values.service.stompPortName }}
+      port: {{ .Values.service.stompPort }}
+      targetPort: stats
+      {{- if eq .Values.service.type "ClusterIP" }}
+      nodePort: null
       {{- else if and (or (eq .Values.service.type "NodePort") (eq .Values.service.type "LoadBalancer")) (not (empty .Values.service.managerNodePort)) }}
       nodePort: {{ .Values.service.managerNodePort }}
       {{- end }}
diff --git a/rabbitmq/values.yaml b/rabbitmq/values.yaml
index e2508da..c66f907 100644
--- a/rabbitmq/values.yaml
+++ b/rabbitmq/values.yaml
@@ -784,6 +784,24 @@ service:
   ## managerNodePort: 30673
   ##
   managerNodePort: ""
+  ## @param service.managerPortEnabled RabbitMQ Manager port
+  ## ref: https://github.com/bitnami/bitnami-docker-rabbitmq#environment-variables
+  ##
+  stompPortEnabled: true
+
+  ## @param service.managerPort RabbitMQ Manager port
+  ##
+  stompPort: 15674
+
+  ## @param service.managerPortName RabbitMQ Manager service port name
+  ##
+  stompPortName: http-stomp
+
+  ## @param service.managerNodePort Node port override for `http-stats` port, if serviceType `NodePort`
+  ## e.g:
+  ## managerNodePort: 30673
+  ##
+  stompNodePort: ""
 
   ## @param service.metricsPort RabbitMQ Prometheues metrics port
   ##