|
|
package migration |
|
|
|
|
|
import ( |
|
|
"fmt" |
|
|
|
|
|
"go.uber.org/zap" |
|
|
"gorm.io/gorm" |
|
|
) |
|
|
|
|
|
// RemoveDeleteAtFields 移除相关表的 delete_at 字段 |
|
|
func RemoveDeleteAtFields(db *gorm.DB, log *zap.Logger) error { |
|
|
log.Info("开始移除相关表的 delete_at 字段...") |
|
|
|
|
|
// 1. 移除 frontend_backend_routes 表的 delete_at 字段 |
|
|
if err := removeDeleteAtFromFrontendBackendRoutes(db, log); err != nil { |
|
|
return fmt.Errorf("移除 frontend_backend_routes 表 delete_at 字段失败: %w", err) |
|
|
} |
|
|
|
|
|
// 2. 移除 frontend_routes 表的 delete_at 字段 |
|
|
if err := removeDeleteAtFromFrontendRoutes(db, log); err != nil { |
|
|
return fmt.Errorf("移除 frontend_routes 表 delete_at 字段失败: %w", err) |
|
|
} |
|
|
log.Info("✅ 所有表的 delete_at 字段移除完成") |
|
|
return nil |
|
|
} |
|
|
|
|
|
// removeDeleteAtFromFrontendBackendRoutes 移除 frontend_backend_routes 表的 delete_at 字段 |
|
|
func removeDeleteAtFromFrontendBackendRoutes(db *gorm.DB, log *zap.Logger) error { |
|
|
log.Info("移除 frontend_backend_routes 表的 delete_at 字段...") |
|
|
|
|
|
// 检查表是否存在 |
|
|
if !db.Migrator().HasTable("frontend_backend_routes") { |
|
|
log.Info("frontend_backend_routes 表不存在，跳过") |
|
|
return nil |
|
|
} |
|
|
|
|
|
// 检查 delete_at 字段是否存在 |
|
|
if !db.Migrator().HasColumn("frontend_backend_routes", "deleted_at") { |
|
|
log.Info("frontend_backend_routes 表没有 deleted_at 字段，跳过") |
|
|
return nil |
|
|
} |
|
|
|
|
|
// 删除 delete_at 字段 |
|
|
if err := db.Exec("ALTER TABLE frontend_backend_routes DROP COLUMN deleted_at").Error; err != nil { |
|
|
return fmt.Errorf("删除 deleted_at 字段失败: %w", err) |
|
|
} |
|
|
|
|
|
log.Info("✅ frontend_backend_routes 表的 deleted_at 字段移除成功") |
|
|
return nil |
|
|
} |
|
|
|
|
|
// removeDeleteAtFromFrontendRoutes 移除 frontend_routes 表的 delete_at 字段 |
|
|
func removeDeleteAtFromFrontendRoutes(db *gorm.DB, log *zap.Logger) error { |
|
|
log.Info("移除 frontend_routes 表的 delete_at 字段...") |
|
|
|
|
|
// 检查表是否存在 |
|
|
if !db.Migrator().HasTable("frontend_routes") { |
|
|
log.Info("frontend_routes 表不存在，跳过") |
|
|
return nil |
|
|
} |
|
|
|
|
|
// 检查 delete_at 字段是否存在 |
|
|
if !db.Migrator().HasColumn("frontend_routes", "deleted_at") { |
|
|
log.Info("frontend_routes 表没有 deleted_at 字段，跳过") |
|
|
return nil |
|
|
} |
|
|
|
|
|
// 删除 delete_at 字段 |
|
|
if err := db.Exec("ALTER TABLE frontend_routes DROP COLUMN deleted_at").Error; err != nil { |
|
|
return fmt.Errorf("删除 deleted_at 字段失败: %w", err) |
|
|
} |
|
|
|
|
|
log.Info("✅ frontend_routes 表的 deleted_at 字段移除成功") |
|
|
return nil |
|
|
} |
|
|
|
|
|
// removeDeleteAtFromRouteMappings 移除 route_mappings 表的 delete_at 字段 |
|
|
func removeDeleteAtFromRouteMappings(db *gorm.DB, log *zap.Logger) error { |
|
|
log.Info("🔧 开始移除 route_mappings 表的 delete_at 字段...") |
|
|
|
|
|
// 检查表是否存在 |
|
|
tableExists := db.Migrator().HasTable("route_mappings") |
|
|
log.Info("📊 表存在检查", zap.Bool("route_mappings_exists", tableExists)) |
|
|
|
|
|
if !tableExists { |
|
|
log.Info("⚠️ route_mappings 表不存在，跳过") |
|
|
return nil |
|
|
} |
|
|
|
|
|
// 检查 delete_at 字段是否存在 |
|
|
columnExists := db.Migrator().HasColumn("route_mappings", "deleted_at") |
|
|
log.Info("📊 字段存在检查", zap.Bool("deleted_at_exists", columnExists)) |
|
|
|
|
|
if !columnExists { |
|
|
log.Info("ℹ️ route_mappings 表没有 deleted_at 字段，跳过") |
|
|
return nil |
|
|
} |
|
|
|
|
|
// 删除 delete_at 字段 |
|
|
ddlQuery := "ALTER TABLE route_mappings DROP COLUMN deleted_at" |
|
|
log.Info("🗑️ 执行DDL", zap.String("query", ddlQuery)) |
|
|
|
|
|
if err := db.Exec(ddlQuery).Error; err != nil { |
|
|
log.Error("❌ 删除 deleted_at 字段失败", zap.Error(err)) |
|
|
return fmt.Errorf("删除 deleted_at 字段失败: %w", err) |
|
|
} |
|
|
|
|
|
log.Info("✅ route_mappings 表的 deleted_at 字段移除成功") |
|
|
return nil |
|
|
}
|
|
|
|
