1 | |
|
2 | |
|
3 | |
|
4 | |
|
5 | |
|
6 | |
|
7 | |
|
8 | |
|
9 | |
|
10 | |
|
11 | |
package org.mule.config; |
12 | |
|
13 | |
import org.mule.impl.work.MuleWorkManager; |
14 | |
import org.mule.umo.manager.UMOWorkManager; |
15 | |
import org.mule.util.MapUtils; |
16 | |
import org.mule.util.concurrent.NamedThreadFactory; |
17 | |
import org.mule.util.concurrent.WaitPolicy; |
18 | |
|
19 | |
import java.util.Map; |
20 | |
|
21 | |
import edu.emory.mathcs.backport.java.util.concurrent.BlockingQueue; |
22 | |
import edu.emory.mathcs.backport.java.util.concurrent.LinkedBlockingDeque; |
23 | |
import edu.emory.mathcs.backport.java.util.concurrent.RejectedExecutionHandler; |
24 | |
import edu.emory.mathcs.backport.java.util.concurrent.SynchronousQueue; |
25 | |
import edu.emory.mathcs.backport.java.util.concurrent.ThreadFactory; |
26 | |
import edu.emory.mathcs.backport.java.util.concurrent.ThreadPoolExecutor; |
27 | |
import edu.emory.mathcs.backport.java.util.concurrent.TimeUnit; |
28 | |
import org.apache.commons.collections.map.CaseInsensitiveMap; |
29 | |
|
30 | |
|
31 | |
|
32 | |
|
33 | |
|
34 | |
|
35 | |
|
36 | |
public class ThreadingProfile |
37 | |
{ |
38 | |
|
39 | |
|
40 | |
|
41 | |
|
42 | |
public static final int DEFAULT_MAX_THREADS_ACTIVE = 16; |
43 | |
|
44 | |
|
45 | |
|
46 | |
|
47 | |
public static final int DEFAULT_MAX_THREADS_IDLE = 1; |
48 | |
|
49 | |
|
50 | |
|
51 | |
|
52 | |
public static final int DEFAULT_MAX_BUFFER_SIZE = 0; |
53 | |
|
54 | |
|
55 | |
|
56 | |
|
57 | |
public static final long DEFAULT_MAX_THREAD_TTL = 60000; |
58 | |
|
59 | |
|
60 | |
|
61 | |
|
62 | |
public static final long DEFAULT_THREAD_WAIT_TIMEOUT = 30000L; |
63 | |
|
64 | |
|
65 | |
|
66 | |
|
67 | |
public static final boolean DEFAULT_DO_THREADING = true; |
68 | |
|
69 | |
|
70 | |
|
71 | |
|
72 | |
public static final int WHEN_EXHAUSTED_WAIT = 0; |
73 | |
public static final int WHEN_EXHAUSTED_DISCARD = 1; |
74 | |
public static final int WHEN_EXHAUSTED_DISCARD_OLDEST = 2; |
75 | |
public static final int WHEN_EXHAUSTED_ABORT = 3; |
76 | |
public static final int WHEN_EXHAUSTED_RUN = 4; |
77 | |
|
78 | |
|
79 | |
|
80 | |
|
81 | |
public static final int DEFAULT_POOL_EXHAUST_ACTION = WHEN_EXHAUSTED_RUN; |
82 | |
|
83 | |
|
84 | 0 | private static final Map POOL_EXHAUSTED_ACTIONS = new CaseInsensitiveMap() |
85 | |
{ |
86 | |
private static final long serialVersionUID = 1L; |
87 | |
|
88 | |
|
89 | 0 | { |
90 | 0 | Integer value = new Integer(WHEN_EXHAUSTED_WAIT); |
91 | 0 | this.put("WHEN_EXHAUSTED_WAIT", value); |
92 | 0 | this.put("WAIT", value); |
93 | |
|
94 | 0 | value = new Integer(WHEN_EXHAUSTED_DISCARD); |
95 | 0 | this.put("WHEN_EXHAUSTED_DISCARD", value); |
96 | 0 | this.put("DISCARD", value); |
97 | |
|
98 | 0 | value = new Integer(WHEN_EXHAUSTED_DISCARD_OLDEST); |
99 | 0 | this.put("WHEN_EXHAUSTED_DISCARD_OLDEST", value); |
100 | 0 | this.put("DISCARD_OLDEST", value); |
101 | |
|
102 | 0 | value = new Integer(WHEN_EXHAUSTED_ABORT); |
103 | 0 | this.put("WHEN_EXHAUSTED_ABORT", value); |
104 | 0 | this.put("ABORT", value); |
105 | |
|
106 | 0 | value = new Integer(WHEN_EXHAUSTED_RUN); |
107 | 0 | this.put("WHEN_EXHAUSTED_RUN", value); |
108 | 0 | this.put("RUN", value); |
109 | 0 | } |
110 | |
}; |
111 | |
|
112 | 0 | private int maxThreadsActive = DEFAULT_MAX_THREADS_ACTIVE; |
113 | 0 | private int maxThreadsIdle = DEFAULT_MAX_THREADS_IDLE; |
114 | 0 | private int maxBufferSize = DEFAULT_MAX_BUFFER_SIZE; |
115 | 0 | private long threadTTL = DEFAULT_MAX_THREAD_TTL; |
116 | 0 | private long threadWaitTimeout = DEFAULT_THREAD_WAIT_TIMEOUT; |
117 | 0 | private int poolExhaustPolicy = DEFAULT_POOL_EXHAUST_ACTION; |
118 | 0 | private boolean doThreading = DEFAULT_DO_THREADING; |
119 | |
|
120 | 0 | private WorkManagerFactory workManagerFactory = new DefaultWorkManagerFactory(); |
121 | |
|
122 | |
private RejectedExecutionHandler rejectedExecutionHandler; |
123 | |
|
124 | |
private ThreadFactory threadFactory; |
125 | |
|
126 | |
public ThreadingProfile() |
127 | |
{ |
128 | 0 | super(); |
129 | 0 | } |
130 | |
|
131 | |
public ThreadingProfile(int maxThreadsActive, |
132 | |
int maxThreadsIdle, |
133 | |
long threadTTL, |
134 | |
int poolExhaustPolicy, |
135 | |
RejectedExecutionHandler rejectedExecutionHandler, |
136 | |
ThreadFactory threadFactory) |
137 | 0 | { |
138 | 0 | this.maxThreadsActive = maxThreadsActive; |
139 | 0 | this.maxThreadsIdle = maxThreadsIdle; |
140 | 0 | this.threadTTL = threadTTL; |
141 | 0 | this.poolExhaustPolicy = poolExhaustPolicy; |
142 | 0 | this.rejectedExecutionHandler = rejectedExecutionHandler; |
143 | 0 | this.threadFactory = threadFactory; |
144 | 0 | } |
145 | |
|
146 | |
public ThreadingProfile(ThreadingProfile tp) |
147 | 0 | { |
148 | 0 | this.maxThreadsActive = tp.getMaxThreadsActive(); |
149 | 0 | this.maxThreadsIdle = tp.getMaxThreadsIdle(); |
150 | 0 | this.maxBufferSize = tp.getMaxBufferSize(); |
151 | 0 | this.threadTTL = tp.getThreadTTL(); |
152 | 0 | this.threadWaitTimeout = tp.getThreadWaitTimeout(); |
153 | 0 | this.poolExhaustPolicy = tp.getPoolExhaustedAction(); |
154 | 0 | this.doThreading = tp.isDoThreading(); |
155 | 0 | this.rejectedExecutionHandler = tp.getRejectedExecutionHandler(); |
156 | 0 | this.threadFactory = tp.getThreadFactory(); |
157 | 0 | this.workManagerFactory = tp.getWorkManagerFactory(); |
158 | 0 | } |
159 | |
|
160 | |
public int getMaxThreadsActive() |
161 | |
{ |
162 | 0 | return maxThreadsActive; |
163 | |
} |
164 | |
|
165 | |
public int getMaxThreadsIdle() |
166 | |
{ |
167 | 0 | return maxThreadsIdle; |
168 | |
} |
169 | |
|
170 | |
public long getThreadTTL() |
171 | |
{ |
172 | 0 | return threadTTL; |
173 | |
} |
174 | |
|
175 | |
public long getThreadWaitTimeout() |
176 | |
{ |
177 | 0 | return threadWaitTimeout; |
178 | |
} |
179 | |
|
180 | |
public int getPoolExhaustedAction() |
181 | |
{ |
182 | 0 | return poolExhaustPolicy; |
183 | |
} |
184 | |
|
185 | |
public RejectedExecutionHandler getRejectedExecutionHandler() |
186 | |
{ |
187 | 0 | return rejectedExecutionHandler; |
188 | |
} |
189 | |
|
190 | |
public ThreadFactory getThreadFactory() |
191 | |
{ |
192 | 0 | return threadFactory; |
193 | |
} |
194 | |
|
195 | |
public void setMaxThreadsActive(int maxThreadsActive) |
196 | |
{ |
197 | 0 | this.maxThreadsActive = maxThreadsActive; |
198 | 0 | } |
199 | |
|
200 | |
public void setMaxThreadsIdle(int maxThreadsIdle) |
201 | |
{ |
202 | 0 | this.maxThreadsIdle = maxThreadsIdle; |
203 | 0 | } |
204 | |
|
205 | |
public void setThreadTTL(long threadTTL) |
206 | |
{ |
207 | 0 | this.threadTTL = threadTTL; |
208 | 0 | } |
209 | |
|
210 | |
public void setThreadWaitTimeout(long threadWaitTimeout) |
211 | |
{ |
212 | 0 | this.threadWaitTimeout = threadWaitTimeout; |
213 | 0 | } |
214 | |
|
215 | |
public void setPoolExhaustedAction(int poolExhaustPolicy) |
216 | |
{ |
217 | 0 | this.poolExhaustPolicy = poolExhaustPolicy; |
218 | 0 | } |
219 | |
|
220 | |
public void setPoolExhaustedActionString(String poolExhaustPolicy) |
221 | |
{ |
222 | 0 | this.poolExhaustPolicy = MapUtils.getIntValue(POOL_EXHAUSTED_ACTIONS, poolExhaustPolicy, |
223 | |
DEFAULT_POOL_EXHAUST_ACTION); |
224 | 0 | } |
225 | |
|
226 | |
public void setRejectedExecutionHandler(RejectedExecutionHandler rejectedExecutionHandler) |
227 | |
{ |
228 | 0 | this.rejectedExecutionHandler = rejectedExecutionHandler; |
229 | 0 | } |
230 | |
|
231 | |
public void setThreadFactory(ThreadFactory threadFactory) |
232 | |
{ |
233 | 0 | this.threadFactory = threadFactory; |
234 | 0 | } |
235 | |
|
236 | |
public int getMaxBufferSize() |
237 | |
{ |
238 | 0 | return maxBufferSize; |
239 | |
} |
240 | |
|
241 | |
public void setMaxBufferSize(int maxBufferSize) |
242 | |
{ |
243 | 0 | this.maxBufferSize = maxBufferSize; |
244 | 0 | } |
245 | |
|
246 | |
public WorkManagerFactory getWorkManagerFactory() |
247 | |
{ |
248 | 0 | return workManagerFactory; |
249 | |
} |
250 | |
|
251 | |
public void setWorkManagerFactory(WorkManagerFactory workManagerFactory) |
252 | |
{ |
253 | 0 | this.workManagerFactory = workManagerFactory; |
254 | 0 | } |
255 | |
|
256 | |
public UMOWorkManager createWorkManager(String name) |
257 | |
{ |
258 | 0 | return workManagerFactory.createWorkManager(this, name); |
259 | |
} |
260 | |
|
261 | |
public ThreadPoolExecutor createPool() |
262 | |
{ |
263 | 0 | return createPool(null); |
264 | |
} |
265 | |
|
266 | |
public ThreadPoolExecutor createPool(String name) |
267 | |
{ |
268 | |
BlockingQueue buffer; |
269 | |
|
270 | 0 | if (maxBufferSize > 0 && maxThreadsActive > 1) |
271 | |
{ |
272 | 0 | buffer = new LinkedBlockingDeque(maxBufferSize); |
273 | |
} |
274 | |
else |
275 | |
{ |
276 | 0 | buffer = new SynchronousQueue(); |
277 | |
} |
278 | |
|
279 | 0 | if (maxThreadsIdle > maxThreadsActive) |
280 | |
{ |
281 | 0 | maxThreadsIdle = maxThreadsActive; |
282 | |
} |
283 | |
|
284 | 0 | ThreadPoolExecutor pool = new ThreadPoolExecutor(maxThreadsIdle, maxThreadsActive, threadTTL, |
285 | |
TimeUnit.MILLISECONDS, buffer); |
286 | |
|
287 | 0 | ThreadFactory tf = threadFactory; |
288 | 0 | if (name != null) |
289 | |
{ |
290 | 0 | tf = new NamedThreadFactory(name); |
291 | |
} |
292 | |
|
293 | 0 | if (tf != null) |
294 | |
{ |
295 | 0 | pool.setThreadFactory(tf); |
296 | |
} |
297 | |
|
298 | 0 | if (rejectedExecutionHandler != null) |
299 | |
{ |
300 | 0 | pool.setRejectedExecutionHandler(rejectedExecutionHandler); |
301 | |
} |
302 | |
else |
303 | |
{ |
304 | 0 | switch (poolExhaustPolicy) |
305 | |
{ |
306 | |
case WHEN_EXHAUSTED_DISCARD_OLDEST : |
307 | 0 | pool.setRejectedExecutionHandler(new ThreadPoolExecutor.DiscardOldestPolicy()); |
308 | 0 | break; |
309 | |
case WHEN_EXHAUSTED_RUN : |
310 | 0 | pool.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy()); |
311 | 0 | break; |
312 | |
case WHEN_EXHAUSTED_ABORT : |
313 | 0 | pool.setRejectedExecutionHandler(new ThreadPoolExecutor.AbortPolicy()); |
314 | 0 | break; |
315 | |
case WHEN_EXHAUSTED_DISCARD : |
316 | 0 | pool.setRejectedExecutionHandler(new ThreadPoolExecutor.DiscardPolicy()); |
317 | 0 | break; |
318 | |
default : |
319 | |
|
320 | 0 | pool.setRejectedExecutionHandler(new WaitPolicy(threadWaitTimeout, TimeUnit.MILLISECONDS)); |
321 | |
break; |
322 | |
} |
323 | |
} |
324 | |
|
325 | 0 | return pool; |
326 | |
} |
327 | |
|
328 | |
public boolean isDoThreading() |
329 | |
{ |
330 | 0 | return doThreading; |
331 | |
} |
332 | |
|
333 | |
public void setDoThreading(boolean doThreading) |
334 | |
{ |
335 | 0 | this.doThreading = doThreading; |
336 | 0 | } |
337 | |
|
338 | |
public String toString() |
339 | |
{ |
340 | 0 | return "ThreadingProfile{" + "maxThreadsActive=" + maxThreadsActive + ", maxThreadsIdle=" |
341 | |
+ maxThreadsIdle + ", maxBufferSize=" + maxBufferSize + ", threadTTL=" + threadTTL |
342 | |
+ ", poolExhaustPolicy=" + poolExhaustPolicy + ", threadWaitTimeout=" |
343 | |
+ threadWaitTimeout + ", doThreading=" + doThreading + ", workManagerFactory=" |
344 | |
+ workManagerFactory + ", rejectedExecutionHandler=" + rejectedExecutionHandler |
345 | |
+ ", threadFactory=" + threadFactory + "}"; |
346 | |
} |
347 | |
|
348 | |
public static interface WorkManagerFactory |
349 | |
{ |
350 | |
UMOWorkManager createWorkManager(ThreadingProfile profile, String name); |
351 | |
} |
352 | |
|
353 | 0 | private class DefaultWorkManagerFactory implements WorkManagerFactory |
354 | |
{ |
355 | |
public UMOWorkManager createWorkManager(ThreadingProfile profile, String name) |
356 | |
{ |
357 | 0 | return new MuleWorkManager(profile, name); |
358 | |
} |
359 | |
} |
360 | |
|
361 | |
} |