红帽白皮书新鲜出炉!点击获取,让你的云战略更胜一筹! 了解详情
写点什么

如何在 iOS 中解决循环引用的问题

  • 2019-12-10
  • 本文字数:4417 字

    阅读完需:约 14 分钟

如何在 iOS 中解决循环引用的问题

关注仓库,及时获得更新:iOS-Source-Code-Analyze


稍有常识的人都知道在 iOS 开发时,我们经常会遇到循环引用的问题,比如两个强指针相互引用,但是这种简单的情况作为稍有经验的开发者都会轻松地查找出来。


但是遇到下面这样的情况,如果只看其实现代码,也很难仅仅凭借肉眼上的观察以及简单的推理就能分析出其中存在的循环引用问题,更何况真实情况往往比这复杂的多:


Objective-C


testObject1.object = testObject2;testObject1.secondObject = testObject3;testObject2.object = testObject4;testObject2.secondObject = testObject5;testObject3.object = testObject1;testObject5.object = testObject6;testObject4.object = testObject1;testObject5.secondObject = testObject7;testObject7.object = testObject2;
复制代码


上述代码确实是存在循环引用的问题:



这一次分享的内容就是用于检测循环引用的框架 FBRetainCycleDetector 我们会分几个部分来分析 FBRetainCycleDetector 是如何工作的:


  1. 检测循环引用的基本原理以及过程

  2. 检测涉及 NSObject 对象的循环引用问题

  3. 检测涉及 Associated Object 关联对象的循环引用问题

  4. 检测涉及 Block 的循环引用问题


这是四篇文章中的第一篇,我们会以类 FBRetainCycleDetector- findRetainCycles 方法为入口,分析其实现原理以及运行过程。


简单介绍一下 FBRetainCycleDetector 的使用方法:


Objective-C


_RCDTestClass *testObject = [_RCDTestClass new];testObject.object = testObject;
FBRetainCycleDetector *detector = [FBRetainCycleDetector new];[detector addCandidate:testObject];NSSet *retainCycles = [detector findRetainCycles];
NSLog(@"%@", retainCycles);
复制代码


  1. 初始化一个 FBRetainCycleDetector 的实例

  2. 调用 - addCandidate: 方法添加潜在的泄露对象

  3. 执行 - findRetainCycles 返回 retainCycles


在控制台中的输出是这样的:


C


2016-07-29 15:26:42.043 xctest[30610:1003493] {(    (    "-> _object -> _RCDTestClass "  ))}
复制代码


说明 FBRetainCycleDetector 在代码中发现了循环引用。

findRetainCycles 的实现

在具体开始分析 FBRetainCycleDetector 代码之前,我们可以先观察一下方法 findRetainCycles 的调用栈:


Objective-C


- (NSSet<NSArray<FBObjectiveCGraphElement *> *> *)findRetainCycles└── - (NSSet<NSArray<FBObjectiveCGraphElement *> *> *)findRetainCyclesWithMaxCycleLength:(NSUInteger)length    └── - (NSSet<NSArray<FBObjectiveCGraphElement *> *> *)_findRetainCyclesInObject:(FBObjectiveCGraphElement *)graphElement stackDepth:(NSUInteger)stackDepth        └── - (instancetype)initWithObject:(FBObjectiveCGraphElement *)object            └── - (FBNodeEnumerator *)nextObject                ├── - (NSArray<FBObjectiveCGraphElement *> *)_unwrapCycle:(NSArray<FBNodeEnumerator *> *)cycle                ├── - (NSArray<FBObjectiveCGraphElement *> *)_shiftToUnifiedCycle:(NSArray<FBObjectiveCGraphElement *> *)array                └── - (void)addObject:(ObjectType)anObject;
复制代码


调用栈中最上面的两个简单方法的实现都是比较容易理解的:


Objective-C


- (NSSet<NSArray<FBObjectiveCGraphElement *> *> *)findRetainCycles {  return [self findRetainCyclesWithMaxCycleLength:kFBRetainCycleDetectorDefaultStackDepth];}
- (NSSet<NSArray<FBObjectiveCGraphElement *> *> *)findRetainCyclesWithMaxCycleLength:(NSUInteger)length { NSMutableSet<NSArray<FBObjectiveCGraphElement *> *> *allRetainCycles = [NSMutableSet new]; for (FBObjectiveCGraphElement *graphElement in _candidates) { NSSet<NSArray<FBObjectiveCGraphElement *> *> *retainCycles = [self _findRetainCyclesInObject:graphElement stackDepth:length]; [allRetainCycles unionSet:retainCycles]; } [_candidates removeAllObjects];
return allRetainCycles;}
复制代码


