iOS开发-如何使用Runtime在分类中添加一个weak属性

使用runtime给分类添加属性

- (HCPushSettingViewController*)HCParentController {
    return objc_getAssociatedObject(self, _cmd);
}

- (void)setHCParentController:(HCPushSettingViewController*)parentController {
    objc_setAssociatedObject(self, @selector(HCParentController), parentController, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}

枚举中并没有weak的

/* Associative References */

/**
 * Policies related to associative references.
 * These are options to objc_setAssociatedObject()
 */
typedef OBJC_ENUM(uintptr_t, objc_AssociationPolicy) {
    OBJC_ASSOCIATION_ASSIGN = 0,           /**< Specifies a weak reference to the associated object. */
    OBJC_ASSOCIATION_RETAIN_NONATOMIC = 1, /**< Specifies a strong reference to the associated object. 
                                            *   The association is not made atomically. */
    OBJC_ASSOCIATION_COPY_NONATOMIC = 3,   /**< Specifies that the associated object is copied. 
                                            *   The association is not made atomically. */
    OBJC_ASSOCIATION_RETAIN = 01401,       /**< Specifies a strong reference to the associated object.
                                            *   The association is made atomically. */
    OBJC_ASSOCIATION_COPY = 01403          /**< Specifies that the associated object is copied.
                                            *   The association is made atomically. */
};

相比而言 OBJC_ASSOCIATION_ASSIGN 更接近，保持弱引用，但是不会在引用计数为0后，将指针置为nil.

使用weak的目的为了避免循环引用，多余的引用

我这里想到了NSProxy，他可以转发消息，又不会造成循环引用的问题，常用于NSTimer和CADisplayLink。

那我使用NSProxy作为属性，通过消息转发就可以实现属性功能，其不会循环引用的特性也符合weak的定义

---

使用NSProxy

.h

@interface HCProxy : NSProxy
@property (nonatomic, weak, readonly) id target;
- (instancetype)initWithTarget:(id)target;
+ (instancetype)proxyWithTarget:(id)target;
@end


@interface UIViewController (HCPush)
//use Proxy to avoid retain cycle
@property (nonatomic, strong) HCPushSettingViewController *HCParentController;

@end

.m

@implementation HCProxy
- (instancetype)initWithTarget:(id)target {
    _target = target;
    return self;
}
+ (instancetype)proxyWithTarget:(id)target {
    return [[HCProxy alloc] initWithTarget:target];
}
- (id)forwardingTargetForSelector:(SEL)selector {
    return _target;
}
- (void)forwardInvocation:(NSInvocation *)invocation {
    void *null = NULL;
    [invocation setReturnValue:&null];
}
- (NSMethodSignature *)methodSignatureForSelector:(SEL)selector {
    return [NSObject instanceMethodSignatureForSelector:@selector(init)];
}
- (BOOL)respondsToSelector:(SEL)aSelector {
    return [_target respondsToSelector:aSelector];
}
- (BOOL)isEqual:(id)object {
    return [_target isEqual:object];
}
- (NSUInteger)hash {
    return [_target hash];
}
- (Class)superclass {
    return [_target superclass];
}
- (Class)class {
    return [_target class];
}
- (BOOL)isKindOfClass:(Class)aClass {
    return [_target isKindOfClass:aClass];
}
- (BOOL)isMemberOfClass:(Class)aClass {
    return [_target isMemberOfClass:aClass];
}
- (BOOL)conformsToProtocol:(Protocol *)aProtocol {
    return [_target conformsToProtocol:aProtocol];
}
- (BOOL)isProxy {
    return YES;
}
- (NSString *)description {
    return [_target description];
}
- (NSString *)debugDescription {
    return [_target debugDescription];
}
@end

@implementation UIViewController (HCPush)

- (HCPushSettingViewController*)HCParentController {
    HCProxy *proxy = objc_getAssociatedObject(self, _cmd);
    return (HCPushSettingViewController*)proxy;
}

- (void)setHCParentController:(HCPushSettingViewController*)parentController {
    HCProxy *proxy = [HCProxy proxyWithTarget:parentController];
    objc_setAssociatedObject(self, @selector(HCParentController), proxy, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}

@end

---

当然在runtime接口判断其class类型或者isa时是会有问题，但是我只要实现消息转发就没问题。

也可以创建一个NSObject的子类对象，将weak对象置为其属性即可，通过strong->weak来实现