Swift带关联值枚举如何自动获取case总数及各case索引

你不需要为每个枚举手动编写switch分支的样板代码，以下三种方案都可以满足需求，适配最新Swift版本：

方案1：基于Swift ABI元数据的运行时实现（仅需声明协议遵守，无额外样板） #

Swift 5之后ABI稳定，枚举的内存布局、元数据结构固定，可直接通过内存读取拿到枚举总case数、当前实例的case索引，支持带关联值的枚举，在可以接受unsafe代码的前提下这是成本最低的实现。
首先定义通用协议和默认实现：

protocol EnumCaseIndexable {
    static var caseCount: Int { get }
    var idx: Int { get }
}

// 按照Swift ABI定义枚举元数据内存结构
private struct EnumMetadata {
    let kind: UnsafeRawPointer
    let typeDescriptor: UnsafePointer<EnumTypeDescriptor>
}

private struct EnumTypeDescriptor {
    let flags: UInt32
    let parent: Int32
    let name: Int32
    let accessFunctionPointer: Int32
    let reflectionFieldDescriptor: Int32
    let payloadSizeAndCaseCount: UInt32
    
    func fetchTotalCaseCount() -> Int {
        let hasAssociatedValueFlag: UInt32 = 0x1F
        if flags & hasAssociatedValueFlag != 0 {
            // 带关联值枚举，case总数存储在字段低24位
            return Int(payloadSizeAndCaseCount & 0xFFFFFF)
        } else {
            // 无关联值枚举，case总数存储在字段低16位
            return Int(payloadSizeAndCaseCount & 0xFFFF)
        }
    }
}

extension EnumCaseIndexable {
    static var caseCount: Int {
        let metadata = unsafeBitCast(Self.self, to: UnsafePointer<EnumMetadata>.self)
        return metadata.pointee.typeDescriptor.pointee.fetchTotalCaseCount()
    }
    
    var idx: Int {
        var instance = self
        return withUnsafeBytes(of: &instance) { buffer in
            // 常规场景下case判别式存储在实例首字节，覆盖绝大多数业务场景
            // 若遇到case数超过256的极端枚举，可对应调整读取位宽
            Int(buffer.load(as: UInt8.self))
        }
    }
}

使用时只需要让目标枚举遵守该协议即可，不需要编写任何额外逻辑：

enum MyEnum: EnumCaseIndexable {
  case Position(x: Int, y: Int)
  case Name(String)
}

// 调用测试
MyEnum.caseCount // 返回2
let val = MyEnum.Position(x: 0, y: 0)
val.idx // 返回0
MyEnum.Name("My awesome name").idx // 返回1

该实现兼容iOS 13+、macOS 10.15+及以上搭载Swift 5运行时的系统。如果遇到特殊内存布局的枚举（比如关联值内存对齐规则特殊导致判别式不在首字节），可以调整idx的读取逻辑，或通过Mirror获取当前实例的case名称，和元数据中读取到的全量case名列表匹配得到索引，仅会增加极少量运行时开销。

方案2：编译期代码生成（无运行时风险，性能最优） #

如果不想使用unsafe代码，可以通过代码生成工具，在编译阶段自动为枚举生成和手写逻辑完全一致的扩展代码：

• 工具会扫描项目中所有需要实现该能力的枚举
• 自动生成caseCount静态属性、idx实例属性的实现，内部自动完成switch分支匹配
• 生成的代码和你手动编写的样板代码完全一致，无任何运行时开销，也不存在ABI兼容问题

方案3：Swift 5.9+ 宏实现 #

如果项目使用Swift 5.9及以上版本，可以自定义编译期附加宏：

• 只需要在枚举定义前添加宏标记
• 宏会在编译时自动为枚举拼接caseCount和idx的实现逻辑
• 不需要额外引入代码生成工具链，也没有运行时开销

补充说明 #

系统提供的CaseIterable无法自动为带关联值枚举生成实现，核心原因是该协议要求返回所有case的实例数组，而带关联值的case初始化需要传入关联值，无法自动生成实例。但你需要的总case数、实例对应索引能力不需要生成全量case实例，因此不受该限制。

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