java-ArrayBlockingQueue详解

在Java并发编程中,ArrayBlockingQueue是一个非常常用的工具类。它是一个由数组支持的有界阻塞队列,提供了线程安全的队列操作。

1.ArrayBlockingQueue概述

ArrayBlockingQueue是一个基于数组实现的阻塞队列,它继承自AbstractQueue并实现了BlockingQueue接口。这个队列在创建时需要指定一个固定的大小,之后这个大小就不能再改变了。当队列满时,如果再有新的元素试图加入队列,那么这个操作会被阻塞;同样地,如果队列为空,那么从队列中取元素的操作也会被阻塞。这种特性使得ArrayBlockingQueue非常适合作为生产者-消费者模式中的缓冲区。

2.ArrayBlockingQueue的核心特性

2.1.线程安全性

ArrayBlockingQueue是线程安全的,它通过内部锁机制保证了在多线程环境下的安全性。因此,在多线程环境中,你可以放心地使用它而不需要担心数据的一致性问题。

2.2.阻塞控制

ArrayBlockingQueue提供了阻塞控制机制。当队列满时,尝试向队列中添加元素的线程会被阻塞,直到队列中有空间可用;同样,当队列为空时,尝试从队列中取出元素的线程也会被阻塞,直到队列中有元素可供消费。这种机制可以有效地控制生产者和消费者的速度,避免资源的浪费。

2.3.有界性

ArrayBlockingQueue的有界性可以防止队列无限制地增长,从而避免内存溢出。在实际应用中,这种有界性可以作为系统的一个流量控制阀,当系统过载时,通过阻塞或拒绝请求来保护系统。

3.ArrayBlockingQueue的使用

3.1.创建ArrayBlockingQueue

创建一个ArrayBlockingQueue非常简单,只需要指定队列的大小即可:

int queueSize = 10;
BlockingQueue<Integer> queue = new ArrayBlockingQueue<>(queueSize);

3.2.生产者-消费者模式

ArrayBlockingQueue常用于生产者-消费者模式。生产者负责生成数据并添加到队列中,而消费者则从队列中取出数据并处理。下面是一个简单的生产者-消费者示例:

import java.util.concurrent.ArrayBlockingQueue;
import java.util.concurrent.BlockingQueue;

