Java定时器简单实现

  • Post author:
  • Post category:java




一、定时器

定时器相当于一个任务管理器。有些任务可能现在执行, 有些任务可能过1个小时,甚至很久才会执行。定时器就是对这些任务进行管理监视, 如果一个任务执行时间到了,定时器就会将这个任务执行。 保证所有的任务都会在合适的时间执行。



二、定时器的实现


对于定时器的实现,我们可以划分为3个部分。

1、 使用一个Task类描述每一个任务(里面包含任务的执行方法, 定时时间)。

2、 使用优先级队列管理这些任务类。

2.1 我们都知道优先级队列底层实现是堆(以小根堆为例), 堆顶的元素是所有的元素的最小值。 我们以任务的定时时间为比较原则构建, 这样就可以保证堆顶元素的任务执行时间是最短的(这样的实现,我们需要在Task类内部定义比较规则-即重写Comparable接口的CompareTo方法)。

2.2 当一个任务执行完毕, 就会从优先级队列取出poll掉, 然后内部重新组织保证新的堆顶元素是定时时间最短的。

2.3 如果说堆顶的任务定时时间还没有到达(当然后续的任务定时时间肯定会更长,不会被执行)

3、使用一个线程循环扫描优先级队列, 相当于一个监控线程,循环判断堆顶任务是否满足执行时间。



三、定时器的组成


1、制定任务类Task

Task类包含任务的 执行方法 和 定时时间。

1.1 执行方法我采用封装Runnable中run方法实现, 这样做是为了后续添加任务时方便写执行逻辑。

1.2 定时时间就是long类型的变量

1.3 制定比较规则, 后续优先级队列中存放的是Task对象(而在内部构建时,需要比较两个Task对象的),对于对象的比较, 我们以对象的定时时间为规则, 制定小根堆。

static class Task implements Comparable<Task>{
        //Runnable类中有一个run方法, 通过这个方法实现任务的执行
        private Runnable command;
        //time表示执行的时间
        private long time;

        //构造方法
        public Task(Runnable command, long time) {
            this.command = command;
            this.time = System.currentTimeMillis() + time;  //将时间转化为绝对时间
        }

        //执行任务的逻辑
        public void run() {
            command.run();
        }

        //定义比较方法 - 方便后续的优先级队列构建

        @Override
        public int compareTo(Task o) {
            return (int)(this.time - o.time);
        }
    }


2、监管线程&定时器对象Timer

监管线程Worker中包含优先级队列(小根堆)queue 和 循环监管的流程。

Timer对象封装了监管线程Woker 和 任务的添加方法schedule()

关于监管线程的优化

2.1 循环监控存在一个弊端,那就是一直循环判断, 占用CPU资源。

(假如堆首任务的执行是1小时后, 再次期间监管线程会跑1小时循环判断。)

解决方法: 可以通过线程阻塞和唤醒来解决。在下面代码有详细注释和实现。

2.1.1 如果任务1小时后执行, 我们让监管线程wait(1小时), 但在此期间如果有新的任务添加进来(可能新的任务需要等30分钟就可以执行,堆首元素发生变化) ,这时需要唤醒监管线程来重新判断。(由于wait和notify方法不在用一个类中实现, 我们通过一个Object(mailBox)来阻塞、唤醒)

//检测线程, 继承Thread类,重写内部run方法,属于线程的创建方法之一。
    static class Worker extends Thread {
    	//优先级队列 - JUC包里面
        private PriorityBlockingQueue<Task> queue = null;
        //为了对监管线程进行阻塞和唤醒,采用同一对象
        private Object mailBox = null;
		
		//构造函数
        public Worker(PriorityBlockingQueue<Task> queue, Object mailBox) {
            this.queue = queue;
            this.mailBox = mailBox;
        }

        @Override
        public void run() {
            //实现具体的执行逻辑
            while(true) {
                try {
                    //1、取优先级队列的队首元素
                    Task task = queue.peek();
                    //2、比较队首的元素的时间是否大于当前时间
                    if(task == null) {
                        continue;
                    }
                    long curTime = System.currentTimeMillis();
                    if(task.time > curTime) {
                        //时间还没有到, 由于取出了任务, 需要重新放置回去
                        //优化1: 空循环等待 - wait(time) 让线程休眠time时间,然后在执行
                        //       如果在等待期间有新的任务添加, 这个时候我们唤醒线程, 继续判断(因为存在新的时间过短需要立即执行)
                        //       这个只需要添加一个新任务时, 唤醒即可
                        //优化2: 访问队首元素而不是取出, 防止无所谓的删除、插入。(维护优先级队列是有消耗的)
                        long gapTime = task.time - curTime;
                        synchronized (mailBox) {
                            mailBox.wait(gapTime);
                        }
                    }
                    else {
                        //直接执行
                        //如果执行到了, 则会删除头部元素, 调用任务的执行过程。
                        task = queue.take();
                        task.run();
                    }
                }
                catch(InterruptedException e) {
                    e.printStackTrace();
                    break;
                }
            }
        }
    }

    //定时器简单实现
    static class Timer {
        //定时器的实现步骤
        //1、用一个类描述任务
        //2、用优先级队列管理这些任务, 比较方法通过任务的制定时间,每次取队首元素
        //   队首元素是执行时间最近的
        private PriorityBlockingQueue<Task> queue = new PriorityBlockingQueue<>();
        //3、用一个线程来循环扫描当前的阻塞队列,判断队首的执行时间, 如果执行时间到了,那就执行。

        //4、创建一个Object对象,用于设置线程阻塞使用的, 存在线程阻塞, 添加任务时唤醒的操作
        private Object mailBox = new Object();

        //构造函数
        public Timer() {
            //创建线程
            Worker worker = new Worker(queue, mailBox);
            worker.start();
        }


        //4、提供一个方法, 让调用者能够把任务安排起来
        public void schedule(Runnable command, long time) {
            Task task = new Task(command, time);
            queue.put(task);
            synchronized (mailBox) {
                mailBox.notify();
            }
        }
    }


3、测试代码

其中添加了4个任务, 分别是2s、5s、7s、10s后执行。

    public static void main(String[] args) {
        Timer timer = new Timer();
        timer.schedule(new Runnable() {
            @Override
            public void run() {
                System.out.println("郝梦武一号任务执行, 执行代号:闪电;  定时时间:2s");
            }
        }, 2000);

        timer.schedule(new Runnable() {
            @Override
            public void run() {
                System.out.println("郝梦武二号任务执行, 执行代号:暴风;  定时时间:5s");
            }
        }, 5000);

        timer.schedule(new Runnable() {
            @Override
            public void run() {
                System.out.println("郝梦武三号任务执行, 执行代号:狂风;  定时时间:7s");
            }
        }, 7000);

        timer.schedule(new Runnable() {
            @Override
            public void run() {
                System.out.println("郝梦武三号任务执行, 执行代号:地震;  定时时间:10s");
            }
        }, 10000);
    }


4、测试结果


在这里插入图片描述



版权声明:本文为weixin_44024891原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。