zoukankan      html  css  js  c++  java
  • Java的优先级任务队列的实践

    队列的基本理解

    在说队列之前说两个名词:Task是任务,TaskExecutor是任务执行器

    而我们今天要说的队列就完全符合某机构这个情况,队列在有Task进来的时候TaskExecutor就立刻开始执行Task,当没有Task的时候TaskExecutor就处于一个阻塞状态,当有很多Task的时候Task也需要排队,TaskExecutor也可以是多个,并且可以指定某几个Task优先执行或者滞后执行。

    综上所说我们得出一个这样的关系:队列相当于某机构TaskExecutor相当于窗口办事者就是Task

    普通队列

    当然很多机构也没有设置什么军人优先的窗口,所以队列也有不带优先级的队列,因此我们先来实现一个非优先级的队列。

    和上述某机构不一样,某机构可以先有机构,再有窗口,再有办事者。但是我们写代码的时候,要想写一个队列,那么务必要在队列中写TaskExecutor,那么就得先写好TaskExecutor类,以此类推就得先有Task类。

    因此我们先写一个Task的接口,也就是办事的人,我把它设计为接口,方便办各种不同事的人进来:

    // 办事的人。
    public interface ITask {
        // 办事,我们把办事的方法给办事的人,也就是你要办什么事,由你自己决定。
        void run();
    }
    

    接下来再写一个TaskExecutor的类,也就是窗口,用来执行Task,认真看注释,非常有助于理解:

    // 窗口
    public class TaskExecutor extends Thread {
    
        // 在窗口拍的队,这个队里面是办事的人。
        private BlockingQueue<ITask> taskQueue;
    
        // 这个办事窗口是否在等待着办事。
        private boolean isRunning = true;
    
        public TaskExecutor(BlockingQueue<ITask> taskQueue) {
            this.taskQueue = taskQueue;
        }
    
        // 下班。
        public void quit() {
            isRunning = false;
            interrupt();
        }
    
        @Override
        public void run() {
            while (isRunning) { // 如果是上班状态就待着。
                ITask iTask;
                try {
                    iTask = taskQueue.take(); // 叫下一个办事的人进来,没有人就等着。
                } catch (InterruptedException e) {
                    if (!isRunning) {
                        // 发生意外了,是下班状态的话就把窗口关闭。
                        interrupt();
                        break; // 如果执行到break,后面的代码就无效了。
                    }
                    // 发生意外了,不是下班状态,那么窗口继续等待。
                    continue;
                }
    
                // 为这个办事的人办事。
                iTask.run();
            }
        }
    }
    

    这里要稍微解释下BlockingQueue<T>#take()方法,这个方法当队列里面的item为空的时候,它会一直处于阻塞状态,当队列中进入item的时候它会立刻有一个返回值,它就和ServerSocket.accept()方法一样,所以我们把它放入一个Thread中,以免阻塞调用它的线程(Android中可能是主线程)。

    办事的人和窗口都有了,下面我们封装一个队列,也就是某机构,用来管理这些窗口:

    // 某机构。
    public class TaskQueue {
    
        // 某机构排的队,队里面是办事的人。
        private BlockingQueue<ITask> mTaskQueue;
        // 好多窗口。
        private TaskExecutor[] mTaskExecutors;
    
        // 在开发者new队列的时候,要指定窗口数量。
        public TaskQueue(int size) {
            mTaskQueue = new LinkedBlockingQueue<>();
            mTaskExecutors = new TaskExecutor[size];
        }
    
        // 开始上班。
        public void start() {
            stop();
            // 把各个窗口都打开,让窗口开始上班。
            for (int i = 0; i < mTaskExecutors.length; i++) {
                mTaskExecutors[i] = new TaskExecutor(mTaskQueue);
                mTaskExecutors[i].start();
            }
        }
    
        // 统一各个窗口下班。
        public void stop() {
            if (mTaskExecutors != null)
                for (TaskExecutor taskExecutor : mTaskExecutors) {
                    if (taskExecutor != null) taskExecutor.quit();
                }
        }
    
        // 开一个门,让办事的人能进来。
        public <T extends ITask> int add(T task) {
            if (!mTaskQueue.contains(task)) {
                mTaskQueue.add(task);
            }
            // 返回排的队的人数,公开透明,让外面的人看的有多少人在等着办事。
            return mTaskQueue.size();
        }
    }
    

    某机构、窗口、办事的人都有了,下面我们就派一个人去一件具体的事,但是上面我的Task是一个接口,所以我们需要用一个类来实现这个接口,来做某一件事:

    // 做一件打印自己的id的事。
    public class PrintTask implements ITask {
    
        private int id;
    
        public PrintTask(int id) {
            this.id = id;
        }
    
        @Override
        public void run() {
            // 为了尽量模拟窗口办事的速度,我们这里停顿两秒。
            try {
                Thread.sleep(2000);
            } catch (InterruptedException ignored) {
            }
    
            System.out.println("我的id是:" + id);
        }
    }
    

    下面就让我们模拟的虚拟世界运行一次:

    public class Main {
    
        public static void main(String... args) {
            // 这里暂时只开一个窗口。
            TaskQueue taskQueue = new TaskQueue(1);
            taskQueue.start();
    
            for (int i = 0; i < 10; i++) {
                PrintTask task = new PrintTask(i);
                taskQueue.add(task);
            }
        }
    
    }
    

    没错,队列按照我们理想的状况打印出来了:

    我的id是:0
    我的id是:1
    我的id是:2
    我的id是:3
    我的id是:4
    我的id是:5
    我的id是:6
    我的id是:7
    我的id是:8
    我的id是:9
    

    上面我门只开了一个窗口,下面我多开几个窗口:

    public class Main {
    
        public static void main(String... args) {
            // 开三个窗口。
            TaskQueue taskQueue = new TaskQueue(3);
            taskQueue.start(); // 某机构开始工作。
    
            for (int i = 0; i < 10; i++) {
                // new 10 个需要办事的人,并且进入某机构办事。
                PrintTask task = new PrintTask(i);
                taskQueue.add(task);
            }
        }
    }
    

    这里要说明一下,在初始化的时候我们开了3个窗口,内部的顺序应该是这样的:

    某机构的大门开了以后,第一个办事的人进去到了第一个窗口,第二个办事的人进去到了第二个窗口,第三个办事的人进去到了第三个窗口,第四个办事的人进去排队在第一位,当第一、第二、第三个窗口中不论哪一个窗口的事办完了,第四个人就去哪一个窗口继续办事,第五个人等待,一次类推。这样子就达到了队列同事并发三个任务的效果。

    这就是一个普通的队列,其它的一些特性也是基于此再次封装的,那么下面我就基于此再把人物的优先级加上,也就是我们上面说的特殊窗口->军人优先!

    优先级队列

    我们排队等待办事的时候,来了一个办事的人,那么如何判断这个办事人是否可以优先办理呢?那就要判断它是否具有优先的权限甚至他可以优先到什么程度。

    所以我们需要让这个Task有一标志,那就是优先级,所以我用一个枚举类标记优先级:

    public enum Priority {
        LOW, // 最低。
        DEFAULT, // 默认级别。
        HIGH, // 高于默认级别。
        Immediately // 立刻执行。
    }
    

    这里我把分了四个等级:最低默认立刻,这个等级肯定要给到我们的办事的人,也就是Task

    public interface ITask {
    
        void run();
    
        void setPriority(Priority priority);
    
        Priority getPriority();
    }
    

    可以设置优先级和可以拿到优先级。

    下面我们要把上面的LinkedBlockingQueue替换成PriorityBlockingQueue<E>,因为它可以自动做到优先级的比较,它要求泛型<E>,也就是我们的Task必须实现Comparable<E>接口,而Comparable<E>有一个compareTo(E)方法可以对两个<T>做比较,因此我们的队列需要改一下实现的方法:

    // 某机构。
    public class TaskQueue {
    
        // 某机构排的队,队里面是办事的人。
        private BlockingQueue<ITask> mTaskQueue;
        // 好多窗口。
        private TaskExecutor[] mTaskExecutors;
    
        // 在开发者new队列的时候,要指定窗口数量。
        public TaskQueue(int size) {
            mTaskQueue = new PriorityBlockingQueue<>();
            mTaskExecutors = new TaskExecutor[size];
        }
    
        ...
    

    然后ITask接口继承Comparable<E>接口:

    public interface ITask extends Comparable<ITask> {
    
        void run();
    
        void setPriority(Priority priority);
    
        Priority getPriority();
    }
    

    因为有setPriority(Priority)方法和getPriority()方法和Comparable<E>compareTo(E)方法,所以我们的每一个Task都需要实现这几个方法,这样就会很麻烦,所以我们封装一个BasicTask:

    public abstract class BasicTask implements ITask {
    
        // 默认优先级。
        private Priority priority = Priority.DEFAULT;
    
        @Override
        public void setPriority(Priority priority) {
            this.priority = priority;
        }
    
        @Override
        public Priority getPriority() {
            return priority;
        }
    
        // 做优先级比较。
        @Override
        public int compareTo(ITask another) {
            final Priority me = this.getPriority();
            final Priority it = another.getPriority();
            return me == it ? [...] : it.ordinal() - me.ordinal();
        }
    }
    

    其它都好说,我们看到compareTo(E)方法就不太理解了,这里说一下这个方法:

    compareTo(E)中传进来的E是另一个Task,如果当前Task比另一个Task更靠前就返回负数,如果比另一个Task靠后,那就返回正数,如果优先级相等,那就返回0。

    这里要特别注意,我们看到上面当两个Task优先级不一样的时候调用了Priority.orinal()方法,并有后面的orinal减去了当前的orinal,怎么理解呢?首先要理解Priority.orinal()方法,在Java中每一个枚举值都有这个方法,这个枚举的值是它的下标+1,也就是[index + 1],所以我们写的Priority类其实可以这样理解:

    public enum Priority {
        1,
        2,
        3,
        4
    }
    

    继续,如果给当前Task比较低,给compareTo(E)中的Task设置的优先级别比较高,那么Priority不一样,那么返回的值就是整数,因此当前Task就会被PriorityBlockingQueue<E>排到后面,如果调换那么返回结果也就调换了。

    但是我们注意到me == it ? [...] : it.ordinal() - me.ordinal();中的[...]是什么鬼啊?因为这里缺一段代码呀哈哈哈(这个作者怎么傻乎乎的),这一段代码的意思是当优先级别一样的时候怎么办,那就是谁先加入队列谁排到前面呗,那么怎样返回值呢,我们怎么知道哪个Task先加入队列呢?这个时候可爱的我就出现了,我给它给一个序列标记它什么时候加入队列的不久完事了,于是我们可以修改下ITask接口,增加两个方法:

    public interface ITask extends Comparable<ITask> {
    
        void run();
    
        void setPriority(Priority priority);
    
        Priority getPriority();
    
        void setSequence(int sequence);
    
        int getSequence();
    }
    

    我们用setSequence(int)标记它加入队列的顺序,然后因为setSequence(int)getSequence()是所有Task都需要实现的,所以我们在BasicTask中实现这两个方法:

    public abstract class BasicTask implements ITask {
    
        // 默认优先级。
        private Priority priority = Priority.DEFAULT;
        private int sequence;
    
        @Override
        public void setPriority(Priority priority) {
            this.priority = priority;
        }
    
        @Override
        public Priority getPriority() {
            return priority;
        }
    
        @Override
        public void setSequence(int sequence) {
            this.sequence = sequence;
        }
    
        @Override
        public int getSequence() {
            return sequence;
        }
    
        // 做优先级比较。
        @Override
        public int compareTo(ITask another) {
            final Priority me = this.getPriority();
            final Priority it = another.getPriority();
            return me == it ?  this.getSequence() - another.getSequence() :
                it.ordinal() - me.ordinal();
        }
    }
    

    看到了吧,刚才的[...]已经变成了this.getSequence() - another.getSequence(),这里需要和上面的it.ordinal() - me.ordinal();的逻辑对应,上面说到如果给当前Task比较低,给compareTo(E)中的Task设置的优先级别比较高,那么Priority不一样,那么返回的值就是整数,因此当前Task就会被PriorityBlockingQueue<E>排到后面,如果调换那么返回结果也就调换了。

    这里的逻辑和上面对应就是和上面的逻辑相反,因为这里是当两个优先级一样时的返回,上面是两个优先级不一样时的返回,所以当优先级别一样时,返回负数表示当前Task在前,返回正数表示当前Task在后,正好上面上的逻辑对应。

    接下来就是给Task设置序列了,于是我们在TaskQueue中的T void add(T)方法做个手脚:

    public class TaskQueue {
    
        private AtomicInteger mAtomicInteger = new AtomicInteger();
    
        ...
    
        public TaskQueue(int size) {
            ...
        }
    
        public void start() {
            ...
        }
    
        public void stop() {
            ...
        }
    
        public <T extends ITask> int add(T task) {
            if (!mTaskQueue.contains(task)) {
                task.setSequence(mAtomicInteger.incrementAndGet()); // 注意这行。
                mTaskQueue.add(task);
            }
            return mTaskQueue.size();
        }
    }
    

    这里我们使用了AtomicInteger类,它的incrementAndGet()方法会每次递增1,其实它相当于:

    mAtomicInteger.addAndGet(1);
    

    其它具体用法请自行搜索,这里不再赘述。

    到此为止,我们的优先级别的队列就实现完毕了,我们来做下测试:

    public static void main(String... args) {
        // 开一个窗口,这样会让优先级更加明显。
        TaskQueue taskQueue = new TaskQueue(1);
        taskQueue.start(); //  // 某机构开始工作。
    
        // 为了显示出优先级效果,我们预添加3个在前面堵着,让后面的优先级效果更明显。
        taskQueue.add(new PrintTask(110));
        taskQueue.add(new PrintTask(112));
        taskQueue.add(new PrintTask(122));
    
        for (int i = 0; i < 10; i++) { // 从第0个人开始。
        PrintTask task = new PrintTask(i);
        if (1 == i) { 
            task.setPriority(Priority.LOW); // 让第2个进入的人最后办事。
        } else if (8 == i) {
            task.setPriority(Priority.HIGH); // 让第9个进入的人第二个办事。
        } else if (9 == i) {
            task.setPriority(Priority.Immediately); // 让第10个进入的人第一个办事。
        } 
        // ... 其它进入的人,按照进入顺序办事。
        taskQueue.add(task);
    }
    

    没错这就是我们看到的效果:

    我的id是:9
    我的id是:8
    我的id是:110
    我的id是:112
    我的id是:122
    我的id是:0
    我的id是:2
    我的id是:3
    我的id是:4
    我的id是:5
    我的id是:6
    我的id是:7
    我的id是:1
    

      

  • 相关阅读:
    关于换行
    WebService
    C#操作XML的通用方法总结
    19个必须知道的Visual Studio快捷键
    Asp.net C# 把 Datatable转换成JSON 字符串
    C#中的多态性
    virtual和abstract
    DataTable的过滤需要的数据
    自定义控件之瀑布流与水波纹实现
    反转链表
  • 原文地址:https://www.cnblogs.com/ganchuanpu/p/9115239.html
Copyright © 2011-2022 走看看