ZBLOG

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

以下是优化后的代码:

#include #include #include

template class LockFreeQueue { public: LockFreeQueue() : head(new Node), tail(head_.load()) {} ~LockFreeQueue() {

while (Node *old_head = head_.load()) {
  Node *next = old_head->next_;
  if (head_.compare_exchange_weak(old_head, next)) {
    delete old_head;
    break;
  }
}

}

void push(const T &value) {

auto new_node = std::make_unique<Node>(value);
auto *new_node_ptr = new_node.get();
Node *tail = tail_.load(std::memory_order_relaxed);
while (true) {
  auto *next = tail->next_.load(std::memory_order_acquire);
  if (!next) {
    if (tail->next_.compare_exchange_weak(next, new_node_ptr,
                                           std::memory_order_release,
                                           std::memory_order_relaxed)) {
      break;
    }
  } else {
    tail_.compare_exchange_weak(tail, next, std::memory_order_release,
                                 std::memory_order_relaxed);
  }
}
tail_.compare_exchange_weak(tail, new_node_ptr, std::memory_order_release);
size_.fetch_add(1, std::memory_order_release);

}

bool try_pop(T &value) {

Node *old_head = head_.load(std::memory_order_relaxed);
while (true) {
  auto *new_head = old_head->next_.load(std::memory_order_acquire);
  if (!new_head) {
    return false;
  }
  if (head_.compare_exchange_weak(old_head, new_head,
                                   std::memory_order_release,
                                   std::memory_order_relaxed)) {
    value = std::move(new_head->value_);
    size_.fetch_sub(1, std::memory_order_release);
    delete old_head;
    return true;
  }
}

}

bool empty() const { return size_.load(std::memory_order_acquire) == 0; }

private: struct Node {

T value_;
std::atomic<Node *> next_{nullptr};
explicit Node(const T &value) : value_(value) {}

};

std::atomic head{nullptr}; std::atomic tail{nullptr}; std::atomic size_{0}; };

优化点:

1.使用std::make_unique构造节点对象,避免手动new和delete

2.在try_pop中使用while循环重试,避免竞争条件(多个线程同时pop)时出现问题

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

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

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

联系我们

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

微信号:3007537140

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

关注我们
x

注册

已经有帐号?