一个extension让你在Swift中用NSTimer更少操心

NSTimer有坑,坑在哪里?

它会持有target的引用计数,不出意外的话你的target还会持有它的引用计数。另外,还会被runloop持有它的引用计数。

是不是感觉特别混乱?往往在这种情况下我们也容易出现循环引用的错误,会导致该释放的资源没有办法被正确释放。

举个例子,通常我们的应用场景是我们的ViewController持有这个timer,想让它定时唤起某个操作。这里首先ViewController持有了timer的引用计数,然后timer又会持有ViewController,那么循环引用就产生了。可能有些同学会在deinit中写上timer.invalidate(),然而由于循环引用的原因,根本就进不了这儿。

好吧,可能有的同学会想些其他的方法来让timer失效。通常做法的要么增加复杂程度要么破坏了封装,所以不见得是个好的做法。

于是,我�把之前在OC中的做法(当然方法来源于大神Matt Galloway)翻译成了swift的版本,直接上代码吧:

public typealias TimerExcuteClosure = @convention(block) () -> ()

extension Timer {
    private class TimerActionBlockWrapper : NSObject {
        var block : TimerExcuteClosure
        init(block: @escaping TimerExcuteClosure) {
            self.block = block
        }
    }

    public class func YQ_scheduledTimerWithTimeInterval(_ ti:TimeInterval, closure: @escaping TimerExcuteClosure, repeats yesOrNo: Bool) -> Timer {
        return self.scheduledTimer(timeInterval: ti, target: self, selector: #selector(Timer.excuteTimerClosure(_:)), userInfo: TimerActionBlockWrapper(block: closure), repeats: true)
    }
    
    @objc private class func excuteTimerClosure(_ timer: Timer) {
        if let action = timer.userInfo as? TimerActionBlockWrapper {
            action.block()
        }
    }
}

这里把想要执行的操作放到了一个闭包里,然后把它设为userInfo。比较关键的地方在于这里的target是NSTimer自己了,这里防止了它去持有外部调用者的引用计数,比如我们的ViewController。切断了之前的联系之后,deinit就能正常调用了,也能正常调用timer.invalidate()了,timer失效的时候也会释放它对target的引用,从而能够正确的释放资源。

这里有一点需要注意的是,NSTimer的userInfo的类型是AnyObject,这意味这你不能直接把closure传给它,需要用unsafeBitCast来转一下。当然既然苹果说了这个东西是unsafe的,那么自然有它的道理,用的时候需要小心。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 创建NSTimer 创建NSTimer的常用方法是: + (NSTimer *)scheduledTimerWit...
    LanWor阅读 1,398评论 0 2
  • 定时器: 需要被添加到Runloop,否则不会运行,当然添加的Runloop不存在也不会运行 还要指定添加到的Ru...
    Sunli_阅读 1,737评论 0 3
  • iOS中计时器常用的有两种方式 使用NSTimer类(Swift 中更名为 Timer) NSTimer 常用的初...
    superDg阅读 1,866评论 0 1
  • NSTimer是iOS最常用的定时器工具之一,在使用的时候常常会遇到各种各样的问题,最常见的是内存泄漏,通常我们使...
    bomo阅读 1,243评论 0 7
  • 你是千堆雪我是长街,怕日出一到彼此瓦解。 无奈的爱情总是让人痛苦,就算是雪和长街,也曾彼此依偎,也会相互分开,如果...
    湛卢圆公子阅读 416评论 0 1