恋恋风辰的官方博客

利用引用计数实现无锁并发栈
Posted on 2023-12-24 11:59:48 by 恋恋风辰 | In C++ | 原文链接

简介

前文我们通过风险指针的方式实现了无锁栈，但是也提出了一些弊端，比如每次pop都要从风险数组中选择一个空闲的节点作为标记。其次删除节点前要遍历风险数组对比节点是否被风险指针所指涉，如果被风险指针指涉则需放入待删列表。最后pop结束时也要回收待删列表中的节点，还要依次将待删列表中的节点和风险数组对比，如果未被风险指针指涉则需删除，否则跳过。

但是这种方式多次遍历风险数组，会有性能损耗，我们提出一种新的解决方式，利用引用计数实现无锁并发的栈。

引用计数

在C++并发编程一书中提出了两个计数，一个外部计数，一个内部计数，二者加起来就是有效的引用计数，很多读者对此费解，为何不用一个引用计数维护呢？那本文就是带着大家如何一步一步去实现并说明单引用计数的不可行性。

那我们先定义一个栈结构，以及它的内部节点结构

  1. template<typename T>
  2. class single_ref_stack {
  3. public:
  4.     single_ref_stack():head(nullptr) {
  6.     }
  8.     ~single_ref_stack() {
  9.         //循环出栈
  10.         while (pop());
  11.     }
  13. private:
  14.     struct ref_node {
  15.         //1 数据域智能指针
  16.         std::shared_ptr<T>  _data;
  17.         //2 引用计数
  18.         std::atomic<int> _ref_count;
  19.         //3  下一个节点
  20.         ref_node* _next;
  21.         ref_node(T const& data_) : _data(std::make_shared<T>(data_)),
  22.             _ref_count(1), _next(nullptr) {}
  23.     };
  25.     //头部节点
  26.     std::atomic<ref_node*> head;
  27. };

1 single_ref_stack为我们定义的栈结构。内部包含一个head节点，head节点为一个ref_node*类型的原子变量。

2 single_ref_stack的构造函数将head设置为nullptr，析构函数循环pop直到栈为空为止。pop我们之后再实现。

3 定义ref_node结构作为每个栈存储的元素。内部包含_data表示数据域， int类型的原子变量表示引用计数。_next表示下一个节点指针。
ref_node的构造函数接收一个T类型的通用数据类型，利用这个参数构造自己的数据域。

接下来我们实现push操作

  1. void push(T const& data) {
  2.     auto new_node = new ref_node(data);
  3.     new_node->next = head.load();
  4.     while (!head.compare_exchange_weak(new_node->next, new_node));
  5. }

push 操作很简单，创建一个ref_node类型的指针对象new_node，将new_node的next指针指向现在的头节点，然后不断地重试(防止其他线程修改head后导致head变化不一致),直到将head更新为new_node.

接下来我们实现pop

  1. std::shared_ptr<T> pop() {
  2.     ref_node* old_head = head.load();
  3.     for (;;) {
  4.         if (!old_head) {
  5.             return std::shared_ptr<T>();
  6.         }
  7.         //1 只要执行pop就对引用计数+1
  8.         ++(old_head->_ref_count);
  9.         //2 比较head和old_head想等则交换否则说明head已经被其他线程更新
  10.         if (head.compare_exchange_strong(old_head, old_head->_next)) {
  11.             auto cur_count = old_head->_ref_count.load();
  12.             auto new_count;
  13.             //3  循环重试保证引用计数安全更新
  14.             do {
  15.                 //4 减去本线程增加的1次和初始的1次
  16.                 new_count = cur_count - 2;
  17.             } while (!old_head->_ref_count.compare_exchange_weak(cur_count,  new_count));
  19.             //返回头部数据
  20.             std::shared_ptr<T> res;
  21.             //5  交换数据
  22.             res.swap(old_head->_data);
  23.             //6
  24.             if (old_head->_ref_count == 0) {
  25.                 delete old_head;
  26.             }
  28.             return res;
  29.         }
  30.         else {
  31.             //7 
  32.             if (old_head->_ref_count.fetch_sub(1) == 1) {
  33.                 delete old_head;
  34.             }
  35.         }
  36.     }
  37. }

