This commit is contained in:
@@ -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);
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user