【多线程案例】定时器

1. 定时器是什么?

定时器也是软件开发中的一个重要组件. 类似于一个 "闹钟". 达到一个设定的时间之后, 就执行某个指定好的代码.

定时器是一种实际开发中非常常用的组件. 比如网络通信中, 如果对方 500ms 内没有返回数据, 则断开连接尝试重连. 比如一个 Map, 希望里面的某个 key 在 3s 之后过期(自动删除). 类似于这样的场景就需要用到定时器.

2. 使用标准库中的定时器

  • 标准库中提供了一个 Timer 类(java.util包下面). Timer 类的核心方法为 schedule .
  • schedule 包含两个参数. 第一个参数指定即将要执行的任务代码, 第二个参数指定多长时间之后 执行 (单位为毫秒).
public static void main(String[] args) {
        Timer timer = new Timer();
        timer.schedule(new TimerTask() {
            @Override
            public void run() {
                System.out.println("hello");
            }
        }, 3000);
    }

其中TimerTask()就是一个实现了Runnable的抽象类:

【多线程案例】定时器_第1张图片

可以把它的作用看成是给定时器一个任务,而第二个参数就是指定多久时间后执行这个任务。

3. 手写代码实现定时器

思考一下定时器的构成需要哪些?

  • 一个带有优先级的阻塞式队列
  • 队列中的每一个元素都是一个“任务”对象
  • “任务”对象中包含两个属性,一个属性用于描述任务,也就是一个Runnable,另一个属性用来定义delay。如此一来对手元素就是最即将要执行的任务。
  • 同时需要有一个线程不停的扫描队首元素。看队首元素是否到了执行时间。

1)写一个任务类,任务类还必须能够按照时间来比大小,因为优先级阻塞队列需要比较大小

//任务类 描述任务和任务的delay时间
    static class Task implements Comparable{
        //任务
        private Runnable command;
        //delay
        private long time;

        public Task(Runnable command,long time){
            this.command = command;
            //时间是在现在的时间的基础上加上delay
            this.time = System.currentTimeMillis() + time;
        }

        public void run(){
            command.run();
        }

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

 2)需要有一个优先级阻塞队列来存放用户注册的任务

//优先级阻塞队列 核心结构
    //队首存放的是最近要执行的任务 time最小
    private PriorityBlockingQueue queue = new PriorityBlockingQueue();

    public void schedule(Runnable command,long time){
        //生成一个任务 然后放进去优先级队列
        Task task = new Task(command, time);
        queue.put(task);
    }

3)在构造方法中整一个线程对队首元素扫描,看是否到了执行时间

public MyTimer(){
        //在构造方法中来一个扫面线程 一直扫描队首的元素是否到了执行时间
        Thread work = new Thread(() -> {
            while (true) {
                try {
                    Task task = queue.take();
                    long curTime = System.currentTimeMillis();
                    if (task.time > curTime) {
                        // 时间还没到, 就把任务再塞回去
                        queue.put(task);
                    } else {
                        // 时间到了, 可以执行任务
                        task.run();
                    }
                } catch (InterruptedException e) {
                    e.printStackTrace();
                    break;
                }
            }
        });
        work.start();
    }

用写一个测试:

    //测试代码
    public static void main(String[] args) {
        MyTimer myTimer = new MyTimer();
        myTimer.schedule(new Runnable() {
            @Override
            public void run() {
                System.out.println("我有第一个任务!");
            }
        },3000);

        myTimer.schedule(new Runnable() {
            @Override
            public void run() {
                System.out.println("我有第二个任务!");
            }
        },1000);

        myTimer.schedule(new Runnable() {
            @Override
            public void run() {
                System.out.println("我有第三个任务!");
            }
        },2000);
    }

此时已经可以按照定时器的工作原理来完成任务了:

【多线程案例】定时器_第2张图片

 但是当前的代码还存在着比较严重的问题,就是在3)中如果时间没有到的话会存在cpu一直比较的情况。举个例子,比如小明九点上班,他七点在床上突然醒了。正常情况下应该是继续睡睡到平时订的闹钟时间,但是如果小明一直看表一直看表知道闹铃响起,这样既没有休息也没有做有意义的事情,是十分愚蠢的行为。代码的问题也就在于此,如果没有到执行时间,不管还有多久还都会一直比较有没有到执行时间是没有意义的,也就是处于”忙等“状态。