1 上面的代码我们先判断old_head是否为空，如果为空则说明栈为空。

2 然后代码1处 对引用计数+1, 因为是原子变量所以可以保证线程安全。

3 然后代码2处 比较head和old_head是否相等，如果相等则将head更新为old_head的next指向的数据。
简而言之就是将head更新为新的栈顶元素。因为存在多个线程并发执行2处代码的情况，导致只有一个线程交换成功，交换成功的线程就承担起返回数据的任务。

并且在4处减少2个引用计数(减去初始的1个加上自己pop开始增加的那一个)，并且在3处循环重试更新引用计数。在6处判断引用计数如果变为0则删除指针。

交换失败的线程是抢占失败的线程，则执行7处代码需减少1次引用计数(因为该线程进入pop时已经增加了1次引用计数)。fetch_sub会将原子变量的数值减1，然后返回减少之前的数值。所以我们判断如果之前的数值为1则说明该线程是最后引用此指针的线程，可以将指针删除。

我们观察上述pop函数，存在严重漏洞

如果线程1和线程2都准备执行1处代码，但是线程2抢先执行，并且更新引用计数_ref_count变为0，则执行删除old_head的操作，此时线程1刚好执行1处代码，引发崩溃。

引发崩溃的原因我们知道了就是old_head被删除了，那我们要做的就是将引用计数提出来，不放在指针里，和指针解耦。

我们将原来的节点结构拆成两个

  1. struct node {
  2.     //1 数据域智能指针
  3.     std::shared_ptr<T>  _data;
  4.     //2  下一个节点
  5.     ref_node _next;
  6.     node(T const& data_) : _data(std::make_shared<T>(data_)) {}
  8. };
  10. struct ref_node {
  11.     // 引用计数
  12.     std::atomic<int> _ref_count;
  14.     node* _node_ptr;
  15.     ref_node( T const & data_):_node_ptr(new node(data_)), _ref_count(1){}
  17.     ref_node():_node_ptr(nullptr),_ref_count(0){}
  18. };

ref_node表示栈存储的节点结构，包括引用计数和节点的指针。而node为实际的节点结构，包括节点的数据域以及下一个节点的地址。

那我们的single_ref_stack结构管理的head是指针类型好还是副本类型好呢？

我们可以假设head存储的是指针类型

  1. //头部节点
  2. std::atomic<ref_node*> head;

那么pop逻辑就要改为

  1. std::shared_ptr<T> pop() {
  2.     //0 处
  3.     ref_node* old_head = head.load();
  4.     for (;;) {
  5.         //1 只要执行pop就对引用计数+1并更新到head中
  6.         ref_node* new_head;
  7.         do {
  8.             new_head = old_head;
  9.             //7 处
  10.             new_head->_ref_count += 1;
  11.         } while (!head.compare_exchange_weak(old_head, new_head));
  12.         //4 
  13.         old_head = new_head;
  15.         auto* node_ptr = old_head->_node_ptr;
  16.         if (node_ptr == nullptr) {
  17.             return  std::shared_ptr<T>();
  18.         }
  20.         //2 比较head和old_head想等则交换否则说明head已经被其他线程更新
  21.         if (head.compare_exchange_strong(old_head, node_ptr->_next)) {
  23.             //要返回的值
  24.             std::shared_ptr<T> res;
  25.             //交换智能指针
  26.             //5 处
  27.             res.swap(node_ptr->_data);
  29.             //6 增加的数量
  30.             int increase_count = old_head->_ref_count.fetch_sub(2);
  32.             //3 处判断仅有当前线程持有指针则删除
  33.             if (increase_count == 2) {
  34.                 delete node_ptr;
  35.             }
  37.             return res;
  38.         }else {
  39.             if (old_head->_ref_count.fetch_sub(1) == 1) {
  40.                 delete node_ptr;
  41.             }
  42.         }
  43.     }
  44. }

