方案一:
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()
}
}
原理是使用一個數(shù)組記錄已經(jīng)執(zhí)行過的閉包,比較簡單,直接看缺點好了。
- 沒有使用外層判斷,每一次都要執(zhí)行加鎖/解鎖操作;
- 鎖對象使用self(即DispatchQueue.self),不同調(diào)用者的閉包無法并發(fā)執(zhí)行,這可能會導(dǎo)致線程死鎖;
- token使用String,調(diào)用者需保證token恒定且不與其他調(diào)用者沖突;
- 數(shù)組需要額外開銷。
方案二:
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)
}
}
}
- 外層的判斷使得當(dāng)token==true時,不執(zhí)行鎖操作;
- 鎖對象和token使用同一個Bool變量,由調(diào)用者提供;此時調(diào)用者只需保證token初始化為false且不與其他調(diào)用者使用同一個變量。