This commit is contained in:
Redkale
2021-01-09 14:42:45 +08:00
parent fb5f82c44f
commit 1982c984ee

View File

@@ -23,25 +23,41 @@ public class ThreadHashExecutor {
private final AtomicInteger index = new AtomicInteger();
private final LinkedBlockingQueue<Runnable>[] queues;
private final ExecutorService[] executors;
public ThreadHashExecutor(int size) {
ExecutorService[] array = new ExecutorService[size];
LinkedBlockingQueue[] ques = new LinkedBlockingQueue[size];
final AtomicInteger counter = new AtomicInteger();
for (int i = 0; i < array.length; i++) {
array[i] = Executors.newSingleThreadExecutor((Runnable r) -> {
Thread t = new Thread(r);
t.setDaemon(true);
t.setName("Redkale-HashThread-" + counter.incrementAndGet());
return t;
});
LinkedBlockingQueue<Runnable> queue = new LinkedBlockingQueue<>();
ques[i] = queue;
array[i] = new ThreadPoolExecutor(1, 1, 0L, TimeUnit.MILLISECONDS, queue,
(Runnable r) -> {
Thread t = new Thread(r);
t.setDaemon(true);
t.setName("Redkale-HashThread-" + counter.incrementAndGet());
return t;
});
}
this.queues = ques;
this.executors = array;
}
public void execute(int hash, Runnable command) {
if (hash < 1) {
this.executors[index.incrementAndGet() % this.executors.length].execute(command);
int k = 0;
int minsize = queues[0].size();
for (int i = 1; i < queues.length; i++) {
int size = queues[i].size();
if (size < minsize) {
minsize = size;
k = i;
}
}
this.executors[k].execute(command);
} else {
this.executors[hash % this.executors.length].execute(command);
}