AQS源码探究_05 Conditon条件队列(手写一个入门的BrokingQueue)

简介: AQS源码探究_05 Conditon条件队列(手写一个入门的BrokingQueue)

1. Condition队列介绍

AQS中还有另一个非常重要的内部类ConditionObject,它实现了Condition接口,主要用于实现条件锁。


ConditionObject中也维护了一个队列,这个队列主要用于等待条件的成立,当条件成立时,其它线程将signal这个队列中的元素,将其移动到AQS的队列中,等待占有锁的线程释放锁后被唤醒。


Condition典型的运用场景是在BlockingQueue中的实现,当队列为空时,获取元素的线程阻塞在notEmpty条件上,一旦队列中添加了一个元素,将通知notEmpty条件,将其队列中的元素移动到AQS队列中等待被唤醒。


2. 手写一个入门的BrokingQueue

2.1 自定义BrokingQueue接口

/**
 * @author csp
 * @date 2021-05-03
 */
public interface BrokingQueue<T> {
    /**
     * 插入数据
     */
    void put(T element) throws InterruptedException;
    /**
     * 获取数据
     */
    T take() throws InterruptedException;
}

2.2 自定义MiniArrayBrokingQueue类

/**
 * @author csp
 * @date 2021-05-03
 */
public class MiniArrayBrokingQueue implements BrokingQueue {
    /**
     * ReentrantLock锁:用于线程并发控制
     */
    private Lock lock = new ReentrantLock();
    /**
     * 生产者线程生产数据时,会先检查当前queues是否已经满了,如果已满,则需要令当前生产者线程调用notFull.await()
     * 进入到notFull条件队列挂起,等待消费者线程消费一个数据时才可以唤醒~
     */
    private Condition notFull = lock.newCondition();
    /**
     * 消费者线程消费数据时,会先检查当前queues中是否有数据,如果没有数据,则需要令当前消费者线程调用notEmpty.await()
     * 进入到notEmpty条件队列挂起,等待生产者生产出数据时才能唤醒!
     */
    private Condition notEmpty = lock.newCondition();
    /**
     * 存储元素的数组队列
     */
    private Object[] queues;
    /**
     * 数组队列的长度
     */
    private int size;
    /**
     * count:当前队列中,可以被消费的数据量
     * putptr:记录生产者存放数据的下一次位置,每个生产者生产完一个数据后,会将 putptr++
     * takeptr:记录消费者消费数据的下一个位置,每个消费者消费完一个数据后,会将takeptr++
     */
    private int count, putptr, takeptr;
    public MiniArrayBrokingQueue(int size) {
        this.size = size;
        this.queues = new Object[size];
    }
    /**
     * 插入数据
     *
     * @param element
     */
    @Override
    public void put(Object element) throws InterruptedException {
        // 获取锁
        lock.lock();
        try {
            // 判断当前queues数组队列是否已满
            if (count == this.size) {
                // 如果已满,则进入到notFull条件队列挂起,等待消费者线程消费一个数据时才可以唤醒~
                notFull.await();
            }
            // 执行到这里,说明数组队列queues未满,可以向队列中存放数据~
            this.queues[putptr] = element;
            putptr++;
            // 新存放数据后,判断是否达到了数组队列的最大值
            if (putptr == this.size) {
                // putptr恢复到0,为什么可以归零呢?
                // 因为:多线程执行条件下,总会有消费者线程在不断消费数据,即使当前生产者线程已经将putptr移动到size的位置,
                // 但是扔有可能有消费者线程将queues数组队列size位置之前的数据给消费掉了,那么此时queues其实并不算已经满了,
                // 那么将putptr恢复到0,就是为了让其不断去寻找queues数组中的空位,并用来存放element
                putptr = 0;
            }
            // 当前队列中,可以被消费的数据量++
            count++;
            // 当向队列中成功放入一个元素之后,需要做什么呢?
            // 需要给notEmpty一个唤醒信号,告诉消费者去消费
            notEmpty.signal();
        } finally {
            // 释放锁
            lock.unlock();
        }
    }
    /**
     * 获取数据
     *
     * @return
     */
    @Override
    public Object take() throws InterruptedException {
        // 获取锁
        lock.lock();
        try {
            // 判断当前queues数组队列是否是空的
            if (count == 0) {
                // 如果queues是空的,则进入到notEmpty队列进行挂起,等待生产者线程生产数据时才可以唤醒~
                notEmpty.await();
            }
            // 执行到这里,说明队列中有数据可以被消费了~
            Object element = this.queues[takeptr];
            takeptr++;
            // 新消费数据后,判断是否达到了数组队列的最大值
            if (takeptr == this.size) {
                // takeptr恢复到0,为什么可以归零呢?
                // 因为:多线程执行条件下,总会有生产者线程在不断生产数据,即使当前消费者线程已经将takeptr移动到size的位置,
                // 但是扔有可能有生产者线程向queues数组队列size之前的位置put数据,那么此时queues其实并不算空了,
                // 那么将takeptr恢复到0,就是为了让其不断去寻找queues数组中能被消费的数据
                takeptr = 0;
            }
            // 当前队列中,可以被消费的数据量--
            count--;
            // 当从队列中成功消费一个元素之后,需要做什么呢?
            // 需要给notFull一个唤醒信号,告诉生产者去生产
            notFull.signal();
            // 返回消费的element
            return element;
        } finally {
            // 释放锁
            lock.unlock();
        }
    }
    // 测试程序:
    public static void main(String[] args) {
        BrokingQueue<Integer> queue = new MiniArrayBrokingQueue(10);
        Thread producer = new Thread(() -> {
            int i = 0;
            while (true) {
                i++;
                if (i == 10) {
                    i = 0;
                }
                try {
                    System.out.println("生产者线程生产数据:" + i);
                    queue.put(Integer.valueOf(i));
                    TimeUnit.MILLISECONDS.sleep(200);
                } catch (InterruptedException e) {
                }
            }
        });
        producer.start();
        Thread consumer = new Thread(() -> {
            while (true) {
                try {
                    Integer result = queue.take();
                    System.out.println("消费者线程消费数据:" + result);
                    TimeUnit.MILLISECONDS.sleep(200);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        });
        consumer.start();
    }
}


相关文章
|
存储 缓存 Java
一文读懂线程池的实现原理
一文读懂线程池的实现原理
185 0
一文读懂线程池的实现原理
|
6月前
|
Java
并发编程之线程池的应用以及一些小细节的详细解析
并发编程之线程池的应用以及一些小细节的详细解析
36 0
|
4月前
|
存储 Java 索引
(十二)彻悟并发之JUC分治思想产物-ForkJoin分支合并框架原理剖析下篇
在《(十二)彻悟并发之JUC分治思想产物-ForkJoin分支合并框架原理剖析上篇》中,我们曾初步了解了ForkJoin分支合并框架的使用,也分析框架的成员构成以及任务提交和创建工作的原理实现,在本篇则会对框架的任务执行、任务扫描、线程挂起、结果合并以及任务窃取的源码实现进行分析。
|
4月前
|
存储 监控 Java
(十一)彻悟并发之JUC分治思想产物-ForkJoin分支合并框架原理剖析上篇
在上篇文章《深入理解并发之Java线程池、工作原理、复用原理及源码分析》中,曾详细谈到了Java的线程池框架。在其中也说到了JDK提供的四种原生线程池以及自定义线程池,而本文则再来详细谈谈JDK1.7中新推出的线程池:ForkJoinPool。
|
3月前
|
Java 调度
【多线程面试题十四】、说一说synchronized的底层实现原理
这篇文章解释了Java中的`synchronized`关键字的底层实现原理,包括它在代码块和方法同步中的实现方式,以及通过`monitorenter`和`monitorexit`指令以及`ACC_SYNCHRONIZED`访问标志来控制线程同步和锁的获取与释放。
|
6月前
|
存储 安全 Java
Java多线程基础-9:代码案例之阻塞队列(二)
Java多线程基础教程系列中,介绍了如何实现一个简单的阻塞队列(非泛型版本)。
55 0
|
6月前
|
消息中间件 存储 负载均衡
Java多线程基础-9:代码案例之阻塞队列(一)
阻塞队列是一种遵循先进先出原则的线程安全数据结构,它在队列满时会阻塞入队操作,队列空时会阻塞出队操作,常用于多线程间的协作,简化同步代码编写。Java中提供了`BlockingQueue`接口及其实现类,如`ArrayBlockingQueue`和`LinkedBlockingQueue`,用于实现生产者-消费者模型,以实现负载均衡和资源的有效利用,如削峰填谷,降低系统压力。
76 0
|
6月前
|
安全 Java
Java并发编程—并发流程控制与AQS原理及相关源码解析
Java并发编程—并发流程控制与AQS原理及相关源码解析
76 0
|
6月前
|
Java
剑指JUC原理-12.手写简易版线程池思路
剑指JUC原理-12.手写简易版线程池思路
41 0
|
Java
高并发编程-自定义简易的线程池(2),体会原理
高并发编程-自定义简易的线程池(2),体会原理
65 0