线程池原理及源码解读

线程池的优点

  1. 线程是稀缺资源,使用线程池可以减少创建和销毁线程的次数,每个工作线程可以重复使用。
  2. 可以根据系统的承受能力,调整线程池中工作线程的数量,防止因为消耗过多内容导致服务器崩溃。

线程池的创建

public ThreadPoolExecutor(int corePoolSize,
                          int maximumPoolSize,
                          long keepAliveTime,
                          TimeUnit unit,
                          BlockingQueue<Runnable> workQueue,
                          RejectedExecutionHandler handler)

corePoolSize:线程池核心线程的数量

maximumPoolSize:线程池最大线程个数

keepAliveTime:当活跃线程大于核心线程数时,空闲的多余线程最大存活时间

unit:存活时间的单位

workQueue:存放任务的队列

handler:超过线程范围和队列容量的任务的处理程序

线程池的实现原理

提交一个任务到线程池中,线程池的处理流程如下:

  1. 判断线程池里的核心线程是否都在执行任务,如果不是(核心线程空闲或核心线程还没有被创建),则创建一个新的工作线程来执行任务,如果核心线程都在执行任务,则进入2;
  2. 线程池判断工作队列是否已满,如果没有满,将新任务存储到这个工作队列中。如果工作队列满了,则进入3;
  3. 判断线程池里的线程是否都处于工作状态,如果没有,则创建一个新的工作线程来执行任务。如果已经满了,则交给饱和策略来处理这个任务。

线程池源码解读

1、ThreadPoolExecutor的 execute()方法

public void execute(Runnable command) {
        if (command == null)
            throw new NullPointerException();       
        //如果线程数大于等于核心线程数或者线程创建失败,将任务加入队列
        if (poolSize >= corePoolSize || !addIfUnderCorePoolSize(command)){       //线程池处于运行状态并且加入队列成功
            if (runState == RUNNING && workQueue.offer(command)) {
                if (runState != RUNNING || poolSize == 0)
                    ensureQueuedTaskHandled(command);
            }         
            //线程池不处于运行状态或者加入队列失败,则创建线程(创建的是非核心线程)
            else if (!addIfUnderMaximumPoolSize(command))           
            //创建线程失败,则采取阻塞处理的方式
            reject(command); // is shutdown or saturated
        }
    }

2、创建线程的方法:addIfUnderCorePoolSize(commod)

private boolean addIfUnderCorePoolSize(Runnable firstTask) {
        Thread t = null;
        final ReentrantLock mainLock = this.mainLock;
        mainLock.lock();
        try {
            if (poolSize < corePoolSize && runState == RUNNING)
                t = addThread(firstTask);
        } finally {
            mainLock.unlock();
        }
        if (t == null)
            return false;
        t.start();
        return true;
    }

重点关注addThread(firstTask)

private Thread addThread(Runnable firstTask) {
    	// 线程封装成工作线程
        Worker w = new Worker(firstTask);
        Thread t = threadFactory.newThread(w);
        if (t != null) {
            w.thread = t;
            // 放入工作线程组
            workers.add(w);
            int nt = ++poolSize;
            if (nt > largestPoolSize)
                largestPoolSize = nt;
        }
        return t;
    }

worker类的run方法:

public void run() {
            try {
                Runnable task = firstTask;
                firstTask = null;
                // getTask()循环获取工作队列里的任务,非空情况下,继续执行
                while (task != null || (task = getTask()) != null) {
                    runTask(task);
                    task = null;
                }
            } finally {
                workerDone(this);
            }
        }

worker在执行完任务后,还会通过getTask方法循环获取工作队列里的任务来执行。

