spinlock.h
代碼位于: https://github.com/cloudwu/skynet/blob/master/skynet-src/spinlock.h
該文件內,根據不同環境提供了 3 種 api 實現:
- pthread_mutex_t 系列函數
- gcc 內置原子操作函數
- std atomic 系列函數
看了下,效率最高的是 std atomic 系列函數的實現:
#include "atomic.h"#define atomic_test_and_set_(ptr) STD_ atomic_exchange_explicit(ptr, 1, STD_ memory_order_acquire)
#define atomic_clear_(ptr) STD_ atomic_store_explicit(ptr, 0, STD_ memory_order_release);
#define atomic_load_relaxed_(ptr) STD_ atomic_load_explicit(ptr, STD_ memory_order_relaxed)#if defined(__x86_64__)
#include <immintrin.h> // For _mm_pause
#define atomic_pause_() _mm_pause()
#else
#define atomic_pause_() ((void)0)
#endifstruct spinlock {STD_ atomic_int lock;
};static inline void
spinlock_init(struct spinlock *lock) {STD_ atomic_init(&lock->lock, 0);
}static inline void
spinlock_lock(struct spinlock *lock) {for (;;) {if (!atomic_test_and_set_(&lock->lock))return;while (atomic_load_relaxed_(&lock->lock))atomic_pause_();}
}static inline int
spinlock_trylock(struct spinlock *lock) {return !atomic_load_relaxed_(&lock->lock) &&!atomic_test_and_set_(&lock->lock);
}static inline void
spinlock_unlock(struct spinlock *lock) {atomic_clear_(&lock->lock);
}static inline void
spinlock_destroy(struct spinlock *lock) {(void) lock;
}
- spinlock_init 初始化 lock 默認值為 0
- spinlock_lock 搶占設置 1 ,失敗 while 自旋;自旋出來,再去搶占(for)
- spinlock_trylock 查看 lock 并嘗試搶占設置 1
- spinlock_unlock lock 設回 0
- spinlock_destroy 占位,沒有可實際釋放的資源
memory_order_x
memory_order_acquire/memory_order_release 比較好理解,請自行度娘
atomic_load_relaxed_ 函數內用的是 memory_order_relaxed 。 memory_order_relaxed 即普通內存訪問
因此 atomic_load_relaxed_ 性能效果等價于普通內存訪問(即性能最高),且可能會返回錯值
妙的地方在于:
- spinlock_lock for - while 雙重循環,允許 atomic_load_relaxed_ 可能的誤判
- spinlock_trylock 函數語義本身就在于 TRY 。因此也允許可能的誤判
_mm_pause
while 空循環的一種性能優化,度娘了解知道就好