Java的优先级任务队列的实践

2018-06-18 00:38:59来源:未知 阅读 ()

新老客户大回馈,云服务器低至5折

队列的基本理解

在说队列之前说两个名词: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

  

标签:

版权申明:本站文章部分自网络,如有侵权,请联系:west999com@outlook.com
特别注意:本站所有转载文章言论不代表本站观点,本站所提供的摄影照片,插画,设计作品,如需使用,请与原作者联系,版权归原作者所有

上一篇:Java学习笔记三:Java的变量、常量、变量的类型及使用规则

下一篇:Spring Cloud 入门 Eureka-Server服务注册