/
gen_mongo_config.go
70 lines (58 loc) · 1.43 KB
/
gen_mongo_config.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
package nested
import (
"context"
"fmt"
"github.com/ezbuy/ezorm/v2/pkg/db"
"github.com/ezbuy/wrapper/database"
"go.mongodb.org/mongo-driver/mongo"
)
type SetupOption struct {
monitor database.Monitor
postHooks []func()
}
type SetupOptionFn func(opts *SetupOption)
func WithStatsDMonitor(app string) SetupOptionFn {
return func(opts *SetupOption) {
opts.monitor = database.NewStatsDPoolMonitor(app)
}
}
func WithPrometheusMonitor(app, gatewayAddress string) SetupOptionFn {
return func(opts *SetupOption) {
opts.monitor = database.NewPrometheusPoolMonitor(app, gatewayAddress)
}
}
func WithPostHooks(fn ...func()) SetupOptionFn {
return func(opts *SetupOption) {
opts.postHooks = append(opts.postHooks, fn...)
}
}
var mongoDriver *db.MongoDriver
func MgoSetup(config *db.MongoConfig, opts ...SetupOptionFn) {
sopt := &SetupOption{}
for _, opt := range opts {
opt(sopt)
}
// setup the indexes
sopt.postHooks = append(sopt.postHooks,
UserIndexesFunc,
)
var dopt []db.MongoDriverOption
if sopt.monitor != nil {
dopt = append(dopt, db.WithPoolMonitor(database.NewMongoDriverMonitor(sopt.monitor)))
}
db.Setup(config)
var err error
mongoDriver, err = db.NewMongoDriver(
context.Background(),
dopt...,
)
if err != nil {
panic(fmt.Errorf("failed to create mongodb driver: %s", err))
}
for _, hook := range sopt.postHooks {
hook()
}
}
func Col(col string) *mongo.Collection {
return mongoDriver.GetCol(col)
}