如何实现DelayQueue中poll与offer操作的同步原子执行？

这问题我之前也碰到过！确实，DelayQueue的单个poll()和offer()都是线程安全的，但把它们放一块做条件判断的时候，就会出现竞态条件——毕竟两个方法各自拿锁、释放锁，中间的空隙完全可能被其他线程插进来捣乱。

为啥内部锁解决不了问题？ #

你可能会好奇，为啥不能直接复用DelayQueue内部的锁？可惜那锁是private权限的，JDK没有对外暴露，硬用反射去获取不仅破坏封装，还容易因为JDK版本变更出问题，完全不推荐。

而DelayQueue的单个方法只保证自身操作的原子性：调用poll()时会获取内部锁，执行完就释放；再调用offer()时又重新获取锁，这两步之间没有任何同步保护，其他线程完全可以在这个间隙修改队列，导致你的条件判断逻辑失效。

更优雅的实现方式 #

你一开始想到的用ReentrantLock包裹的思路是完全正确的，觉得“冗余”大概率是不想在业务代码里重复写锁的模板代码。那最优解就是把这个原子操作封装成一个工具类，把锁和DelayQueue绑定在一起，统一管理同步逻辑：

import java.util.concurrent.DelayQueue;
import java.util.concurrent.Delayed;
import java.util.concurrent.locks.ReentrantLock;

public class AtomicDelayQueue<E extends Delayed> {
    private final DelayQueue<E> delayQueue = new DelayQueue<>();
    private final ReentrantLock lock = new ReentrantLock();

    // 原子执行：poll成功则offer新元素
    public boolean pollAndOffer(E newElement) {
        lock.lock();
        try {
            E polledElement = delayQueue.poll();
            if (polledElement != null) {
                delayQueue.offer(newElement);
                return true;
            }
            return false;
        } finally {
            lock.unlock();
        }
    }

    // 按需封装其他需要原子执行的操作，比如单独的poll/offer
    public E poll() {
        lock.lock();
        try {
            return delayQueue.poll();
        } finally {
            lock.unlock();
        }
    }

    public boolean offer(E element) {
        lock.lock();
        try {
            return delayQueue.offer(element);
        } finally {
            lock.unlock();
        }
    }
}

注意事项 #

• 所有对队列的操作都必须通过这个封装类的方法，不能直接操作内部的DelayQueue，否则会破坏同步逻辑，引发线程安全问题。
• 如果你的场景需要阻塞式获取元素（比如用take()替代poll()），也可以在封装类里添加对应的原子方法，逻辑和上面一致。

说白了，这种“多个并发方法组合成原子操作”的场景，本身就需要额外的同步机制，这不是冗余，而是保证逻辑正确性的必要步骤。封装之后，业务代码会清爽很多，也能避免重复造轮子。

内容的提问来源于stack exchange，提问作者Orest