- findRetainCycles 调用了 - findRetainCyclesWithMaxCycleLength: 传入了 kFBRetainCycleDetectorDefaultStackDepth 参数来限制查找的深度,如果超过该深度(默认为 10)就不会继续处理下去了(查找的深度的增加会对性能有非常严重的影响)。


- findRetainCyclesWithMaxCycleLength: 中,我们会遍历所有潜在的内存泄露对象 candidate,执行整个框架中最核心的方法 - _findRetainCyclesInObject:stackDepth:,由于这个方法的实现太长,这里会分几块对其进行介绍,并会省略其中的注释:


Objective-C


- (NSSet<NSArray<FBObjectiveCGraphElement *> *> *)_findRetainCyclesInObject:(FBObjectiveCGraphElement *)graphElement                                 stackDepth:(NSUInteger)stackDepth {  NSMutableSet<NSArray<FBObjectiveCGraphElement *> *> *retainCycles = [NSMutableSet new];  FBNodeEnumerator *wrappedObject = [[FBNodeEnumerator alloc] initWithObject:graphElement];
NSMutableArray<FBNodeEnumerator *> *stack = [NSMutableArray new];
NSMutableSet<FBNodeEnumerator *> *objectsOnPath = [NSMutableSet new];
...}
复制代码


其实整个对象的相互引用情况可以看做一个有向图,对象之间的引用就是图的 Edge,每一个对象就是 Vertex查找循环引用的过程就是在整个有向图中查找环的过程,所以在这里我们使用 DFS 来扫面图中的环,这些环就是对象之间的循环引用。


文章中并不会介绍 DFS 的原理,如果对 DFS 不了解的读者可以看一下这个视频,或者找以下相关资料了解一下 DFS 的实现。


接下来就是 DFS 的实现:


Objective-C


- (NSSet<NSArray<FBObjectiveCGraphElement *> *> *)_findRetainCyclesInObject:(FBObjectiveCGraphElement *)graphElement                                 stackDepth:(NSUInteger)stackDepth {  ...  [stack addObject:wrappedObject];
while ([stack count] > 0) { @autoreleasepool { FBNodeEnumerator *top = [stack lastObject]; [objectsOnPath addObject:top];
FBNodeEnumerator *firstAdjacent = [top nextObject]; if (firstAdjacent) {
BOOL shouldPushToStack = NO;
if ([objectsOnPath containsObject:firstAdjacent]) { NSUInteger index = [stack indexOfObject:firstAdjacent]; NSInteger length = [stack count] - index;
if (index == NSNotFound) { shouldPushToStack = YES; } else { NSRange cycleRange = NSMakeRange(index, length); NSMutableArray<FBNodeEnumerator *> *cycle = [[stack subarrayWithRange:cycleRange] mutableCopy]; [cycle replaceObjectAtIndex:0 withObject:firstAdjacent];
[retainCycles addObject:[self _shiftToUnifiedCycle:[self _unwrapCycle:cycle]]]; } } else { shouldPushToStack = YES; }
if (shouldPushToStack) { if ([stack count] < stackDepth) { [stack addObject:firstAdjacent]; } } } else { [stack removeLastObject]; [objectsOnPath removeObject:top]; } } } return retainCycles;}
复制代码


这里其实就是对 DFS 的具体实现,其中比较重要的有两点,一是使用 nextObject 获取下一个需要遍历的对象,二是对查找到的环进行处理和筛选;在这两点之中,第一点相对重要,因为 nextObject 的实现是调用 allRetainedObjects 方法获取被当前对象持有的对象,如果没有这个方法,我们就无法获取当前对象的邻接结点,更无从谈起遍历了:


Objective-C


- (FBNodeEnumerator *)nextObject {  if (!_object) {    return nil;  } else if (!_retainedObjectsSnapshot) {    _retainedObjectsSnapshot = [_object allRetainedObjects];    _enumerator = [_retainedObjectsSnapshot objectEnumerator];  }
FBObjectiveCGraphElement *next = [_enumerator nextObject];
if (next) { return [[FBNodeEnumerator alloc] initWithObject:next]; }
return nil;}
复制代码


基本上所有图中的对象 FBObjectiveCGraphElement 以及它的子类 FBObjectiveCBlock FBObjectiveCObjectFBObjectiveCNSCFTimer 都实现了这个方法返回其持有的对象数组。获取数组之后,就再把其中的对象包装成新的 FBNodeEnumerator 实例,也就是下一个 Vertex