优化的话,应该让系统在看到当前队首任务还没有到达执行时间的时候就执行wait(时间差)。但是此时还存在另外一个问题,系统wait一段时候之后确实会执行队首的任务,但是如果在wait的时间中又来了新的任务并且新的任务重新处于了队首,此时就会出bug了。正确的做法是在每次有新的任务被注册的时候都通知一下结束wait。

修改代码:

1.引入一个lock对象,借助该对象的wait/notify来解决忙等状态

private Object lock = new Object();

2.修改构造方法中的work的工作方法

public MyTimer(){
        //在构造方法中来一个扫面线程 一直扫描队首的元素是否到了执行时间
        Thread work = new Thread(() -> {
            while (true) {
                try {
                    Task task = queue.take();
                    long curTime = System.currentTimeMillis();
                    if (task.time > curTime) {
                        // 时间还没到, 就把任务再塞回去
                        queue.put(task);
                        // 等待一段时间
                        synchronized (lock){
                            lock.wait(task.time - curTime);
                        }
                    } else {
                        // 时间到了, 可以执行任务
                        task.run();
                    }
                } catch (InterruptedException e) {
                    e.printStackTrace();
                    break;
                }
            }
        });
        work.start();
    }

3. 修改 Timer 的 schedule 方法, 每次有新任务到来的时候唤醒一下 worker 线程. (因为新插入的任务可能是需要马上执行的).

public void schedule(Runnable command,long time){
        //生成一个任务 然后放进去优先级队列
        Task task = new Task(command, time);
        queue.put(task);

        //有新任务来了 唤醒work 检测是否有更新的工作需要执行
        synchronized (lock){
            lock.notify();
        }
    }

完整代码:

public class MyTimer {
    //任务类 描述任务和任务的delay时间
    static class Task implements Comparable{
        //任务
        private Runnable command;
        //delay
        private long time;

        public Task(Runnable command,long time){
            this.command = command;
            //时间是在现在的时间的基础上加上delay
            this.time = System.currentTimeMillis() + time;
        }

        public void run(){
            command.run();
        }

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

    //优先级阻塞队列 核心结构
    //队首存放的是最近要执行的任务 time最小
    private PriorityBlockingQueue queue = new PriorityBlockingQueue();

    public void schedule(Runnable command,long time){
        //生成一个任务 然后放进去优先级队列
        Task task = new Task(command, time);
        queue.put(task);

        //有新任务来了 唤醒work 检测是否有更新的工作需要执行
        synchronized (lock){
            lock.notify();
        }
    }

    private Object lock = new Object();

    public MyTimer(){
        //在构造方法中来一个扫面线程 一直扫描队首的元素是否到了执行时间
        Thread work = new Thread(() -> {
            while (true) {
                try {
                    Task task = queue.take();
                    long curTime = System.currentTimeMillis();
                    if (task.time > curTime) {
                        // 时间还没到, 就把任务再塞回去
                        queue.put(task);
                        // 等待一段时间
                        synchronized (lock){
                            lock.wait(task.time - curTime);
                        }
                    } else {
                        // 时间到了, 可以执行任务
                        task.run();
                    }
                } catch (InterruptedException e) {
                    e.printStackTrace();
                    break;
                }
            }
        });
        work.start();
    }

    //测试代码
    public static void main(String[] args) {
        MyTimer myTimer = new MyTimer();
        myTimer.schedule(new Runnable() {
            @Override
            public void run() {
                System.out.println("我有第一个任务!");
            }
        },3000);

        myTimer.schedule(new Runnable() {
            @Override
            public void run() {
                System.out.println("我有第二个任务!");
            }
        },1000);

        myTimer.schedule(new Runnable() {
            @Override
            public void run() {
                System.out.println("我有第三个任务!");
            }
        },2000);
    }
}

此时代码还有问题吗????

理论上说说代码中还是有一点小问题的。(烧脑啊.....)上图:

【多线程案例】定时器_第3张图片

了解了上述问题之后,就不难发现,问题出现的原因,是因为当前 take 操作,和 wait 操作,并非是原子的如果在 take 和 wait 之间加上锁,保证在这个过程中,不会有新的任务过来,问题自然解决(换句话说只要保证每次 notify 时确实都正在 wait )

【多线程案例】定时器_第4张图片

你可能感兴趣的:(Java,java,开发语言)