refac: mongo connection

Author: lawRathod

connections/database/interface.go

@@ -4,7 +4,6 @@ import (
 	"fmt"
 	"net/http"
 
-	"go.mongodb.org/mongo-driver/mongo"
 	"gorm.io/gorm"
 )
 
@@ -23,13 +22,12 @@ type DBConnection interface {
 }
 
 type IDBQueryClient interface {
-	Connect() error
 	GetDB() interface{}
+	Connect() error
 }
 
 type DBQueryClient struct {
-	Client *mongo.Client
-	Params *DBConnectionParams
+	Client IDBQueryClient
 }
 
 type DBConnectionParams struct {

connections/database/mongoDB.go

@@ -3,39 +3,46 @@ package database
 import (
 	"context"
 	"fmt"
-	"os"
 	"time"
 
 	"go.mongodb.org/mongo-driver/mongo"
 	"go.mongodb.org/mongo-driver/mongo/options"
 	"go.mongodb.org/mongo-driver/mongo/readpref"
 )
 
-func (c *DBQueryClient) Connect() error {
-	uri := os.Getenv("MONGO_URI")
+type MongoConnection struct {
+	db *mongo.Client
+}
+
+func NewMongoConnection(params *DBConnectionParams) (*MongoConnection, error) {
+	uri := params.URI
 	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
 	defer cancel()
 
 	client, err := mongo.Connect(ctx, options.Client().ApplyURI(uri))
 	if err != nil {
-		fmt.Println("Failed to connect to db")
+		fmt.Printf("Failed to connect to db: %v \n", err)
+		return nil, err
 	}
 
 	// Ping the primary
 	if err := client.Ping(ctx, readpref.Primary()); err != nil {
-		fmt.Println("Failed to ping after connecting to the db")
+		return nil, err
 	}
 	fmt.Println("Successfully connected and pinged.")
 
-	c.Client = client
-	return err
+	return &MongoConnection{db: client}, nil
+}
+
+func (c *MongoConnection) GetDB() *mongo.Client {
+	return c.db
 }
 
-func (c *DBQueryClient) GetDB() *mongo.Client {
-	err := c.Connect()
+func (c *DBQueryClient) Connect(params *DBConnectionParams) (*mongo.Client, error) {
+	conn, err := NewMongoConnection(params)
 	if err != nil {
-		panic(err)
+		return nil, err
 	}
 
-	return c.Client
+	return conn.GetDB(), nil
 }

connections/database/mongoDB_test.go

@@ -3,6 +3,7 @@ package database
 import (
 	"context"
 	"fmt"
+	"os"
 	"testing"
 
 	"go.mongodb.org/mongo-driver/bson"
@@ -12,13 +13,14 @@ import (
 
 func TestBuffer_InsertRead(t *testing.T) {
 	var c DBQueryClient
-	err := c.Connect()
+	params := &DBConnectionParams{URI: os.Getenv("MONGO_URI")}
+	client, err := c.Connect(params)
 	if err != nil {
 		fmt.Println(err)
 		t.Fail()
 	}
 
-	coll := c.Client.Database("test").Collection("sample")
+	coll := client.Database("test").Collection("sample")
 	res, err := coll.InsertOne(context.TODO(), bson.D{{Key: "name", Value: "Alice"}})
 	if err != nil {
 		fmt.Println(err)