1 files changed, 79 insertions, 0 deletions

diff --git a/src/mongo.go b/src/mongo.go
new file mode 100644
index 0000000..d00abd2
--- /dev/null
+++ b/src/mongo.go
@@ -0,0 +1,79 @@
+package main
+
+import (
+        "context"
+        "time"
+
+        "go.mongodb.org/mongo-driver/bson"
+        "go.mongodb.org/mongo-driver/mongo"
+        "go.mongodb.org/mongo-driver/mongo/options"
+)
+
+type mongoClient struct {
+        dbName, colName string
+        client          *mongo.Client
+        col             *mongo.Collection
+}
+
+func (mc *mongoClient) Connect(dbName, colName string) error {
+        var err error
+        mc.client, err = mongo.NewClient(options.Client().ApplyURI("mongodb://localhost:27017"))
+
+        if err != nil {
+                return err
+        }
+
+        ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
+        err = mc.client.Connect(ctx)
+        if err != nil {
+                return err
+        }
+
+        mc.col = mc.client.Database(dbName).Collection(colName)
+        mc.dbName = dbName
+        mc.colName = colName
+        return nil
+}
+
+func (mc *mongoClient) InsertOne(document interface{}) error {
+        _, err := mc.col.InsertOne(context.Background(), document)
+        return err
+}
+
+func (mc *mongoClient) UpdateOne(filter, update interface{}) error {
+        _, err := mc.col.UpdateOne(context.Background(), filter, update)
+        return err
+}
+
+func (mc *mongoClient) UpdateMany(filter, update interface{}) error {
+        _, err := mc.col.UpdateMany(context.Background(), filter, update)
+        return err
+}
+
+func (mc *mongoClient) Finddoc(filter bson.M) ([]bson.M, error) {
+        cur, err := mc.col.Find(context.Background(), filter)
+        if err != nil {
+                return nil, err
+        }
+
+        var results []bson.M
+        err = cur.All(context.Background(), &results)
+
+        return results, err
+}
+
+func (mc *mongoClient) Drop() error {
+        return mc.col.Drop(context.Background())
+}
+
+func (mc *mongoClient) Disconnect() error {
+        err := mc.client.Disconnect(context.Background())
+        if err != nil {
+                return err
+        }
+        mc.col = nil
+        mc.client = nil
+        mc.dbName = ""
+        mc.colName = ""
+        return nil
+}