java - 如何设置 Java DelayQueue 的容量

Question

我正在使用 LinkedBlockingQueue 队列来实现用于 TCP/IP 事件传输的生产者-消费者模式，我正在使用 boolean offer(e)这意味着一旦队列达到其容量，新传入的事件将被忽略(丢弃)并返回 false。

现在我必须将事件保持一段可配置的时间(比如 2 秒)，所以我决定使用 DelayQueue，它可以保持元素并仅在时间到期时释放元素。

不幸的是，DelayQueue 是无界。我想知道是否有办法为 DelayQueue 或 delayQueue.size() == CAPACITY 设置容量总是可靠的？

Answer 1

这是一个相当大的问题，因为如果我们要对其进行子类化或委托(delegate)给它并且锁在 DelayQueue 中是私有(private)的，我们需要访问 DelayQueue 使用的内部锁。

我们不能使用第二个锁，因为这会导致 take 出现问题。您也许可以自己实现它，但这样做您已经完成了自己实现 DelayQueue 的一半以上，所以这可能不是您想要的。

我们可以使用反射访问锁。但是请注意，这不是最好的想法，因为它依赖于 DelayQueue 的实现细节。它可能无法在所有 JRE 上运行，如果您更改正在运行的 JRE 的版本，它甚至可能会中断。也就是说，我认为这是解决您的问题的绝对最简单的解决方案，尽管有点脏。

/**
 * Bounded implementation of {@link DelayQueue}. This implementation uses
 * reflection to access the internal lock in {@link DelayQueue} so might
 * only work on the Oracle 1.8 JRE.
 * @param <T>
 */
public class BoundedDelayQueue<T extends Delayed> extends DelayQueue<T> {

    // Lock used to synchronize every operation
    private final transient ReentrantLock lock;

    // The limit
    private final int limit;

    BoundedDelayQueue(int limit) {
        try {
            // Grab the private lock in DelayQueue using reflection so we can use it
            // to gain exclusive access outside of DelayQueue
            Field lockField = DelayQueue.class.getDeclaredField("lock");
            lockField.setAccessible(true);
            this.lock = (ReentrantLock) lockField.get(this);
        } catch (NoSuchFieldException | IllegalAccessException e) {
            throw new Error("Could not access lock field", e);
        }
        this.limit = limit;
    }

    @Override
    // All the various ways of adding items in DelayQueue delegate to
    // offer, so we only have to override it and not the other methods
    public boolean offer(final T t) {
        // Lock the lock
        lock.lock();
        try {
            // Check the size limit
            if(size() == limit) {
                return false;
            }
            // Forward to superclass
            return super.offer(t);
        } finally {
            lock.unlock();
        }
    }
}

请注意，这不会实现带有超时的offer，如果您需要，您必须自己完成。