public class ProducerConsumerExample {
    public static void main(String[] args) {
        BlockingQueue<Integer> queue = new ArrayBlockingQueue<>(5);

        Thread producer = new Thread(() -> {
            for (int i = 0; i < 10; i++) {
                try {
                    System.out.println("生产者生产了数据:" + i);
                    queue.put(i);
                    Thread.sleep(200);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        });

        Thread consumer = new Thread(() -> {
            while (true) {
                try {
                    Integer data = queue.take();
                    System.out.println("消费者消费了数据:" + data);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        });

        producer.start();
        consumer.start();
    }
}

运行结果:

生产者生产了数据:0
消费者消费了数据:0
生产者生产了数据:1
消费者消费了数据:1
生产者生产了数据:2
消费者消费了数据:2
生产者生产了数据:3
消费者消费了数据:3
生产者生产了数据:4
消费者消费了数据:4
生产者生产了数据:5
消费者消费了数据:5
生产者生产了数据:6
消费者消费了数据:6
生产者生产了数据:7
消费者消费了数据:7
生产者生产了数据:8
消费者消费了数据:8
生产者生产了数据:9
消费者消费了数据:9

在这个示例中,我们创建了一个大小为5的ArrayBlockingQueue,然后启动了一个生产者线程和一个消费者线程。生产者线程会生成10个数据,并尝试将它们添加到队列中;消费者线程则会不断地从队列中取出数据并处理。由于队列的大小只有5,因此当生产者生产了5个数据后,它会被阻塞,直到消费者消费了一些数据释放出空间。同样地,当队列为空时,消费者线程也会被阻塞,直到生产者生产了新的数据。

4.ArrayBlockingQueue的最佳实践

4.1.选择合适的队列大小

队列的大小应根据具体的应用场景来设置。如果设置得太小,可能会导致频繁的阻塞和上下文切换,影响性能;如果设置得太大,可能会浪费内存资源。因此,在选择队列大小时,需要综合考虑系统的负载、内存资源和性能要求等因素。

4.2.合理使用阻塞方法

ArrayBlockingQueue提供了多种阻塞方法,如puttakeofferpoll等。在使用这些方法时,需要根据具体的需求来选择合适的方法。例如,如果你希望当队列满时生产者线程能够阻塞等待空间可用,那么可以使用put方法;如果你希望生产者线程在队列满时能够立即返回并做其他处理,那么可以使用offer方法。

4.3.避免死锁

在使用ArrayBlockingQueue时,需要注意避免死锁的发生。例如,不要在持有其他锁的情况下调用ArrayBlockingQueue的阻塞方法,否则可能会导致死锁。此外,还需要注意避免循环等待和饥饿等问题。

4.4.考虑使用公平策略

ArrayBlockingQueue的构造函数允许指定一个公平性参数。如果设置为true,等待时间最长的线程将优先获得访问队列的机会。但需要注意的是,公平性可能会降低性能。因此,在决定是否使用公平策略时,需要综合考虑系统的性能和公平性要求。

5.源码详解

5.1.主要属性

// 用于存储队列元素的数组
final Object[] items;

// 队列的容量
int count;

// 控制并发访问的锁
final ReentrantLock lock;

// 队列不满时的等待条件
private final Condition notFull;

// 队列不为空时的等待条件
private final Condition notEmpty;

// 队列中等待取数据的线程数
final AtomicInteger waitingConsumers = new AtomicInteger();

// 队列中等待插入数据的线程数
final AtomicInteger waitingProducers = new AtomicInteger();

5.2.构造函数

ArrayBlockingQueue 提供了几种构造函数,其中最基本的两个是接受队列容量和指定是否公平的构造函数。

public ArrayBlockingQueue(int capacity) {
    this(capacity, false);
}

public ArrayBlockingQueue(int capacity, boolean fair) {
    if (capacity <= 0)
        throw new IllegalArgumentException();
    this.items = new Object[capacity];
    lock = new ReentrantLock(fair);
    notEmpty = lock.newCondition();
    notFull = lock.newCondition();
}

5.3.入队操作

put(E e)offer(E e) 是两种入队操作,其中 put 方法在队列满时会阻塞,而 offer 方法在队列满时会立即返回失败或者根据提供的超时时间等待。

public void put(E e) throws InterruptedException {
    checkNotNull(e);
    final ReentrantLock lock = this.lock;
    lock.lockInterruptibly();
    try {
        while (count == items.length)
            notFull.await();
        enqueue(e);
    } finally {
        lock.unlock();
    }
}

public boolean offer(E e, long timeout, TimeUnit unit)
    throws InterruptedException {

    checkNotNull(e);
    long nanos = unit.toNanos(timeout);
    final ReentrantLock lock = this.lock;
    lock.lockInterruptibly();
    try {
        while (count == items.length) {
            if (nanos <= 0)
                return false;
            nanos = notFull.awaitNanos(nanos);
        }
        enqueue(e);
        return true;
    } finally {
        lock.unlock();
    }
}

private void enqueue(E x) {
    // 队列尾部插入元素
    final Object[] items = this.items;
    items[putIndex] = x;
    if (++putIndex == items.length)
        putIndex = 0;
    count++;
    // 通知可能在等待的消费者线程
    notEmpty.signal();
}

5.4.出队操作

take()poll() 是两种出队操作,其中 take 方法在队列空时会阻塞,而 poll 方法在队列空时会立即返回 null 或者根据提供的超时时间等待。

public E take() throws InterruptedException {
    final ReentrantLock lock = this.lock;
    lock.lockInterruptibly();
    try {
        while (count == 0)
            notEmpty.await();
        return dequeue();
    } finally {
        lock.unlock();
    }
}

public E poll(long timeout, TimeUnit unit) throws InterruptedException {
    long nanos = unit.toNanos(timeout);
    final ReentrantLock lock = this.lock;
    lock.lockInterruptibly();
    try {
        while (count == 0) {
            if (nanos <= 0)
                return null;
            nanos = notEmpty.awaitNanos(nanos);
        }
        return dequeue();
    } finally {
        lock.unlock();
    }
}

private E dequeue() {
    // 队列头部取出元素
    final Object[] items = this.items;
    @SuppressWarnings("unchecked")
    E x = (E) items[takeIndex];
    items[takeIndex] = null;
    if (++takeIndex == items.length)
        takeIndex = 0;
    count--;
    if (itrs != null)
        itrs.elementDequeued();
    // 通知可能在等待的生产者线程
    notFull.signal();
    return x;
}

6.总结

ArrayBlockingQueue是Java并发编程中一个非常实用的工具类。它提供了线程安全的阻塞队列实现,支持生产者-消费者模式,并允许通过队列的大小来控制系统的流量。在使用ArrayBlockingQueue时,需要注意选择合适的队列大小、合理使用阻塞方法、避免死锁和考虑使用公平策略等问题。通过合理地使用ArrayBlockingQueue,可以有效地提高系统的并发性能和稳定性。

你可能感兴趣的:(多线程,java)