-
-
Notifications
You must be signed in to change notification settings - Fork 44
/
ThreadManager.java
160 lines (142 loc) · 4.05 KB
/
ThreadManager.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
package me.nallar.tickthreading.minecraft;
import java.util.HashSet;
import java.util.Set;
import java.util.concurrent.BlockingQueue;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.atomic.AtomicInteger;
import cpw.mods.fml.common.FMLCommonHandler;
import me.nallar.tickthreading.Log;
import me.nallar.tickthreading.collections.ConcurrentIterableArrayList;
import net.minecraft.profiler.Profiler;
import net.minecraft.server.MinecraftServer;
import net.minecraft.server.ThreadMinecraftServer;
import net.minecraft.util.AxisAlignedBB;
public final class ThreadManager {
private static final Profiler profiler = MinecraftServer.getServer().theProfiler;
private final boolean isServer = FMLCommonHandler.instance().getEffectiveSide().isServer();
private final BlockingQueue<Runnable> taskQueue = new LinkedBlockingQueue<Runnable>();
private final String namePrefix;
private final Set<Thread> workThreads = new HashSet<Thread>();
private final Object readyLock = new Object();
private final AtomicInteger waiting = new AtomicInteger();
private final Runnable killTask = new KillRunnable();
public long endTime = 0;
private final Runnable workerTask = new Runnable() {
@SuppressWarnings ("FieldRepeatedlyAccessedInMethod")
@Override
public void run() {
while (true) {
try {
Runnable runnable;
synchronized (taskQueue) {
runnable = taskQueue.take();
}
if (runnable == killTask) {
workThreads.remove(Thread.currentThread());
return;
}
runnable.run();
} catch (InterruptedException ignored) {
} catch (Exception e) {
Log.severe("Unhandled exception in worker thread " + Thread.currentThread().getName(), e);
}
if (waiting.decrementAndGet() == 0) {
endTime = System.nanoTime();
synchronized (readyLock) {
readyLock.notify();
}
}
}
}
};
private void newThread(String name) {
Thread workThread = isServer ? new ServerWorkThread() : new Thread(workerTask);
workThread.setName(name);
workThread.setDaemon(true);
workThread.start();
workThreads.add(workThread);
}
public ThreadManager(int threads, String name) {
namePrefix = name;
addThreads(threads);
}
public long waitForCompletion() {
profiler.startSection(namePrefix);
synchronized (readyLock) {
while (waiting.get() > 0) {
try {
readyLock.wait(1L);
} catch (InterruptedException ignored) {
}
}
}
profiler.endSection();
return endTime;
}
public void runList(final ConcurrentIterableArrayList<? extends Runnable> tasks) {
tasks.reset();
Runnable arrayRunnable = new Runnable() {
@Override
public void run() {
Runnable r;
while ((r = tasks.next()) != null) {
r.run();
}
AxisAlignedBB.getAABBPool().cleanPool();
}
};
for (int i = 0, len = workThreads.size(); i < len; i++) {
run(arrayRunnable);
}
}
public void run(Iterable<? extends Runnable> tasks) {
for (Runnable runnable : tasks) {
run(runnable);
}
}
public void runAll(Runnable runnable) {
for (int i = 0; i < size(); i++) {
run(runnable);
}
}
public void run(Runnable runnable) {
if (taskQueue.add(runnable)) {
waiting.incrementAndGet();
} else {
Log.severe("Failed to add " + runnable);
}
}
private void addThreads(int number) {
number += workThreads.size();
for (int i = workThreads.size() + 1; i <= number; i++) {
newThread(namePrefix + " - " + i);
}
}
private void killThreads(int number) {
for (int i = 0; i < number; i++) {
taskQueue.add(killTask);
}
}
public int size() {
return workThreads.size();
}
public void stop() {
killThreads(workThreads.size());
}
private static class KillRunnable implements Runnable {
KillRunnable() {
}
@Override
public void run() {
}
}
private class ServerWorkThread extends ThreadMinecraftServer {
public ServerWorkThread() {
super(null, "");
}
@Override
public void run() {
workerTask.run();
}
}
}