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