自旋锁——代码在循环中“自旋”。
自旋锁（spin lock）是一种非阻塞锁，如果某线程需要获取锁，但该锁已经被其他线程占用时，该线程不会被挂起，而是在不断的消耗 CPU 的时间，不停的试图获取锁。
互斥量（mutex）是阻塞锁，当某线程无法获取锁时，该线程会被直接挂起，该线程不再消耗CPU时间，当其他线程释放锁后，操作系统会激活那个被挂起的线程，让其投入运行。
因此，多核 CPU 才能用自旋锁。
其他细节，详看下面代码注释。
#ifndef AMYSPINLOCK_H#define AMYSPINLOCK_H#include <atomic>// 采用 std::atomic_flag 实现自旋锁互斥，即 TAS 算法（Test And Set）class AMySpinLock1{public:AMySpinLock1() = default;AMySpinLock1(const AMySpinLock1&) = delete;AMySpinLock1& operator=(const AMySpinLock1&) = delete;void lock(){while(flag.test_and_set(std::memory_order_acquire));}void unlock(){flag.clear(std::memory_order_release);}private:// std::atomic_flag 类型的对象必须由宏 ATOMIC_FLAG_INIT 初始化，它把标志初始化为置零状态。std::atomic_flag flag{ATOMIC_FLAG_INIT};};// 采用 std::atomic<bool> 和 compare_exchange_strong 实现自旋锁互斥，即 CAS 算法（Compare And Swap）// 比较-交换操作是原子类型的编程基石。// 使用者给定一个期望值，原子变量将它和自身的值比较，如果相等，就存入另一既定的值，否则，更新期望值所属的变量，向它赋予原子变量的值。// 比较-交换函数返回布尔类型，如果完成了保存动作（前提是两值相等），则操作成功，函数返回 true；反之操作失败，函数返回 false。class AMySpinLock2{public:AMySpinLock2() = default;AMySpinLock2(const AMySpinLock2&) = delete;AMySpinLock2& operator=(const AMySpinLock2&) = delete;void lock(){// 判断 flag 对象封装的 bool 值是否为期望值(false)：// ① 若 bool 值为 false，与期望值相等，说明自旋锁空闲。此时，flag 对象写入 true，返回 true，即上锁成功。// ② 若 bool 值为 true，与期望值不相等，说明自旋锁被锁。此时，while 将一直循环，直到返回 true 为止。bool expected = false;while(false == pare_exchange_strong(expected, true)){// 当 compare_exchange_strong 返回 false 时，// 证明 expected 与 flag 不相等，此时 expected 为false，flag 为 true；// 则将 expected 赋值为 flag 的值，即此时 expected 为 true 了，// 因此要修改为 false 后，才能进入下一次循环。expected = false;}}void unlock(){flag.store(false);}private:// flag 对象所封装的 bool 值为 false 时，说明自旋锁未被线程占有。std::atomic<bool> flag{false};};#endif // AMYSPINLOCK_H
测试程序如下：
#include <iostream>#include <thread>#include "./lock/amyspinlock.h"AMySpinLock1 m;//AMySpinLock2 m;void testSpinLock(){static int i = 0, j = 0;for(int k=0; k<1000; ++k){m.lock();++i;std::this_thread::sleep_for(std::chrono::milliseconds(1));++j;if(i != j)printf("[%d] i:%d, j:%d \n", std::this_thread::get_id(), i, j);m.unlock();}}int main(int argc, char *argv[]){std::cout << "Test Begin...\n";std::thread t1(testSpinLock);std::thread t2(testSpinLock);if(t1.joinable())t1.join();if(t2.joinable())t2.join();std::cout << "Test End...\n";return 0;}
