破解 NSTimer 的强引用

问题

很多人都知道, Timer 会通过 Runloop 对 target 进行强引用, 需要手动对 Timer 进行 invalidate 以便释放强引用.
所以很多人就写了下面的代码

...
override func viewDidLoad() {
    super.viewDidLoad()
    self.timer = Timer.scheduledTimer(timeInterval: 1, target: self, selector: #selector(tick), userInfo: nil, repeats: true)
}
deinit {
    self.timer.invalidate()
}
@objc func tick() {
    print("tick")
}

这样写, 很快就会发现问题, 因为你的 ViewController 在弹出之后, 依然在打印 tick
简单分析一下, 因为 Timer 对 target 有一个强引用, 导致无法释放, 我们代码中对 Timer 的释放函数却在析构函数中, 而析构函数根本就无法被调用, 也就造成了引用无法释放, Timer 也无法结束.

错误的解决方案:

看到引用无法释放, 你可能会想到使用弱引用解决, 于是写出了下面的代码

override func viewDidLoad() {
    super.viewDidLoad()
    weak var weakself = self
    self.timer = Timer.scheduledTimer(timeInterval: 1, target: weakself, selector: #selector(tick), userInfo: nil, repeats: true)
}
...

结果是令人失望的, 即使这样, 控制器也无法被释放. 强引用的是变量指向的对象, 而不是变量本身, 弱引用并不能解决这个问题.

正确的姿势

既然无法直接解决, 那我们可以通过一个局部变量作为 target, 然后在局部变量中存放一个 weakself 来解决引用问题.
先来定义一个类型用来存放 weakself 和接收 Timer 事件

class TimerTarget: NSObject {
    weak var target: AnyObject?
    let timerSelctor: Selector
    init(target: AnyObject, selector: Selector) {
        self.timerSelctor = selector
        super.init()
        self.target = target
    }
    @objc func timerTick() {
        _ = self.target?.perform(self.timerSelctor)
    }
}

这里面通过一个 weak 属性指向了控制器, 这样, 即使 Timer 对 TimerTarget 增加一个强引用, 也不会对控制器的引用计数产生影响

override func viewDidLoad() {
    super.viewDidLoad()
    self.timer = Timer.scheduledTimer(timeInterval: 1, target: TimerTarget(target: self, selector: #selector(tick)), selector: #selector(TimerTarget.timerTick), userInfo: nil, repeats: true)
}

这样操作后, 控制器就可以正常释放了. 而且, 由于控制器会在释放时, 主动释放 Timer, 这样, TimerTarget 也可以正常释放

进阶, 更简洁的解决方案

上面的代码看起来略显累赘, 有没有跟简单的办法?
当然有, 使用 NSProxy 类, 将 Timer 触发的方法转发到我们的控制器中.
不过遗憾的是, Swift 由于不支持 NSInvocation 以及构造函数无法构造(NSProxy 没有 init 方法), 只能使用混编来解决.
我们来创建一个 OC 的类
WeakTarget.h 文件

#import <Foundation/Foundation.h>

@interface WeakTarget : NSProxy
@property(weak, nonatomic) id target;
- (instancetype)initWithTarget:(id)target;
@end

WeakTarget.m 文件

#import "WeakTarget.h"

@implementation WeakTarget
- (instancetype)initWithTarget:(id)target{
    self.target = target;
    return self;
}
-(void)forwardInvocation:(NSInvocation *)invocation {
    [invocation invokeWithTarget:self.target];
}
-(NSMethodSignature *)methodSignatureForSelector:(SEL)sel {
    return [self.target methodSignatureForSelector:sel];
}
@end

另外, 没有用过混编的同学注意, 在询问是否需要创建bridge header 的时候, 一定要选择创建, 否则就需要自己创建了.
在 XXX-Bridging-Header.h 中添加对这个头文件的引用.

#import "WeakTarget.h"

这样准备工作就好了.
使用方式如下

self.timer = Timer.scheduledTimer(timeInterval: 1, target: WeakTarget(target: self), selector: #selector(tick), userInfo: nil, repeats: true)

本质上还是一样的, 都是在WeakTarget 里面存放一个弱引用的对象, 避免对控制器进行强引用.
使用NSProxy 可以将方法重定向到控制器中, 代码更加整洁.