mirror of
https://github.com/go-eagle/eagle.git
synced 2025-10-06 09:06:55 +08:00
40 lines
631 B
Go
40 lines
631 B
Go
package model
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"gorm.io/gorm"
|
|
|
|
"github.com/go-eagle/eagle/pkg/config"
|
|
"github.com/go-eagle/eagle/pkg/storage/orm"
|
|
)
|
|
|
|
// DB 数据库全局变量
|
|
var DB *gorm.DB
|
|
|
|
// Init 初始化数据库
|
|
func Init() *gorm.DB {
|
|
cfg, err := loadConf()
|
|
if err != nil {
|
|
panic(fmt.Sprintf("load orm conf err: %v", err))
|
|
}
|
|
|
|
DB = orm.NewMySQL(cfg)
|
|
return DB
|
|
}
|
|
|
|
// GetDB 返回默认的数据库
|
|
func GetDB() *gorm.DB {
|
|
return DB
|
|
}
|
|
|
|
// loadConf load gorm config
|
|
func loadConf() (ret *orm.Config, err error) {
|
|
var cfg orm.Config
|
|
if err := config.Load("database", &cfg); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &cfg, nil
|
|
}
|