ZBLOG

#include #include template class LockFreeQueue { public: LockFreeQueue() : head_(new Node), tail_(head_.load()) {} ~LockFreeQueue() { while (Node *old_head = head_.load()) { ...

这个队列实现上可能存在ABA问题。具体来说,如果一个节点在某一时刻被弹出,然后又有一个新节点插入并且其指针恰好和该节点相同,那么后续的操作会将该新节点误认为是之前弹出的节点,从而导致问题。

解决方法是使用带有版本号的指针,在CAS操作时同时比较版本号。以下给出改进后的代码:

#include #include

template class LockFreeQueue { public:

  1. LockFreeQueue() : head_(new Node), tail_(head_.load()) {}
  2. ~LockFreeQueue()
  3. {
  4. while (Node *old_head = head_.load())
  5. {
  6. Node *next = old_head->next_;
  7. if (head_.compare_exchange_weak(old_head, next))
  8. {
  9. delete old_head;
  10. break;
  11. }
  12. }
  13. }
  14. void push(const T &value)
  15. {
  16. Node *new_node = new Node(value);
  17. while (true)
  18. {
  19. Tail tail = tail_.load();
  20. Node *tail_ptr = tail.ptr;
  21. uint64_t tail_version = tail.version;
  22. Node *next = tail_ptr->next_.load(std::memory_order_relaxed);
  23. if (!next)
  24. {
  25. if (tail_ptr->next_.compare_exchange_weak(next, new_node,
  26. std::memory_order_release,
  27. std::memory_order_relaxed))
  28. {
  29. Tail new_tail{new_node, tail_version + 1};
  30. tail_.compare_exchange_weak(tail, new_tail,
  31. std::memory_order_release,
  32. std::memory_order_relaxed);
  33. size_.fetch_add(1, std::memory_order_release);
  34. return;
  35. }
  36. }
  37. else
  38. {
  39. Tail new_tail{next, tail_version + 1};
  40. tail_.compare_exchange_weak(tail, new_tail,
  41. std::memory_order_release,
  42. std::memory_order_relaxed);
  43. }
  44. }
  45. }
  46. bool try_pop(T &value)
  47. {
  48. while (true)
  49. {
  50. Head head = head_.load();
  51. Tail tail = tail_.load();
  52. Node *head_ptr = head.ptr;
  53. uint64_t head_version = head.version;
  54. uint64_t tail_version = tail.version;
  55. Node *next = head_ptr->next_.load(std::memory_order_relaxed);
  56. if (head.ptr == tail.ptr)
  57. {
  58. if (!next)
  59. {
  60. return false;
  61. }
  62. Tail new_tail{next, tail_version + 1};
  63. tail_.compare_exchange_weak(tail, new_tail,
  64. std::memory_order_release,
  65. std::memory_order_relaxed);
  66. }
  67. else
  68. {
  69. value = next->value_;
  70. Head new_head{next, head_version + 1};
  71. if (head_.compare_exchange_weak(head, new_head,
  72. std::memory_order_release,
  73. std::memory_order_relaxed))
  74. {
  75. size_.fetch_sub(1, std::memory_order_release);
  76. delete head_ptr;
  77. return true;
  78. }
  79. }
  80. }
  81. }
  82. bool empty() const
  83. {
  84. return size_.load(std::memory_order_acquire) == 0;
  85. }

private:

  1. struct Node
  2. {
  3. T value_;
  4. std::atomic<Node *> next_;
  5. Node() : value_(T()), next_(nullptr) {}
  6. explicit Node(const T &value) : value_(value), next_(nullptr) {}
  7. };
  8. struct Head
  9. {
  10. Node *ptr;
  11. uint64_t version;
  12. };
  13. struct Tail
  14. {
  15. Node *ptr;
  16. uint64_t version;
  17. };
  18. std::atomic<Head> head_;
  19. std::atomic<Tail> tail_;
  20. std::atomic<size_t> size_{0};

};

本站部分文章来源于网络,版权归原作者所有,如有侵权请联系站长删除。
转载请注明出处:https://golang.0voice.com/?id=1631

分享:
扫描分享到社交APP
上一篇
下一篇
发表列表
游客 游客
此处应有掌声~
评论列表

还没有评论,快来说点什么吧~

联系我们

在线咨询: 点击这里给我发消息

微信号:3007537140

上班时间: 10:30-22:30

关注我们
x

注册

已经有帐号?