37 |
|
import java.util.concurrent.Callable; |
38 |
|
import java.util.concurrent.CountDownLatch; |
39 |
|
import java.util.concurrent.CyclicBarrier; |
40 |
+ |
import java.util.concurrent.ExecutionException; |
41 |
+ |
import java.util.concurrent.Executors; |
42 |
|
import java.util.concurrent.ExecutorService; |
43 |
|
import java.util.concurrent.Future; |
44 |
|
import java.util.concurrent.RecursiveAction; |
729 |
|
/** |
730 |
|
* Waits out termination of a thread pool or fails doing so. |
731 |
|
*/ |
732 |
< |
void joinPool(ExecutorService exec) { |
732 |
> |
void joinPool(ExecutorService pool) { |
733 |
|
try { |
734 |
< |
exec.shutdown(); |
735 |
< |
if (!exec.awaitTermination(2 * LONG_DELAY_MS, MILLISECONDS)) |
736 |
< |
fail("ExecutorService " + exec + |
734 |
> |
pool.shutdown(); |
735 |
> |
if (!pool.awaitTermination(2 * LONG_DELAY_MS, MILLISECONDS)) |
736 |
> |
fail("ExecutorService " + pool + |
737 |
|
" did not terminate in a timely manner"); |
738 |
|
} catch (SecurityException ok) { |
739 |
|
// Allowed in case test doesn't have privs |
742 |
|
} |
743 |
|
} |
744 |
|
|
745 |
+ |
/** Like Runnable, but with the freedom to throw anything */ |
746 |
+ |
interface Thunk { public void run() throws Throwable; } |
747 |
+ |
|
748 |
+ |
/** |
749 |
+ |
* Runs all the given tasks in parallel, failing if any fail. |
750 |
+ |
* Useful for running multiple variants of tests that are |
751 |
+ |
* necessarily individually slow because they must block. |
752 |
+ |
*/ |
753 |
+ |
void testInParallel(Thunk ... thunks) { |
754 |
+ |
ExecutorService pool = Executors.newCachedThreadPool(); |
755 |
+ |
try { |
756 |
+ |
ArrayList<Future<?>> futures = new ArrayList<>(thunks.length); |
757 |
+ |
for (final Thunk thunk : thunks) |
758 |
+ |
futures.add(pool.submit(new CheckedRunnable() { |
759 |
+ |
public void realRun() throws Throwable { thunk.run();}})); |
760 |
+ |
for (Future<?> future : futures) |
761 |
+ |
try { |
762 |
+ |
assertNull(future.get(LONG_DELAY_MS, MILLISECONDS)); |
763 |
+ |
} catch (ExecutionException ex) { |
764 |
+ |
threadUnexpectedException(ex.getCause()); |
765 |
+ |
} catch (Exception ex) { |
766 |
+ |
threadUnexpectedException(ex); |
767 |
+ |
} |
768 |
+ |
} finally { |
769 |
+ |
joinPool(pool); |
770 |
+ |
} |
771 |
+ |
} |
772 |
+ |
|
773 |
|
/** |
774 |
|
* A debugging tool to print all stack traces, as jstack does. |
775 |
|
*/ |