Ý tưởng ở đây là sử dụng PriorityBlockingQueue trong trình thực thi. Đối với điều này:
- Tạo một so sánh so sánh tương lai của chúng tôi.
- Tạo proxy cho Tương lai để giữ mức độ ưu tiên.
- Ghi đè 'newTaskFor' để bao bọc mọi tương lai trong proxy của chúng tôi.
Trước tiên, bạn cần phải giữ ưu tiên về tương lai của bạn:
class PriorityFuture<T> implements RunnableFuture<T> {
private RunnableFuture<T> src;
private int priority;
public PriorityFuture(RunnableFuture<T> other, int priority) {
this.src = other;
this.priority = priority;
}
public int getPriority() {
return priority;
}
public boolean cancel(boolean mayInterruptIfRunning) {
return src.cancel(mayInterruptIfRunning);
}
public boolean isCancelled() {
return src.isCancelled();
}
public boolean isDone() {
return src.isDone();
}
public T get() throws InterruptedException, ExecutionException {
return src.get();
}
public T get(long timeout, TimeUnit unit) throws InterruptedException, ExecutionException, TimeoutException {
return src.get();
}
public void run() {
src.run();
}
}
Tiếp theo, bạn cần phải xác định so sánh rằng sẽ sắp xếp một cách chính xác tương lai ưu tiên:
class PriorityFutureComparator implements Comparator<Runnable> {
public int compare(Runnable o1, Runnable o2) {
if (o1 == null && o2 == null)
return 0;
else if (o1 == null)
return -1;
else if (o2 == null)
return 1;
else {
int p1 = ((PriorityFuture<?>) o1).getPriority();
int p2 = ((PriorityFuture<?>) o2).getPriority();
return p1 > p2 ? 1 : (p1 == p2 ? 0 : -1);
}
}
}
Tiếp theo chúng ta hãy giả sử chúng ta có một công việc dài như thế này:
class LenthyJob implements Callable<Long> {
private int priority;
public LenthyJob(int priority) {
this.priority = priority;
}
public Long call() throws Exception {
System.out.println("Executing: " + priority);
long num = 1000000;
for (int i = 0; i < 1000000; i++) {
num *= Math.random() * 1000;
num /= Math.random() * 1000;
if (num == 0)
num = 1000000;
}
return num;
}
public int getPriority() {
return priority;
}
}
Sau đó, để thực hiện các công việc này, mã sẽ trông giống như sau:
public class TestPQ {
public static void main(String[] args) throws InterruptedException, ExecutionException {
int nThreads = 2;
int qInitialSize = 10;
ExecutorService exec = new ThreadPoolExecutor(nThreads, nThreads, 0L, TimeUnit.MILLISECONDS,
new PriorityBlockingQueue<Runnable>(qInitialSize, new PriorityFutureComparator())) {
protected <T> RunnableFuture<T> newTaskFor(Callable<T> callable) {
RunnableFuture<T> newTaskFor = super.newTaskFor(callable);
return new PriorityFuture<T>(newTaskFor, ((LenthyJob) callable).getPriority());
}
};
for (int i = 0; i < 20; i++) {
int priority = (int) (Math.random() * 100);
System.out.println("Scheduling: " + priority);
LenthyJob job = new LenthyJob(priority);
exec.submit(job);
}
}
}
Đây là mã rất nhiều nhưng đó là cách duy nhất có thể thực hiện được.
Trên máy tính của tôi đầu ra là như sau:
Scheduling: 39
Scheduling: 90
Scheduling: 88
Executing: 39
Scheduling: 75
Executing: 90
Scheduling: 15
Scheduling: 2
Scheduling: 5
Scheduling: 24
Scheduling: 82
Scheduling: 81
Scheduling: 3
Scheduling: 23
Scheduling: 7
Scheduling: 40
Scheduling: 77
Scheduling: 49
Scheduling: 34
Scheduling: 22
Scheduling: 97
Scheduling: 33
Executing: 2
Executing: 3
Executing: 5
Executing: 7
Executing: 15
Executing: 22
Executing: 23
Executing: 24
Executing: 33
Executing: 34
Executing: 40
Executing: 49
Executing: 75
Executing: 77
Executing: 81
Executing: 82
Executing: 88
Executing: 97
+1 PriorityBlockingQueue là con đường để đi. Bạn có thể thực hiện một Comparator hoặc tự thực hiện các nhiệm vụ Comparable. –
Bài viết này là một tài liệu tham khảo tuyệt vời: http://binkley.blogspot.fr/2009/04/jumping-work-queue-in-executor.html – Snicolas
Lệnh giải quyết tác vụ của tôi theo mức độ ưu tiên, nhưng vẫn giữ nguyên thứ tự gửi ở cùng mức ưu tiên: http://stackoverflow.com/a/42831172/1386911 –