NSTimer循环引用原理及解决方案

NSTimer的循环引用和Block循环引用有什么区别?

对于block型的循环引用,我们可以通过weakSelf来解决(点击了解block循环引用) ,但是对于timer,我们使用weakSelf确是无效的,比如下面的代码:

 __weak typeof(self) weakSelf = self;
    self.timer = [NSTimer timerWithTimeInterval:1 target:weakSelf selector:@selector(timerAction) userInfo:nil repeats:YES];

这里的self是不会被释放的,为什么?
原因在于block中引用的weakSelf这个指针,而NSTimer引用的是weakSelf指向的值,这就意味着,你外边的指针不管怎么释放,只要timer还在,它就会一直持有一个当前对象,而因为timer被runloop强持有,所以timer一直没有被释放。因此,我们解决timer的强引用不能像block一样加个弱引用就行了。要最终解决timer强引用的问题,还得通过timer本身。接下来就看看解决NSTimer强引用的几个方案。

一、在适当的地方手动调用invalidate

比如常见的在didMoveToParentViewController或viewWillDisappear方法中调用。在viewWillDisappear方法中调用invalidate的,如果我们push 到下一层,timer就无效了,返回就不走了;didMoveToParentViewController虽然能解决viewWillDisappear中的问题,但是还是显得比较麻烦。

- (void)didMoveToParentViewController:(UIViewController *)parent{
    // 无论push 进来 还是 pop 出去 正常跑
    // 就算继续push 到下一层 pop 回去还是继续
    if (parent == nil) {
       [self.timer invalidate];
        self.timer = nil;
    }
}

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

二、使用block方式

block方式代码实现

使用NSTimer的Category加block的方式。主要有以下几个步骤:

1、创建一个NSTimer类的Category(BlockTimer),自定义一个类方法,传递timer需要的参数以及一个block回调。这个block的参数是NSTimer,后面有用。
2、通过把NSTimer类作为target,而类对象本身就是单利,不怕强持有;
3、实现一个类方法handler:处理timer的回调;
4、通过userInfo传递外部block,在handler: 中把block读取出来调用,同时把timer传回target;

实现代码如下:

@implementation NSTimer (BlockTimer)

+ (NSTimer *)bt_timerWithTimeInterval:(NSTimeInterval)interval repeats:(BOOL)repeats block:(void(^)(NSTimer *timer))block{
    
    return [self scheduledTimerWithTimeInterval:interval target:self selector:@selector(handler:) userInfo:[block copy] repeats:repeats];
}

+ (void)handler:(NSTimer *)timer{
   
    void (^block)(NSTimer *timer) = timer.userInfo;
    if (block) {
        block(timer);
    }
}
@end
block方式正确使用

比如我们在viewController中,可以这样使用:

__weak typeof(self) weakSelf = self;
    self.timer = [NSTimer bt_timerWithTimeInterval:1 repeats:YES block:^(NSTimer * _Nonnull timer) {
        __strong typeof(weakSelf) strongSelf = weakSelf;
        if (strongSelf) {
            [strongSelf timerAction];
        }else
        {
            [timer invalidate];
        }
    }];

使用解析:这里虽然解决了self被NSTimer强引用的问题,self也能正常被释放掉,但是由于timer还没被释放掉,所以block会一直被执行,所以这里判断self为空时调用释放timer。当然,如果我们不在block里面释放,我们也可以在dealloc中释放:

- (void)dealloc
{
    [self.timer invalidate];
    self.timer = nil;
    NSLog(@"%@-%s", [self class], __func__);
}

三、使用中间对象,自定义Timer类

  • 实现步骤如下:
    1、自定义一个类CustomTimer,定义两个属性,包括timer和target,timer就是真实的NSTimer对象,这里的target不在是timer的target,而是我们外面真正实现业务的对象:
@property (nonatomic, strong) NSTimer *timer;
@property (nonatomic, weak) id target;

target使用weak修饰,不在CustomTimer强持有。

