C++ 线程安全队列

#pragma once
#include <algorithm>
#include <cstddef>
#include <stdexcept>

template<typename T>
class Queue {
public:
    explicit Queue(const size_t capacity = 4) : capacity(capacity), size(0),
                                                data(new std::byte[capacity * sizeof(T)]) {
    }

    virtual ~Queue() {
        for (size_t i = 0; i < size; i++) {
            reinterpret_cast<T*>(data)[i].~T();
        }
        delete[] data;
    }

    void Push(const T&value) {
        std::unique_lock lock(mutex);
        if (capacity <= size) {
            capacity *= 2;
            auto* newData = new std::byte[sizeof(T) * capacity];
            for (size_t i = 0; i < size; i++) {
                new (newData + i * sizeof(T)) T(std::move(reinterpret_cast<T*>(data)[i]));
                reinterpret_cast<T*>(data)[i].~T();
            }
            delete[] data;
            data = newData;
        }
        new(data + size * sizeof(T)) T(value);
        size++;
    }

    void Pop(T&value) {
        std::unique_lock lock(mutex);
        if (size == 0) {
            throw std::out_of_range("Queue::Pop() when size = 0");
        }
        size--;
        value = std::move(*reinterpret_cast<T *>(data + sizeof(T)));
    }

    T Pop() {
        std::unique_lock lock(mutex);
        if (size == 0) {
            throw std::out_of_range("Queue::Pop() when size = 0");
        }
        size--;
        return std::move(*reinterpret_cast<T *>(data + sizeof(T) * size));
    }

    void Clear() {
        std::unique_lock lock(mutex);
        for (size_t i = 0; i < size; i++) {
            reinterpret_cast<T*>(data + i * sizeof(T))->~T();
        }
        size = 0;
    }

    [[nodiscard]] size_t Size() const {
        return size;
    }

    [[nodiscard]] size_t Capacity() const {
        return capacity;
    }

    [[nodiscard]] bool Empty() const {
        return size == 0;
    }

private:
    std::byte* data;
    size_t size;
    size_t capacity;
    std::mutex mutex;
};

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值