解释一下上面的逻辑：

在1处head调用比较交换和old_head做比较，比较分为两个方面，一个是引用计数一个是node*的值。

那我们假设线程1和线程2依次通过了比较交换逻辑(假设线程1先于线程2)，那么假设线程1在4处看到的old_head的引用计数为2，线程2在4处看到old_head的引用计数为3.

而head最后被更新的引用计数为3.所以在2处的判断逻辑里，线程2会进入if的逻辑，线程1会进入else的逻辑,最后会有一个线程回收node_ptr节点，这么看来是正常的。

但是我们仔细分析，看上面的代码有很大漏洞

1 假设线程1比线程2先执行，线程1在2处执行比较交换后head会被更新为新的值。线程2执行比较交换操作会失败，则进入else处理, old_head会被更新为新的head值， 此时old_head的引用计数为1则会被线程2误删，因为线程2此时读到的old_head正是新的head指向的数据。而且没有弹出和修改head的值。这样其他线程pop头部元素时会崩溃。

https://cdn.llfc.club/1703215490981.jpg

2 线程1和线程2都执行完0处代码，二者读取的old_head值相同。假设线程1比线程2先执行，线程2因未抢占到cpu的时间片停顿在1处，线程1按次序依次执行最后执行到3处将node_ptr删除。而且现在的head已经指向了新的栈顶元素即old_head的下一个元素。此时线程2抢占到时间片，执行1处代码又将old_head更新为head当前值了，只不过引用计数加了1变为2，但是指向的是下一个节点，所以这种情况下进入仍会进入if条件，对新的old_head节点删除。这种情况倒是正常。

https://cdn.llfc.club/1703215559019.jpg

3 还是假设线程1和线程2都执行完0处代码，线程1抢先执行完5处。准备执行6处时，线程2抢占CPU执行了7处代码，尽管会被while比较old_head和head不同而重试，进而更新old_head。但是线程2的do逻辑中第一次的old_head和线程1的old_head指向的是同一个，线程2修改了old_head中的引用计数，导致线程1执行6处代码时不会进入if逻辑。又因为线程2在2处之后while会不断重试，线程2的head已经和old_head指向不同了，导致线程2也不会回收old_head内部节点指向的数据，导致内存泄漏。

https://cdn.llfc.club/1703215758466.jpg

这就告诉我们当我们设计pop逻辑的时候尽量不要存储指针，存储指针意味着存在多个线程操作同一块内存的情况。

所以我们得出以下结论

1 head的类型修改为ref_node类型而不是指针。

2 现有的引用保留，我们用其表示增加的引用计数，该引用计数可以用原子类型，也可以不用原子类型。为简化和节省效率我们用普通int类型。

3 新增一个表示减少的引用计数，因为这个表示减少的引用计数要在多个线程中同步，并且要保证安全性，那我们将其放入node类里, 因为node类的指针被存储在栈的节点中，所以可以达到多个线程修改这个减少的引用计数的效果。

4 一个节点能否被回收取决于整体的引用计数是否为0。

改进引用节点

按照上面的推论，我们新增_dec_count表示减少的引用计数，放在node结构里。

  1.     struct ref_node;
  2.     struct node {
  3.         //1 数据域智能指针
  4.         std::shared_ptr<T>  _data;
  5.         //2  下一个节点
  6.         ref_node _next;
  7.         node(T const& data_) : _data(std::make_shared<T>(data_)) {}
  9.         //减少的数量
  10.         std::atomic<int>  _dec_count;
  11.     };
  13.     struct ref_node {
  14.         // 引用计数
  15.         int _ref_count;
  17.         node* _node_ptr;
  18.         ref_node( T const & data_):_node_ptr(new node(data_)), _ref_count(1){}
  20.         ref_node():_node_ptr(nullptr),_ref_count(0){}
  21.     };

