欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

YYClassInfo.h.m

程序员文章站 2024-02-02 15:03:58
...
/**
 Class information for a class.    
一个类的类信息模型
 */
@interface YYClassInfo : NSObject

///< class object   本类类对象
@property (nonatomic, assign, readonly) Class cls; 

///< super class object 父类类对象
@property (nullable, nonatomic, assign, readonly) Class superCls; 

///< class's meta class object  元类类对象
@property (nullable, nonatomic, assign, readonly) Class metaCls;  

///< whether this class is meta class   是否是一个元类
@property (nonatomic, readonly) BOOL isMeta; 

///< class name   类名
@property (nonatomic, strong, readonly) NSString *name; 

///< super class's class info  父类的类信息
@property (nullable, nonatomic, strong, readonly) YYClassInfo *superClassInfo; 

// 变量列表(数组)
@property (nullable, nonatomic, strong, readonly) NSDictionary<NSString *, YYClassIvarInfo *> *ivarInfos; ///< ivars

// 方法列表(数组)
@property (nullable, nonatomic, strong, readonly) NSDictionary<NSString *, YYClassMethodInfo *> *methodInfos; ///< methods

// 属性列表(数组)
@property (nullable, nonatomic, strong, readonly) NSDictionary<NSString *, YYClassPropertyInfo *> *propertyInfos; ///< properties

/**
 If the class is changed (for example: you add a method to this class with
 'class_addMethod()'), you should call this method to refresh the class info cache.
 如果这个类是改变了的(例如: 你通过调用 class_addMethod() 给这类添加了方法 ),你就需要调用这个方法来刷新类的缓存信息。
 
 After called this method, `needUpdate` will returns `YES`, and you should call 
 'classInfoWithClass' or 'classInfoWithClassName' to get the updated class info.
调用这个方法后, needUpdate 将返回 yes, 你将调用 classInfoWithClass 或者 classInfoWithClassName 来根系类信息。
 */
- (void)setNeedUpdate;

/**
 If this method returns `YES`, you should stop using this instance and call
 `classInfoWithClass` or `classInfoWithClassName` to get the updated class info.
如果这个方法返回 yes , 你需要停止使用这个类的实例并调用 classInfoWithClass 或  classInfoWithClassName 去更新类信息。
 
 @return Whether this class info need update.
 */
- (BOOL)needUpdate;

/**
 Get the class info of a specified Class.
 获取指定类的类信息。
 
 @discussion This method will cache the class info and super-class info
 at the first access to the Class. This method is thread-safe.
在第一次使用这个类的时候,这个方法将存储本类的类信息和父类的类信息。 这个方法是线程安全的。
 
 @param cls A class.
 @return A class info, or nil if an error occurs.  
 一个类的类信息,如果是 nil 就会发生一个错误。
 */
+ (nullable instancetype)classInfoWithClass:(Class)cls;


这个方法和上面的方法是一样的。
/**
 Get the class info of a specified Class.
 
 @discussion This method will cache the class info and super-class info
 at the first access to the Class. This method is thread-safe.
 
 @param className A class name.
 @return A class info, or nil if an error occurs.
 */
+ (nullable instancetype)classInfoWithClassName:(NSString *)className;
@implementation YYClassInfo {

    // 是否需要更新类信息
    BOOL _needUpdate;
}

- (instancetype)initWithClass:(Class)cls {

    // cls 为 nil 就直接返回 nil
    if (!cls) return nil;

    // 初始化父类
    self = [super init];
    
    // 记录 “类”
    _cls = cls;

    // 记录“父类”
    _superCls = class_getSuperclass(cls);

    // 判断是否是 "元" 类
    /*
      判断是一个 元类,主要是要确定是一个 “类“ 不是一个 ”实例“
      */ 
    _isMeta = class_isMetaClass(cls);

    // 不是 “元” 类
    if (!_isMeta) {

      // 获取当前类的 “元” 类
        _metaCls = objc_getMetaClass(class_getName(cls));
    }

    // 获取类名
    _name = NSStringFromClass(cls);

    // 调用 update 方法
    [self _update];

    // 获取父类的类信息
    _superClassInfo = [self.class classInfoWithClass:_superCls];
    return self;
}

