IT数码 购物 网址 头条 软件 日历 阅读 图书馆
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
图片批量下载器
↓批量下载图片,美女图库↓
图片自动播放器
↓图片自动播放器↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁
 
   -> 数据结构与算法 -> 【C++】实现自旋锁互斥(TAS 算法和 CAS 算法) -> 正文阅读

[数据结构与算法]【C++】实现自旋锁互斥(TAS 算法和 CAS 算法)

自旋锁——代码在循环中“自旋”。

自旋锁(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 == flag.compare_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;
}

  数据结构与算法 最新文章
【力扣106】 从中序与后续遍历序列构造二叉
leetcode 322 零钱兑换
哈希的应用:海量数据处理
动态规划|最短Hamilton路径
华为机试_HJ41 称砝码【中等】【menset】【
【C与数据结构】——寒假提高每日练习Day1
基础算法——堆排序
2023王道数据结构线性表--单链表课后习题部
LeetCode 之 反转链表的一部分
【题解】lintcode必刷50题<有效的括号序列
上一篇文章      下一篇文章      查看所有文章
加:2022-05-08 08:20:57  更:2022-05-08 08:22:46 
 
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁

360图书馆 购物 三丰科技 阅读网 日历 万年历 2024年11日历 -2024/11/26 3:32:18-

图片自动播放器
↓图片自动播放器↓
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
图片批量下载器
↓批量下载图片,美女图库↓
  网站联系: qq:121756557 email:121756557@qq.com  IT数码