我发现自己重复这种模式,并经常想知道它是否在Java中是惯用的,或者有更好的方法来实现这种行为。
问题:给定生产者/消费者设置,消费者想要处理批量的项目,因此它使用drainTo()
,但是drainTo()
将轮询现有项目并且可能无法获得任何项目,避免这种情况我在排水管前加上take()
,以确保它阻塞,直到至少有一个项目可用。
对于特定数据集,我得到的一个问题是,在许多用例中,批量大小通常在(1,N,1,N)之间交替变化。一般来说,这是解决此问题的常用方法:
示例:
ArrayBlockingQueue<Foo> queue;
function void produce() {
while(true) {
queue.put(createFoo());
}
}
function void consumeBatchSpin() {
while(true) {
List<Foo> batch = Lists.newLinkedList();
queue.drainTo(batch);
doSomething(batch);
//the problem here is that if nothing is being produced, this loop will spin
}
}
function void consumeBatchTake() {
while(true) {
List<Foo> batch = Lists.newLinkedList();
batch.add(queue.take()); //force at least one item to be there
queue.drainTo(batch);
doSomething(batch);
}
}