tcjlj/OpenFlare
1
0
Fork 0
mirror of https://github.com/Rain-kl/OpenFlare.git synced 2026-10-09 00:56:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

[优化] 修复迁移

Browse Source
This commit is contained in:
ryan
2026-06-01 17:50:49 +08:00
parent d3ce26414c
commit 8f08a962e6
1 changed files with 13 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
openflare_server/model/migrate/v21.go
+13 -2
View File
@@ -32,9 +32,20 @@ func migrateV21(ctx Context, db *gorm.DB, backend string) error {
migrator := db.Migrator() migrator := db.Migrator()
// Rename agent_token → access_token (target may already exist if AutoMigrate ran earlier)
if migrator.HasColumn(&nodeV21{}, "agent_token") { if migrator.HasColumn(&nodeV21{}, "agent_token") {
if err := migrator.RenameColumn(&nodeV21{}, "agent_token", "access_token"); err != nil { if migrator.HasColumn(&nodeV21{}, "access_token") {
return fmt.Errorf("failed to rename agent_token to access_token: %w", err) slog.Info("v21: access_token column already exists, backfilling from agent_token")
if err := db.Exec(`UPDATE nodes SET access_token = agent_token WHERE access_token IS NULL OR access_token = ''`).Error; err != nil {
return fmt.Errorf("failed to backfill access_token from agent_token: %w", err)
}
if err := migrator.DropColumn(&nodeV21{}, "agent_token"); err != nil {
slog.Warn("failed to drop agent_token after backfill", "error", err)
}
} else {
if err := migrator.RenameColumn(&nodeV21{}, "agent_token", "access_token"); err != nil {
return fmt.Errorf("failed to rename agent_token to access_token: %w", err)
}
} }
} }