方案一:
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且不与其他调用者使用同一个变量。
网友评论