Swift中能否自动实现枚举类型映射，替代switch语句？

问题场景 #

现有Swift项目中存在如下代码，getUrlFromPageType方法通过Switch语句实现PageType枚举到UrlConfig?的映射。该方法包含20余个case，除2个返回nil的例外情况外，其余输入枚举值的名称与返回枚举值名称完全一致，仅类型不同：

static func getUrlFromPageType(_ type: PageType) -> UrlConfig? {
    switch type {
    case .helpScreen:
        return .helpScreen
    case .contact:
        return .contact
    case .termsAndConditions:
        return .termsAndConditions
    // etc
    }
}

假设无法重构为更合理的设计方案，如何自动实现这类枚举间的映射，避免冗长的Switch语句？

可行解决方案 #

方法1：利用枚举RawValue关联（推荐） #

如果两个枚举的对应case可以使用相同类型的RawValue（比如String），这是最简洁高效的方式：

1. 先确保两个枚举定义一致的RawValue：

enum PageType: String {
    case helpScreen
    case contact
    case termsAndConditions
    case exceptionCase1 // 需要返回nil的例外case
    case exceptionCase2
}

enum UrlConfig: String {
    case helpScreen
    case contact
    case termsAndConditions
    // 无需包含例外case
}

2. 简化映射方法：

static func getUrlFromPageType(_ type: PageType) -> UrlConfig? {
    // 统一维护例外列表
    let exceptionCases: [PageType] = [.exceptionCase1, .exceptionCase2]
    guard !exceptionCases.contains(type) else { return nil }
    // 通过RawValue直接完成类型转换
    return UrlConfig(rawValue: type.rawValue)
}

方法2：利用Swift反射获取case名称 #

如果枚举无法定义RawValue，可通过反射读取枚举case的名称，再初始化目标枚举：

1. 实现映射方法：

static func getUrlFromPageType(_ type: PageType) -> UrlConfig? {
    let exceptionCases: [PageType] = [.exceptionCase1, .exceptionCase2]
    guard !exceptionCases.contains(type) else { return nil }
    
    // 反射获取当前case的名称
    let mirror = Mirror(reflecting: type)
    guard let caseName = mirror.children.first?.label else { return nil }
    
    // 尝试通过case名称初始化UrlConfig（需先给UrlConfig添加对应初始化方法）
    return UrlConfig.init(caseName: caseName)
}

2. 给UrlConfig添加自定义初始化方法：

extension UrlConfig {
    init?(caseName: String) {
        switch caseName {
        case "helpScreen": self = .helpScreen
        case "contact": self = .contact
        case "termsAndConditions": self = .termsAndConditions
        // 补充其他case...
        default: return nil
        }
    }
}

方法3：自定义协议统一映射逻辑 #

通过定义通用协议，让两个枚举遵循，实现可复用的转换逻辑：

1. 定义协议：

protocol CaseNameConvertible {
    var caseName: String { get }
    init?(caseName: String)
}

extension CaseNameConvertible {
    // 默认实现：通过反射获取case名称
    var caseName: String {
        let mirror = Mirror(reflecting: self)
        return mirror.children.first?.label ?? ""
    }
}

2. 让两个枚举遵循协议：

extension PageType: CaseNameConvertible {}

extension UrlConfig: CaseNameConvertible {
    // 实现UrlConfig的case名称初始化逻辑
    init?(caseName: String) {
        switch caseName {
        case "helpScreen": self = .helpScreen
        case "contact": self = .contact
        case "termsAndConditions": self = .termsAndConditions
        // 补充其他case...
        default: return nil
        }
    }
}

3. 简化映射方法：

static func getUrlFromPageType(_ type: PageType) -> UrlConfig? {
    let exceptionCases: [PageType] = [.exceptionCase1, .exceptionCase2]
    guard !exceptionCases.contains(type) else { return nil }
    return UrlConfig(caseName: type.caseName)
}

注意事项 #

• 无论采用哪种方式，都需要保证两个枚举的对应case名称完全一致，否则映射会失败；
• 反射方式的性能略低于RawValue转换，但在20余个case的场景下，性能差异可忽略；
• 例外情况通过统一列表维护，比在Switch中单独写case更便于后续修改。

内容的提问来源于stack exchange，提问作者Cassiterite