mirror of https://gitee.com/godoos/godoos.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
28 lines
522 B
28 lines
522 B
package model
|
|
|
|
import (
|
|
"godo/libs"
|
|
|
|
_ "github.com/ncruces/go-sqlite3/embed"
|
|
"github.com/ncruces/go-sqlite3/gormlite"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
var Db *gorm.DB
|
|
|
|
func InitDB() {
|
|
dbPath := libs.GetSystemDb()
|
|
db, err := gorm.Open(gormlite.Open(dbPath), &gorm.Config{})
|
|
if err != nil {
|
|
return
|
|
}
|
|
Db = db
|
|
// 自动迁移模式
|
|
db.AutoMigrate(&SysDisk{})
|
|
// 初始化 SysDisk 记录
|
|
initSysDisk(db)
|
|
db.AutoMigrate(&LocalProxy{})
|
|
db.AutoMigrate(&SysUser{})
|
|
db.AutoMigrate(&ClientUser{})
|
|
db.AutoMigrate(&ServerUser{})
|
|
}
|
|
|