然后将栈中的head结构变为ref_node类型的原子变量。

  1. //头部节点
  2. std::atomic<ref_node> head;

我们重新实现push

  1.     void push(T const& data) {
  2.         auto new_node =  ref_node(data);
  3.         new_node._node_ptr->_next = head.load();
  4.         while (!head.compare_exchange_weak(new_node._node_ptr->_next, new_node));
  5.     }

我们重新实现pop

  1. std::shared_ptr<T> pop() {
  2.     ref_node old_head = head.load();
  3.     for (;;) {
  4.         //1 只要执行pop就对引用计数+1并更新到head中
  5.         ref_node new_head;
  7.         //2
  8.         do {
  9.             new_head = old_head;
  10.             new_head._ref_count += 1;
  11.         } while (!head.compare_exchange_weak(old_head, new_head));
  13.         old_head = new_head;
  14.         //3
  15.         auto* node_ptr = old_head._node_ptr;
  16.         if (node_ptr == nullptr) {
  17.             return  std::shared_ptr<T>();
  18.         }
  20.         //4 比较head和old_head相等则交换否则说明head已经被其他线程更新
  21.         if (head.compare_exchange_strong(old_head, node_ptr->_next)) {
  23.             //要返回的值
  24.             std::shared_ptr<T> res;
  25.             //交换智能指针
  26.             res.swap(node_ptr->_data);
  28.             //5  增加的数量
  29.             int increase_count = old_head._ref_count - 2;
  30.             //6  
  31.             if (node_ptr->_dec_count.fetch_add(increase_count) == -increase_count) {
  32.                 delete node_ptr;
  33.             }
  35.             return res;
  36.         }else {
  37.             //7
  38.             if (node_ptr->_dec_count.fetch_sub(1) == 1) {
  39.                 delete node_ptr;
  40.             }
  41.         }
  42.     }
  43. }

1 多个线程并发pop如果有线程在2处重试，可能时head和old_head的引用计数不同或者node的地址不同，不过无论如何我们的head采用的是副本存储，所以重试失败增加的引用计数不会影响到其他线程。

2 在代码3处我们将old_head的node地址取出来，留作node_ptr,这样我们以后可以对node_ptr内部的引用计数做减少，因为多个线程操作node_ptr指向的数据，所以引用计数是原子变量，并且多个线程是可见的。

3 在4处进行判断，由于我们的head存储的是ref_node类型，所以多个线程看到的old_head的值可能不一样，但我们能保证仅有一个线程进入if逻辑，进入的线程就是old_head和head匹配的那个，我们定义了res用来返回数据。在5处对增加的引用计数减2操作，获取除了自己以外并行操作这个old_head的线程数。然后我们说过增加引用计数和减少引用计数相加为0就说明可以删除节点。那我们在6处利用fetch_add操作返回操作之前的值，让fetch_add增加increase_count，并且fetch_add返回增加之前_dec_count的值，如果这个值是负的increase_count即表示当前仅有该线程操作这个old_head节点，即可删除。

为了让大家了解这个过程我们假设线程1和线程2都执行到4处之前，线程2没抢占到cpu暂停，而线程1抢占并且执行了4处的交换进入if条件，而此时线程2抢占cpu继续执行else逻辑，将_dec_count减少1，原来的_dec_count为0，减少后变为-1，fetch_sub返回之前的值为0不满足if条件所以线程2不会删除node_ptr。此时线程1继续抢占cpu执行到5处_ref_count为3，increse_count为1，_dec_count为-1，_dec_count进行fetch_add之后变为0,但是fetch_add返回的时相加之前的值即为-1，而increase_count恰好为1，所以线程1回收这个node_ptr。

测试和验证

