多线程 Java 应用程序中的数据缓冲

Question

我有一个多线程应用程序，它有一个生产者线程和几个消费者线程。数据存储在一个共享的线程安全集合中，当缓冲区中有足够的数据时，就刷新到数据库中。

来自爪哇文档 -

BlockingQueue<E>

一个队列，它还支持在检索元素时等待队列变为非空，并在存储元素时等待队列中的空间变为可用。

take()

检索并删除此队列的头部，如有必要，请等待元素可用。

我的问题是-

< li >是否有另一个集合具有E[] take(int n)方法？即阻塞队列等待，直到元素可用。我想要的是，它应该等到100或200个元素可用时。 < li >或者，我是否可以使用另一种方法来解决这个问题，而无需轮询？

Answer 1

我不确定标准库中是否有类似的类具有 take（int n） 类型方法，但您应该能够包装默认的 BlockingQueue 以添加该函数而不会有太多麻烦，你不觉得吗？

另一种方案是触发一个操作，在该操作中您将元素放入集合中，您设置的阈值将触发刷新。

Answer 2

< code>drainTo方法并不完全是您想要的，但是它能满足您的目的吗？

http://docs.oracle.com/javase/6/docs/api/java/util/concurrent/BlockingQueue.html#drainTo（java.util.Collection，int）

编辑

您可以使用take和drainTo的组合来实现性能稍高的批处理阻塞takemin：

public <E> void drainTo(final BlockingQueue<E> queue, final List<E> list, final int min) throws InterruptedException
{
  int drained = 0;
  do 
  {
    if (queue.size() > 0)
      drained += queue.drainTo(list, min - drained);
    else
    {
      list.add(queue.take());
      drained++;
    }
  }
  while (drained < min);
}

Answer 3

我认为唯一的方法是要么扩展< code>BlockingQueue的某些实现，要么使用< code>take创建某种实用方法:

public <E> void take(BlockingQueue<E> queue, List<E> to, int max) 
        throws InterruptedException {

    for (int i = 0; i < max; i++)
        to.add(queue.take());
}