/
store.go
72 lines (59 loc) · 1.93 KB
/
store.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
package store
import (
"context"
"github.com/nehaal10/ecommerce-api/internal/conf"
"github.com/nehaal10/ecommerce-api/internal/utils"
"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"
)
type Collection struct {
NewUser *mongo.Collection
Cutomer *mongo.Collection
Product *mongo.Collection
VendorRegister *mongo.Collection
Vendor *mongo.Collection
}
var db Collection
func SetupDBConeection(cfg conf.Config) {
clientoption := options.Client().ApplyURI(cfg.DbHost)
client, err := mongo.Connect(context.TODO(), clientoption)
utils.Checkerr(err)
dbLogin := client.Database("ecommerce").Collection("register")
dbCustomer := client.Database("ecommerce").Collection("user")
dbProduct := client.Database("ecommerce").Collection("product")
dbVendorRegister := client.Database("ecommerce").Collection("vendor_register")
dbVendor := client.Database("ecommerce").Collection("vendor")
if err := client.Ping(context.TODO(), readpref.Primary()); err != nil {
panic(err)
}
registerModel := mongo.IndexModel{
Keys: bson.M{"email_id": 1},
Options: options.Index().SetUnique(true),
}
_, err = dbLogin.Indexes().CreateOne(context.TODO(), registerModel)
utils.Checkerr(err)
vendormodel := mongo.IndexModel{
Keys: bson.M{"email_id": 1},
Options: options.Index().SetUnique(true),
}
_, err = dbVendorRegister.Indexes().CreateOne(context.TODO(), vendormodel)
utils.Checkerr(err)
vendormodeld := mongo.IndexModel{
Keys: bson.M{"email_id": 1},
Options: options.Index().SetUnique(true),
}
_, err = dbVendor.Indexes().CreateOne(context.TODO(), vendormodeld)
utils.Checkerr(err)
db = Collection{
NewUser: dbLogin,
Cutomer: dbCustomer,
Product: dbProduct,
VendorRegister: dbVendorRegister,
Vendor: dbVendor,
}
}
func GetdbConnection() Collection {
return db
}