自定义线程池

1、自定义线程池的实现

自定义线程池应包括 线程池 + 阻塞队列
在这里插入图片描述

实现

/**
 * @author houChen
 * @date 2022/6/14 21:36
 * @Description: 自定义线程池
 */
@Slf4j(topic = "c.main")
public class TestPool {
    public static void main(String[] args) {
        ThreadPool threadPool = new ThreadPool(2, 1000, TimeUnit.MILLISECONDS, 10);
        for (int i = 0; i < 5; i++) {
            int j = i;
            threadPool.execute(() -> {
                log.debug("{}", j);
            });
        }
    }
}

//线程池
@Slf4j(topic = "c.testPool")
class ThreadPool {
    //阻塞队列
    private BlockingQueue<Runnable> taskQueue;
    //线程集合
    private HashSet<Worker> workers = new HashSet();
    //核心线程数
    private int coreSize;
    //超时时间
    private long timeOut;

    private TimeUnit timeUnit;

    //执行任务
    public void execute(Runnable task) {
        synchronized (workers) {
            //【注意】 当任务个数大于核心线程数时,先向任务队列push, 当任务队列满了,再创建max - core的线程数,最后再是任务队列的丢弃策略
            //如果任务数小于核心线程数,直接交给worker对象执行
            //当任务数超过核心线程数,加入任务队列暂存
            if (workers.size() < coreSize) {
                Worker worker = new Worker(task);
                log.debug("新增worker{},task{}", worker, task);
                workers.add(worker);
                worker.start();
            } else {
                log.debug("加入任务队列{}", task);
                taskQueue.push(task);
            }
        }
    }

    //构造方法
    public ThreadPool(int coreSize, long timeOut, TimeUnit timeUnit, int capcity) {
        this.coreSize = coreSize;
        this.timeOut = timeOut;
        this.timeUnit = timeUnit;
        this.taskQueue = new BlockingQueue<>(capcity);
    }

    class Worker extends Thread {

        private Runnable task;

        //构造器
        public Worker(Runnable task) {
            this.task = task;
        }

        @Override
        public void run() {
            //执行任务
            //1)task不为空时,执行任务
            //2)task为空时,从任务队列取出任务并执行
            if (task != null || (task = taskQueue.take()) != null) {
                try {
                    log.debug("正在执行...{}", task);
                    this.task.run();
                } catch (Exception e) {
                    e.printStackTrace();
                } finally {
                    task = null;
                }
            }

            synchronized (workers) {
                log.debug("worker{} 被移除",this);
                workers.remove(this);
            }
        }
    }
}

//阻塞队列
class BlockingQueue<T> {
    //1.任务队列
    private Deque<T> queue;

    //2.锁 当多个线程来取队列中的任务时,保证其互斥性
    private ReentrantLock lock = new ReentrantLock();

    //3.生产者条件变量
    private Condition fullWaitSet = lock.newCondition();

    //4.消费者条件变量
    private Condition emptyWaitSet = lock.newCondition();

    //5.容量
    private int capcity;

    public BlockingQueue(int capcity) {
        this.queue = new ArrayDeque<>(capcity);
    }


