在排空ArrayBlockingQueue之前阻塞

时间:2013-03-04 00:32:20

标签: java concurrency

我发现自己重复这种模式,并经常想知道它是否在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);
    }
 }

1 个答案:

答案 0 :(得分:1)

您是否考虑过添加到列表并在get上获取整个列表。

我最近发布了一个here。它正在进行代码审查here,但我的测试表明它很强大。

基本上,当您执行put时,将新元素添加到当前列表中。获得整个列表并以原子方式将其替换为新的空列表。

无需使用drainTo,也不需要旋转。