143 lines
3.6 KiB
Go
143 lines
3.6 KiB
Go
package startup
|
|
|
|
import (
|
|
"fmt"
|
|
"honoka-chan/internal/model/user"
|
|
"honoka-chan/internal/schema/api/profile"
|
|
"honoka-chan/internal/schema/api/unit"
|
|
"honoka-chan/internal/utils"
|
|
"honoka-chan/pkg/db"
|
|
"strconv"
|
|
"time"
|
|
)
|
|
|
|
func InitUserData(userId int) {
|
|
userList := []user.Users{}
|
|
if userId != 0 {
|
|
err := db.UserEng.Table("users").Where("user_id = ?", userId).Find(&userList)
|
|
utils.CheckErr(err)
|
|
} else {
|
|
err := db.UserEng.Table("users").Asc("id").Find(&userList)
|
|
utils.CheckErr(err)
|
|
}
|
|
|
|
// 同步用户表
|
|
// db.UserEng.ShowSQL(true)
|
|
// db.UserEng.Table("user_info_m").Sync2(new(user.UserInfo))
|
|
// db.UserEng.Table("user_pref").Sync2(new(UserPref))
|
|
|
|
// 开始会话
|
|
session := db.UserEng.NewSession()
|
|
defer session.Close()
|
|
|
|
if err := session.Begin(); err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
for _, u := range userList {
|
|
// 检查用户配置
|
|
exists, err := db.UserEng.Table("user_pref").Where("user_id = ?", u.UserID).Exist()
|
|
utils.CheckErr(err)
|
|
|
|
if !exists {
|
|
// 默认中心成员（）
|
|
var oId int
|
|
_, err = db.MainEng.Table("common_unit_m").Cols("unit_owning_user_id").Where("unit_id = ?", 31).Get(&oId)
|
|
utils.CheckErr(err)
|
|
fmt.Println("Center UnitOwningUserId:", oId)
|
|
userPref := user.UserPref{
|
|
UserID: u.UserID,
|
|
AwardID: 1, // 音乃木坂学生
|
|
BackgroundID: 1, // 初始背景
|
|
UnitOwningUserID: oId,
|
|
UserName: "音乃木坂学生",
|
|
UserLevel: 1,
|
|
UserDesc: "你好。",
|
|
InviteCode: strconv.Itoa(u.UserID),
|
|
UpdateTime: time.Now().Unix(),
|
|
}
|
|
_, err = session.Table("user_pref").Insert(&userPref)
|
|
utils.CheckErr(err)
|
|
// fmt.Println("UserPref Id", userPref.Id)
|
|
}
|
|
|
|
// 检查用户信息
|
|
exists, err = db.UserEng.Table("user_info_m").Where("user_id = ?", u.UserID).Exist()
|
|
|
|
// 检查用户卡组配置
|
|
exists, err = db.UserEng.Table("user_deck").Where("user_id = ?", u.UserID).Asc("deck_id").Exist()
|
|
utils.CheckErr(err)
|
|
// fmt.Println("UserDeck exists:", exists)
|
|
|
|
if !exists {
|
|
userDeck := unit.UserDeckData{
|
|
DeckID: 1,
|
|
MainFlag: 1,
|
|
DeckName: "队伍A",
|
|
UserID: u.UserID,
|
|
InsertDate: time.Now().Unix(),
|
|
}
|
|
|
|
// 默认队伍
|
|
_, err = session.Table("user_deck").Insert(&userDeck)
|
|
if err != nil {
|
|
session.Rollback()
|
|
panic(err)
|
|
}
|
|
userDeckId := userDeck.ID
|
|
// fmt.Println("New UserDeck:", userDeckId)
|
|
|
|
// 默认卡组
|
|
unitIds := []int{}
|
|
err = db.MainEng.Table("unit_m").Cols("unit_id").Where("album_series_id = ?", 615).Find(&unitIds)
|
|
if err != nil {
|
|
session.Rollback()
|
|
panic(err)
|
|
}
|
|
|
|
unitData := []profile.UnitData{}
|
|
err = db.MainEng.Table("common_unit_m").In("unit_id", unitIds).Find(&unitData)
|
|
if err != nil {
|
|
session.Rollback()
|
|
panic(err)
|
|
}
|
|
// fmt.Println(unitData)
|
|
|
|
position := 1
|
|
for _, u := range unitData {
|
|
unitDeckData := unit.UnitDeckData{
|
|
UserDeckID: userDeckId,
|
|
UnitOwningUserID: u.UnitOwningUserID,
|
|
UnitID: u.UnitID,
|
|
Position: position,
|
|
Level: 100,
|
|
LevelLimitID: 1,
|
|
DisplayRank: 2,
|
|
Love: 1000,
|
|
UnitSkillLevel: 8,
|
|
IsRankMax: true,
|
|
IsLoveMax: true,
|
|
IsLevelMax: true,
|
|
IsSigned: u.IsSigned,
|
|
BeforeLove: 1000,
|
|
MaxLove: 1000,
|
|
InsertData: time.Now().Unix(),
|
|
}
|
|
_, err = session.Table("user_deck_unit").Insert(&unitDeckData)
|
|
if err != nil {
|
|
session.Rollback()
|
|
panic(err)
|
|
}
|
|
// fmt.Println("New DeckUnit:", unitDeckData.Id)
|
|
|
|
position++
|
|
}
|
|
|
|
// 结束会话
|
|
if err = session.Commit(); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
}
|
|
}
|