manta/gofaster/backend/internal/auth/migration/remove_delete_at_fields.go

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
}