2、创建一个timer,self作为timer的target,实现timer回调方法handler:,同时创建一个NSInvocation对象,用于外部真正target的方法调用,invocation通过userInfo参数传递:

- (instancetype)initWithTimeInterval:(NSTimeInterval)interval
                                   target:(id)target
                                 selector:(SEL)selector
                                  repeats:(BOOL)repeat
{
    self = [super init];
    if (self) {
        NSMethodSignature *methodSignature = [target methodSignatureForSelector:selector];
        NSInvocation *invocation = [NSInvocation invocationWithMethodSignature:methodSignature];
        invocation.selector = selector;
        invocation.target = target;
        self.target = target;
        self.timer = [NSTimer scheduledTimerWithTimeInterval:interval target:self selector:@selector(handler:) userInfo:invocation repeats:repeat];
    }
    return self;
}

3、处理timer回调
在handler:中判断弱引用属性target是否为空,不为空则通过invocation调用target方法,如果为空则释放timer。

- (void)handler:(NSTimer *)timer
{
    NSInvocation *invocation = [timer userInfo];
    if (self.target) {
        [invocation invoke];
    }else{
        [self invalidate];
    }
}

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

如果我们不想在handler:中进行释放,也可以放到target对象(这里是ViewController)的dealloc方法释放:

- (void)dealloc
{
    [self.customTimer invalidate];
}

完整代码实现如下:

@interface CustomTimer ()

@property (nonatomic, strong) NSTimer *timer;

@property (nonatomic, weak) id target;

@end

@implementation CustomTimer

- (instancetype)initWithTimeInterval:(NSTimeInterval)interval
                                   target:(id)target
                                 selector:(SEL)selector
                                  repeats:(BOOL)repeat
{
    
    
    self = [super init];
    if (self) {
        NSMethodSignature *methodSignature = [target methodSignatureForSelector:selector];
        NSInvocation *invocation = [NSInvocation invocationWithMethodSignature:methodSignature];
        invocation.selector = selector;
        invocation.target = target;
        self.target = target;
        self.timer = [NSTimer scheduledTimerWithTimeInterval:interval target:self selector:@selector(handler:) userInfo:invocation repeats:repeat];
    }
    return self;
}


- (void)handler:(NSTimer *)timer
{
    NSInvocation *invocation = [timer userInfo];
    if (self.target) {
        [invocation invoke];
    }else{
        [self invalidate];
    }
}

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

@end

  • 使用示例
self.customTimer = [[CustomTimer alloc] initWithTimeInterval:1 target:self selector:@selector(timerAction) repeats:YES];

- (void)dealloc
{
    [self.customTimer invalidate];
}

如果我们不在handler:中进行释放,也可以放到target对象(这里是ViewController)的dealloc方法释放。

使用NSProxy对象

了解NSProxy对象请前往官方文档。NSProxy是虚拟基类,通过实现NSProxy子类,实现消息转发。

  • 代码实现
@interface CustomProxy ()

@property(nonatomic, weak) id target;

@end

@implementation CustomProxy


+ (instancetype)proxyWithTarget:(id)target{
    CustomProxy *proxy = [CustomProxy alloc];
    proxy.target = target;
    return proxy;
}

-(id)forwardingTargetForSelector:(SEL)aSelector {
    
    return self.target;
}
@end

这里target要用弱引用。实现forwardingTargetForSelector方法,返回target,这样通过CustomProxy调用的方法都交给target处理。

  • 代码示例
- (void)startTimer
{
self.timer = [NSTimer scheduledTimerWithTimeInterval:1 target:[CustomProxy proxyWithTarget:self] selector:@selector(timerAction) userInfo:nil repeats:YES];
}

- (void)dealloc
{
    [self.timer invalidate];
    self.timer = nil;
    NSLog(@"%@-%s", [self class], __func__);
}

由于self不会被强引用,所以可以被释放,调用dealloc。此时我们在dealloc中释放timer,这个必须得做,不然会崩溃。

你可能感兴趣的:(NSTimer循环引用原理及解决方案)