因为使用 - subarrayWithRange: 方法获取的数组中的对象都是 FBNodeEnumerator 的实例,还需要一定的处理才能返回:


  1. (NSArray<FBObjectiveCGraphElement *> *)_unwrapCycle:(NSArray<FBNodeEnumerator *> *)cycle

  2. (NSArray<FBObjectiveCGraphElement *> *)_shiftToUnifiedCycle:(NSArray<FBObjectiveCGraphElement *> *)array


- _unwrapCycle: 的作用是将数组中的每一个 FBNodeEnumerator 实例转换成 FBObjectiveCGraphElement


Objective-C


- (NSArray<FBObjectiveCGraphElement *> *)_unwrapCycle:(NSArray<FBNodeEnumerator *> *)cycle {  NSMutableArray *unwrappedArray = [NSMutableArray new];  for (FBNodeEnumerator *wrapped in cycle) {    [unwrappedArray addObject:wrapped.object];  }
return unwrappedArray;}
复制代码


- _shiftToUnifiedCycle: 方法将每一个环中的元素按照地址递增以及字母顺序来排序,方法签名很好的说明了它们的功能,两个方法的代码就不展示了,它们的实现没有什么值得注意的地方:


Objective-C


- (NSArray<FBObjectiveCGraphElement *> *)_shiftToUnifiedCycle:(NSArray<FBObjectiveCGraphElement *> *)array {  return [self _shiftToLowestLexicographically:[self _shiftBufferToLowestAddress:array]];}
复制代码


方法的作用是防止出现相同环的不同表示方式,比如说下面的两个环其实是完全相同的:


-> object1 -> object2-> object2 -> object1
复制代码


在获取图中的环并排序好之后,就可以讲这些环 union 一下,去除其中重复的元素,最后返回所有查找到的循环引用了。

总结

到目前为止整个 FBRetainCycleDetector 的原理介绍大概就结束了,其原理完全是基于 DFS 算法:把整个对象的之间的引用情况当做图进行处理,查找其中的环,就找到了循环引用。不过原理真的很简单,如果这个 lib 的实现仅仅是这样的话,我也不会写几篇文章来专门分析这个框架,真正让我感兴趣的还是 - allRetainedObjects 方法在各种对象以及 block 中获得它们强引用的对象的过程,这也是之后的文章要分析的主要内容。


关注仓库,及时获得更新:iOS-Source-Code-Analyze


本文转载自 Draveness 技术博客。


原文链接:https://draveness.me/retain-cycle1


2019-12-10 17:57666

评论

发布
暂无评论
发现更多内容

极客大学架构师训练营--食堂就餐系统架构设计⽂档 -- 第一次作业

John(易筋)

极客时间 极客大学 极客大学架构师训练营

食堂就餐卡系统设计

mh

系统设计

命题作业—第一周

于江水

极客大学架构师训练营

食堂就餐卡系统设计

GalaxyCreater

第一周学习总结

Geek_5d0795

极客大学架构师训练营 第一周总结

食堂就餐卡系统架构设计文档

朱月俊

为啥成为架构师总是少数人?

朱月俊

学习总结—第一周

于江水

极客大学架构师训练营

Week1-架构是什么&如何做架构

wyzwlj

极客大学架构师训练营

架构师训练营 Week01 学习心得

极客大学架构师训练营

架构师训练营第一周

Melo

架构师训练营第一周总结

小树林

Week 01-作业一:食堂就餐卡系统设计

dean

架构师训练营第一周总结

Geek_2dfa9a

Week 01-作业二:学习总结

dean

架构师训练营 Week 01

Wancho

Week1-Homework

深入理解JVM垃圾回收机制 - 对象的内存布局

SkyeDance

深入理解JVM 内存布局 垃圾回收

架构师训练营第0期-第一周-命题作业1

sljoai

作业

架构师训练营第0期-第一周学习总结

sljoai

作业

食堂就餐卡系统架构设计⽂档

Geek_2dfa9a

食堂就餐卡系统设计

dapaul

架构设计 极客大学架构师训练营

使用IBM Blockchain Platform extension开发你的第一个fabric智能合约

程序那些事

区块链 blockchain hyperledger fabric ibm

第1周 - 食堂就餐卡系统

大海

极客大学架构师训练营

小师妹学JavaIO之:Buffer和Buff

程序那些事

io nio 小师妹 buffer buff

第一周学习总结

战峰

第1周小结

龙7

极客大学架构师训练营

食堂就餐卡系统设计

_MISSYOURLOVE

极客大学架构师训练营 第一周命题作业

在IBM Cloud中运行Fabric

程序那些事

vscode blockchain hyperledger fabric ibm

架构师训练营学习总结(week-01)

Justin

架构师训练营作业(week-01)

Justin

如何在 iOS 中解决循环引用的问题_语言 & 开发_Draveness_InfoQ精选文章