- (void)_update {
    _ivarInfos = nil;
    _methodInfos = nil;
    _propertyInfos = nil;
    

    // 实例一个 "类" 制作
    Class cls = self.cls;

    // 定义变量记录方法的个数
    unsigned int methodCount = 0;

    // 获取方法列表信息
    Method *methods = class_copyMethodList(cls, &methodCount);
    if (methods) {

        // 获取方法信息
        NSMutableDictionary *methodInfos = [NSMutableDictionary new];
        _methodInfos = methodInfos;

        for (unsigned int i = 0; i < methodCount; i++) {
            YYClassMethodInfo *info = [[YYClassMethodInfo alloc] initWithMethod:methods[i]];
            if (info.name) methodInfos[info.name] = info;
        }
        free(methods);
    }


    // 获取属性列表信息
    unsigned int propertyCount = 0;
    objc_property_t *properties = class_copyPropertyList(cls, &propertyCount);
    if (properties) {
        NSMutableDictionary *propertyInfos = [NSMutableDictionary new];
        _propertyInfos = propertyInfos;
        for (unsigned int i = 0; i < propertyCount; i++) {
            YYClassPropertyInfo *info = [[YYClassPropertyInfo alloc] initWithProperty:properties[i]];
            if (info.name) propertyInfos[info.name] = info;
        }
        free(properties);
    }
    

    // 获取变量信息
    unsigned int ivarCount = 0;
    Ivar *ivars = class_copyIvarList(cls, &ivarCount);
    if (ivars) {
        NSMutableDictionary *ivarInfos = [NSMutableDictionary new];
        _ivarInfos = ivarInfos;
        for (unsigned int i = 0; i < ivarCount; i++) {
            YYClassIvarInfo *info = [[YYClassIvarInfo alloc] initWithIvar:ivars[i]];
            if (info.name) ivarInfos[info.name] = info;
        }
        free(ivars);
    }
    

    // 当获取的方法列表信息,属性列表信息,变量列表信息为空的时候对相应的存储变量置空
    if (!_ivarInfos) _ivarInfos = @{};
    if (!_methodInfos) _methodInfos = @{};
    if (!_propertyInfos) _propertyInfos = @{};
    
    // 更新是否更新的标记
    _needUpdate = NO;
}

- (void)setNeedUpdate {
    _needUpdate = YES;
}

- (BOOL)needUpdate {
    return _needUpdate;
}


// 一个类方法
+ (instancetype)classInfoWithClass:(Class)cls {
    // 判断类是否有值
    if (!cls) return nil;


    // 缓存池用的是全局静态变量
    // 类缓存池
    static CFMutableDictionaryRef classCache;
    // 元类缓存池
    static CFMutableDictionaryRef metaCache;

    // 锁
    static dispatch_once_t onceToken;
    static dispatch_semaphore_t lock;

    // 类缓存池,和 元类缓存池保证只初始化一次
    dispatch_once(&onceToken, ^{
        classCache = CFDictionaryCreateMutable(CFAllocatorGetDefault(), 0, &kCFTypeDictionaryKeyCallBacks, &kCFTypeDictionaryValueCallBacks);
        metaCache = CFDictionaryCreateMutable(CFAllocatorGetDefault(), 0, &kCFTypeDictionaryKeyCallBacks, &kCFTypeDictionaryValueCallBacks);
        lock = dispatch_semaphore_create(1);
    });
    dispatch_semaphore_wait(lock, DISPATCH_TIME_FOREVER);

    // 获取缓存中的信息
    YYClassInfo *info = CFDictionaryGetValue(class_isMetaClass(cls) ? metaCache : classCache, (__bridge const void *)(cls));
    if (info && info->_needUpdate) {
        [info _update];
    }
    dispatch_semaphore_signal(lock);

    // 没有从缓存池中获取信息直接实例化一个
    if (!info) {
        info = [[YYClassInfo alloc] initWithClass:cls];
        if (info) {
            dispatch_semaphore_wait(lock, DISPATCH_TIME_FOREVER);

            // 将信息保存到缓存池中
            CFDictionarySetValue(info.isMeta ? metaCache : classCache, (__bridge const void *)(cls), (__bridge const void *)(info));
            dispatch_semaphore_signal(lock);
        }
    }
    return info;
}

+ (instancetype)classInfoWithClassName:(NSString *)className {

    // 通过字符串实例化一个类
    Class cls = NSClassFromString(className);
    return [self classInfoWithClass:cls];
}

@end

推荐阅读