NekoX/TMessagesProj/src/main/java/org/telegram/DispatchQueuePriority.java

package org.telegram;

import java.util.Comparator;
import java.util.concurrent.PriorityBlockingQueue;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;

public class DispatchQueuePriority {

    ThreadPoolExecutor threadPoolExecutor = new ThreadPoolExecutor(1, 1, 60, TimeUnit.SECONDS, new PriorityBlockingQueue<>(10, new Comparator<Runnable>() {

        @Override
        public int compare(Runnable o1, Runnable o2) {
            int priority1 = 1;
            int priority2 = 1;
            if (o1 instanceof PriorityRunnable) {
                priority1 = ((PriorityRunnable) o1).priority;
            }
            if (o2 instanceof PriorityRunnable) {
                priority2 = ((PriorityRunnable) o2).priority;
            }
            return priority2 - priority1;
        }
    }));


    public DispatchQueuePriority(String threadName) {

    }

    public static Runnable wrap(Runnable runnable, int priority) {
        if (priority == 1) {
            return runnable;
        } else {
            return new PriorityRunnable(priority, runnable);
        }
    }

    public void postRunnable(Runnable runnable) {
        threadPoolExecutor.execute(runnable);
    }

    public Runnable postRunnable(Runnable runnable, int priority) {
        if (priority == 1) {
            postRunnable(runnable);
            return runnable;
        } else {
            PriorityRunnable priorityRunnable = new PriorityRunnable(priority, runnable);
            threadPoolExecutor.execute(priorityRunnable);
            return priorityRunnable;
        }
    }

    public void cancelRunnable(Runnable runnable) {
        if (runnable == null) {
            return;
        }
        threadPoolExecutor.remove(runnable);

    }

    private static class PriorityRunnable implements Runnable {
        final int priority;
        final Runnable runnable;

        private PriorityRunnable(int priority, Runnable runnable) {
            this.priority = priority;
            this.runnable = runnable;
        }

        @Override
        public void run() {
            runnable.run();
        }
    }
}