java - 超时排空队列

Question

我有一个流程，多个生产者填充队列，单个消费者处理从该队列检索的数据。

为了提高效率，消费者使用 BlockingQueue#drainTo API 从队列中排出数据。但有一个问题，当队列为空时，消费者会尝试在循环中耗尽数据，而不会延迟，从而消耗大量 CPU。

是否有 API 允许使用类似于 BlockingQueue#poll(long timeout, TimeUnit unit) 的超时耗尽队列？

示例消费者代码:

while (threadIsActive) {
    List<Event> events = new ArrayList<>();
    queue.drainTo(events);
    processEvents(events);
}

Answer 1

我最终使用了 @Kayaman 建议的 poll/drainTo 组合，我不太喜欢它，它看起来很丑。但我会接受任何其他提出更好解决方案的答案:

while (threadIsActive) {
    Event firstEvent = queue.poll(queuePollTimeout, TimeUnit.MILLISECONDS);
    if (firstEvent != null) {
        // reserve enough space to fit first event, current queue capacity and
        // new events which occur while we are draining
        int drainCapacity = queue.size() * 2;
        List<Event> events = new ArrayList<>(drainCapacity);

        events.add(firstEvent);
        queue.drainTo(events);

        processEvents(events);
    }
}