31 lines
506 B
Go
31 lines
506 B
Go
package global
|
|
|
|
import (
|
|
"gorm.io/gorm"
|
|
"work_cation/cfg"
|
|
"work_cation/models"
|
|
"work_cation/models/repo"
|
|
"work_cation/pkg/gormx"
|
|
)
|
|
|
|
var DB *gorm.DB
|
|
|
|
func InitDB() {
|
|
var err error
|
|
DB, err = gormx.New(gormx.Config{DSN: cfg.T.DbPath, Type: "sqlite3"})
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
err = DB.AutoMigrate(
|
|
&models.ErlangCards{},
|
|
&models.ErlangCardCollections{},
|
|
&models.History{},
|
|
&models.Users{},
|
|
&models.UserFollows{},
|
|
)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
repo.UserRepo.GetUserInfo(DB)
|
|
}
|