YumeMichi
f6466989c6
- Add user_live_status to store hi score, hi combo count, and clear count per chart - Expand user_live_goal to persist achieved live_goal_reward_id entries instead of only coarse goal types - Update live reward handling to refresh live status and sync newly achieved live goals on every clear - Return live status and achieved goal ids from stored user data in /api/live/status - Aggregate profile liveCnt from stored user_live_status instead of hardcoded values - Load complete-rank thresholds from live info for goal evaluation - Migrate legacy user_live_record data into user_live_status and backfill achievable live goals on startup Signed-off-by: Sean Du <do4suki@gmail.com>
488 lines
14 KiB
Go
488 lines
14 KiB
Go
package startup
|
|
|
|
import (
|
|
"honoka-chan/internal/constant"
|
|
ghomemodel "honoka-chan/internal/model/ghome"
|
|
loginmodel "honoka-chan/internal/model/login"
|
|
unitmodel "honoka-chan/internal/model/unit"
|
|
usermodel "honoka-chan/internal/model/user"
|
|
"honoka-chan/pkg/db"
|
|
"log"
|
|
"sort"
|
|
"time"
|
|
|
|
"xorm.io/xorm"
|
|
)
|
|
|
|
var (
|
|
userEng *xorm.Session
|
|
)
|
|
|
|
func CreateTables() {
|
|
// db.UserEng.ShowSQL(true)
|
|
db.UserEng.Sync2(new(ghomemodel.DeviceKey))
|
|
db.UserEng.Sync2(new(loginmodel.AuthKey))
|
|
db.UserEng.Sync2(new(usermodel.UserAccessoryWear))
|
|
db.UserEng.Sync2(new(usermodel.UserDeck))
|
|
db.UserEng.Sync2(new(usermodel.UserDeckUnit))
|
|
db.UserEng.Sync2(new(usermodel.UserKey))
|
|
db.UserEng.Sync2(new(usermodel.UserLiveGoal))
|
|
db.UserEng.Sync2(new(usermodel.UserLiveStatus))
|
|
db.UserEng.Sync2(new(usermodel.UserLiveInProgress))
|
|
db.UserEng.Sync2(new(usermodel.UserLiveRecord))
|
|
db.UserEng.Sync2(new(usermodel.UserFriend))
|
|
db.UserEng.Sync2(new(usermodel.UserGreet))
|
|
db.UserEng.Sync2(new(usermodel.UserPref))
|
|
db.UserEng.Sync2(new(usermodel.Users))
|
|
db.UserEng.Sync2(new(usermodel.UserUnit))
|
|
db.UserEng.Sync2(new(usermodel.UserUnitSkillEquip))
|
|
|
|
MigrateUserPref()
|
|
MigrateUserLiveData()
|
|
}
|
|
|
|
func MigrateUserPref() {
|
|
session := db.UserEng.NewSession()
|
|
defer session.Close()
|
|
|
|
prefList := []usermodel.UserPref{}
|
|
err := session.Table(new(usermodel.UserPref)).
|
|
Where("profile_version < ?", usermodel.CurrentUserPrefProfileVersion).
|
|
Or("profile_version IS NULL").
|
|
Find(&prefList)
|
|
if err != nil {
|
|
log.Fatalln("迁移 user_pref 失败:", err.Error())
|
|
}
|
|
|
|
for _, pref := range prefList {
|
|
pref.ApplyProfileDefaults()
|
|
_, err = session.Table(new(usermodel.UserPref)).
|
|
ID(pref.ID).
|
|
Cols(usermodel.UserPrefProfileColumns()...).
|
|
Update(&pref)
|
|
if err != nil {
|
|
log.Fatalln("迁移 user_pref 失败:", err.Error())
|
|
}
|
|
}
|
|
}
|
|
|
|
type liveGoalRewardMigrationRow struct {
|
|
LiveGoalRewardID int `xorm:"live_goal_reward_id"`
|
|
LiveDifficultyID int `xorm:"live_difficulty_id"`
|
|
LiveGoalType int `xorm:"live_goal_type"`
|
|
Rank int `xorm:"rank"`
|
|
}
|
|
|
|
type liveGoalInfoMigrationRow struct {
|
|
LiveDifficultyID int `xorm:"live_difficulty_id"`
|
|
CRankScore int `xorm:"c_rank_score"`
|
|
BRankScore int `xorm:"b_rank_score"`
|
|
ARankScore int `xorm:"a_rank_score"`
|
|
SRankScore int `xorm:"s_rank_score"`
|
|
CRankCombo int `xorm:"c_rank_combo"`
|
|
BRankCombo int `xorm:"b_rank_combo"`
|
|
ARankCombo int `xorm:"a_rank_combo"`
|
|
SRankCombo int `xorm:"s_rank_combo"`
|
|
CRankComplete int `xorm:"c_rank_complete"`
|
|
BRankComplete int `xorm:"b_rank_complete"`
|
|
ARankComplete int `xorm:"a_rank_complete"`
|
|
SRankComplete int `xorm:"s_rank_complete"`
|
|
}
|
|
|
|
func liveRankForMigration(value int, cRank int, bRank int, aRank int, sRank int) int {
|
|
switch {
|
|
case value >= sRank:
|
|
return 1
|
|
case value >= aRank:
|
|
return 2
|
|
case value >= bRank:
|
|
return 3
|
|
case value >= cRank:
|
|
return 4
|
|
default:
|
|
return 5
|
|
}
|
|
}
|
|
|
|
func MigrateUserLiveData() {
|
|
session := db.UserEng.NewSession()
|
|
defer session.Close()
|
|
|
|
recordRows := []usermodel.UserLiveRecord{}
|
|
if err := session.Table(new(usermodel.UserLiveRecord)).Find(&recordRows); err != nil {
|
|
log.Fatalln("迁移 user_live_status 失败:", err.Error())
|
|
}
|
|
|
|
existingStatusRows := []usermodel.UserLiveStatus{}
|
|
if err := session.Table(new(usermodel.UserLiveStatus)).Find(&existingStatusRows); err != nil {
|
|
log.Fatalln("迁移 user_live_status 失败:", err.Error())
|
|
}
|
|
|
|
type liveStatusKey struct {
|
|
UserID int
|
|
LiveDifficultyID int
|
|
}
|
|
|
|
statusMap := make(map[liveStatusKey]usermodel.UserLiveStatus, len(existingStatusRows))
|
|
for _, row := range existingStatusRows {
|
|
statusMap[liveStatusKey{UserID: row.UserID, LiveDifficultyID: row.LiveDifficultyID}] = row
|
|
}
|
|
|
|
now := time.Now().Unix()
|
|
for _, row := range recordRows {
|
|
key := liveStatusKey{UserID: row.UserID, LiveDifficultyID: row.LiveDifficultyID}
|
|
status, ok := statusMap[key]
|
|
if !ok {
|
|
status = usermodel.UserLiveStatus{
|
|
UserID: row.UserID,
|
|
LiveDifficultyID: row.LiveDifficultyID,
|
|
HiScore: row.TotalScore,
|
|
HiComboCount: row.MaxCombo,
|
|
ClearCnt: 1,
|
|
InsertDate: now,
|
|
UpdateDate: now,
|
|
}
|
|
if _, err := session.Insert(&status); err != nil {
|
|
log.Fatalln("迁移 user_live_status 失败:", err.Error())
|
|
}
|
|
statusMap[key] = status
|
|
continue
|
|
}
|
|
|
|
updated := false
|
|
if row.TotalScore > status.HiScore {
|
|
status.HiScore = row.TotalScore
|
|
updated = true
|
|
}
|
|
if row.MaxCombo > status.HiComboCount {
|
|
status.HiComboCount = row.MaxCombo
|
|
updated = true
|
|
}
|
|
if status.ClearCnt <= 0 {
|
|
status.ClearCnt = 1
|
|
updated = true
|
|
}
|
|
if updated {
|
|
status.UpdateDate = now
|
|
if _, err := session.Table(new(usermodel.UserLiveStatus)).
|
|
Where("user_id = ? AND live_difficulty_id = ?", status.UserID, status.LiveDifficultyID).
|
|
Cols("hi_score", "hi_combo_count", "clear_cnt", "update_date").
|
|
Update(&status); err != nil {
|
|
log.Fatalln("迁移 user_live_status 失败:", err.Error())
|
|
}
|
|
statusMap[key] = status
|
|
}
|
|
}
|
|
|
|
liveGoalInfoMap := map[int]liveGoalInfoMigrationRow{}
|
|
loadLiveGoalInfoRows := func(table string) {
|
|
rows := []liveGoalInfoMigrationRow{}
|
|
err := db.MainEng.Table(table).Alias("live").
|
|
Join("LEFT", "live_setting_m setting", "live.live_setting_id = setting.live_setting_id").
|
|
Select(`
|
|
live.live_difficulty_id,
|
|
setting.c_rank_score,
|
|
setting.b_rank_score,
|
|
setting.a_rank_score,
|
|
setting.s_rank_score,
|
|
setting.c_rank_combo,
|
|
setting.b_rank_combo,
|
|
setting.a_rank_combo,
|
|
setting.s_rank_combo,
|
|
live.c_rank_complete,
|
|
live.b_rank_complete,
|
|
live.a_rank_complete,
|
|
live.s_rank_complete
|
|
`).
|
|
Find(&rows)
|
|
if err != nil {
|
|
log.Fatalln("迁移 user_live_goal 失败:", err.Error())
|
|
}
|
|
for _, row := range rows {
|
|
liveGoalInfoMap[row.LiveDifficultyID] = row
|
|
}
|
|
}
|
|
loadLiveGoalInfoRows("normal_live_m")
|
|
loadLiveGoalInfoRows("special_live_m")
|
|
|
|
goalRewardRows := []liveGoalRewardMigrationRow{}
|
|
if err := db.MainEng.Table("live_goal_reward_m").
|
|
OrderBy("live_difficulty_id ASC, live_goal_type ASC, rank ASC, live_goal_reward_id ASC").
|
|
Find(&goalRewardRows); err != nil {
|
|
log.Fatalln("迁移 user_live_goal 失败:", err.Error())
|
|
}
|
|
|
|
goalRewardMap := map[int][]liveGoalRewardMigrationRow{}
|
|
for _, row := range goalRewardRows {
|
|
goalRewardMap[row.LiveDifficultyID] = append(goalRewardMap[row.LiveDifficultyID], row)
|
|
}
|
|
|
|
existingGoalRows := []usermodel.UserLiveGoal{}
|
|
if err := session.Table(new(usermodel.UserLiveGoal)).
|
|
Where("live_goal_reward_id > 0").
|
|
Find(&existingGoalRows); err != nil {
|
|
log.Fatalln("迁移 user_live_goal 失败:", err.Error())
|
|
}
|
|
|
|
existingGoalMap := make(map[liveStatusKey]map[int]struct{})
|
|
for _, row := range existingGoalRows {
|
|
key := liveStatusKey{UserID: row.UserID, LiveDifficultyID: row.LiveDifficultyID}
|
|
if existingGoalMap[key] == nil {
|
|
existingGoalMap[key] = map[int]struct{}{}
|
|
}
|
|
existingGoalMap[key][row.LiveGoalRewardID] = struct{}{}
|
|
}
|
|
|
|
statusKeys := make([]liveStatusKey, 0, len(statusMap))
|
|
for key := range statusMap {
|
|
statusKeys = append(statusKeys, key)
|
|
}
|
|
sort.Slice(statusKeys, func(i, j int) bool {
|
|
if statusKeys[i].UserID == statusKeys[j].UserID {
|
|
return statusKeys[i].LiveDifficultyID < statusKeys[j].LiveDifficultyID
|
|
}
|
|
return statusKeys[i].UserID < statusKeys[j].UserID
|
|
})
|
|
|
|
for _, key := range statusKeys {
|
|
status := statusMap[key]
|
|
liveInfo, ok := liveGoalInfoMap[key.LiveDifficultyID]
|
|
if !ok {
|
|
continue
|
|
}
|
|
|
|
scoreRank := liveRankForMigration(status.HiScore, liveInfo.CRankScore, liveInfo.BRankScore, liveInfo.ARankScore, liveInfo.SRankScore)
|
|
comboRank := liveRankForMigration(status.HiComboCount, liveInfo.CRankCombo, liveInfo.BRankCombo, liveInfo.ARankCombo, liveInfo.SRankCombo)
|
|
clearRank := liveRankForMigration(status.ClearCnt, liveInfo.CRankComplete, liveInfo.BRankComplete, liveInfo.ARankComplete, liveInfo.SRankComplete)
|
|
|
|
for _, reward := range goalRewardMap[key.LiveDifficultyID] {
|
|
achieved := false
|
|
switch reward.LiveGoalType {
|
|
case 1:
|
|
achieved = scoreRank <= reward.Rank
|
|
case 2:
|
|
achieved = comboRank <= reward.Rank
|
|
case 3:
|
|
achieved = clearRank <= reward.Rank
|
|
}
|
|
if !achieved {
|
|
continue
|
|
}
|
|
|
|
if existingGoalMap[key] != nil {
|
|
if _, ok := existingGoalMap[key][reward.LiveGoalRewardID]; ok {
|
|
continue
|
|
}
|
|
}
|
|
|
|
if _, err := session.Insert(&usermodel.UserLiveGoal{
|
|
UserID: key.UserID,
|
|
LiveDifficultyID: key.LiveDifficultyID,
|
|
LiveGoalRewardID: reward.LiveGoalRewardID,
|
|
GoalType: constant.LiveGoalType(reward.LiveGoalType),
|
|
Rank: reward.Rank,
|
|
CompletedAt: now,
|
|
}); err != nil {
|
|
log.Fatalln("迁移 user_live_goal 失败:", err.Error())
|
|
}
|
|
if existingGoalMap[key] == nil {
|
|
existingGoalMap[key] = map[int]struct{}{}
|
|
}
|
|
existingGoalMap[key][reward.LiveGoalRewardID] = struct{}{}
|
|
}
|
|
}
|
|
}
|
|
|
|
func LoadUnitData() {
|
|
userEng = db.UserEng.NewSession()
|
|
defer userEng.Close()
|
|
|
|
err := userEng.Begin()
|
|
CheckErr(err)
|
|
|
|
commonUnitExist, err := userEng.IsTableExist(new(unitmodel.CommonUnitData))
|
|
CheckErr(err)
|
|
userUnitExist, err := userEng.IsTableExist(new(usermodel.UserUnitData))
|
|
CheckErr(err)
|
|
|
|
if !commonUnitExist || !userUnitExist {
|
|
log.Println("卡片数据不存在，正在同步...")
|
|
|
|
userEng.DropTable(new(unitmodel.CommonUnitData))
|
|
userEng.CreateTable(new(unitmodel.CommonUnitData))
|
|
userEng.DropTable(new(usermodel.UserUnitData))
|
|
userEng.CreateTable(new(usermodel.UserUnitData))
|
|
|
|
var unitData []unitmodel.UnitM
|
|
err = db.MainEng.Table(new(unitmodel.UnitM)).OrderBy("unit_id ASC").Find(&unitData)
|
|
CheckErr(err)
|
|
|
|
checked := false
|
|
for _, u := range unitData {
|
|
// 判断卡片最大等级
|
|
var unitMaxLevel, nextExp, sumExp int
|
|
_, err = db.MainEng.Table("unit_level_up_pattern_m").
|
|
Where("unit_level_up_pattern_id = ?", u.UnitLevelUpPatternId).
|
|
Select("MAX(unit_level),next_exp").Get(&unitMaxLevel, &nextExp)
|
|
CheckErr(err)
|
|
|
|
// 计算突破前的经验总和
|
|
_, err = db.MainEng.Table("unit_level_up_pattern_m").
|
|
Where("unit_level_up_pattern_id = ?", u.UnitLevelUpPatternId).
|
|
Where("unit_level = ?", unitMaxLevel-1).Cols("next_exp").Get(&sumExp)
|
|
CheckErr(err)
|
|
|
|
// 计算突破前的属性
|
|
var smileMax, pureMax, coolMax int
|
|
smileMax = u.SmileMax
|
|
pureMax = u.PureMax
|
|
coolMax = u.CoolMax
|
|
|
|
// 如果 nexpExp 不为零，则说明卡片等级没有达到上限
|
|
if nextExp != 0 {
|
|
// 计算突破后的经验总和
|
|
_, err = db.MainEng.Table("unit_level_limit_pattern_m").
|
|
Where("unit_level_limit_id = 1 AND unit_level = 349").
|
|
Cols("next_exp").Get(&sumExp)
|
|
CheckErr(err)
|
|
|
|
// 突破后最大等级
|
|
unitMaxLevel = 350
|
|
|
|
// 计算突破后的属性
|
|
smileMax += 6000
|
|
pureMax += 6000
|
|
coolMax += 6000
|
|
}
|
|
|
|
// 计算绊值、技能等级、技能经验
|
|
var maxLove, skillLevel, skillExp, removableSkillCapacity, levelLimitID int
|
|
switch u.Rarity {
|
|
case 1:
|
|
maxLove = 50
|
|
skillExp = 0
|
|
skillLevel = 0
|
|
removableSkillCapacity = 0
|
|
levelLimitID = 0
|
|
case 2:
|
|
maxLove = 200
|
|
skillExp = 490
|
|
skillLevel = 8
|
|
removableSkillCapacity = 1
|
|
levelLimitID = 0
|
|
case 3:
|
|
maxLove = 500
|
|
skillExp = 4900
|
|
skillLevel = 8
|
|
removableSkillCapacity = 2
|
|
levelLimitID = 0
|
|
case 4:
|
|
maxLove = 1000
|
|
skillExp = 29900
|
|
skillLevel = 8
|
|
removableSkillCapacity = 8
|
|
levelLimitID = 1
|
|
case 5:
|
|
maxLove = 750
|
|
skillExp = 12700
|
|
skillLevel = 8
|
|
removableSkillCapacity = 3
|
|
levelLimitID = 0
|
|
}
|
|
|
|
// 针对技能卡等应援卡片
|
|
if smileMax == 1 {
|
|
maxLove = 0
|
|
skillExp = 0
|
|
skillLevel = 0
|
|
removableSkillCapacity = 0
|
|
}
|
|
|
|
// 检查是否签名卡
|
|
var isSigned bool
|
|
exist, err := db.MainEng.Table("unit_sign_asset_m").Where("unit_id = ?", u.UnitId).Exist()
|
|
CheckErr(err)
|
|
if exist {
|
|
isSigned = true
|
|
}
|
|
|
|
// 生成公共卡片
|
|
unitCommon := unitmodel.CommonUnitData{
|
|
UnitNumber: u.UnitNumber,
|
|
UnitID: u.UnitId,
|
|
UnitTypeID: u.UnitTypeId,
|
|
Name: *u.NameEn,
|
|
Eponym: u.EponymEn,
|
|
Rarity: u.Rarity,
|
|
Attribute: u.AttributeId,
|
|
Smile: smileMax,
|
|
Cute: pureMax,
|
|
Cool: coolMax,
|
|
Exp: sumExp,
|
|
Level: unitMaxLevel,
|
|
MaxLevel: unitMaxLevel,
|
|
LevelLimitID: levelLimitID,
|
|
Rank: u.RankMin,
|
|
MaxRank: u.RankMax,
|
|
Love: maxLove,
|
|
MaxLove: maxLove,
|
|
UnitSkillExp: skillExp,
|
|
UnitSkillLevel: skillLevel,
|
|
MaxHp: u.HpMax,
|
|
UnitRemovableSkillCapacity: removableSkillCapacity,
|
|
IsRankMax: true,
|
|
IsLoveMax: true,
|
|
IsLevelMax: true,
|
|
IsSigned: isSigned,
|
|
IsSkillLevelMax: true,
|
|
IsRemovableSkillCapacityMax: true,
|
|
InsertDate: time.Now().Unix(),
|
|
}
|
|
|
|
_, err = userEng.Insert(&unitCommon)
|
|
CheckErr(err)
|
|
|
|
var userID []int
|
|
err = db.UserEng.Table(new(usermodel.Users)).Cols("user_id").Find(&userID)
|
|
CheckErr(err)
|
|
|
|
for _, id := range userID {
|
|
userUnit := usermodel.UserUnitData{
|
|
UnitID: u.UnitId,
|
|
FavoriteFlag: false,
|
|
DisplayRank: u.RankMax,
|
|
UserID: id,
|
|
InsertDate: time.Now().Unix(),
|
|
}
|
|
|
|
// 检查表里是否已经有数据
|
|
if !checked {
|
|
ct, err := userEng.Table(new(usermodel.UserUnitData)).Count()
|
|
CheckErr(err)
|
|
|
|
if ct == 0 {
|
|
userUnit.UnitOwningUserID = 38383
|
|
}
|
|
|
|
checked = true
|
|
}
|
|
|
|
_, err = userEng.Insert(&userUnit)
|
|
CheckErr(err)
|
|
}
|
|
}
|
|
|
|
err = userEng.Commit()
|
|
CheckErr(err)
|
|
|
|
log.Println("同步完成！")
|
|
}
|
|
}
|
|
|
|
func CheckErr(err error) {
|
|
if err != nil {
|
|
userEng.Rollback()
|
|
log.Fatalln("同步失败:", err.Error())
|
|
}
|
|
}
|