Files
honoka-chan/internal/startup/database.go
T
2026-06-28 01:20:05 +08:00

709 lines
21 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.UserAccessory))
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.UserLiveRandom))
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()
MigrateUserAccessories()
MigrateUserLiveData()
ForceAllUsersRelogin()
}
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())
}
}
}
func ForceAllUsersRelogin() {
if _, err := db.UserEng.Table(new(usermodel.UserPref)).
Cols("force_relogin").
Update(&usermodel.UserPref{ForceRelogin: true}); err != nil {
log.Fatalln("设置全员重新登录失败:", err.Error())
}
}
func MigrateLegacyUnitTables() {
if err := db.UserEng.Sync2(new(unitmodel.CommonUnitData), new(usermodel.UserUnitData)); err != nil {
log.Fatalln("迁移卡片历史数据表失败:", err.Error())
}
}
type accessoryOwningMapRow struct {
AccessoryOwningUserID int `xorm:"accessory_owning_user_id"`
AccessoryID int `xorm:"accessory_id"`
}
func MigrateUserAccessories() {
session := db.UserEng.NewSession()
defer session.Close()
if err := session.Begin(); err != nil {
log.Fatalln("迁移 user_accessory 失败:", err.Error())
}
userIDs := []int{}
if err := session.Table(new(usermodel.Users)).Cols("user_id").Find(&userIDs); err != nil {
session.Rollback()
log.Fatalln("迁移 user_accessory 失败:", err.Error())
}
templateAccessoryIDs, err := usermodel.LoadAccessoryTemplateIDs(db.MainEng)
if err != nil {
session.Rollback()
log.Fatalln("迁移 user_accessory 失败:", err.Error())
}
legacyAccessoryMap := map[int]int{}
hasLegacyTable, err := db.MainEng.IsTableExist("common_accessory_m")
if err != nil {
session.Rollback()
log.Fatalln("迁移 user_accessory 失败:", err.Error())
}
if hasLegacyTable {
legacyRows := []accessoryOwningMapRow{}
if err := db.MainEng.Table("common_accessory_m").
Cols("accessory_owning_user_id,accessory_id").
Find(&legacyRows); err != nil {
session.Rollback()
log.Fatalln("迁移 user_accessory 失败:", err.Error())
}
for _, row := range legacyRows {
legacyAccessoryMap[row.AccessoryOwningUserID] = row.AccessoryID
}
}
wearRows := []usermodel.UserAccessoryWear{}
if err := session.Table(new(usermodel.UserAccessoryWear)).Find(&wearRows); err != nil {
session.Rollback()
log.Fatalln("迁移 user_accessory 失败:", err.Error())
}
existingAccessories := []usermodel.UserAccessory{}
if err := session.Table(new(usermodel.UserAccessory)).Find(&existingAccessories); err != nil {
session.Rollback()
log.Fatalln("迁移 user_accessory 失败:", err.Error())
}
currentAccessoryMap := make(map[int]map[int]int)
for _, row := range existingAccessories {
if currentAccessoryMap[row.UserID] == nil {
currentAccessoryMap[row.UserID] = map[int]int{}
}
currentAccessoryMap[row.UserID][row.AccessoryOwningUserID] = row.AccessoryID
}
requiredCountsByUser := make(map[int]map[int]int, len(userIDs))
for _, row := range wearRows {
accessoryID, ok := resolveAccessoryIDForWear(currentAccessoryMap, legacyAccessoryMap, templateAccessoryIDs, row.UserID, row.AccessoryOwningUserID)
if !ok {
continue
}
if requiredCountsByUser[row.UserID] == nil {
requiredCountsByUser[row.UserID] = map[int]int{}
}
requiredCountsByUser[row.UserID][accessoryID]++
}
for _, userID := range userIDs {
if err := usermodel.EnsureUserAccessories(session, db.MainEng, userID, requiredCountsByUser[userID]); err != nil {
session.Rollback()
log.Fatalln("迁移 user_accessory 失败:", err.Error())
}
}
updatedAccessories := []usermodel.UserAccessory{}
if err := session.Table(new(usermodel.UserAccessory)).
OrderBy("user_id ASC, accessory_id ASC, accessory_owning_user_id ASC").
Find(&updatedAccessories); err != nil {
session.Rollback()
log.Fatalln("迁移 user_accessory 失败:", err.Error())
}
currentAccessoryMap = make(map[int]map[int]int)
availableOwningIDs := make(map[int]map[int][]int)
for _, row := range updatedAccessories {
if currentAccessoryMap[row.UserID] == nil {
currentAccessoryMap[row.UserID] = map[int]int{}
}
if availableOwningIDs[row.UserID] == nil {
availableOwningIDs[row.UserID] = map[int][]int{}
}
currentAccessoryMap[row.UserID][row.AccessoryOwningUserID] = row.AccessoryID
availableOwningIDs[row.UserID][row.AccessoryID] = append(availableOwningIDs[row.UserID][row.AccessoryID], row.AccessoryOwningUserID)
}
usedOwningIDs := make(map[int]map[int]struct{})
for _, row := range wearRows {
if currentAccessoryMap[row.UserID] == nil {
continue
}
if _, ok := currentAccessoryMap[row.UserID][row.AccessoryOwningUserID]; !ok {
continue
}
if usedOwningIDs[row.UserID] == nil {
usedOwningIDs[row.UserID] = map[int]struct{}{}
}
usedOwningIDs[row.UserID][row.AccessoryOwningUserID] = struct{}{}
}
for i := range wearRows {
row := wearRows[i]
if row.AccessoryOwningUserID <= 0 {
continue
}
if currentAccessoryMap[row.UserID] != nil {
if _, ok := currentAccessoryMap[row.UserID][row.AccessoryOwningUserID]; ok {
continue
}
}
accessoryID, ok := resolveAccessoryIDForWear(currentAccessoryMap, legacyAccessoryMap, templateAccessoryIDs, row.UserID, row.AccessoryOwningUserID)
if !ok {
continue
}
candidates := availableOwningIDs[row.UserID][accessoryID]
if len(candidates) == 0 {
continue
}
if usedOwningIDs[row.UserID] == nil {
usedOwningIDs[row.UserID] = map[int]struct{}{}
}
newOwningID := 0
for _, candidate := range candidates {
if _, used := usedOwningIDs[row.UserID][candidate]; used {
continue
}
newOwningID = candidate
break
}
if newOwningID == 0 {
newOwningID = candidates[0]
}
if _, err := session.Table(new(usermodel.UserAccessoryWear)).
ID(row.ID).
Cols("accessory_owning_user_id").
Update(&usermodel.UserAccessoryWear{AccessoryOwningUserID: newOwningID}); err != nil {
session.Rollback()
log.Fatalln("迁移 user_accessory 失败:", err.Error())
}
wearRows[i].AccessoryOwningUserID = newOwningID
usedOwningIDs[row.UserID][newOwningID] = struct{}{}
}
if err := session.Commit(); err != nil {
session.Rollback()
log.Fatalln("迁移 user_accessory 失败:", err.Error())
}
}
func resolveAccessoryIDForWear(currentAccessoryMap map[int]map[int]int, legacyAccessoryMap map[int]int, accessoryIDs []int, userID int, accessoryOwningUserID int) (int, bool) {
if accessoryOwningUserID <= 0 {
return 0, false
}
if currentAccessoryMap[userID] != nil {
if accessoryID, ok := currentAccessoryMap[userID][accessoryOwningUserID]; ok {
return accessoryID, true
}
}
if accessoryID, ok := legacyAccessoryMap[accessoryOwningUserID]; ok {
return accessoryID, true
}
return legacyAccessoryOwningUserIDToAccessoryID(accessoryIDs, accessoryOwningUserID)
}
func legacyAccessoryOwningUserIDToAccessoryID(accessoryIDs []int, accessoryOwningUserID int) (int, bool) {
if accessoryOwningUserID <= 0 {
return 0, false
}
index := (accessoryOwningUserID - 1) / 9
if index < 0 || index >= len(accessoryIDs) {
return 0, false
}
return accessoryIDs[index], true
}
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())
}
}