检查给定的元类型是否为枚举

Question

给定方法

func enumCaseCount<T: Hashable>(ofType type: T.Type) -> Int {
    // Needed check if type is an enum type

   return 3
}

如下使用

private enum SimpleEnum: String {
    case a = "A"
    case b = "B"
    case c = "C"
}

enumCaseCount(ofType: SimpleEnum.self)

知道如何检查给定的元类型是否为枚举吗？

---

可以通过这种方式测试类

class Test {}
Test.self is AnyClass // returns true

Answer 1

为了好玩，作为（解决方法）hack，我们可以实例化T 的实例并使用Mirror 对其执行运行时自省，特别是它的displayStyle 属性。在我们继续之前，我们注意到我们只会将其用于调试目的

游乐场和调试器使用镜像。

我还要指出，当我们求助于运行时来查询编译时已知的事物（至少是编译器）时，我们真的在追尾。

---

无论如何，首先，我将enumCaseCount(...) 重命名为isEnum(...)，因为这个问题只涉及查询元类型是否为enum。对于查询给定enum 的案例数量的类似（有点脆弱）黑客，请参阅：

• How do I get the count of a Swift enum?

现在，isEnum(...) 中的通用占位符 T 只知道它是符合 Hashable 的类型，这并没有为我们提供任何直接的方法来实例化 T 的实例（如果 @ 987654341@ 蓝图，比如说，一个初始化器init()，我们可以很容易地构造一个T 的实例并对其执行运行时自省）。相反，我们将求助于为单个 T 实例 (UnsafeMutableRawPointer.allocate(bytes:alignedTo:)) 手动分配原始内存，将其绑定到 T (bindMemory(to:capacity:))，最后释放内存 (deallocate(bytes:alignedTo:))完成了对指向绑定内存的指针引用的实例的运行时自省。至于运行时自省，我们简单地使用Mirror来检查它的displayStyle是否为enum。

func isEnum<T: Hashable>(_: T.Type) -> Bool {
    var result = false
    // Allocate memory with size and alignment matching T.
    let bytesPointer = UnsafeMutableRawPointer.allocate(
        bytes: MemoryLayout<T>.size,
        alignedTo: MemoryLayout<T>.alignment)
    // Bind memory to T and perform introspection on the instance
    // reference to by the bound memory.
    if case .some(.`enum`) = Mirror(reflecting:
        bytesPointer.bindMemory(to: T.self, capacity: 1).pointee)
        .displayStyle {
        print("Is an enum")
        result = true
    } else { print("Is not an enum") }
    // Deallocate the manually allocate memory.
    bytesPointer.deallocate(bytes: MemoryLayout<T>.size,
                            alignedTo: MemoryLayout<T>.alignment)
    return result
}

示例用法：

enum SimpleEnum { case a, b, c }

enum SimpleStrEnum: String {
    case a = "A"
    case b = "B"
    case c = "C"
}

enum SimpleExplicitIntEnum: Int { case a, b, c }

struct SimpleStruct: Hashable {
    let i: Int
    // Hashable
    var hashValue: Int { return 0 }
    static func ==(lhs: SimpleStruct, rhs: SimpleStruct) -> Bool { return true }
}

print(isEnum(SimpleEnum.self))            // true
print(isEnum(SimpleStrEnum.self))         // true
print(isEnum(SimpleExplicitIntEnum.self)) // true
print(isEnum(SimpleStruct.self))          // false

Answer 2

正如其他人所提到的，在 Swift 中没有很好的非 hacky 方式来做到这一点。但是，它是 Sourcery 的示例用例之一，这是一个元编程库（这意味着它会分析您的代码以生成其他代码）。您编写一个 Stencil 模板来描述其行为，并在 Xcode 中作为构建阶段执行。它可以为您项目中找到的任何枚举自动生成此代码。

AutoCases enum example

Answer 3

要检查某个类型是否实际上是 Enum，您可以使用：

func isEnum<T>(_ type: T.Type) -> Bool {
    let ptr = unsafeBitCast(T.self as Any.Type, to: UnsafeRawPointer.self)
    return ptr.load(as: Int.self) == 513
}

您可以像这样简单地使用它：

enum MyEnum {}
struct MyStruct {}
class MyClass {}

isEnum(MyEnum.self)   // this returns true
isEnum(MyStruct.self) // this returns false
isEnum(MyClass.self)  // this returns false

在你的例子中，你会像这样使用它：

func enumCaseCount<T: Hashable>(ofType type: T.Type) -> Int {
   isEnum(T.self) // returns true if enum
   return 3
}