当前位置：首页 > news >正文

iOS弱引用

news 来源：原创 2024/9/20 19:40:14

背景：在面试过程中被问到如果两个对象已经发生循环引用了，该如何将他们剪断，在运行态的时候。

由于这个场景比较抽象，我理解面试官是希望我通过运行时的方法和方式来解决循环引用。

解决方案一:

重写setter用关联对象来实现weak引用。由于objc_setAssociatedObject中的策略不支持weak修饰对象属性。如果我们可以借助中间类或者block持有弱引用对象来实现。

- (void)setMyObject:(id)myObject
{id __weak weakObject = myObject;id (^block)(void) = ^ {return weakObject;};objc_setAssociatedObject(self, @selector(myObject), block, OBJC_ASSOCIATION_COPY_NONATOMIC);
}- (id)myObject
{id (^block)(void) = objc_getAssociatedObject(self, _cmd);return block();
}

解决方案二:

还是重写setter通过关联对象实现弱引用，但是弱引用的实现不是通过中间对象的方式，而是通过runtime运行时重写value对象的子类的delloc方法，在这个方法中将关联对象的value设置成ni；

void weak_setAssociatedObject(id _Nonnull object,const void * _Nonnull key,id _Nullable value) {//派生一个子类，类名为WeakObjWrapper+value对应的类名const char *clsName = [[NSString stringWithFormat:@"WeakObjWrapper%@", [value class]] UTF8String];//获取派生的子类Class childCls = objc_getClass(clsName);//如果子类不存在，利用runtime动态的创建一个子类if (!childCls) {childCls = objc_allocateClassPair([value class], clsName, 0);objc_registerClassPair(childCls);}//注册dealloc方法SELSEL sel = sel_registerName("dealloc");//获取dealloc对应的类型编码const char *deallocEncoding = method_getTypeEncoding(class_getInstanceMethod([value class], sel));// 注意：内部持有value此处需要弱引用处理一下__weak typeof(value) weakValue = value;// 创建一个指向在调用dealloc方法时调用指定block的函数指针IMP deallocImp = imp_implementationWithBlock(^(id _childCls) {//在子类的dealloc方法中将value设置为nil，避免崩溃objc_setAssociatedObject(object, key, nil, OBJC_ASSOCIATION_ASSIGN);//派生的子类的dealloc方法会被调用，父类的不再被调用，故在此处调用一下父类的((void (*)(id, SEL))(void *)objc_msgSend)(weakValue, sel);});//给子类添加dealloc方法class_addMethod(childCls, sel, deallocImp, deallocEncoding);//将value对应的isa指向子类object_setClass(value, childCls);//设置关联对象objc_setAssociatedObject(object, key, value, OBJC_ASSOCIATION_ASSIGN);
}- (id)anthorObj
{return objc_getAssociatedObject(self, _cmd);
}- (void)setAnthorObj:(id)anthorObj
{weak_setAssociatedObject(self, @selector(anthorObj), anthorObj);
}

测试代码如下：

Engine * a = [[Engine alloc] init];{Tire * b = [[Tire alloc] init];Tire * c = [[Tire alloc] init];a.myObject = b;a.anthorObj = c;NSLog(@"myObject%@", a.myObject);NSLog(@"anthorObject%@", a.anthorObj);}NSLog(@"myObject outer%@", a.myObject);NSLog(@"anthorObject outer%@", a.anthorObj);