mirror of
https://github.com/go-eagle/eagle.git
synced 2025-09-26 20:41:26 +08:00
38 lines
618 B
Go
38 lines
618 B
Go
package model
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"gorm.io/gorm"
|
|
|
|
"github.com/go-eagle/eagle/pkg/storage/orm"
|
|
)
|
|
|
|
const (
|
|
// DefaultDatabase default database
|
|
DefaultDatabase = "default"
|
|
// UserDatabase user database
|
|
UserDatabase = "user"
|
|
)
|
|
|
|
// Init 初始化数据库
|
|
func Init() {
|
|
err := orm.New(
|
|
DefaultDatabase,
|
|
UserDatabase,
|
|
)
|
|
if err != nil {
|
|
panic(fmt.Sprintf("new orm database err: %v", err))
|
|
}
|
|
}
|
|
|
|
// GetDB 返回默认的数据库
|
|
func GetDB() (*gorm.DB, error) {
|
|
return orm.GetDB(DefaultDatabase)
|
|
}
|
|
|
|
// GetUserDB 获取用户数据库实例
|
|
func GetUserDB() (*gorm.DB, error) {
|
|
return orm.GetDB(UserDatabase)
|
|
}
|