Swift3.0中dispatch_once/after的替代实现

方案一:

public extension DispatchQueue {
    private static var _onceTracker = [String]()
    public class func once(token: String, block:()->Void) {
        objc_sync_enter(self)
        defer { objc_sync_exit(self) }
        if _onceTracker.contains(token) { return }
        _onceTracker.append(token)
        block()
    }  
}

原理是使用一个数组记录已经执行过的闭包,比较简单,直接看缺点好了。

  • 没有使用外层判断,每一次都要执行加锁/解锁操作;
  • 锁对象使用self(即DispatchQueue.self),不同调用者的闭包无法并发执行,这可能会导致线程死锁;
  • token使用String,调用者需保证token恒定且不与其他调用者冲突;
  • 数组需要额外开销。
方案二:
public extension DispatchQueue {
    
    /// dispatch_after
    public class func after(_ after: Double, _ block: @escaping ()->Void) {
        self.main.asyncAfter(deadline: DispatchTime.now() + after, execute: block)
    }
    
    /// dispatch_once
    public class func once(_ token: inout Bool, _ block: ()->Void) {
        if token == false {
            objc_sync_enter(token)
            if token == false {
                block()
                token = true
            }
            objc_sync_exit(token)
        }
    }
}
  • 外层的判断使得当token==true时,不执行锁操作;
  • 锁对象和token使用同一个Bool变量,由调用者提供;此时调用者只需保证token初始化为false且不与其他调用者使用同一个变量。

你可能感兴趣的:(Swift3.0中dispatch_once/after的替代实现)