-
Notifications
You must be signed in to change notification settings - Fork 0
/
adapter.go
95 lines (70 loc) · 1.84 KB
/
adapter.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
package mongodb_adapter
import (
"context"
"errors"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/mongo/options"
"log"
"sync"
)
type MongoClient struct {
Client *mongo.Client
Collection *mongo.Collection
}
var once sync.Once
var conn *mongo.Client
func ConnectMongo(mongoUrl string, dbName, collectionName string) *MongoClient {
var err error
once.Do(func() {
conn, err = mongo.Connect(context.TODO(), options.Client().ApplyURI(mongoUrl))
if err != nil {
log.Fatal(err)
}
})
collection := conn.Database(dbName).Collection(collectionName)
return &MongoClient{Client: conn, Collection: collection}
}
func (c *MongoClient) GetData(filter bson.D, returnData interface{}) error {
err := c.Collection.FindOne(context.TODO(), filter).Decode(returnData)
if err != nil {
err = errors.New("Document Not Found")
}
return err
}
func (c *MongoClient) GetAll(filter bson.D, returnData interface{}) error {
findOptions := options.Find()
findOptions.SetLimit(100)
cursor, err := c.Collection.Find(context.TODO(), filter, findOptions)
if err != nil {
err = errors.New("Document Not Found:" + err.Error())
} else {
err = cursor.All(context.TODO(), returnData)
if err != nil {
err = errors.New("Document Cursor")
}
}
return err
}
func (c *MongoClient) AddData(insert interface{}) error {
_, err := c.Collection.InsertOne(context.TODO(), insert)
if err != nil {
return err
}
return nil
}
func (c *MongoClient) UpdateData(filter bson.D, update bson.D) error {
_, err := c.Collection.UpdateOne(context.TODO(), filter, update)
if err != nil {
return err
}
return nil
}
func (c *MongoClient) DeleteData(filter bson.D) error {
options := options.Delete()
_, err := c.Collection.DeleteOne(context.TODO(), filter, options)
if err != nil {
return err
}
return nil
}