|
|
|
|
|
package dbm
|
|
|
|
|
|
|
|
|
|
|
|
import (
|
|
|
|
|
|
"fmt"
|
|
|
|
|
|
"github.com/han-joker/moo-layout/api/moo/confm"
|
|
|
|
|
|
"github.com/han-joker/moo-layout/api/moo/logm"
|
|
|
|
|
|
"github.com/han-joker/moo-layout/api/moo/toolm"
|
|
|
|
|
|
"gorm.io/driver/sqlite"
|
|
|
|
|
|
"gorm.io/gorm"
|
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
const defaultType = "sqlite"
|
|
|
|
|
|
|
|
|
|
|
|
type DSN = string
|
|
|
|
|
|
|
|
|
|
|
|
var pool = map[DSN]*gorm.DB{}
|
|
|
|
|
|
|
|
|
|
|
|
type Option struct {
|
|
|
|
|
|
DSN
|
|
|
|
|
|
Options []gorm.Option
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
// New 创建对象
|
|
|
|
|
|
func New(options ...Option) *gorm.DB {
|
|
|
|
|
|
option := verifyOption(options)
|
|
|
|
|
|
return create(option)
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
// Get 存在直接返回，否则创建、存储再返回
|
|
|
|
|
|
func Get(options ...Option) *gorm.DB {
|
|
|
|
|
|
option := verifyOption(options)
|
|
|
|
|
|
if !Has(option.DSN) {
|
|
|
|
|
|
pool[option.DSN] = create(option)
|
|
|
|
|
|
}
|
|
|
|
|
|
return pool[option.DSN]
|
|
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
func verifyOption(options []Option) Option {
|
|
|
|
|
|
t := toolm.StringDefault(confm.Get().String("app.databaseType"), defaultType)
|
|
|
|
|
|
option := Option{
|
|
|
|
|
|
DSN: confm.Get().String(fmt.Sprintf("%s.DSN", t)),
|
|
|
|
|
|
Options: []gorm.Option{},
|
|
|
|
|
|
}
|
|
|
|
|
|
if len(options) > 0 {
|
|
|
|
|
|
if options[0].DSN != "" {
|
|
|
|
|
|
option.DSN = options[0].DSN
|
|
|
|
|
|
}
|
|
|
|
|
|
if options[0].Options != nil {
|
|
|
|
|
|
option.Options = options[0].Options
|
|
|
|
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
return option
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
// Has 存在返回 true，否则返回 false
|
|
|
|
|
|
func Has(dialector DSN) bool {
|
|
|
|
|
|
_, has := pool[dialector]
|
|
|
|
|
|
return has && pool[dialector] != nil
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
func create(option Option) *gorm.DB {
|
|
|
|
|
|
db, err := gorm.Open(sqlite.Open(option.DSN), option.Options...)
|
|
|
|
|
|
if err != nil {
|
|
|
|
|
|
logm.Get().Error(err)
|
|
|
|
|
|
}
|
|
|
|
|
|
return db
|
|
|
|
|
|
}
|