为了测试安全性，效率就不测了，这个无锁的栈后期还要完善，目前我们只要测试安全性即可。

我们启动三个线程t1,t2,t3，t1用来向栈中压入元素,t2和t3用来从栈中弹出元素。

  1. void TestSingleRefStack() {
  2.     single_ref_stack<int>  single_ref_stack;
  3.     std::set<int>  rmv_set;
  4.     std::mutex set_mtx;
  6.     std::thread t1([&]() {
  7.         for (int i = 0; i < 20000; i++) {
  8.             single_ref_stack.push(i);
  9.             std::cout << "push data " << i << " success!" << std::endl;
  10.             std::this_thread::sleep_for(std::chrono::milliseconds(5));
  11.         }
  12.         });
  14.     std::thread t2([&]() {
  15.         for (int i = 0; i < 10000;) {
  16.             auto head = single_ref_stack.pop();
  17.             if (!head) {
  18.                 std::this_thread::sleep_for(std::chrono::milliseconds(10));
  19.                 continue;
  20.             }
  21.             std::lock_guard<std::mutex> lock(set_mtx);
  22.             rmv_set.insert(*head);
  23.             std::cout << "pop data " << *head << " success!" << std::endl;
  24.             i++;
  25.         }
  26.         });
  28.     std::thread t3([&]() {
  29.         for (int i = 0; i < 10000;) {
  30.             auto head = single_ref_stack.pop();
  31.             if (!head) {
  32.                 std::this_thread::sleep_for(std::chrono::milliseconds(10));
  33.                 continue;
  34.             }
  35.             std::lock_guard<std::mutex> lock(set_mtx);
  36.             rmv_set.insert(*head);
  37.             std::cout << "pop data " << *head << " success!" << std::endl;
  38.             i++;
  39.         }
  40.         });
  42.     t1.join();
  43.     t2.join();
  44.     t3.join();
  46.     assert(rmv_set.size() == 20000);
  47. }

上面的代码中t1负责压入两万个元素，t2和t3分别从栈中弹出元素
我们在assert处打个断点，可以看到集合大小确实为两万个，而且不存在重复元素，不存在缺失的元素。

https://cdn.llfc.club/1703226074849.jpg

总结

源码链接：
https://gitee.com/secondtonone1/boostasio-learn/tree/master/concurrent/day17-LockFreeStack

视频链接：
https://space.bilibili.com/271469206/channel/collectiondetail?sid=1623290

热门评论
点击查看更多评论 没有更多评论了,点击收起 展开评论

热门文章

  1. Qt环境搭建
    喜欢(517) 浏览(23067)

  2. vscode搭建windows C++开发环境
    喜欢(596) 浏览(77924)

  3. 聊天项目(28) 分布式服务通知好友申请
    喜欢(507) 浏览(5631)

  4. 使用hexo搭建个人博客
    喜欢(533) 浏览(11090)

  5. Linux环境搭建和编码
    喜欢(594) 浏览(12756)

