From c945620f54664f7507bc98ac6e92b895692ca4b7 Mon Sep 17 00:00:00 2001
From: Arnob Kumar Saha <arnob@appscode.com>
Date: Sat, 16 Mar 2024 15:23:35 +0600
Subject: [PATCH] Register dbs in extractDatabaseInfo for petset (#1178)

Signed-off-by: Arnob kumar saha <arnob@appscode.com>
---
 pkg/controller/petset/util.go | 73 +++++++++++++++++++++++++++++++++--
 1 file changed, 69 insertions(+), 4 deletions(-)

diff --git a/pkg/controller/petset/util.go b/pkg/controller/petset/util.go
index 5d6c4e27e6..7088201740 100644
--- a/pkg/controller/petset/util.go
+++ b/pkg/controller/petset/util.go
@@ -61,14 +61,79 @@ func (c *Controller) extractDatabaseInfo(ps *petsetapps.PetSet) (*databaseInfo,
 		Version: gv.Version,
 	}
 	switch owner.Kind {
+	case api.ResourceKindDruid:
+		dbInfo.opts.GVR.Resource = api.ResourcePluralDruid
+		dr, err := c.DBClient.KubedbV1alpha2().Druids(dbInfo.opts.Namespace).Get(context.TODO(), dbInfo.opts.Name, metav1.GetOptions{})
+		if err != nil {
+			return nil, err
+		}
+		dbInfo.replicasReady, dbInfo.msg, err = dr.ReplicasAreReady(c.PSLister)
+		if err != nil {
+			return nil, err
+		}
+
+	case api.ResourceKindFerretDB:
+		dbInfo.opts.GVR.Resource = api.ResourcePluralFerretDB
+		fr, err := c.DBClient.KubedbV1alpha2().FerretDBs(dbInfo.opts.Namespace).Get(context.TODO(), dbInfo.opts.Name, metav1.GetOptions{})
+		if err != nil {
+			return nil, err
+		}
+		dbInfo.replicasReady, dbInfo.msg, err = fr.ReplicasAreReady(c.PSLister)
+		if err != nil {
+			return nil, err
+		}
+
+	case api.ResourceKindPgpool:
+		dbInfo.opts.GVR.Resource = api.ResourcePluralPgpool
+		pp, err := c.DBClient.KubedbV1alpha2().Pgpools(dbInfo.opts.Namespace).Get(context.TODO(), dbInfo.opts.Name, metav1.GetOptions{})
+		if err != nil {
+			return nil, err
+		}
+		dbInfo.replicasReady, dbInfo.msg, err = pp.ReplicasAreReady(c.PSLister)
+		if err != nil {
+			return nil, err
+		}
+
+	case api.ResourceKindRabbitmq:
+		dbInfo.opts.GVR.Resource = api.ResourcePluralRabbitmq
+		mq, err := c.DBClient.KubedbV1alpha2().RabbitMQs(dbInfo.opts.Namespace).Get(context.TODO(), dbInfo.opts.Name, metav1.GetOptions{})
+		if err != nil {
+			return nil, err
+		}
+		dbInfo.replicasReady, dbInfo.msg, err = mq.ReplicasAreReady(c.PSLister)
+		if err != nil {
+			return nil, err
+		}
+
+	case api.ResourceKindSinglestore:
+		dbInfo.opts.GVR.Resource = api.ResourcePluralSinglestore
+		sdb, err := c.DBClient.KubedbV1alpha2().Singlestores(dbInfo.opts.Namespace).Get(context.TODO(), dbInfo.opts.Name, metav1.GetOptions{})
+		if err != nil {
+			return nil, err
+		}
+		dbInfo.replicasReady, dbInfo.msg, err = sdb.ReplicasAreReady(c.PSLister)
+		if err != nil {
+			return nil, err
+		}
+
+	case api.ResourceKindSolr:
+		dbInfo.opts.GVR.Resource = api.ResourcePluralSolr
+		sl, err := c.DBClient.KubedbV1alpha2().Solrs(dbInfo.opts.Namespace).Get(context.TODO(), dbInfo.opts.Name, metav1.GetOptions{})
+		if err != nil {
+			return nil, err
+		}
+		dbInfo.replicasReady, dbInfo.msg, err = sl.ReplicasAreReady(c.PSLister)
+		if err != nil {
+			return nil, err
+		}
 
-	case api.ResourceKindPostgres:
-		dbInfo.opts.GVR.Resource = api.ResourcePluralPostgres
-		pg, err := c.DBClient.KubedbV1alpha2().Postgreses(dbInfo.opts.Namespace).Get(context.TODO(), dbInfo.opts.Name, metav1.GetOptions{})
+	case api.ResourceKindZooKeeper:
+		dbInfo.opts.GVR.Resource = api.ResourcePluralZooKeeper
+		zk, err := c.DBClient.KubedbV1alpha2().ZooKeepers(dbInfo.opts.Namespace).Get(context.TODO(), dbInfo.opts.Name, metav1.GetOptions{})
 		if err != nil {
 			return nil, err
 		}
-		dbInfo.replicasReady, dbInfo.msg, err = pg.ReplicasAreReady(c.StsLister)
+		dbInfo.replicasReady, dbInfo.msg, err = zk.ReplicasAreReady(c.PSLister)
 		if err != nil {
 			return nil, err
 		}