From 6a00c5da86b8c0aebb7fa9da75a1321eb7af6b14 Mon Sep 17 00:00:00 2001
From: raihankhan <raihan@appscode.com>
Date: Wed, 17 Jan 2024 18:21:41 +0600
Subject: [PATCH] Add Rabbitmq API

Signed-off-by: raihankhan <raihan@appscode.com>
---
 apis/kubedb/v1alpha2/constants.go     | 18 ++++++++++++++++++
 apis/kubedb/v1alpha2/druid_helpers.go |  6 ++++++
 2 files changed, 24 insertions(+)

diff --git a/apis/kubedb/v1alpha2/constants.go b/apis/kubedb/v1alpha2/constants.go
index 25403684e8..2853b6a8fb 100644
--- a/apis/kubedb/v1alpha2/constants.go
+++ b/apis/kubedb/v1alpha2/constants.go
@@ -732,6 +732,24 @@ const (
 	KafkaJMXMetricReporter        = "org.apache.kafka.common.metrics.JmxReporter"
 )
 
+const (
+	RabbitMQAMQPPort          = 5672
+	RabbitMQPeerDiscoveryPort = 4369
+	RabbitMQManagementUIPort  = 15672
+	RabbitMQInterNodePort     = 25672
+
+	RabbitMQVolumeData         = "data"
+	RabbitMQVolumeConfig       = "rabbitmqconfig"
+	RabbitMQVolumeTempConfig   = "temp-config"
+	RabbitMQVolumeCustomConfig = "custom-config"
+
+	RabbitMQDataDir       = "/var/lib/rabbitmq/mnesia"
+	RabbitMQPluginsDir    = "/etc/rabbitmq/"
+	RabbitMQCertDir       = "/var/private/ssl"
+	RabbitMQConfigDir     = "/config/"
+	RabbitMQTempConfigDir = "/tmp/config/"
+)
+
 // =========================== Druid Constants ============================
 const (
 	DruidConfigDirCommon              = "/opt/druid/conf/druid/cluster/_common"
diff --git a/apis/kubedb/v1alpha2/druid_helpers.go b/apis/kubedb/v1alpha2/druid_helpers.go
index 4dd48f7147..994b0beb4d 100644
--- a/apis/kubedb/v1alpha2/druid_helpers.go
+++ b/apis/kubedb/v1alpha2/druid_helpers.go
@@ -25,12 +25,14 @@ import (
 	"kubedb.dev/apimachinery/apis"
 	catalog "kubedb.dev/apimachinery/apis/catalog/v1alpha1"
 	"kubedb.dev/apimachinery/apis/kubedb"
+	"kubedb.dev/apimachinery/crds"
 
 	"gomodules.xyz/pointer"
 	v1 "k8s.io/api/core/v1"
 	meta "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"k8s.io/apimachinery/pkg/types"
 	"k8s.io/klog/v2"
+	"kmodules.xyz/client-go/apiextensions"
 	coreutil "kmodules.xyz/client-go/core/v1"
 	meta_util "kmodules.xyz/client-go/meta"
 	"kmodules.xyz/client-go/policy/secomp"
@@ -38,6 +40,10 @@ import (
 	ofst "kmodules.xyz/offshoot-api/api/v2"
 )
 
+func (d *Druid) CustomResourceDefinition() *apiextensions.CustomResourceDefinition {
+	return crds.MustCustomResourceDefinition(SchemeGroupVersion.WithResource(ResourcePluralDruid))
+}
+
 func (d *Druid) Owner() *meta.OwnerReference {
 	return meta.NewControllerRef(d, SchemeGroupVersion.WithKind(d.ResourceKind()))
 }