最新评论

  1. 堆排序 secondtonone1:堆排序非常实用，定时器就是这个原理制作的。
  2. 处理网络粘包问题 zyouth: //消息的长度小于头部规定的长度，说明数据未收全，则先将部分消息放到接收节点里 if (bytes_transferred < data_len) { memcpy(_recv_msg_node->_data + _recv_msg_node->_cur_len, _data + copy_len, bytes_transferred); _recv_msg_node->_cur_len += bytes_transferred; ::memset(_data, 0, MAX_LENGTH); _socket.async_read_some(boost::asio::buffer(_data, MAX_LENGTH), std::bind(&CSession::HandleRead, this, std::placeholders::_1, std::placeholders::_2, shared_self)); //头部处理完成 _b_head_parse = true; return; } 把_b_head_parse = true;放在_socket.async_read_some前面是不是更好
  3. 类和对象 陈宇航:支持！！！！
  4. 解决博客回复区被脚本注入的问题 secondtonone1:走到现在我忽然明白一个道理，无论工作也好生活也罢，最重要的是开心，即使一份安稳的工作不能给我带来事业上的积累也要合理的舍弃，所以我还是想去做喜欢的方向。
  5. 无锁并发队列 TenThousandOne:_head  和 _tail  替换为原子变量。那里pop的逻辑，val = _data[h] 可以移到循环外面吗
  6. 答疑汇总(thread,async源码分析) Yagus:如果引用计数为0，则会执行 future 的析构进而等待任务执行完成，那么看到的输出将是 这边应该不对吧，std::future析构只在这三种情况都满足的时候才回block: 1.共享状态是std::async 创造的(类型是_Task_async_state) 2.共享状态没有ready 3.这个future是共享状态的最后一个引用 这边共享状态类型是“_Package_state”，引用计数即使为0也不应该block啊
  7. 构造函数 secondtonone1:构造函数是类的基础知识，要着重掌握
  8. Qt MVC结构之QItemDelegate介绍 胡歌-此生不换:gpt, google
  9. 聊天项目(7) visualstudio配置grpc diablorrr:cmake文件得改一下 find_package(Boost REQUIRED COMPONENTS system filesystem)，要加上filesystem。在target_link_libraries中也同样加上
  10. interface应用 secondtonone1:interface是万能类型，但是使用时要转换为实际类型来使用。interface丰富了go的多态特性，也降低了传统面向对象语言的耦合性。
  11. 网络编程学习方法和图书推荐 Corleone:啥程度可以找工作
  12. 创建项目和编译 secondtonone1:谢谢支持
  13. visual studio配置boost库 一giao里我离giaogiao:请问是修改成这样吗：.\b2.exe toolset=MinGW
  14. 聊天项目(9) redis服务搭建 pro_lin:redis线程池的析构函数，除了pop出队列，还要free掉redis连接把
  15. protobuf配置和使用 熊二:你可以把dll放到系统目录，也可以配置环境变量，还能把dll丢到lib里
  16. boost::asio之socket的创建和连接 项空月:发现一些错别字 :每隔vector存储  是不是是每个. asio::mutable_buffers_1 o或者    是不是多打了个o
  17. 聊天项目(15) 客户端实现TCP管理者 lkx:已经在&QTcpSocket::readyRead 回调函数中做了处理了的。
  18. C++ 并发三剑客future, promise和async Yunfei:大佬您好，如果这个线程池中加入的异步任务的形参如果有右值引用，这个commit中的返回类型推导和bind绑定就会出现问题，请问实际工程中，是不是不会用到这种任务，如果用到了，应该怎么解决？
  19. string类 WangQi888888:确实错了，应该是！isspace(sind[index]). 否则不进入循环，还是原来的字符串“some string”
  20. Qt 对话框 Spade2077:QDialog w(); //这里是不是不需要带括号
  21. 再谈单例模式 secondtonone1:是的，C++11以后返回局部static变量对象能保证线程安全了。
  22. 利用栅栏实现同步 Dzher:作者你好！我觉得 std::thread a(write_x); std::thread b(write_y); std::thread c(read_x_then_y); std::thread d(read_y_then_x); 这个例子中的assert fail并不会发生，原子变量设定了非relaxed内存序后一个线程的原子变量被写入，那么之后的读取一定会被同步的，c和d线程中只可能同时发生一个z++未执行的情况，最终z不是1就是2了，我测试了很多次都没有assert，请问我这个观点有什么错误，谢谢！
  23. 面试题汇总(一) secondtonone1:看到网络上经常提问的go的问题，做了一下汇总，结合自己的经验给出的答案，如有纰漏，望指正批评。
  24. 可变参数模板+异步队列实现异步打印功能 熊二:学到了%%%

其他链接

  1. GitHub
  2. 个人博客
  3. 博客园
  4. CSDN

个人公众号

个人微信