示例(观察线程池的工作原理)

  1. 创建一个线程

    public class ThreadPoolTest implements Runnable
    {
        @Override
        public void run()
        {
            try
            {
                Thread.sleep(300);
            }
            catch (InterruptedException e)
            {
                e.printStackTrace();
            }
        }
    }
    
  2. 线程池循环运行16个线程

    public static void main(String[] args)
        {
            LinkedBlockingQueue<Runnable> queue =
                new LinkedBlockingQueue<Runnable>(5);
        	// 创建线程池
            ThreadPoolExecutor threadPool = new ThreadPoolExecutor(5, 10, 60, TimeUnit.SECONDS, queue);
            for (int i = 0; i < 16 ; i++)
            {
                threadPool.execute(
                    new Thread(new ThreadPoolTest(), "Thread".concat(i + "")));
                System.out.println("线程池中活跃的线程数: " + threadPool.getPoolSize());
                if (queue.size() > 0)
                {
                    System.out.println("----------------队列中阻塞的线程数" + queue.size());
                }
            }
            threadPool.shutdown();
        }
    

    执行结果:

    线程池中活跃的线程数: 1
    线程池中活跃的线程数: 2
    线程池中活跃的线程数: 3
    线程池中活跃的线程数: 4
    线程池中活跃的线程数: 5
    线程池中活跃的线程数: 5
    ----------------队列中阻塞的线程数1
    线程池中活跃的线程数: 5
    ----------------队列中阻塞的线程数2
    线程池中活跃的线程数: 5
    ----------------队列中阻塞的线程数3
    线程池中活跃的线程数: 5
    ----------------队列中阻塞的线程数4
    线程池中活跃的线程数: 5
    ----------------队列中阻塞的线程数5
    线程池中活跃的线程数: 6
    ----------------队列中阻塞的线程数5
    线程池中活跃的线程数: 7
    ----------------队列中阻塞的线程数5
    线程池中活跃的线程数: 8
    ----------------队列中阻塞的线程数5
    线程池中活跃的线程数: 9
    ----------------队列中阻塞的线程数5
    线程池中活跃的线程数: 10
    ----------------队列中阻塞的线程数5
    Exception in thread "main" java.util.concurrent.RejectedExecutionException: Task Thread[Thread15,5,main] rejected from java.util.concurrent.ThreadPoolExecutor@232204a1[Running, pool size = 10, active threads = 10, queued tasks = 5, completed tasks = 0]
        at java.util.concurrent.ThreadPoolExecutor$AbortPolicy.rejectedExecution(ThreadPoolExecutor.java:2047)
        at java.util.concurrent.ThreadPoolExecutor.reject(ThreadPoolExecutor.java:823)
        at java.util.concurrent.ThreadPoolExecutor.execute(ThreadPoolExecutor.java:1369)
        at test.ThreadTest.main(ThreadTest.java:17)
    

从结果可以观察到:

  1. 创建的线程池具体配置为:核心线程数为5个,全部线程数为10个,工作队列长度为5;

  2. queue.size()可以获取到工作队列中的任务数;

  3. 运行原理解读:

    刚开始都是在创建新的线程,达到核心线程数5个后,新的任务进来后不再创建新的线程,而是将任务加入到工作队列。当任务队列达到上限5个后,新的任务又会创建新的普通线程,直到达到线程池最大的线程数量10个。再后面的任务则根据配置的饱和策略来处理,这里没有具体配置,因此使用的是默认的策略AbortPolicy:直接抛出异常。

    这里线程都利用休眠导致线程资源没有被释放掉!

RejectedExecutionHandler:饱和策略

当队列和线程池都满了,说明线程池处于饱和状态,那么必须对新提交的任务采用一种特殊的策略来进行处理。这个策略默认配置是AbortPolicy,表示无法处理新的任务而抛出异常。

Java提供了4种策略:

  1. AbortPolicy:直接抛出异常
  2. CallerRunsPolicy:只用调用所在的线程运行任务
  3. DiscardOldestPolicy:丢弃队列里最近的一个任务,并执行当前任务
  4. DiscardPolicy:不处理,丢弃掉

我们用第四种策略来作为示例:

public static void main(String[] args)
    {
        LinkedBlockingQueue<Runnable> queue =
            new LinkedBlockingQueue<Runnable>(3);
    	// 拒绝策略
        RejectedExecutionHandler handler = new ThreadPoolExecutor.DiscardPolicy();
		// 初始化线程池
        ThreadPoolExecutor threadPool = new ThreadPoolExecutor(2, 5, 60, TimeUnit.SECONDS, queue,handler);
        for (int i = 0; i < 9 ; i++)
        {
            threadPool.execute(
                new Thread(new ThreadPoolTest(), "Thread".concat(i + "")));
            System.out.println("线程池中活跃的线程数: " + threadPool.getPoolSize());
            if (queue.size() > 0)
            {
                System.out.println("----------------队列中阻塞的线程数" + queue.size());
            }
        }
        threadPool.shutdown();
    }

执行结果:

线程池中活跃的线程数: 1
线程池中活跃的线程数: 2
线程池中活跃的线程数: 2
----------------队列中阻塞的线程数1
线程池中活跃的线程数: 2
----------------队列中阻塞的线程数2
线程池中活跃的线程数: 2
----------------队列中阻塞的线程数3
线程池中活跃的线程数: 3
----------------队列中阻塞的线程数3
线程池中活跃的线程数: 4
----------------队列中阻塞的线程数3
线程池中活跃的线程数: 5
----------------队列中阻塞的线程数3
线程池中活跃的线程数: 5
----------------队列中阻塞的线程数3

这里采用了丢弃策略后,就没有抛出异常,而是直接丢弃。但是在某些场景下,可以采用记录日志或存储到数据库中,而不应该直接丢弃。

设置策略有两种方式:

// 第一种
RejectedExecutionHandler handler = new ThreadPoolExecutor.DiscardPolicy();
ThreadPoolExecutor threadPool = new ThreadPoolExecutor(2, 5, 60, TimeUnit.SECONDS, queue,handler);

// 第二种
ThreadPoolExecutor threadPool = new ThreadPoolExecutor(2, 5, 60, TimeUnit.SECONDS, queue);
threadPool.setRejectedExecutionHandler(new ThreadPoolExecutor.AbortPolicy());

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值