如何限制 LinkedTransferQueue 的大小？

Question

我正在实现生产者/消费者模式并使用 LinkedTransferQueue 集合。

我不希望我的制作人超出某些内存限制。

目前我正在使用此检查，但从documentation 开始，大小操作需要 O(N) 遍历。对于我目前的实现，这很好，

但是有没有比我目前使用的更好的方法 使用？

LinkedTransferQueue<String> producerStringLinkedTransferQueue = new LinkedTransferQueue<String>();

if (producerStringLinkedTransferQueue.size() <= 5000) {
    producerStringLinkedTransferQueue.add(<some data>);
}

Answer 1

如果您不想在队列已满时不执行任何操作（而不是像 Apache 公共集合 CircularFifoQueue 那样替换其最旧的项目），那么您可以将 LinkedTransferQueue 包装在您的自定义类中并实现计算队列所需的方法尺寸：

public class LimitLinkedTransferQueue<E> {
    LinkedTransferQueue<E> queue = new LinkedTransferQueue<>();
    private final long maxSize;
    private long size = 0;

    public LimitLinkedTransferQueue(long maxSize) {
        super();
        this.maxSize = maxSize;
    }

    public boolean add(E e) {
        if (this.size == this.maxSize){
            return false;
        }
        boolean result = queue.add(e);
        if (result) {
            size++;
        }
        return result;
    }

    public E take() throws InterruptedException {
        E item = queue.take();
        size--;
        return item;
    }
    // other need methods
}