NSTimer迴圈引用的幾種解決方案

zhouyangyng發表於2018-08-03

前言

在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引用了中介軟體，這樣通過弱引用來解決了相互引用，如圖：

NSTimer迴圈引用的幾種解決方案

接下來看看怎麼實現這個中介軟體，直接上程式碼：

@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

相關文章