swift 列出一个类的所有子类

wnrlj8wa  于 2024-01-05  发布在  Swift
关注(0)|答案(4)|浏览(126)

我可以返回一个类的所有子类的列表吗?例如:

  1. class Mother {
  2. }
  3. class ChildFoo: Mother {
  4. }
  5. class ChildBar: Mother {
  6. }
  7. let motherSubclasses = ... // TODO
  8. print(motherSubclasses) // should to return [ChildFoo.self, ChildBar.self]

字符串

pxy2qtax

pxy2qtax1#

令人惊讶的是,即使Swift类不是NSObject的子类,Swift中的所有类似乎都是从SwiftObject派生的。SwiftObject本身没有超类。
首先,处理ObjC运行时函数的 Package 器结构:

  1. import Foundation
  2. struct ClassInfo : CustomStringConvertible, Equatable {
  3. let classObject: AnyClass
  4. let className: String
  5. init?(_ classObject: AnyClass?) {
  6. guard classObject != nil else { return nil }
  7. self.classObject = classObject!
  8. let cName = class_getName(classObject)!
  9. self.className = String(cString: cName)
  10. }
  11. var superclassInfo: ClassInfo? {
  12. let superclassObject: AnyClass? = class_getSuperclass(self.classObject)
  13. return ClassInfo(superclassObject)
  14. }
  15. var description: String {
  16. return self.className
  17. }
  18. static func ==(lhs: ClassInfo, rhs: ClassInfo) -> Bool {
  19. return lhs.className == rhs.className
  20. }
  21. }

字符串
以下是您可以如何使用它:

  1. class Mother { }
  2. class ChildFoo: Mother { }
  3. class ChildBar: Mother { }
  4. class AnIrrelevantClass { }
  5. let motherClassInfo = ClassInfo(Mother.self)!
  6. var subclassList = [ClassInfo]()
  7. var count = UInt32(0)
  8. let classList = objc_copyClassList(&count)!
  9. for i in 0..<Int(count) {
  10. if let classInfo = ClassInfo(classList[i]),
  11. let superclassInfo = classInfo.superclassInfo,
  12. superclassInfo == motherClassInfo
  13. {
  14. subclassList.append(classInfo)
  15. }
  16. }
  17. print(subclassList)


这只执行一个浅层搜索,所以它不会清理孙子类,但你明白的想法。

展开查看全部
uwopmtnx

uwopmtnx2#

Jean Le Moignan代码的优化版本

  1. static func subclasses<T>(of theClass: T) -> [T] {
  2. var count: UInt32 = 0, result: [T] = []
  3. let allClasses = objc_copyClassList(&count)!
  4. let classPtr = address(of: theClass)
  5. for n in 0 ..< count {
  6. let someClass: AnyClass = allClasses[Int(n)]
  7. guard let someSuperClass = class_getSuperclass(someClass), address(of: someSuperClass) == classPtr else { continue }
  8. result.append(someClass as! T)
  9. }
  10. return result
  11. }

个字符
对于每个Type,在运行时只有一个元类示例,因此,它们上的指针是唯一的。出于某种原因,操作符===不允许用于AnyClass,但我们可以直接比较指针
性能测试:

  1. let start = CFAbsoluteTimeGetCurrent()
  2. let found = RuntimeUtils.subclasses(of:UIViewController.self)
  3. let diff = CFAbsoluteTimeGetCurrent() - start
  4. print("Took \(diff) seconds, \(found.count) found")


产出:
字符串(描述:theClass):Took 1.0465459823608398 seconds, 174 found
地址(of:theClass):Took 0.2642860412597656 seconds, 174 found

展开查看全部
svmlkihl

svmlkihl3#

以下是基于Code Different之前的答案的变体,但更短:

  1. func subclasses<T>(of theClass: T) -> [T] {
  2. var count: UInt32 = 0, result: [T] = []
  3. let allClasses = objc_copyClassList(&count)!
  4. for n in 0 ..< count {
  5. let someClass: AnyClass = allClasses[Int(n)]
  6. guard let someSuperClass = class_getSuperclass(someClass), String(describing: someSuperClass) == String(describing: theClass) else { continue }
  7. result.append(someClass as! T)
  8. }
  9. return result
  10. }

字符串
返回类型将根据接收变量的类型,这要归功于泛型。
我想把它作为AnyClass的一个扩展来写,但不幸的是Swift不允许。

展开查看全部
sxissh06

sxissh064#

objc_copyClassList返回的结果不再是可验证的。这是有效的:

  1. public func subclasses<T>(of theClass: T) -> [T] {
  2. var count: UInt32 = 0, result: [T] = []
  3. let classList = objc_copyClassList(&count)!
  4. defer { free(UnsafeMutableRawPointer(classList)) }
  5. let classes = UnsafeBufferPointer(start: classList, count: Int(count))
  6. let classPtr = address(of: theClass)
  7. for someClass in classes {
  8. guard let someSuperClass = class_getSuperclass(someClass), address(of: someSuperClass) == classPtr else { continue }
  9. result.append(someClass as! T)
  10. }
  11. return result
  12. }
  13. public func address(of object: Any?) -> UnsafeMutableRawPointer {
  14. return Unmanaged.passUnretained(object as AnyObject).toOpaque()
  15. }

字符串

展开查看全部

相关问题