11 |
|
import static java.util.concurrent.TimeUnit.SECONDS; |
12 |
|
|
13 |
|
import java.util.ArrayList; |
14 |
+ |
import java.util.Collection; |
15 |
+ |
import java.util.Collections; |
16 |
|
import java.util.List; |
17 |
|
import java.util.concurrent.ArrayBlockingQueue; |
18 |
|
import java.util.concurrent.BlockingQueue; |
29 |
|
import java.util.concurrent.SynchronousQueue; |
30 |
|
import java.util.concurrent.ThreadFactory; |
31 |
|
import java.util.concurrent.ThreadPoolExecutor; |
32 |
+ |
import java.util.concurrent.ThreadPoolExecutor.AbortPolicy; |
33 |
+ |
import java.util.concurrent.ThreadPoolExecutor.CallerRunsPolicy; |
34 |
+ |
import java.util.concurrent.ThreadPoolExecutor.DiscardPolicy; |
35 |
+ |
import java.util.concurrent.ThreadPoolExecutor.DiscardOldestPolicy; |
36 |
|
import java.util.concurrent.atomic.AtomicInteger; |
37 |
+ |
import java.util.concurrent.atomic.AtomicReference; |
38 |
|
|
39 |
|
import junit.framework.Test; |
40 |
|
import junit.framework.TestSuite; |
97 |
|
final Runnable task = new CheckedRunnable() { |
98 |
|
public void realRun() { done.countDown(); }}; |
99 |
|
p.execute(task); |
100 |
< |
assertTrue(done.await(LONG_DELAY_MS, MILLISECONDS)); |
100 |
> |
await(done); |
101 |
|
} |
102 |
|
} |
103 |
|
|
191 |
|
public void realRun() throws InterruptedException { |
192 |
|
threadStarted.countDown(); |
193 |
|
assertEquals(0, p.getCompletedTaskCount()); |
194 |
< |
threadProceed.await(); |
194 |
> |
await(threadProceed); |
195 |
|
threadDone.countDown(); |
196 |
|
}}); |
197 |
|
await(threadStarted); |
198 |
|
assertEquals(0, p.getCompletedTaskCount()); |
199 |
|
threadProceed.countDown(); |
200 |
< |
threadDone.await(); |
200 |
> |
await(threadDone); |
201 |
|
long startTime = System.nanoTime(); |
202 |
|
while (p.getCompletedTaskCount() != 1) { |
203 |
|
if (millisElapsedSince(startTime) > LONG_DELAY_MS) |
289 |
|
LONG_DELAY_MS, MILLISECONDS, |
290 |
|
new ArrayBlockingQueue<Runnable>(10)); |
291 |
|
try (PoolCleaner cleaner = cleaner(p)) { |
292 |
< |
assertTrue(p.getRejectedExecutionHandler() |
286 |
< |
instanceof ThreadPoolExecutor.AbortPolicy); |
292 |
> |
assertTrue(p.getRejectedExecutionHandler() instanceof AbortPolicy); |
293 |
|
} |
294 |
|
} |
295 |
|
|
476 |
|
assertFalse(p.awaitTermination(Long.MIN_VALUE, MILLISECONDS)); |
477 |
|
assertFalse(p.awaitTermination(-1L, NANOSECONDS)); |
478 |
|
assertFalse(p.awaitTermination(-1L, MILLISECONDS)); |
479 |
< |
assertFalse(p.awaitTermination(0L, NANOSECONDS)); |
480 |
< |
assertFalse(p.awaitTermination(0L, MILLISECONDS)); |
479 |
> |
assertFalse(p.awaitTermination(randomExpiredTimeout(), |
480 |
> |
randomTimeUnit())); |
481 |
|
long timeoutNanos = 999999L; |
482 |
|
long startTime = System.nanoTime(); |
483 |
|
assertFalse(p.awaitTermination(timeoutNanos, NANOSECONDS)); |
1151 |
|
LONG_DELAY_MS, |
1152 |
|
MILLISECONDS, |
1153 |
|
new ArrayBlockingQueue<Runnable>(1), |
1154 |
< |
new ThreadPoolExecutor.CallerRunsPolicy()); |
1154 |
> |
new CallerRunsPolicy()); |
1155 |
|
try (PoolCleaner cleaner = cleaner(p)) { |
1156 |
|
final CountDownLatch done = new CountDownLatch(1); |
1157 |
|
Runnable blocker = new CheckedRunnable() { |
1183 |
|
new ThreadPoolExecutor(1, 1, |
1184 |
|
LONG_DELAY_MS, MILLISECONDS, |
1185 |
|
new ArrayBlockingQueue<Runnable>(1), |
1186 |
< |
new ThreadPoolExecutor.DiscardPolicy()); |
1186 |
> |
new DiscardPolicy()); |
1187 |
|
try (PoolCleaner cleaner = cleaner(p, done)) { |
1188 |
|
p.execute(awaiter(done)); |
1189 |
|
|
1209 |
|
new ThreadPoolExecutor(1, 1, |
1210 |
|
LONG_DELAY_MS, MILLISECONDS, |
1211 |
|
new ArrayBlockingQueue<Runnable>(1), |
1212 |
< |
new ThreadPoolExecutor.DiscardOldestPolicy()); |
1212 |
> |
new DiscardOldestPolicy()); |
1213 |
|
try (PoolCleaner cleaner = cleaner(p, done)) { |
1214 |
|
assertEquals(LatchAwaiter.NEW, r1.state); |
1215 |
|
assertEquals(LatchAwaiter.NEW, r2.state); |
1247 |
|
* execute using CallerRunsPolicy drops task on shutdown |
1248 |
|
*/ |
1249 |
|
public void testCallerRunsOnShutdown() { |
1250 |
< |
RejectedExecutionHandler h = new ThreadPoolExecutor.CallerRunsPolicy(); |
1250 |
> |
RejectedExecutionHandler h = new CallerRunsPolicy(); |
1251 |
|
final ThreadPoolExecutor p = |
1252 |
|
new ThreadPoolExecutor(1, 1, |
1253 |
|
LONG_DELAY_MS, MILLISECONDS, |
1269 |
|
new ThreadPoolExecutor(1, 1, |
1270 |
|
LONG_DELAY_MS, MILLISECONDS, |
1271 |
|
new ArrayBlockingQueue<Runnable>(1), |
1272 |
< |
new ThreadPoolExecutor.DiscardPolicy()); |
1272 |
> |
new DiscardPolicy()); |
1273 |
|
|
1274 |
|
try { p.shutdown(); } catch (SecurityException ok) { return; } |
1275 |
|
try (PoolCleaner cleaner = cleaner(p)) { |
1287 |
|
new ThreadPoolExecutor(1, 1, |
1288 |
|
LONG_DELAY_MS, MILLISECONDS, |
1289 |
|
new ArrayBlockingQueue<Runnable>(1), |
1290 |
< |
new ThreadPoolExecutor.DiscardOldestPolicy()); |
1290 |
> |
new DiscardOldestPolicy()); |
1291 |
|
|
1292 |
|
try { p.shutdown(); } catch (SecurityException ok) { return; } |
1293 |
|
try (PoolCleaner cleaner = cleaner(p)) { |
1501 |
|
} |
1502 |
|
|
1503 |
|
/** |
1504 |
< |
* invokeAny(empty collection) throws IAE |
1504 |
> |
* invokeAny(empty collection) throws IllegalArgumentException |
1505 |
|
*/ |
1506 |
|
public void testInvokeAny2() throws Exception { |
1507 |
|
final ExecutorService e = |
1591 |
|
} |
1592 |
|
|
1593 |
|
/** |
1594 |
< |
* invokeAll(empty collection) returns empty collection |
1594 |
> |
* invokeAll(empty collection) returns empty list |
1595 |
|
*/ |
1596 |
|
public void testInvokeAll2() throws InterruptedException { |
1597 |
|
final ExecutorService e = |
1598 |
|
new ThreadPoolExecutor(2, 2, |
1599 |
|
LONG_DELAY_MS, MILLISECONDS, |
1600 |
|
new ArrayBlockingQueue<Runnable>(10)); |
1601 |
+ |
final Collection<Callable<String>> emptyCollection |
1602 |
+ |
= Collections.emptyList(); |
1603 |
|
try (PoolCleaner cleaner = cleaner(e)) { |
1604 |
< |
List<Future<String>> r = e.invokeAll(new ArrayList<Callable<String>>()); |
1604 |
> |
List<Future<String>> r = e.invokeAll(emptyCollection); |
1605 |
|
assertTrue(r.isEmpty()); |
1606 |
|
} |
1607 |
|
} |
1676 |
|
new ArrayBlockingQueue<Runnable>(10)); |
1677 |
|
try (PoolCleaner cleaner = cleaner(e)) { |
1678 |
|
try { |
1679 |
< |
e.invokeAny(null, MEDIUM_DELAY_MS, MILLISECONDS); |
1679 |
> |
e.invokeAny(null, randomTimeout(), randomTimeUnit()); |
1680 |
|
shouldThrow(); |
1681 |
|
} catch (NullPointerException success) {} |
1682 |
|
} |
1694 |
|
List<Callable<String>> l = new ArrayList<>(); |
1695 |
|
l.add(new StringTask()); |
1696 |
|
try { |
1697 |
< |
e.invokeAny(l, MEDIUM_DELAY_MS, null); |
1697 |
> |
e.invokeAny(l, randomTimeout(), null); |
1698 |
|
shouldThrow(); |
1699 |
|
} catch (NullPointerException success) {} |
1700 |
|
} |
1701 |
|
} |
1702 |
|
|
1703 |
|
/** |
1704 |
< |
* timed invokeAny(empty collection) throws IAE |
1704 |
> |
* timed invokeAny(empty collection) throws IllegalArgumentException |
1705 |
|
*/ |
1706 |
|
public void testTimedInvokeAny2() throws Exception { |
1707 |
|
final ExecutorService e = |
1711 |
|
try (PoolCleaner cleaner = cleaner(e)) { |
1712 |
|
try { |
1713 |
|
e.invokeAny(new ArrayList<Callable<String>>(), |
1714 |
< |
MEDIUM_DELAY_MS, MILLISECONDS); |
1714 |
> |
randomTimeout(), randomTimeUnit()); |
1715 |
|
shouldThrow(); |
1716 |
|
} catch (IllegalArgumentException success) {} |
1717 |
|
} |
1718 |
|
} |
1719 |
|
|
1720 |
|
/** |
1721 |
< |
* timed invokeAny(c) throws NPE if c has null elements |
1721 |
> |
* timed invokeAny(c) throws NullPointerException if c has null elements |
1722 |
|
*/ |
1723 |
|
public void testTimedInvokeAny3() throws Exception { |
1724 |
|
final CountDownLatch latch = new CountDownLatch(1); |
1731 |
|
l.add(latchAwaitingStringTask(latch)); |
1732 |
|
l.add(null); |
1733 |
|
try { |
1734 |
< |
e.invokeAny(l, MEDIUM_DELAY_MS, MILLISECONDS); |
1734 |
> |
e.invokeAny(l, randomTimeout(), randomTimeUnit()); |
1735 |
|
shouldThrow(); |
1736 |
|
} catch (NullPointerException success) {} |
1737 |
|
latch.countDown(); |
1789 |
|
new ArrayBlockingQueue<Runnable>(10)); |
1790 |
|
try (PoolCleaner cleaner = cleaner(e)) { |
1791 |
|
try { |
1792 |
< |
e.invokeAll(null, MEDIUM_DELAY_MS, MILLISECONDS); |
1792 |
> |
e.invokeAll(null, randomTimeout(), randomTimeUnit()); |
1793 |
|
shouldThrow(); |
1794 |
|
} catch (NullPointerException success) {} |
1795 |
|
} |
1807 |
|
List<Callable<String>> l = new ArrayList<>(); |
1808 |
|
l.add(new StringTask()); |
1809 |
|
try { |
1810 |
< |
e.invokeAll(l, MEDIUM_DELAY_MS, null); |
1810 |
> |
e.invokeAll(l, randomTimeout(), null); |
1811 |
|
shouldThrow(); |
1812 |
|
} catch (NullPointerException success) {} |
1813 |
|
} |
1814 |
|
} |
1815 |
|
|
1816 |
|
/** |
1817 |
< |
* timed invokeAll(empty collection) returns empty collection |
1817 |
> |
* timed invokeAll(empty collection) returns empty list |
1818 |
|
*/ |
1819 |
|
public void testTimedInvokeAll2() throws InterruptedException { |
1820 |
|
final ExecutorService e = |
1821 |
|
new ThreadPoolExecutor(2, 2, |
1822 |
|
LONG_DELAY_MS, MILLISECONDS, |
1823 |
|
new ArrayBlockingQueue<Runnable>(10)); |
1824 |
+ |
final Collection<Callable<String>> emptyCollection |
1825 |
+ |
= Collections.emptyList(); |
1826 |
|
try (PoolCleaner cleaner = cleaner(e)) { |
1827 |
< |
List<Future<String>> r = e.invokeAll(new ArrayList<Callable<String>>(), |
1828 |
< |
MEDIUM_DELAY_MS, MILLISECONDS); |
1827 |
> |
List<Future<String>> r = |
1828 |
> |
e.invokeAll(emptyCollection, randomTimeout(), randomTimeUnit()); |
1829 |
|
assertTrue(r.isEmpty()); |
1830 |
|
} |
1831 |
|
} |
1843 |
|
l.add(new StringTask()); |
1844 |
|
l.add(null); |
1845 |
|
try { |
1846 |
< |
e.invokeAll(l, MEDIUM_DELAY_MS, MILLISECONDS); |
1846 |
> |
e.invokeAll(l, randomTimeout(), randomTimeUnit()); |
1847 |
|
shouldThrow(); |
1848 |
|
} catch (NullPointerException success) {} |
1849 |
|
} |
1951 |
|
public void realRun() { |
1952 |
|
done.countDown(); |
1953 |
|
}}); |
1954 |
< |
assertTrue(done.await(LONG_DELAY_MS, MILLISECONDS)); |
1954 |
> |
await(done); |
1955 |
|
} |
1956 |
|
} |
1957 |
|
|
2044 |
|
} |
2045 |
|
} |
2046 |
|
// enough time to run all tasks |
2047 |
< |
assertTrue(done.await(nTasks * SHORT_DELAY_MS, MILLISECONDS)); |
2047 |
> |
await(done, nTasks * SHORT_DELAY_MS); |
2048 |
|
} |
2049 |
|
} |
2050 |
|
|
2085 |
|
} |
2086 |
|
} |
2087 |
|
|
2088 |
+ |
/** Directly test simple ThreadPoolExecutor RejectedExecutionHandlers. */ |
2089 |
+ |
public void testStandardRejectedExecutionHandlers() { |
2090 |
+ |
final ThreadPoolExecutor p = |
2091 |
+ |
new ThreadPoolExecutor(1, 1, 1, SECONDS, |
2092 |
+ |
new ArrayBlockingQueue<Runnable>(1)); |
2093 |
+ |
final AtomicReference<Thread> thread = new AtomicReference<>(); |
2094 |
+ |
final Runnable r = new Runnable() { public void run() { |
2095 |
+ |
thread.set(Thread.currentThread()); }}; |
2096 |
+ |
|
2097 |
+ |
try { |
2098 |
+ |
new AbortPolicy().rejectedExecution(r, p); |
2099 |
+ |
shouldThrow(); |
2100 |
+ |
} catch (RejectedExecutionException success) {} |
2101 |
+ |
assertNull(thread.get()); |
2102 |
+ |
|
2103 |
+ |
new DiscardPolicy().rejectedExecution(r, p); |
2104 |
+ |
assertNull(thread.get()); |
2105 |
+ |
|
2106 |
+ |
new CallerRunsPolicy().rejectedExecution(r, p); |
2107 |
+ |
assertSame(Thread.currentThread(), thread.get()); |
2108 |
+ |
|
2109 |
+ |
// check that pool was not perturbed by handlers |
2110 |
+ |
assertTrue(p.getRejectedExecutionHandler() instanceof AbortPolicy); |
2111 |
+ |
assertEquals(0, p.getTaskCount()); |
2112 |
+ |
assertTrue(p.getQueue().isEmpty()); |
2113 |
+ |
} |
2114 |
+ |
|
2115 |
|
} |