77 lines
1.3 KiB
Go
77 lines
1.3 KiB
Go
|
package moto
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"git.nobla.cn/golang/moto/common/db"
|
||
|
|
"git.nobla.cn/golang/moto/models"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
defaultUsers = map[string]string{
|
||
|
|
"1000": "admin",
|
||
|
|
}
|
||
|
|
|
||
|
|
defaultDepartments = map[string]string{
|
||
|
|
"admin": "客服部",
|
||
|
|
}
|
||
|
|
|
||
|
|
defaultRoles = map[string]string{
|
||
|
|
"admin": "管理员",
|
||
|
|
}
|
||
|
|
)
|
||
|
|
|
||
|
|
func migrate(ctx context.Context) (err error) {
|
||
|
|
var (
|
||
|
|
count int64
|
||
|
|
deptID string
|
||
|
|
roleID string
|
||
|
|
)
|
||
|
|
tx := db.DB().WithContext(ctx)
|
||
|
|
tx.Model(&models.Department{}).Count(&count)
|
||
|
|
if count <= 0 {
|
||
|
|
for _, v := range defaultDepartments {
|
||
|
|
deptModel := &models.Department{
|
||
|
|
Parent: "",
|
||
|
|
Name: v,
|
||
|
|
}
|
||
|
|
if err = tx.Save(deptModel).Error; err == nil {
|
||
|
|
if deptID == "" {
|
||
|
|
deptID = deptModel.ID
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
tx.Model(&models.Role{}).Count(&count)
|
||
|
|
if count <= 0 {
|
||
|
|
for _, v := range defaultRoles {
|
||
|
|
roleModel := &models.Role{
|
||
|
|
Name: v,
|
||
|
|
Permissions: "[]",
|
||
|
|
}
|
||
|
|
if err = tx.Save(roleModel).Error; err == nil {
|
||
|
|
if roleID == "" {
|
||
|
|
roleID = roleModel.ID
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
tx.Model(&models.User{}).Count(&count)
|
||
|
|
if count <= 0 {
|
||
|
|
for k, v := range defaultUsers {
|
||
|
|
tx.Save(&models.User{
|
||
|
|
UID: k,
|
||
|
|
Department: deptID,
|
||
|
|
Role: roleID,
|
||
|
|
Username: v,
|
||
|
|
Password: "Passwd#" + k,
|
||
|
|
Email: k + "@echo.me",
|
||
|
|
Avatar: "/profile/avatar/" + k,
|
||
|
|
Gender: "man",
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|