43 lines
932 B
Go
43 lines
932 B
Go
|
package initialize
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/glebarez/sqlite"
|
||
|
|
"gorm.io/gorm"
|
||
|
|
"miniapp/config"
|
||
|
|
"miniapp/global"
|
||
|
|
"miniapp/initialize/internal"
|
||
|
|
)
|
||
|
|
|
||
|
|
// GormSqlite 初始化Sqlite数据库
|
||
|
|
func GormSqlite() *gorm.DB {
|
||
|
|
s := global.GVA_CONFIG.Sqlite
|
||
|
|
if s.Dbname == "" {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
if db, err := gorm.Open(sqlite.Open(s.Dsn()), internal.Gorm.Config(s.Prefix, s.Singular)); err != nil {
|
||
|
|
panic(err)
|
||
|
|
} else {
|
||
|
|
sqlDB, _ := db.DB()
|
||
|
|
sqlDB.SetMaxIdleConns(s.MaxIdleConns)
|
||
|
|
sqlDB.SetMaxOpenConns(s.MaxOpenConns)
|
||
|
|
return db
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// GormSqliteByConfig 初始化Sqlite数据库用过传入配置
|
||
|
|
func GormSqliteByConfig(s config.Sqlite) *gorm.DB {
|
||
|
|
if s.Dbname == "" {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
if db, err := gorm.Open(sqlite.Open(s.Dsn()), internal.Gorm.Config(s.Prefix, s.Singular)); err != nil {
|
||
|
|
panic(err)
|
||
|
|
} else {
|
||
|
|
sqlDB, _ := db.DB()
|
||
|
|
sqlDB.SetMaxIdleConns(s.MaxIdleConns)
|
||
|
|
sqlDB.SetMaxOpenConns(s.MaxOpenConns)
|
||
|
|
return db
|
||
|
|
}
|
||
|
|
}
|