    //阻塞获取 (从队列中获取一个任务)
    public T take() {
        lock.lock();
        try {
            while (queue.isEmpty()) { //当队列为空时,阻塞
                try {
                    emptyWaitSet.wait();
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
            T t = queue.removeFirst();
            fullWaitSet.signal();
            return t;
        } finally {
            lock.unlock();
        }
    }

    //超时阻塞获取
    public T poll(long timeout, TimeUnit unit) {
        lock.lock();
        try {
            //将timeout统一转换成纳秒
            long nanos = unit.toNanos(timeout);
            while (queue.isEmpty()) { //当队列为空时,阻塞
                try {
                    if (nanos < 0) {
                        return null;
                    }
                    //返回值是剩余等待时间
                    nanos = emptyWaitSet.awaitNanos(nanos);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
            T t = queue.removeFirst();
            fullWaitSet.signal();
            return t;
        } finally {
            lock.unlock();
        }
    }

    //阻塞添加
    public void push(T element) {
        lock.lock();
        try {
            while (queue.size() == capcity) { //当队列为空时,阻塞
                try {
                    fullWaitSet.wait();
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
            queue.addLast(element);
            emptyWaitSet.signal();
        } finally {
            lock.unlock();
        }
    }
    
    public void push(T element) {
        
    }


    //获取队列的大小
    public int size() {
        lock.lock();
        try {
            return queue.size();
        } finally {
            lock.unlock();
        }
    }
}

2、当任务数大于 core + 任务队列的长度时

当线程池需要执行的任务数 > 核心线程数 + 任务队列的容量时, 剩余的 任务应该怎么处理,由线程池的拒绝策略决定

3、任务队列push方法增强 - 带超时时间的阻塞添加

//带超时时间的阻塞添加
public boolean pushP(T element,long timeOut,TimeUnit timeUnit) {
    lock.lock();
    try {
        //将timeout统一转换成纳秒
        long nanos = timeUnit.toNanos(timeOut);
        while (queue.size() == capcity) { //当队列处于满
            try {
                if (nanos < 0) {
                    return false;
                }
                //返回值是剩余等待时间
                nanos = fullWaitSet.awaitNanos(nanos);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
        log.debug("加入任务队列{}", element);
        queue.addLast(element);
        emptyWaitSet.signal();
        return true;
    } finally {
        lock.unlock();
    }
}

4、自定义线程池拒绝策略

1、定义一个函数式接口(也就是一个拒绝策略)

//线程池的拒绝策略
@FunctionalInterface
interface RejectPolicy<T> {
	// 需要传入阻塞队列是因为拒绝这个行为是发生在队列上的
	// 需要传入一个t,是因为阻塞队列对t进行处理
    void reject(BlockingQueue<T> queue,T t);
}

2、线程池注入该策略

class ThreadPool {
    //阻塞队列
    private BlockingQueue<Runnable> taskQueue;
    //线程集合
    private HashSet<Worker> workers = new HashSet();
    //核心线程数
    private int coreSize;
    //超时时间
    private long timeOut;

    private TimeUnit timeUnit;

    //线程池的拒绝策略
    private RejectPolicy<Runnable> rejectPilicy;

    //执行任务
    public void execute(Runnable task) {
        synchronized (workers) {
            //【注意】 当任务个数大于核心线程数时,先向任务队列push, 当任务队列满了,再创建max - core的线程数,最后再是任务队列的丢弃策略
            //如果任务数小于核心线程数,直接交给worker对象执行
            //当任务数超过核心线程数,加入任务队列暂存
            if (workers.size() < coreSize) {
                Worker worker = new Worker(task);
                log.debug("新增worker{},task{}", worker, task);
                workers.add(worker);
                worker.start();
            } else {
                //taskQueue.push(task);
                //当任务队列满了,有以下几种方式
                //1)死等
                //2)超时等待 (加入任务队列超时,就放弃该任务)
                //3)让调用者放弃任务执行
                //4)让调用者抛出异常

                //让任务队列来决定怎么处理该任务
                taskQueue.tryPut(rejectPilicy,task);
            }
        }
    }

    //构造方法
    public ThreadPool(int coreSize, long timeOut, TimeUnit timeUnit, int capcity, RejectPolicy<Runnable> rejectPilicy) {
        this.coreSize = coreSize;
        this.timeOut = timeOut;
        this.timeUnit = timeUnit;
        this.taskQueue = new BlockingQueue<>(capcity);
        this.rejectPilicy = rejectPilicy;
    }

    class Worker extends Thread {

        private Runnable task;

        //构造器
        public Worker(Runnable task) {
            this.task = task;
        }

        @Override
        public void run() {
            //执行任务
            //1)task不为空时,执行任务
            //2)task为空时,从任务队列取出任务并执行
            if (task != null || (task = taskQueue.take()) != null) {
                try {
                    log.debug("正在执行...{}", task);
                    this.task.run();
                } catch (Exception e) {
                    e.printStackTrace();
                } finally {
                    task = null;
                }
            }

            synchronized (workers) {
                log.debug("worker{} 被移除",this);
                workers.remove(this);
            }
        }
    }
}

3、在阻塞队列的方法 tryPut中,根据传入的策略对 task进行处理

class BlockingQueue<T> {
    //1.任务队列
    private Deque<T> queue;

    //2.锁 当多个线程来取队列中的任务时,保证其互斥性
    private ReentrantLock lock = new ReentrantLock();

    //3.生产者条件变量
    private Condition fullWaitSet = lock.newCondition();

    //4.消费者条件变量
    private Condition emptyWaitSet = lock.newCondition();

    //5.容量
    private int capcity;

    public BlockingQueue(int capcity) {
        this.queue = new ArrayDeque<>(capcity);
    }


    //阻塞获取 (从队列中获取一个任务)
    public T take() {
        lock.lock();
        try {
            while (queue.isEmpty()) { //当队列为空时,阻塞
                try {
                    emptyWaitSet.wait();
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
            T t = queue.removeFirst();
            fullWaitSet.signal();
            return t;
        } finally {
            lock.unlock();
        }
    }

    //超时阻塞获取
    public T poll(long timeout, TimeUnit unit) {
        lock.lock();
        try {
            //将timeout统一转换成纳秒
            long nanos = unit.toNanos(timeout);
            while (queue.isEmpty()) { //当队列为空时,阻塞
                try {
                    if (nanos < 0) {
                        return null;
                    }
                    //返回值是剩余等待时间
                    nanos = emptyWaitSet.awaitNanos(nanos);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
            T t = queue.removeFirst();
            fullWaitSet.signal();
            return t;
        } finally {
            lock.unlock();
        }
    }

    //阻塞添加
    public void push(T element) {
        lock.lock();
        try {
            while (queue.size() == capcity) { //当队列为空时,阻塞
                try {
                    fullWaitSet.wait();
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
            log.debug("加入任务队列{}", element);
            queue.addLast(element);
            emptyWaitSet.signal();
        } finally {
            lock.unlock();
        }
    }

    //带超时时间的阻塞添加
    public boolean pushP(T element,long timeOut,TimeUnit timeUnit) {
        lock.lock();
        try {
            //将timeout统一转换成纳秒
            long nanos = timeUnit.toNanos(timeOut);
            while (queue.size() == capcity) { //当队列处于满
                try {
                    if (nanos < 0) {
                        return false;
                    }
                    //返回值是剩余等待时间
                    nanos = fullWaitSet.awaitNanos(nanos);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
            log.debug("加入任务队列{}", element);
            queue.addLast(element);
            emptyWaitSet.signal();
            return true;
        } finally {
            lock.unlock();
        }
    }


    //获取队列的大小
    public int size() {
        lock.lock();
        try {
            return queue.size();
        } finally {
            lock.unlock();
        }
    }

    //线程池任务队列的拒绝策略
    public void tryPut(RejectPolicy<T> rejectPilicy, T task) {
        lock.lock();

        try {
            if(queue.size() == capcity) {   //阻塞队列满了,就使用策略来处理任务
                rejectPilicy.reject(this, task);

            } else {  // 阻塞队列没满,则加入队列
                log.debug("加入任务队列{}", task);
                queue.addLast(task);
                emptyWaitSet.signal();
            }
        } finally {
            lock.unlock();
        }

    }
}

4、测试时,构造线程池时,需要传入策略的具体实现

public class TestPool {
    public static void main(String[] args) {
        ThreadPool threadPool = new ThreadPool(2, 1000, TimeUnit.MILLISECONDS, 10, (queue,task) -> {
            queue.push(task);
        });
        for (int i = 0; i < 5; i++) {
            int j = i;
            threadPool.execute(() -> {
                log.debug("{}", j);
            });
        }
    }
}
  • 2
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
提供的源码资源涵盖了安卓应用、小程序、Python应用和Java应用等多个领域,每个领域都包含了丰富的实例和项目。这些源码都是基于各自平台的最新技术和标准编写,确保了在对应环境下能够无缝运行。同时,源码中配备了详细的注释和文档,帮助用户快速理解代码结构和实现逻辑。 适用人群: 这些源码资源特别适合大学生群体。无论你是计算机相关专业的学生,还是对其他领域编程感兴趣的学生,这些资源都能为你提供宝贵的学习和实践机会。通过学习和运行这些源码,你可以掌握各平台开发的基础知识,提升编程能力和项目实战经验。 使用场景及目标: 在学习阶段,你可以利用这些源码资源进行课程实践、课外项目或毕业设计。通过分析和运行源码,你将深入了解各平台开发的技术细节和最佳实践,逐步培养起自己的项目开发和问题解决能力。此外,在求职或创业过程中,具备跨平台开发能力的大学生将更具竞争力。 其他说明: 为了确保源码资源的可运行性和易用性,特别注意了以下几点:首先,每份源码都提供了详细的运行环境和依赖说明,确保用户能够轻松搭建起开发环境;其次,源码中的注释和文档都非常完善,方便用户快速上手和理解代码;最后,我会定期更新这些源码资源,以适应各平台技术的最新发展和市场需求。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值