go mongo pkg

vinodnextcoder · vinodnextcoder · commit 4295edb2163c · 2024-01-19T11:10:22.000+05:30
diff --git a/main.go b/main.go
@@ -8,6 +8,7 @@ import (
 	"context"
     "time"
  
+    "github.com/vinodnextcoder/golang-mongo-server/mongoconnect"
     "go.mongodb.org/mongo-driver/mongo"
     "go.mongodb.org/mongo-driver/mongo/options"
     "go.mongodb.org/mongo-driver/mongo/readpref"
@@ -17,6 +18,7 @@ import (
 	swaggerFiles "github.com/swaggo/files"
 	ginSwagger "github.com/swaggo/gin-swagger"
 	_ "github.com/vinodnextcoder/golang-mongo-server/docs"
+  
 )
 
 // @title           Swagger Example API
@@ -42,18 +44,17 @@ func main() {
 
   // Get Client, Context, CancelFunc and 
     // err from connect method.
-    client, ctx, cancel, err := connect("mongodb://localhost:27017/test")
+    client, ctx, cancel, err := mongoconnect.Connect("mongodb://localhost:27017/test")
     if err != nil {
         panic(err)
     }
      
     // Release resource when the main
     // function is returned.
-    defer close(client, ctx, cancel)
+    defer mongoconnect.Close(client, ctx, cancel)
 	
-     
     // Ping mongoDB with Ping method
-    ping(client, ctx)
+    mongoconnect.Ping(client, ctx)
 
   port := os.Getenv("PORT")
 
diff --git a/mongoconnect/config.go b/mongoconnect/config.go
@@ -0,0 +1,67 @@
+package mongoconnect
+
+import (
+	"fmt"
+	"context"
+    "time"
+    "go.mongodb.org/mongo-driver/mongo"
+    "go.mongodb.org/mongo-driver/mongo/options"
+    "go.mongodb.org/mongo-driver/mongo/readpref"
+)
+
+
+// This is a user defined method to close resources.
+// This method closes mongoDB connection and cancel context.
+func Close(client *mongo.Client, ctx context.Context,
+	cancel context.CancelFunc){
+	 
+// CancelFunc to cancel to context
+defer cancel()
+
+// client provides a method to close 
+// a mongoDB connection.
+defer func(){
+
+ // client.Disconnect method also has deadline.
+ // returns error if any,
+ if err := client.Disconnect(ctx); err != nil{
+	 panic(err)
+ }
+}()
+}
+
+// This is a user defined method that returns mongo.Client, 
+// context.Context, context.CancelFunc and error.
+// mongo.Client will be used for further database operation.
+// context.Context will be used set deadlines for process.
+// context.CancelFunc will be used to cancel context and 
+// resource associated with it.
+
+func Connect(uri string)(*mongo.Client, context.Context, 
+				   context.CancelFunc, error) {
+					
+// ctx will be used to set deadline for process, here 
+// deadline will of 30 seconds.
+ctx, cancel := context.WithTimeout(context.Background(), 
+								30 * time.Second)
+
+// mongo.Connect return mongo.Client method
+client, err := mongo.Connect(ctx, options.Client().ApplyURI(uri))
+return client, ctx, cancel, err
+}
+
+// This is a user defined method that accepts 
+// mongo.Client and context.Context
+// This method used to ping the mongoDB, return error if any.
+func Ping(client *mongo.Client, ctx context.Context) error{
+
+// mongo.Client has Ping to ping mongoDB, deadline of 
+// the Ping method will be determined by cxt
+// Ping method return error if any occurred, then
+// the error can be handled.
+if err := client.Ping(ctx, readpref.Primary()); err != nil {
+ return err
+}
+fmt.Println("connected successfully")
+return nil
+}
