db golang added

vinodnextcoder · vinodnextcoder · commit b6552a7387e5 · 2024-01-20T20:00:47.000+05:30
diff --git a/.env.dev b/.env.dev
@@ -1,3 +1,4 @@
 S3_BUCKET=YOURS3BUCKET
 SECRET_KEY=YOURSECRETKEYGOESHERE
-PORT=3001
+PORT=3001
+DBNAME=test
diff --git a/main.go b/main.go
@@ -5,13 +5,13 @@ import (
 	"log"
 	"net/http"
 	"os"
-	"context"
-    "time"
- 
+	// "context"
+    // "time"
+    "go.mongodb.org/mongo-driver/bson"
     "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"
+    // "go.mongodb.org/mongo-driver/mongo"
+    // "go.mongodb.org/mongo-driver/mongo/options"
+    // "go.mongodb.org/mongo-driver/mongo/readpref"
 
     "github.com/joho/godotenv"
 	"github.com/gin-gonic/gin"
@@ -56,6 +56,37 @@ func main() {
     // Ping mongoDB with Ping method
     mongoconnect.Ping(client, ctx)
 
+    
+    // Create  a object of type interface to  store
+    // the bson values, that  we are inserting into database.
+    var document interface{}
+     
+     
+    document = bson.D{
+        {"rollNo", 175},
+        {"maths", 80},
+        {"science", 90},
+        {"computer", 95},
+    }
+     
+    dbname := os.Getenv("DBNAME")
+    // insertOne accepts client , context, database
+    // name collection name and an interface that 
+    // will be inserted into the  collection.
+    // insertOne returns an error and a result of 
+    // insert in a single document into the collection.
+    insertOneResult, err := mongoconnect.Insertdata(client, ctx, dbname, "marks", document)
+     
+    // handle the error
+    if err != nil {
+        panic(err)
+    }
+     
+    // print the insertion id of the document, 
+    // if it is inserted.
+    fmt.Println("Result of InsertOne")
+    fmt.Println(insertOneResult.InsertedID)
+
   port := os.Getenv("PORT")
 
 	router := gin.Default()
@@ -81,58 +112,3 @@ func main() {
 func helloCall(c *gin.Context) {
 	c.JSON(http.StatusOK, gin.H{"message": "Hello, You created a Web App!"})
 }
-// 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
-}
diff --git a/mongoconnect/config.go b/mongoconnect/config.go
@@ -4,6 +4,7 @@ import (
 	"fmt"
 	"context"
     "time"
+    // "go.mongodb.org/mongo-driver/bson"
     "go.mongodb.org/mongo-driver/mongo"
     "go.mongodb.org/mongo-driver/mongo/options"
     "go.mongodb.org/mongo-driver/mongo/readpref"
@@ -65,3 +66,18 @@ if err := client.Ping(ctx, readpref.Primary()); err != nil {
 fmt.Println("connected successfully")
 return nil
 }
+ 
+// insertOne is a user defined method, used to insert
+// documents into collection returns result of InsertOne
+// and error if any.
+func Insertdata(client *mongo.Client, ctx context.Context, dataBase, col string, doc interface{}) (*mongo.InsertOneResult, error) {
+ 
+    // select database and collection ith Client.Database method 
+    // and Database.Collection method
+    collection := client.Database(dataBase).Collection(col)
+     
+    // InsertOne accept two argument of type Context 
+    // and of empty interface   
+    result, err := collection.InsertOne(ctx, doc)
+    return result, err
+}
