ZBLOG

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

如果不想使用Node结构体,可以考虑使用std::atomic来代替指向Node的指针。这样每个元素直接存储在std::atomic中,并且也不需要在try_pop()函数中进行delete操作。

以下是修改后的LockFreeQueue代码:

  1. #include <atomic>
  2. #include <condition_variable>
  3. #include <iostream>
  4. #include <thread>
  5. template <typename T>
  6. class LockFreeQueue
  7. {
  8. public:
  9. LockFreeQueue() : head_(nullptr), tail_(nullptr) {}
  10. ~LockFreeQueue()
  11. {
  12. while (T *const old_head = head_)
  13. {
  14. head_ = old_head->next_;
  15. delete old_head;
  16. }
  17. }
  18. void push(const T &value)
  19. {
  20. T *new_node = new T(value);
  21. new_node->next_ = nullptr;
  22. T *tail = tail_.load(std::memory_order_relaxed);
  23. if (tail)
  24. {
  25. tail->next_ = new_node;
  26. }
  27. else
  28. {
  29. // first element
  30. head_ = new_node;
  31. }
  32. tail_.store(new_node, std::memory_order_release);
  33. size_.fetch_add(1, std::memory_order_release);
  34. }
  35. bool try_pop(T &value)
  36. {
  37. T *const old_head = head_.load(std::memory_order_relaxed);
  38. if (!old_head)
  39. {
  40. return false;
  41. }
  42. T *const new_head = old_head->next_;
  43. value = *old_head;
  44. head_.store(new_head, std::memory_order_release);
  45. size_.fetch_sub(1, std::memory_order_release);
  46. return true;
  47. }
  48. bool empty() const
  49. {
  50. return size_.load(std::memory_order_acquire) == 0;
  51. }
  52. private:
  53. struct Node
  54. {
  55. T value_;
  56. std::atomic<Node*> next_;
  57. Node() : next_(nullptr) {}
  58. explicit Node(const T &value) : value_(value), next_(nullptr) {}
  59. };
  60. std::atomic<T*> head_;
  61. std::atomic<T*> tail_;
  62. std::atomic<size_t> size_{0};
  63. std::condition_variable cv_;
  64. };

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

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

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

联系我们

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

微信号:3007537140

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

关注我们
x

注册

已经有帐号?