Java中的ArrayBlockingQueue drainTo() 方法

drainTo() 方法是 Java 语言中 ArrayBlockingQueue 类的一个方法，用来从队列中"掏空"元素到一个集合中去。在本文中，我们将讲解 drainTo() 方法的使用场景、方法签名、参数、返回值、示例和注意事项。

使用场景

当我们需要从一个阻塞队列中读取并处理大量元素时，使用 ArrayBlockingQueue 是一个不错的选择，但当队列元素过多时，依次取出每个元素并进行处理会导致程序效率低下，因此可以使用 drainTo() 方法一次性取出一批元素到集合中，然后在集合中进行处理，以提高程序效率。

方法签名

public int drainTo(Collection<? super E> c)

参数

• c：集合，用于保存取出的元素。

返回值

方法返回取出的元素数量。

示例

import java.util.ArrayList;
import java.util.concurrent.ArrayBlockingQueue;

public class Example {

   public static void main(String[] args) {
   
       ArrayBlockingQueue<Integer> queue = new ArrayBlockingQueue<>(10);
       for (int i = 0; i < 10; i++) {
           queue.offer(i);
       }

       ArrayList<Integer> list = new ArrayList<>();
       int count = queue.drainTo(list);
       System.out.println("取出了 " + count + " 个元素，集合大小为 " + list.size());

       for (int i = 0; i < list.size(); i++) {
           System.out.println(list.get(i));
       }
   }
}

输出结果：

取出了 10 个元素，集合大小为 10
0
1
2
3
4
5
6
7
8
9

注意事项

• 集合必须自己创建，否则会抛出 NullPointerException 异常。
• drainTo() 方法会一次性取出队列中的元素，如果队列中的元素数量少于集合大小，返回的元素数量会等于队列中的元素数量，而不是集合大小。
• 当多线程同时使用 ArrayBlockingQueue 的 drainTo() 方法时，需要注意竞态条件。