dispatch_once 详解
- [dispatch_once 详解](#dispatch_once 详解)
dispatch_once 详解
我们常用dispatch_once创建单例对象,先创建一个静态变量onceToken,调用dispatch_once函数将onceToken和block做参数传入,在block完成静态对象变量instance的初始化。
objc
+ (instancetype)sharedInstance
{
static dispatch_once_t onceToken;
static id instance = nil;
dispatch_once(&onceToken, ^{
instance = [[NSObject alloc] init];
});
return instance;
}
dispatch_once的源码:
objc
void dispatch_once(dispatch_once_t *val, dispatch_block_t block)
{
dispatch_once_f(val, block, _dispatch_Block_invoke(block));
}
// 实现代码在下面
DISPATCH_NOINLINE
void dispatch_once_f(dispatch_once_t *val, void *ctxt, dispatch_function_t func)
{
dispatch_once_gate_t l = (dispatch_once_gate_t)val;
#if !DISPATCH_ONCE_INLINE_FASTPATH || DISPATCH_ONCE_USE_QUIESCENT_COUNTER
uintptr_t v = os_atomic_load(&l->dgo_once, acquire);
if (likely(v == DLOCK_ONCE_DONE)) {
return;
}
#if DISPATCH_ONCE_USE_QUIESCENT_COUNTER
if (likely(DISPATCH_ONCE_IS_GEN(v))) {
return _dispatch_once_mark_done_if_quiesced(l, v);
}
#endif
#endif
if (_dispatch_once_gate_tryenter(l)) {
return _dispatch_once_callout(l, ctxt, func);
}
return _dispatch_once_wait(l);
}
通过包装token作为唯一标识,判断(v == DLOCK_ONCE_DONE)是否已经完成初始化,实际就是判断是否调用过block(func内部调用)。如果已经初始化完成,那么v == DLOCK_ONCE_DONE,直接返回;如果还未初始化完成则往下走。
如果没有初始化,这里进来会先通过token获取锁。如果锁已被占用,说明其他线程在初始化,往下进入等待状态;如果获取锁成功则通过_dispatch_once_callout调用block,进行block回调:
objc
DISPATCH_NOINLINE
static void _dispatch_once_callout(dispatch_once_gate_t l, void *ctxt,
dispatch_function_t func)
{
_dispatch_client_callout(ctxt, func);
_dispatch_once_gate_broadcast(l);
}
DISPATCH_ALWAYS_INLINE
static inline void _dispatch_client_callout(void *ctxt, dispatch_function_t f)
{
return f(ctxt);
}
block调用完成之后通过_dispatch_once_gate_broadcast广播通知初始化完成,把v标记为DLOCK_ONCE_DONE,并释放锁。
如果block正在调用,有其他线程进来,这时候它们尝试获取锁失败,进入等待状态。等到第一次进来初始化完成之后发出广播,它们收到消息立即返回。
总结:dispatch_once首先通过onceToken作为一个唯一标识用来判断对象是否已经初始化(实际就是block有没有执行完成)。然后在第一次调用block(实际上就是初始化对象的时候)加锁,防止其他线程进入。其他线程进来时发现锁被占用就会进入等待状态。当第一次调用block初始化完成之后,就会把onceToken标记的值更新未初始化,然后释放锁,同时会发出通知,让其他等待的线程返回。当二次或多次调用dispatch_once的时候,判断对象已经被初始化,就直接返回,不会再调用block。