ZBLOG

用原子变量实现并发安全的队列

使用原子变量实现并发安全的队列可以采用以下方法:

  1. 定义一个节点结构,包含数据和指向下一个节点的指针。
  2. 使用原子指针作为队列的头部和尾部。
  3. 使用原子操作来保证对队列进行并发操作时的一致性。

下面是一个简单示例的C++代码实现:

#include <atomic>

template<typename T>
class ConcurrentQueue {
private:
    struct Node {
        T data;
        Node* next;

        Node(const T& value) : data(value), next(nullptr) {}
    };

    std::atomic<Node*> head;
    std::atomic<Node*> tail;

public:
    ConcurrentQueue() : head(nullptr), tail(nullptr) {}

    void enqueue(const T& value) {
        Node* newNode = new Node(value);
        newNode->next = nullptr;

        // 将新节点添加到尾部,并更新尾部指针
        Node* prevTail = tail.exchange(newNode);
        if (prevTail == nullptr) {
            // 队列为空,同时更新头部指针
            head.store(newNode);
        }
        else {
            // 队列不为空,将新节点链接到前一个尾节点
            prevTail->next = newNode;
        }
    }

    bool dequeue(T& value) {
        // 从头部开始出队
        Node* currentHead = head.load();
        
        while (currentHead != nullptr && currentHead != tail.load()) {
            if (head.compare_exchange_strong(currentHead, currentHead->next)) {
                // 成功更新头部指针,取出数据并删除节点
                value = currentHead->data;
                delete currentHead;
                return true;
            }
        }

        // 队列为空或者只剩下一个节点
        if (currentHead != nullptr) {
            // 只有一个节点,更新尾部指针以保持一致性
            tail.store(nullptr);
            
            // 获取数据并删除节点
            value = currentHead->data;
            delete currentHead;
            return true;
        }

        return false;  // 队列为空,无法出队
    }
};

使用该ConcurrentQueue类可以在多个线程中安全地进行队列操作。enqueue函数用于向队列中添加元素,dequeue函数用于从队列中取出元素。原子操作的使用确保了对队列头部和尾部指针的并发访问的正确性和一致性。


内容由零声教学AI助手提供,问题来源于学员提问

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

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

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

联系我们

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

微信号:3007537140

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

关注我们
x

注册

已经有帐号?