NSTimer循环引用的几种解决方案

前言

在ios中，nstimer的使用是非常频繁的，但是nstimer在使用中需要注意，避免循环引用的问题。之前经常这样写：

- (void)setuptimer {
    self.timer = [nstimer scheduledtimerwithtimeinterval:1 target:self selector:@selector(timeraction) userinfo:nil repeats:yes];
}

- (void)dealloc {
    [self.timer invalidate];
    self.timer = nil;
}

由于self强引用了timer，同时timer也强引用了self，所以循环引用造成dealloc方法根本不会走，self和timer都不会被释放，造成内存泄漏。

下面介绍一下几种解决timer循环引用的方法。

1. 选择合适的时机手动释放timer(该方法并不太合理)

在之前自己就是这样解决循环引用的：

• 控制器中

- (void)viewdiddisappear:(bool)animated {
    [super viewdiddisappear:animated];
    
    [self.timer invalidate];
    self.timer = nil;
}

 

• view中

- (void)removefromsuperview {
    [super removefromsuperview];
    
    [self.timer invalidate];
    self.timer = nil;
}

 

在某些情况下，这种做法是可以解决问题的，但是有时却会引起其他问题，比如控制器push到下一个控制器，viewdiddisappear后，timer被释放，此时再回来，timer已经不复存在了。

所以，这种"方案"并不是合理的。

 

2. timer使用block方式添加target-action

这里我们需要自己在nstimer的分类中添加类方法:

@implementation nstimer (blcoktimer)

+ (nstimer *)bl_scheduledtimerwithtimeinterval:(nstimeinterval)interval block:(void (^)(void))block repeats:(bool)repeats {
    
    return [self scheduledtimerwithtimeinterval:interval target:self selector:@selector(bl_blockselector:) userinfo:[block copy] repeats:repeats];
}

+ (void)bl_blockselector:(nstimer *)timer {
    
    void(^block)(void) = timer.userinfo;
    if (block) {
        block();
    }
}
@end

通过block的方式，获取action，实际的target设置为self，即nstimer类。这样我们在使用timer时，由于target的改变，就不再有循环引用了。 使用中还需要注意block可能引起的循环引用，所以使用weakself：

__weak typeof(self) weakself = self;
self.timer = [nstimer bl_scheduledtimerwithtimeinterval:1 block:^{
     [weakself changetext];
} repeats:yes];

虽然没有了循环引用，但是还是应该记得在dealloc时释放timer。

3. 给self添加中间件proxy

考虑到循环引用的原因，改方案就是需要打破这些相互引用关系，因此添加一个中间件，弱引用self，同时timer引用了中间件，这样通过弱引用来解决了相互引用，如图：

 

接下来看看怎么实现这个中间件，直接上代码：

@interface zyweakobject()

@property (weak, nonatomic) id weakobject;

@end

@implementation zyweakobject

- (instancetype)initwithweakobject:(id)obj {
    _weakobject = obj;
    return self;
}

+ (instancetype)proxywithweakobject:(id)obj {
    return [[zyweakobject alloc] initwithweakobject:obj];
}

@interface zyweakobject()

@property (weak, nonatomic) id weakobject;

@end

@implementation zyweakobject

- (instancetype)initwithweakobject:(id)obj {
    _weakobject = obj;
    return self;
}

+ (instancetype)proxywithweakobject:(id)obj {
    return [[zyweakobject alloc] initwithweakobject:obj];
}

 

仅仅添加了weak类型的属性还不够，为了保证中间件能够响应外部self的事件，需要通过消息转发机制，让实际的响应target还是外部self，这一步至关重要，主要涉及到runtime的消息机制。

/**
 * 消息转发，让_weakobject响应事件
 */
- (id)forwardingtargetforselector:(sel)aselector {
    return _weakobject;
}

- (void)forwardinvocation:(nsinvocation *)invocation {
    void *null = null;
    [invocation setreturnvalue:&null];
}

- (bool)respondstoselector:(sel)aselector {
    return [_weakobject respondstoselector:aselector];
}

 

接下来就可以这样使用中间件了：

// target要设置成weakobj，实际响应事件的是self
zyweakobject *weakobj = [zyweakobject proxywithweakobject:self];
self.timer = [nstimer scheduledtimerwithtimeinterval:1 target:weakobj selector:@selector(changetext) userinfo:nil repeats:yes];

结论

经测试，以上两种方案都是可以解决timer的循环引用问题

代码请移步github: demo

作者：zhouyangyng
链接：https://juejin.im/post/5b641fc46fb9a04fd16033e7
来源：掘金
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。