/
odm_mock.go
35 lines (30 loc) · 957 Bytes
/
odm_mock.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
package mongo
import (
"context"
"go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/mongo/options"
"log"
"os"
)
// GetMockDmManager returns dmManager.
func GetMockDmManager() *dmManager {
onceDmManager.Do(func() {
log.Println("[INFO] Starting Initializing Singleton DB Manager")
singletonDmManager = &dmManager{}
singletonDmManager.initMockMongoConnection()
})
return singletonDmManager
}
func (dm *dmManager) initMockMongoConnection() {
ctx := context.Background()
dm.Ctx = ctx
clientOpts := options.Client().ApplyURI("mongodb://" + os.Getenv("MONGO_USERNAME") + ":" + os.Getenv("MONGO_PASSWORD") + "@" + os.Getenv("MONGO_SERVER") + ":" + os.Getenv("MONGO_PORT"))
client, err := mongo.Connect(ctx, clientOpts)
if err != nil {
log.Println("[ERROR] DB Connection error:", err.Error())
return
}
db := client.Database(os.Getenv("DATABASE_NAME"))
dm.Db = db
log.Println("[INFO] Initialized Singleton DB Manager")
}