1 |
|
/* |
2 |
|
* Written by Doug Lea with assistance from members of JCP JSR-166 |
3 |
|
* Expert Group and released to the public domain, as explained at |
4 |
< |
* http://creativecommons.org/licenses/publicdomain |
4 |
> |
* http://creativecommons.org/publicdomain/zero/1.0/ |
5 |
|
*/ |
6 |
|
|
7 |
|
import junit.framework.*; |
22 |
|
import java.util.concurrent.ForkJoinWorkerThread; |
23 |
|
import java.util.concurrent.RecursiveTask; |
24 |
|
import java.util.concurrent.TimeUnit; |
25 |
+ |
import java.util.concurrent.atomic.AtomicBoolean; |
26 |
|
import java.util.concurrent.locks.ReentrantLock; |
27 |
|
import static java.util.concurrent.TimeUnit.MILLISECONDS; |
28 |
|
import java.security.AccessControlException; |
165 |
|
try { |
166 |
|
assertSame(ForkJoinPool.defaultForkJoinWorkerThreadFactory, |
167 |
|
p.getFactory()); |
167 |
– |
assertTrue(p.isQuiescent()); |
168 |
|
assertFalse(p.getAsyncMode()); |
169 |
|
assertEquals(0, p.getActiveThreadCount()); |
170 |
|
assertEquals(0, p.getStealCount()); |
199 |
|
} catch (NullPointerException success) {} |
200 |
|
} |
201 |
|
|
202 |
– |
|
202 |
|
/** |
203 |
|
* getParallelism returns size set in constructor |
204 |
|
*/ |
232 |
|
* performs its defined action |
233 |
|
*/ |
234 |
|
public void testSetUncaughtExceptionHandler() throws InterruptedException { |
235 |
< |
final CountDownLatch uncaughtExceptionHappened = new CountDownLatch(1); |
235 |
> |
final CountDownLatch uehInvoked = new CountDownLatch(1); |
236 |
|
final Thread.UncaughtExceptionHandler eh = |
237 |
|
new Thread.UncaughtExceptionHandler() { |
238 |
|
public void uncaughtException(Thread t, Throwable e) { |
239 |
< |
uncaughtExceptionHappened.countDown(); |
239 |
> |
uehInvoked.countDown(); |
240 |
|
}}; |
241 |
|
ForkJoinPool p = new ForkJoinPool(1, new FailingThreadFactory(), |
242 |
|
eh, false); |
243 |
|
try { |
244 |
|
assertSame(eh, p.getUncaughtExceptionHandler()); |
245 |
< |
p.execute(new FailingTask()); |
246 |
< |
uncaughtExceptionHappened.await(); |
245 |
> |
p.execute(new FibTask(8)); |
246 |
> |
assertTrue(uehInvoked.await(MEDIUM_DELAY_MS, MILLISECONDS)); |
247 |
|
} finally { |
248 |
< |
//p.shutdownNow(); |
248 |
> |
p.shutdownNow(); // failure might have prevented processing task |
249 |
|
joinPool(p); |
250 |
|
} |
251 |
|
} |
252 |
|
|
253 |
|
/** |
254 |
< |
* After invoking a single task, isQuiescent is true, |
255 |
< |
* queues are empty, threads are not active, and |
256 |
< |
* construction parameters continue to hold |
254 |
> |
* After invoking a single task, isQuiescent eventually becomes |
255 |
> |
* true, at which time queues are empty, threads are not active, |
256 |
> |
* the task has completed successfully, and construction |
257 |
> |
* parameters continue to hold |
258 |
|
*/ |
259 |
< |
public void testisQuiescent() throws InterruptedException { |
259 |
> |
public void testisQuiescent() throws Exception { |
260 |
|
ForkJoinPool p = new ForkJoinPool(2); |
261 |
|
try { |
262 |
< |
p.invoke(new FibTask(20)); |
262 |
> |
assertTrue(p.isQuiescent()); |
263 |
> |
long startTime = System.nanoTime(); |
264 |
> |
FibTask f = new FibTask(20); |
265 |
> |
p.invoke(f); |
266 |
|
assertSame(ForkJoinPool.defaultForkJoinWorkerThreadFactory, |
267 |
|
p.getFactory()); |
268 |
< |
Thread.sleep(MEDIUM_DELAY_MS); |
268 |
> |
while (! p.isQuiescent()) { |
269 |
> |
if (millisElapsedSince(startTime) > LONG_DELAY_MS) |
270 |
> |
throw new AssertionFailedError("timed out"); |
271 |
> |
assertFalse(p.getAsyncMode()); |
272 |
> |
assertFalse(p.isShutdown()); |
273 |
> |
assertFalse(p.isTerminating()); |
274 |
> |
assertFalse(p.isTerminated()); |
275 |
> |
Thread.yield(); |
276 |
> |
} |
277 |
> |
|
278 |
|
assertTrue(p.isQuiescent()); |
279 |
|
assertFalse(p.getAsyncMode()); |
280 |
|
assertEquals(0, p.getActiveThreadCount()); |
284 |
|
assertFalse(p.isShutdown()); |
285 |
|
assertFalse(p.isTerminating()); |
286 |
|
assertFalse(p.isTerminated()); |
287 |
+ |
assertTrue(f.isDone()); |
288 |
+ |
assertEquals(6765, (int) f.get()); |
289 |
|
} finally { |
290 |
|
joinPool(p); |
291 |
|
} |
329 |
|
try { |
330 |
|
ReentrantLock lock = new ReentrantLock(); |
331 |
|
ManagedLocker locker = new ManagedLocker(lock); |
332 |
< |
ForkJoinTask<Integer> f = new LockingFibTask(30, locker, lock); |
332 |
> |
ForkJoinTask<Integer> f = new LockingFibTask(20, locker, lock); |
333 |
|
p.execute(f); |
334 |
< |
assertEquals(832040, (int) f.get()); |
334 |
> |
assertEquals(6765, (int) f.get()); |
335 |
|
} finally { |
336 |
|
p.shutdownNow(); // don't wait out shutdown |
337 |
|
} |
341 |
|
* pollSubmission returns unexecuted submitted task, if present |
342 |
|
*/ |
343 |
|
public void testPollSubmission() { |
344 |
+ |
final CountDownLatch done = new CountDownLatch(1); |
345 |
|
SubFJP p = new SubFJP(); |
346 |
|
try { |
347 |
< |
ForkJoinTask a = p.submit(new MediumRunnable()); |
348 |
< |
ForkJoinTask b = p.submit(new MediumRunnable()); |
349 |
< |
ForkJoinTask c = p.submit(new MediumRunnable()); |
347 |
> |
ForkJoinTask a = p.submit(awaiter(done)); |
348 |
> |
ForkJoinTask b = p.submit(awaiter(done)); |
349 |
> |
ForkJoinTask c = p.submit(awaiter(done)); |
350 |
|
ForkJoinTask r = p.pollSubmission(); |
351 |
|
assertTrue(r == a || r == b || r == c); |
352 |
|
assertFalse(r.isDone()); |
353 |
|
} finally { |
354 |
+ |
done.countDown(); |
355 |
|
joinPool(p); |
356 |
|
} |
357 |
|
} |
360 |
|
* drainTasksTo transfers unexecuted submitted tasks, if present |
361 |
|
*/ |
362 |
|
public void testDrainTasksTo() { |
363 |
+ |
final CountDownLatch done = new CountDownLatch(1); |
364 |
|
SubFJP p = new SubFJP(); |
365 |
|
try { |
366 |
< |
ForkJoinTask a = p.submit(new MediumRunnable()); |
367 |
< |
ForkJoinTask b = p.submit(new MediumRunnable()); |
368 |
< |
ForkJoinTask c = p.submit(new MediumRunnable()); |
366 |
> |
ForkJoinTask a = p.submit(awaiter(done)); |
367 |
> |
ForkJoinTask b = p.submit(awaiter(done)); |
368 |
> |
ForkJoinTask c = p.submit(awaiter(done)); |
369 |
|
ArrayList<ForkJoinTask> al = new ArrayList(); |
370 |
|
p.drainTasksTo(al); |
371 |
|
assertTrue(al.size() > 0); |
374 |
|
assertFalse(r.isDone()); |
375 |
|
} |
376 |
|
} finally { |
377 |
+ |
done.countDown(); |
378 |
|
joinPool(p); |
379 |
|
} |
380 |
|
} |
381 |
|
|
364 |
– |
|
382 |
|
// FJ Versions of AbstractExecutorService tests |
383 |
|
|
384 |
|
/** |
387 |
|
public void testExecuteRunnable() throws Throwable { |
388 |
|
ExecutorService e = new ForkJoinPool(1); |
389 |
|
try { |
390 |
< |
TrackedShortRunnable task = new TrackedShortRunnable(); |
391 |
< |
assertFalse(task.done); |
390 |
> |
final AtomicBoolean done = new AtomicBoolean(false); |
391 |
> |
CheckedRunnable task = new CheckedRunnable() { |
392 |
> |
public void realRun() { |
393 |
> |
done.set(true); |
394 |
> |
}}; |
395 |
|
Future<?> future = e.submit(task); |
396 |
< |
future.get(); |
397 |
< |
assertTrue(task.done); |
396 |
> |
assertNull(future.get()); |
397 |
> |
assertNull(future.get(0, MILLISECONDS)); |
398 |
> |
assertTrue(done.get()); |
399 |
> |
assertTrue(future.isDone()); |
400 |
> |
assertFalse(future.isCancelled()); |
401 |
|
} finally { |
402 |
|
joinPool(e); |
403 |
|
} |
404 |
|
} |
405 |
|
|
383 |
– |
|
406 |
|
/** |
407 |
|
* Completed submit(callable) returns result |
408 |
|
*/ |
410 |
|
ExecutorService e = new ForkJoinPool(1); |
411 |
|
try { |
412 |
|
Future<String> future = e.submit(new StringTask()); |
413 |
< |
String result = future.get(); |
414 |
< |
assertSame(TEST_STRING, result); |
413 |
> |
assertSame(TEST_STRING, future.get()); |
414 |
> |
assertTrue(future.isDone()); |
415 |
> |
assertFalse(future.isCancelled()); |
416 |
|
} finally { |
417 |
|
joinPool(e); |
418 |
|
} |
425 |
|
ExecutorService e = new ForkJoinPool(1); |
426 |
|
try { |
427 |
|
Future<?> future = e.submit(new NoOpRunnable()); |
428 |
< |
future.get(); |
428 |
> |
assertNull(future.get()); |
429 |
|
assertTrue(future.isDone()); |
430 |
+ |
assertFalse(future.isCancelled()); |
431 |
|
} finally { |
432 |
|
joinPool(e); |
433 |
|
} |
440 |
|
ExecutorService e = new ForkJoinPool(1); |
441 |
|
try { |
442 |
|
Future<String> future = e.submit(new NoOpRunnable(), TEST_STRING); |
443 |
< |
String result = future.get(); |
444 |
< |
assertSame(TEST_STRING, result); |
443 |
> |
assertSame(TEST_STRING, future.get()); |
444 |
> |
assertTrue(future.isDone()); |
445 |
> |
assertFalse(future.isCancelled()); |
446 |
|
} finally { |
447 |
|
joinPool(e); |
448 |
|
} |
449 |
|
} |
450 |
|
|
426 |
– |
|
451 |
|
/** |
452 |
< |
* A submitted privileged action to completion |
452 |
> |
* A submitted privileged action runs to completion |
453 |
|
*/ |
454 |
< |
public void testSubmitPrivilegedAction() throws Throwable { |
455 |
< |
Policy savedPolicy = null; |
456 |
< |
try { |
457 |
< |
savedPolicy = Policy.getPolicy(); |
434 |
< |
AdjustablePolicy policy = new AdjustablePolicy(); |
435 |
< |
policy.addPermission(new RuntimePermission("getContextClassLoader")); |
436 |
< |
policy.addPermission(new RuntimePermission("setContextClassLoader")); |
437 |
< |
Policy.setPolicy(policy); |
438 |
< |
} catch (AccessControlException ok) { |
439 |
< |
return; |
440 |
< |
} |
441 |
< |
|
442 |
< |
try { |
443 |
< |
ExecutorService e = new ForkJoinPool(1); |
444 |
< |
try { |
454 |
> |
public void testSubmitPrivilegedAction() throws Exception { |
455 |
> |
Runnable r = new CheckedRunnable() { |
456 |
> |
public void realRun() throws Exception { |
457 |
> |
ExecutorService e = new ForkJoinPool(1); |
458 |
|
Future future = e.submit(Executors.callable(new PrivilegedAction() { |
459 |
|
public Object run() { |
460 |
|
return TEST_STRING; |
461 |
|
}})); |
462 |
|
|
463 |
< |
Object result = future.get(); |
464 |
< |
assertSame(TEST_STRING, result); |
465 |
< |
} finally { |
466 |
< |
joinPool(e); |
467 |
< |
} |
455 |
< |
} finally { |
456 |
< |
Policy.setPolicy(savedPolicy); |
457 |
< |
} |
463 |
> |
assertSame(TEST_STRING, future.get()); |
464 |
> |
}}; |
465 |
> |
|
466 |
> |
runWithPermissions(r, |
467 |
> |
new RuntimePermission("modifyThread")); |
468 |
|
} |
469 |
|
|
470 |
|
/** |
471 |
< |
* A submitted a privileged exception action runs to completion |
471 |
> |
* A submitted privileged exception action runs to completion |
472 |
|
*/ |
473 |
< |
public void testSubmitPrivilegedExceptionAction() throws Throwable { |
474 |
< |
Policy savedPolicy = null; |
475 |
< |
try { |
476 |
< |
savedPolicy = Policy.getPolicy(); |
467 |
< |
AdjustablePolicy policy = new AdjustablePolicy(); |
468 |
< |
policy.addPermission(new RuntimePermission("getContextClassLoader")); |
469 |
< |
policy.addPermission(new RuntimePermission("setContextClassLoader")); |
470 |
< |
Policy.setPolicy(policy); |
471 |
< |
} catch (AccessControlException ok) { |
472 |
< |
return; |
473 |
< |
} |
474 |
< |
|
475 |
< |
try { |
476 |
< |
ExecutorService e = new ForkJoinPool(1); |
477 |
< |
try { |
473 |
> |
public void testSubmitPrivilegedExceptionAction() throws Exception { |
474 |
> |
Runnable r = new CheckedRunnable() { |
475 |
> |
public void realRun() throws Exception { |
476 |
> |
ExecutorService e = new ForkJoinPool(1); |
477 |
|
Future future = e.submit(Executors.callable(new PrivilegedExceptionAction() { |
478 |
|
public Object run() { |
479 |
|
return TEST_STRING; |
480 |
|
}})); |
481 |
|
|
482 |
< |
Object result = future.get(); |
483 |
< |
assertSame(TEST_STRING, result); |
484 |
< |
} finally { |
485 |
< |
joinPool(e); |
487 |
< |
} |
488 |
< |
} finally { |
489 |
< |
Policy.setPolicy(savedPolicy); |
490 |
< |
} |
482 |
> |
assertSame(TEST_STRING, future.get()); |
483 |
> |
}}; |
484 |
> |
|
485 |
> |
runWithPermissions(r, new RuntimePermission("modifyThread")); |
486 |
|
} |
487 |
|
|
488 |
|
/** |
489 |
|
* A submitted failed privileged exception action reports exception |
490 |
|
*/ |
491 |
< |
public void testSubmitFailedPrivilegedExceptionAction() throws Throwable { |
492 |
< |
Policy savedPolicy = null; |
493 |
< |
try { |
494 |
< |
savedPolicy = Policy.getPolicy(); |
500 |
< |
AdjustablePolicy policy = new AdjustablePolicy(); |
501 |
< |
policy.addPermission(new RuntimePermission("getContextClassLoader")); |
502 |
< |
policy.addPermission(new RuntimePermission("setContextClassLoader")); |
503 |
< |
Policy.setPolicy(policy); |
504 |
< |
} catch (AccessControlException ok) { |
505 |
< |
return; |
506 |
< |
} |
507 |
< |
|
508 |
< |
try { |
509 |
< |
ExecutorService e = new ForkJoinPool(1); |
510 |
< |
try { |
491 |
> |
public void testSubmitFailedPrivilegedExceptionAction() throws Exception { |
492 |
> |
Runnable r = new CheckedRunnable() { |
493 |
> |
public void realRun() throws Exception { |
494 |
> |
ExecutorService e = new ForkJoinPool(1); |
495 |
|
Future future = e.submit(Executors.callable(new PrivilegedExceptionAction() { |
496 |
|
public Object run() throws Exception { |
497 |
|
throw new IndexOutOfBoundsException(); |
498 |
|
}})); |
499 |
|
|
500 |
< |
Object result = future.get(); |
501 |
< |
shouldThrow(); |
502 |
< |
} catch (ExecutionException success) { |
503 |
< |
assertTrue(success.getCause() instanceof IndexOutOfBoundsException); |
504 |
< |
} finally { |
505 |
< |
joinPool(e); |
506 |
< |
} |
507 |
< |
} finally { |
524 |
< |
Policy.setPolicy(savedPolicy); |
525 |
< |
} |
500 |
> |
try { |
501 |
> |
future.get(); |
502 |
> |
shouldThrow(); |
503 |
> |
} catch (ExecutionException success) { |
504 |
> |
assertTrue(success.getCause() instanceof IndexOutOfBoundsException); |
505 |
> |
}}}; |
506 |
> |
|
507 |
> |
runWithPermissions(r, new RuntimePermission("modifyThread")); |
508 |
|
} |
509 |
|
|
510 |
|
/** |
512 |
|
*/ |
513 |
|
public void testExecuteNullRunnable() { |
514 |
|
ExecutorService e = new ForkJoinPool(1); |
533 |
– |
TrackedShortRunnable task = null; |
515 |
|
try { |
516 |
< |
Future<?> future = e.submit(task); |
516 |
> |
Future<?> future = e.submit((Runnable) null); |
517 |
|
shouldThrow(); |
518 |
|
} catch (NullPointerException success) { |
519 |
|
} finally { |
521 |
|
} |
522 |
|
} |
523 |
|
|
543 |
– |
|
524 |
|
/** |
525 |
|
* submit(null callable) throws NullPointerException |
526 |
|
*/ |
527 |
|
public void testSubmitNullCallable() { |
528 |
|
ExecutorService e = new ForkJoinPool(1); |
549 |
– |
StringTask t = null; |
529 |
|
try { |
530 |
< |
Future<String> future = e.submit(t); |
530 |
> |
Future<String> future = e.submit((Callable) null); |
531 |
|
shouldThrow(); |
532 |
|
} catch (NullPointerException success) { |
533 |
|
} finally { |
535 |
|
} |
536 |
|
} |
537 |
|
|
559 |
– |
|
538 |
|
/** |
539 |
|
* submit(callable).get() throws InterruptedException if interrupted |
540 |
|
*/ |
763 |
|
} |
764 |
|
} |
765 |
|
|
788 |
– |
|
766 |
|
/** |
767 |
|
* timed invokeAny(null) throws NullPointerException |
768 |
|
*/ |