forked from ariandi/gocom
/
db.go
99 lines (65 loc) · 1.76 KB
/
db.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
96
97
98
99
package gocom
import (
"sync"
"github.com/frizkey/gocom/config"
"go.uber.org/zap"
"gorm.io/driver/postgres"
"gorm.io/gorm"
)
var dbMap map[string]*gorm.DB = map[string]*gorm.DB{}
var dbMutex sync.Mutex
var dbCreatorMap map[string]DBCreatorFunc = map[string]DBCreatorFunc{}
type DBCreatorFunc func(dsn string) (*gorm.DB, error)
func RegDBCreator(typeName string, creator DBCreatorFunc) {
dbCreatorMap[typeName] = creator
}
func DB(name ...string) *gorm.DB {
var targetName string
if len(name) > 0 {
targetName = name[0]
}
if targetName == "" {
targetName = "default"
}
ret, ok := dbMap[targetName]
if !ok {
dbMutex.Lock()
defer dbMutex.Unlock()
// check whether other thread already create the db
ret, ok = dbMap[targetName]
if !ok {
if config.HasConfig("app.db."+targetName+".type") && config.HasConfig("app.db."+targetName+".dsn") {
dbType := config.Get("app.db." + targetName + ".type")
creator, ok := dbCreatorMap[dbType]
if ok {
dsn := config.Get("app.db." + targetName + ".dsn")
poolSize := config.GetInt("app.db."+targetName+".poolSize", 20)
debug := config.GetBool("app.db."+targetName+".debug", false)
var err error
ret, err = creator(dsn)
if err == nil {
dbMap[targetName] = ret
if debug {
ret.Debug()
}
if poolSize > 0 {
db, errDB := ret.DB()
if errDB == nil {
db.SetMaxOpenConns(poolSize)
}
}
Logger().Info("Conected to DB", zap.Any("dbname", targetName))
} else {
Logger().Info("Error Conected to DB", zap.Error(err))
}
}
}
}
}
return ret
}
func init() {
RegDBCreator("postgresql", func(dsn string) (*gorm.DB, error) {
return gorm.Open(postgres.Open(dsn), &gorm.Config{})
})
}