Skip to content

Commit

Permalink
refactor: 添加更友好的提示和判断，并在失败时回滚数据库处理 (#1175)
Browse files Browse the repository at this point in the history 
Co-authored-by: bugtower100 <[email protected]>
  • Loading branch information
@PaienNate @bugtower100
PaienNate and bugtower100 authored Dec 30, 2024
1 parent f08f1ba commit 17f4b9e
Show file tree
Hide file tree
Showing 2 changed files with 106 additions and 77 deletions.
18 changes: 11 additions & 7 deletions main.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -331,7 +331,7 @@ func main() {
log.Warn("检测到 auto_update.exe，即将自动退出当前程序并进行升级")
log.Warn("程序目录下会出现“升级日志.log”，这代表升级正在进行中，如果失败了请检查此文件。")

err := CheckUpdater(diceManager)
err = CheckUpdater(diceManager)
if err != nil {
log.Error("升级程序检查失败: ", err.Error())
} else {
Expand All @@ -353,7 +353,7 @@ func main() {
}

if doNext {
err := CheckUpdater(diceManager)
err = CheckUpdater(diceManager)
if err != nil {
log.Error("升级程序检查失败: ", err.Error())
} else {
Expand All @@ -368,7 +368,7 @@ func main() {
removeUpdateFiles()

if opts.UpdateTest {
err := CheckUpdater(diceManager)
err = CheckUpdater(diceManager)
if err != nil {
log.Error("升级程序检查失败: ", err.Error())
} else {
Expand All @@ -378,7 +378,7 @@ func main() {

// 先临时放这里，后面再整理一下升级模块
diceManager.UpdateSealdiceByFile = func(packName string, log *log.Helper) bool {
err := CheckUpdater(diceManager)
err = CheckUpdater(diceManager)
if err != nil {
log.Error("升级程序检查失败: ", err.Error())
return false
Expand All @@ -399,7 +399,8 @@ func main() {

useBuiltinUI := false
checkFrontendExists := func() bool {
stat, err := os.Stat("./frontend_overwrite")
var stat os.FileInfo
stat, err = os.Stat("./frontend_overwrite")
return err == nil && stat.IsDir()
}
if !checkFrontendExists() {
Expand Down Expand Up @@ -438,8 +439,11 @@ func main() {
log.Fatalf("移除旧帮助文档时出错，%v", migrateErr)
}
// v150升级
if !migrate.V150Upgrade() {
return
err = migrate.V150Upgrade()
if err != nil {
// Fatalf将会退出程序...或许应该用Errorf一类的吗？
log.Fatalf("您的146数据库可能存在问题，为保护数据，已经停止执行150升级命令。请尝试联系开发者，并提供你的日志。\n"+
"数据已回滚，您可暂时使用旧版本等待进一步的修复和更新。您的报错内容为: %v", err)
}

if !opts.ShowConsole || opts.MultiInstanceOnWindows {
Expand Down
165 changes: 95 additions & 70 deletions migrate/v150_attrs.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,7 @@ import (
"sealdice-core/dice"
"sealdice-core/dice/model"
"sealdice-core/utils"
log "sealdice-core/utils/kratos"
)

func convertToNew(name string, ownerId string, data []byte, updatedAt int64) (*model.AttributesItemModel, error) {
Expand Down Expand Up @@ -391,37 +392,10 @@ func checkTableExists(db *sqlx.DB, tableName string) (bool, error) {
}
}

func V150Upgrade() bool {
dbDataPath, _ := filepath.Abs("./data/default/data.db")
if _, err := os.Stat(dbDataPath); errors.Is(err, os.ErrNotExist) {
return true
}

db, err := openDB(dbDataPath)
if err != nil {
fmt.Fprintln(os.Stdout, "升级失败，无法打开数据库:", err)
return false
}
defer func() {
_ = db.Close()
}()

exists, err := checkTableExists(db, "attrs")
if err != nil {
fmt.Fprintln(os.Stdout, "V150数据转换未知错误:", err.Error())
return false
}
if exists {
// 表格已经存在，说明转换完成，退出
return true
}

fmt.Fprintln(os.Stdout, "1.5 数据迁移")
sheetIdBindByGroupUserId = map[string]string{}
// Pinenutn: 2024-10-28 我要把这个注释全文背诵，它扰乱了GORM的初始化逻辑
// -- 坏，Get这个方法太严格了，所有的字段都要有默认值，不然无法反序列化
sqls := []string{
`
// Pinenutn: 2024-10-28 我要把这个注释全文背诵，它扰乱了GORM的初始化逻辑
// -- 坏，Get这个方法太严格了，所有的字段都要有默认值，不然无法反序列化
var v150sqls = []string{
`
CREATE TABLE IF NOT EXISTS attrs (
id TEXT PRIMARY KEY,
data BYTEA,
Expand All @@ -435,63 +409,114 @@ CREATE TABLE IF NOT EXISTS attrs (
updated_at INTEGER default 0
);
`,
`create index if not exists idx_attrs_binding_sheet_id on attrs (binding_sheet_id);`,
`create index if not exists idx_attrs_owner_id_id on attrs (owner_id);`,
`create index if not exists idx_attrs_attrs_type_id on attrs (attrs_type);`,
`create index if not exists idx_attrs_binding_sheet_id on attrs (binding_sheet_id);`,
`create index if not exists idx_attrs_owner_id_id on attrs (owner_id);`,
`create index if not exists idx_attrs_attrs_type_id on attrs (attrs_type);`,
}

func V150Upgrade() error {
dbDataPath, _ := filepath.Abs("./data/default/data.db")
if _, err := os.Stat(dbDataPath); errors.Is(err, os.ErrNotExist) {
return errors.New("未能查找到旧版本数据库")
}
for _, i := range sqls {
_, _ = db.Exec(i)

db, err := openDB(dbDataPath)
if err != nil {
return fmt.Errorf("升级失败，无法打开数据库: %w", err)
}
defer db.Close()

tx, err := db.Beginx()
if err != nil {
fmt.Fprintln(os.Stdout, "V150数据转换创建事务失败:", err.Error())
return false
return fmt.Errorf("创建事务失败: %w", err)
}
defer func() {
if p := recover(); p != nil {
err = tx.Rollback()
if err != nil {
log.Errorf("回滚事务时出错: %v", err)
}
panic(p) // 继续传播 panic
} else if err != nil {
log.Errorf("日志处理时出现异常行为: %v", err)
err = tx.Rollback()
if err != nil {
log.Errorf("回滚事务时出错: %v", err)
return
}
} else {
err = tx.Commit()
if err != nil {
log.Errorf("提交事务时出错: %v", err)
}
}
}()

exists, err := checkTableExists(db, "attrs")
if err != nil {
return fmt.Errorf("检查表是否存在时出错: %w", err)
}
// 特判146->150的倒霉蛋
exists146, err := checkTableExists(db, "attrs_group")

if exists, _ := checkTableExists(db, "attrs_user"); exists {
count, countSheetsNum, countFailed, err2 := attrsUserMigrate(tx)
fmt.Fprintf(os.Stdout, "数据卡转换 - 角色卡，成功人数%d 失败人数 %d 卡数 %d\n", count, countFailed, countSheetsNum)
if err2 != nil {
fmt.Fprintln(os.Stdout, "异常", err2.Error())
return false
if exists {
if exists146 {
log.Errorf("1.4.6的数据部分迁移！您可能是150部分版本的受害者，请联系开发者")
return errors.New("150和146的数据库共同存在，请联系开发者")
}
// 表格已经存在，说明转换完成
return nil
}

if exists, _ := checkTableExists(db, "attrs_group_user"); exists {
count, countFailed, err2 := attrsGroupUserMigrate(tx)
fmt.Fprintf(os.Stdout, "数据卡转换 - 群组个人数据，成功%d 失败 %d\n", count, countFailed)
if err2 != nil {
fmt.Fprintln(os.Stdout, "异常", err2.Error())
return false
log.Info("1.5 数据迁移")
sheetIdBindByGroupUserId = map[string]string{}

for _, singleSql := range v150sqls {
if _, err = tx.Exec(singleSql); err != nil {
return fmt.Errorf("执行 SQL 出错: %w", err)
}
}

if exists, _ := checkTableExists(db, "attrs_group"); exists {
if exists, _ = checkTableExists(db, "attrs_user"); exists {
count, countSheetsNum, countFailed, err0 := attrsUserMigrate(tx)
log.Infof("数据卡转换 - 角色卡，成功人数%d 失败人数 %d 卡数 %d\n", count, countFailed, countSheetsNum)
if err0 != nil {
return fmt.Errorf("角色卡转换出错: %w", err0)
}
}

if exists, _ = checkTableExists(db, "attrs_group_user"); exists {
count, countFailed, err1 := attrsGroupUserMigrate(tx)
log.Infof("数据卡转换 - 群组个人数据，成功%d 失败 %d\n", count, countFailed)
if err1 != nil {
return fmt.Errorf("群组个人数据转换出错: %w", err1)
}
}

if exists, _ = checkTableExists(db, "attrs_group"); exists {
count, countFailed, err2 := attrsGroupMigrate(tx)
fmt.Fprintf(os.Stdout, "数据卡转换 - 群数据，成功%d 失败 %d\n", count, countFailed)
log.Infof("数据卡转换 - 群数据，成功%d 失败 %d\n", count, countFailed)
if err2 != nil {
fmt.Fprintln(os.Stdout, "异常", err2.Error())
return false
return fmt.Errorf("群数据转换出错: %w", err2)
}
}

// 删档
fmt.Fprintln(os.Stdout, "删除旧版本数据")
_, _ = tx.Exec("drop table attrs_group")
_, _ = tx.Exec("drop table attrs_group_user")
_, _ = tx.Exec("drop table attrs_user")
// 删除旧版本数据
log.Info("删除旧版本数据")
deleteSQLs := []string{
"drop table attrs_group",
"drop table attrs_group_user",
"drop table attrs_user",
}
for _, deleteSQL := range deleteSQLs {
if _, err = tx.Exec(deleteSQL); err != nil {
return fmt.Errorf("删除旧数据时出错: %w", err)
}
}
// 放在这里保证能执行
_, _ = db.Exec("PRAGMA wal_checkpoint(TRUNCATE);")
_, _ = tx.Exec("VACUUM;") // 收尾

_, _ = db.Exec("VACUUM;")
sheetIdBindByGroupUserId = nil

err = tx.Commit()
if err != nil {
fmt.Fprintln(os.Stdout, "V150 数据转换失败:", err.Error())
return false
}

fmt.Fprintln(os.Stdout, "V150 数据转换完成")
return true
log.Info("V150 数据转换完成")
return nil
}

0 comments on commit 17f4b9e

Please sign in to comment.