76 |
|
import java.util.concurrent.CountDownLatch; |
77 |
|
import java.util.concurrent.CyclicBarrier; |
78 |
|
import java.util.concurrent.ExecutionException; |
79 |
+ |
import java.util.concurrent.Executor; |
80 |
|
import java.util.concurrent.Executors; |
81 |
|
import java.util.concurrent.ExecutorService; |
82 |
|
import java.util.concurrent.ForkJoinPool; |
83 |
|
import java.util.concurrent.Future; |
84 |
+ |
import java.util.concurrent.FutureTask; |
85 |
|
import java.util.concurrent.RecursiveAction; |
86 |
|
import java.util.concurrent.RecursiveTask; |
87 |
+ |
import java.util.concurrent.RejectedExecutionException; |
88 |
|
import java.util.concurrent.RejectedExecutionHandler; |
89 |
|
import java.util.concurrent.Semaphore; |
90 |
+ |
import java.util.concurrent.ScheduledExecutorService; |
91 |
+ |
import java.util.concurrent.ScheduledFuture; |
92 |
|
import java.util.concurrent.SynchronousQueue; |
93 |
|
import java.util.concurrent.ThreadFactory; |
94 |
|
import java.util.concurrent.ThreadLocalRandom; |
95 |
|
import java.util.concurrent.ThreadPoolExecutor; |
96 |
+ |
import java.util.concurrent.TimeUnit; |
97 |
|
import java.util.concurrent.TimeoutException; |
98 |
|
import java.util.concurrent.atomic.AtomicBoolean; |
99 |
|
import java.util.concurrent.atomic.AtomicReference; |
644 |
|
public static long MEDIUM_DELAY_MS; |
645 |
|
public static long LONG_DELAY_MS; |
646 |
|
|
647 |
+ |
private static final long RANDOM_TIMEOUT; |
648 |
+ |
private static final long RANDOM_EXPIRED_TIMEOUT; |
649 |
+ |
private static final TimeUnit RANDOM_TIMEUNIT; |
650 |
+ |
static { |
651 |
+ |
ThreadLocalRandom rnd = ThreadLocalRandom.current(); |
652 |
+ |
long[] timeouts = { Long.MIN_VALUE, -1, 0, 1, Long.MAX_VALUE }; |
653 |
+ |
RANDOM_TIMEOUT = timeouts[rnd.nextInt(timeouts.length)]; |
654 |
+ |
RANDOM_EXPIRED_TIMEOUT = timeouts[rnd.nextInt(3)]; |
655 |
+ |
TimeUnit[] timeUnits = TimeUnit.values(); |
656 |
+ |
RANDOM_TIMEUNIT = timeUnits[rnd.nextInt(timeUnits.length)]; |
657 |
+ |
} |
658 |
+ |
|
659 |
+ |
/** |
660 |
+ |
* Returns a timeout for use when any value at all will do. |
661 |
+ |
*/ |
662 |
+ |
static long randomTimeout() { return RANDOM_TIMEOUT; } |
663 |
+ |
|
664 |
+ |
/** |
665 |
+ |
* Returns a timeout that means "no waiting", i.e. not positive. |
666 |
+ |
*/ |
667 |
+ |
static long randomExpiredTimeout() { return RANDOM_EXPIRED_TIMEOUT; } |
668 |
+ |
|
669 |
+ |
/** |
670 |
+ |
* Returns a random non-null TimeUnit. |
671 |
+ |
*/ |
672 |
+ |
static TimeUnit randomTimeUnit() { return RANDOM_TIMEUNIT; } |
673 |
+ |
|
674 |
|
/** |
675 |
|
* Returns the shortest timed delay. This can be scaled up for |
676 |
|
* slow machines using the jsr166.delay.factor system property, |
1116 |
|
} |
1117 |
|
|
1118 |
|
/** |
1086 |
– |
* Checks that thread does not terminate within the default |
1087 |
– |
* millisecond delay of {@code timeoutMillis()}. |
1088 |
– |
*/ |
1089 |
– |
void assertThreadStaysAlive(Thread thread) { |
1090 |
– |
assertThreadStaysAlive(thread, timeoutMillis()); |
1091 |
– |
} |
1092 |
– |
|
1093 |
– |
/** |
1094 |
– |
* Checks that thread does not terminate within the given millisecond delay. |
1095 |
– |
*/ |
1096 |
– |
void assertThreadStaysAlive(Thread thread, long millis) { |
1097 |
– |
try { |
1098 |
– |
// No need to optimize the failing case via Thread.join. |
1099 |
– |
delay(millis); |
1100 |
– |
assertTrue(thread.isAlive()); |
1101 |
– |
} catch (InterruptedException fail) { |
1102 |
– |
threadFail("Unexpected InterruptedException"); |
1103 |
– |
} |
1104 |
– |
} |
1105 |
– |
|
1106 |
– |
/** |
1119 |
|
* Checks that future.get times out, with the default timeout of |
1120 |
|
* {@code timeoutMillis()}. |
1121 |
|
*/ |
1658 |
|
public String call() { throw new NullPointerException(); } |
1659 |
|
} |
1660 |
|
|
1649 |
– |
public static class CallableOne implements Callable<Integer> { |
1650 |
– |
public Integer call() { return one; } |
1651 |
– |
} |
1652 |
– |
|
1653 |
– |
public class ShortRunnable extends CheckedRunnable { |
1654 |
– |
protected void realRun() throws Throwable { |
1655 |
– |
delay(SHORT_DELAY_MS); |
1656 |
– |
} |
1657 |
– |
} |
1658 |
– |
|
1659 |
– |
public class ShortInterruptedRunnable extends CheckedInterruptedRunnable { |
1660 |
– |
protected void realRun() throws InterruptedException { |
1661 |
– |
delay(SHORT_DELAY_MS); |
1662 |
– |
} |
1663 |
– |
} |
1664 |
– |
|
1665 |
– |
public class SmallRunnable extends CheckedRunnable { |
1666 |
– |
protected void realRun() throws Throwable { |
1667 |
– |
delay(SMALL_DELAY_MS); |
1668 |
– |
} |
1669 |
– |
} |
1670 |
– |
|
1661 |
|
public class SmallPossiblyInterruptedRunnable extends CheckedRunnable { |
1662 |
|
protected void realRun() { |
1663 |
|
try { |
1666 |
|
} |
1667 |
|
} |
1668 |
|
|
1679 |
– |
public class SmallCallable extends CheckedCallable { |
1680 |
– |
protected Object realCall() throws InterruptedException { |
1681 |
– |
delay(SMALL_DELAY_MS); |
1682 |
– |
return Boolean.TRUE; |
1683 |
– |
} |
1684 |
– |
} |
1685 |
– |
|
1686 |
– |
public class MediumRunnable extends CheckedRunnable { |
1687 |
– |
protected void realRun() throws Throwable { |
1688 |
– |
delay(MEDIUM_DELAY_MS); |
1689 |
– |
} |
1690 |
– |
} |
1691 |
– |
|
1692 |
– |
public class MediumInterruptedRunnable extends CheckedInterruptedRunnable { |
1693 |
– |
protected void realRun() throws InterruptedException { |
1694 |
– |
delay(MEDIUM_DELAY_MS); |
1695 |
– |
} |
1696 |
– |
} |
1697 |
– |
|
1669 |
|
public Runnable possiblyInterruptedRunnable(final long timeoutMillis) { |
1670 |
|
return new CheckedRunnable() { |
1671 |
|
protected void realRun() { |
1675 |
|
}}; |
1676 |
|
} |
1677 |
|
|
1707 |
– |
public class MediumPossiblyInterruptedRunnable extends CheckedRunnable { |
1708 |
– |
protected void realRun() { |
1709 |
– |
try { |
1710 |
– |
delay(MEDIUM_DELAY_MS); |
1711 |
– |
} catch (InterruptedException ok) {} |
1712 |
– |
} |
1713 |
– |
} |
1714 |
– |
|
1715 |
– |
public class LongPossiblyInterruptedRunnable extends CheckedRunnable { |
1716 |
– |
protected void realRun() { |
1717 |
– |
try { |
1718 |
– |
delay(LONG_DELAY_MS); |
1719 |
– |
} catch (InterruptedException ok) {} |
1720 |
– |
} |
1721 |
– |
} |
1722 |
– |
|
1678 |
|
/** |
1679 |
|
* For use as ThreadFactory in constructors |
1680 |
|
*/ |
1688 |
|
boolean isDone(); |
1689 |
|
} |
1690 |
|
|
1736 |
– |
public static TrackedRunnable trackedRunnable(final long timeoutMillis) { |
1737 |
– |
return new TrackedRunnable() { |
1738 |
– |
private volatile boolean done = false; |
1739 |
– |
public boolean isDone() { return done; } |
1740 |
– |
public void run() { |
1741 |
– |
try { |
1742 |
– |
delay(timeoutMillis); |
1743 |
– |
done = true; |
1744 |
– |
} catch (InterruptedException ok) {} |
1745 |
– |
} |
1746 |
– |
}; |
1747 |
– |
} |
1748 |
– |
|
1749 |
– |
public static class TrackedShortRunnable implements Runnable { |
1750 |
– |
public volatile boolean done = false; |
1751 |
– |
public void run() { |
1752 |
– |
try { |
1753 |
– |
delay(SHORT_DELAY_MS); |
1754 |
– |
done = true; |
1755 |
– |
} catch (InterruptedException ok) {} |
1756 |
– |
} |
1757 |
– |
} |
1758 |
– |
|
1759 |
– |
public static class TrackedSmallRunnable implements Runnable { |
1760 |
– |
public volatile boolean done = false; |
1761 |
– |
public void run() { |
1762 |
– |
try { |
1763 |
– |
delay(SMALL_DELAY_MS); |
1764 |
– |
done = true; |
1765 |
– |
} catch (InterruptedException ok) {} |
1766 |
– |
} |
1767 |
– |
} |
1768 |
– |
|
1769 |
– |
public static class TrackedMediumRunnable implements Runnable { |
1770 |
– |
public volatile boolean done = false; |
1771 |
– |
public void run() { |
1772 |
– |
try { |
1773 |
– |
delay(MEDIUM_DELAY_MS); |
1774 |
– |
done = true; |
1775 |
– |
} catch (InterruptedException ok) {} |
1776 |
– |
} |
1777 |
– |
} |
1778 |
– |
|
1779 |
– |
public static class TrackedLongRunnable implements Runnable { |
1780 |
– |
public volatile boolean done = false; |
1781 |
– |
public void run() { |
1782 |
– |
try { |
1783 |
– |
delay(LONG_DELAY_MS); |
1784 |
– |
done = true; |
1785 |
– |
} catch (InterruptedException ok) {} |
1786 |
– |
} |
1787 |
– |
} |
1788 |
– |
|
1691 |
|
public static class TrackedNoOpRunnable implements Runnable { |
1692 |
|
public volatile boolean done = false; |
1693 |
|
public void run() { |
1695 |
|
} |
1696 |
|
} |
1697 |
|
|
1796 |
– |
public static class TrackedCallable implements Callable { |
1797 |
– |
public volatile boolean done = false; |
1798 |
– |
public Object call() { |
1799 |
– |
try { |
1800 |
– |
delay(SMALL_DELAY_MS); |
1801 |
– |
done = true; |
1802 |
– |
} catch (InterruptedException ok) {} |
1803 |
– |
return Boolean.TRUE; |
1804 |
– |
} |
1805 |
– |
} |
1806 |
– |
|
1698 |
|
/** |
1699 |
|
* Analog of CheckedRunnable for RecursiveAction |
1700 |
|
*/ |
1761 |
|
assertEquals(0, q.size()); |
1762 |
|
assertNull(q.peek()); |
1763 |
|
assertNull(q.poll()); |
1764 |
< |
assertNull(q.poll(0, MILLISECONDS)); |
1764 |
> |
assertNull(q.poll(randomExpiredTimeout(), randomTimeUnit())); |
1765 |
|
assertEquals(q.toString(), "[]"); |
1766 |
|
assertTrue(Arrays.equals(q.toArray(), new Object[0])); |
1767 |
|
assertFalse(q.iterator().hasNext()); |
1912 |
|
static <T> void shuffle(T[] array) { |
1913 |
|
Collections.shuffle(Arrays.asList(array), ThreadLocalRandom.current()); |
1914 |
|
} |
1915 |
+ |
|
1916 |
+ |
/** |
1917 |
+ |
* Returns the same String as would be returned by {@link |
1918 |
+ |
* Object#toString}, whether or not the given object's class |
1919 |
+ |
* overrides toString(). |
1920 |
+ |
* |
1921 |
+ |
* @see System#identityHashCode |
1922 |
+ |
*/ |
1923 |
+ |
static String identityString(Object x) { |
1924 |
+ |
return x.getClass().getName() |
1925 |
+ |
+ "@" + Integer.toHexString(System.identityHashCode(x)); |
1926 |
+ |
} |
1927 |
+ |
|
1928 |
+ |
// --- Shared assertions for Executor tests --- |
1929 |
+ |
|
1930 |
+ |
/** |
1931 |
+ |
* Returns maximum number of tasks that can be submitted to given |
1932 |
+ |
* pool (with bounded queue) before saturation (when submission |
1933 |
+ |
* throws RejectedExecutionException). |
1934 |
+ |
*/ |
1935 |
+ |
static final int saturatedSize(ThreadPoolExecutor pool) { |
1936 |
+ |
BlockingQueue<Runnable> q = pool.getQueue(); |
1937 |
+ |
return pool.getMaximumPoolSize() + q.size() + q.remainingCapacity(); |
1938 |
+ |
} |
1939 |
+ |
|
1940 |
+ |
@SuppressWarnings("FutureReturnValueIgnored") |
1941 |
+ |
void assertNullTaskSubmissionThrowsNullPointerException(Executor e) { |
1942 |
+ |
try { |
1943 |
+ |
e.execute((Runnable) null); |
1944 |
+ |
shouldThrow(); |
1945 |
+ |
} catch (NullPointerException success) {} |
1946 |
+ |
|
1947 |
+ |
if (! (e instanceof ExecutorService)) return; |
1948 |
+ |
ExecutorService es = (ExecutorService) e; |
1949 |
+ |
try { |
1950 |
+ |
es.submit((Runnable) null); |
1951 |
+ |
shouldThrow(); |
1952 |
+ |
} catch (NullPointerException success) {} |
1953 |
+ |
try { |
1954 |
+ |
es.submit((Runnable) null, Boolean.TRUE); |
1955 |
+ |
shouldThrow(); |
1956 |
+ |
} catch (NullPointerException success) {} |
1957 |
+ |
try { |
1958 |
+ |
es.submit((Callable) null); |
1959 |
+ |
shouldThrow(); |
1960 |
+ |
} catch (NullPointerException success) {} |
1961 |
+ |
|
1962 |
+ |
if (! (e instanceof ScheduledExecutorService)) return; |
1963 |
+ |
ScheduledExecutorService ses = (ScheduledExecutorService) e; |
1964 |
+ |
try { |
1965 |
+ |
ses.schedule((Runnable) null, |
1966 |
+ |
randomTimeout(), randomTimeUnit()); |
1967 |
+ |
shouldThrow(); |
1968 |
+ |
} catch (NullPointerException success) {} |
1969 |
+ |
try { |
1970 |
+ |
ses.schedule((Callable) null, |
1971 |
+ |
randomTimeout(), randomTimeUnit()); |
1972 |
+ |
shouldThrow(); |
1973 |
+ |
} catch (NullPointerException success) {} |
1974 |
+ |
try { |
1975 |
+ |
ses.scheduleAtFixedRate((Runnable) null, |
1976 |
+ |
randomTimeout(), LONG_DELAY_MS, MILLISECONDS); |
1977 |
+ |
shouldThrow(); |
1978 |
+ |
} catch (NullPointerException success) {} |
1979 |
+ |
try { |
1980 |
+ |
ses.scheduleWithFixedDelay((Runnable) null, |
1981 |
+ |
randomTimeout(), LONG_DELAY_MS, MILLISECONDS); |
1982 |
+ |
shouldThrow(); |
1983 |
+ |
} catch (NullPointerException success) {} |
1984 |
+ |
} |
1985 |
+ |
|
1986 |
+ |
void setRejectedExecutionHandler( |
1987 |
+ |
ThreadPoolExecutor p, RejectedExecutionHandler handler) { |
1988 |
+ |
p.setRejectedExecutionHandler(handler); |
1989 |
+ |
assertSame(handler, p.getRejectedExecutionHandler()); |
1990 |
+ |
} |
1991 |
+ |
|
1992 |
+ |
void assertTaskSubmissionsAreRejected(ThreadPoolExecutor p) { |
1993 |
+ |
final RejectedExecutionHandler savedHandler = p.getRejectedExecutionHandler(); |
1994 |
+ |
final long savedTaskCount = p.getTaskCount(); |
1995 |
+ |
final long savedCompletedTaskCount = p.getCompletedTaskCount(); |
1996 |
+ |
final int savedQueueSize = p.getQueue().size(); |
1997 |
+ |
final boolean stock = (p.getClass().getClassLoader() == null); |
1998 |
+ |
|
1999 |
+ |
Runnable r = () -> {}; |
2000 |
+ |
Callable<Boolean> c = () -> Boolean.TRUE; |
2001 |
+ |
|
2002 |
+ |
class Recorder implements RejectedExecutionHandler { |
2003 |
+ |
public volatile Runnable r = null; |
2004 |
+ |
public volatile ThreadPoolExecutor p = null; |
2005 |
+ |
public void reset() { r = null; p = null; } |
2006 |
+ |
public void rejectedExecution(Runnable r, ThreadPoolExecutor p) { |
2007 |
+ |
assertNull(this.r); |
2008 |
+ |
assertNull(this.p); |
2009 |
+ |
this.r = r; |
2010 |
+ |
this.p = p; |
2011 |
+ |
} |
2012 |
+ |
} |
2013 |
+ |
|
2014 |
+ |
// check custom handler is invoked exactly once per task |
2015 |
+ |
Recorder recorder = new Recorder(); |
2016 |
+ |
setRejectedExecutionHandler(p, recorder); |
2017 |
+ |
for (int i = 2; i--> 0; ) { |
2018 |
+ |
recorder.reset(); |
2019 |
+ |
p.execute(r); |
2020 |
+ |
if (stock && p.getClass() == ThreadPoolExecutor.class) |
2021 |
+ |
assertSame(r, recorder.r); |
2022 |
+ |
assertSame(p, recorder.p); |
2023 |
+ |
|
2024 |
+ |
recorder.reset(); |
2025 |
+ |
assertFalse(p.submit(r).isDone()); |
2026 |
+ |
if (stock) assertTrue(!((FutureTask) recorder.r).isDone()); |
2027 |
+ |
assertSame(p, recorder.p); |
2028 |
+ |
|
2029 |
+ |
recorder.reset(); |
2030 |
+ |
assertFalse(p.submit(r, Boolean.TRUE).isDone()); |
2031 |
+ |
if (stock) assertTrue(!((FutureTask) recorder.r).isDone()); |
2032 |
+ |
assertSame(p, recorder.p); |
2033 |
+ |
|
2034 |
+ |
recorder.reset(); |
2035 |
+ |
assertFalse(p.submit(c).isDone()); |
2036 |
+ |
if (stock) assertTrue(!((FutureTask) recorder.r).isDone()); |
2037 |
+ |
assertSame(p, recorder.p); |
2038 |
+ |
|
2039 |
+ |
if (p instanceof ScheduledExecutorService) { |
2040 |
+ |
ScheduledExecutorService s = (ScheduledExecutorService) p; |
2041 |
+ |
ScheduledFuture<?> future; |
2042 |
+ |
|
2043 |
+ |
recorder.reset(); |
2044 |
+ |
future = s.schedule(r, randomTimeout(), randomTimeUnit()); |
2045 |
+ |
assertFalse(future.isDone()); |
2046 |
+ |
if (stock) assertTrue(!((FutureTask) recorder.r).isDone()); |
2047 |
+ |
assertSame(p, recorder.p); |
2048 |
+ |
|
2049 |
+ |
recorder.reset(); |
2050 |
+ |
future = s.schedule(c, randomTimeout(), randomTimeUnit()); |
2051 |
+ |
assertFalse(future.isDone()); |
2052 |
+ |
if (stock) assertTrue(!((FutureTask) recorder.r).isDone()); |
2053 |
+ |
assertSame(p, recorder.p); |
2054 |
+ |
|
2055 |
+ |
recorder.reset(); |
2056 |
+ |
future = s.scheduleAtFixedRate(r, randomTimeout(), LONG_DELAY_MS, MILLISECONDS); |
2057 |
+ |
assertFalse(future.isDone()); |
2058 |
+ |
if (stock) assertTrue(!((FutureTask) recorder.r).isDone()); |
2059 |
+ |
assertSame(p, recorder.p); |
2060 |
+ |
|
2061 |
+ |
recorder.reset(); |
2062 |
+ |
future = s.scheduleWithFixedDelay(r, randomTimeout(), LONG_DELAY_MS, MILLISECONDS); |
2063 |
+ |
assertFalse(future.isDone()); |
2064 |
+ |
if (stock) assertTrue(!((FutureTask) recorder.r).isDone()); |
2065 |
+ |
assertSame(p, recorder.p); |
2066 |
+ |
} |
2067 |
+ |
} |
2068 |
+ |
|
2069 |
+ |
// Checking our custom handler above should be sufficient, but |
2070 |
+ |
// we add some integration tests of standard handlers. |
2071 |
+ |
final AtomicReference<Thread> thread = new AtomicReference<>(); |
2072 |
+ |
final Runnable setThread = () -> thread.set(Thread.currentThread()); |
2073 |
+ |
|
2074 |
+ |
setRejectedExecutionHandler(p, new ThreadPoolExecutor.AbortPolicy()); |
2075 |
+ |
try { |
2076 |
+ |
p.execute(setThread); |
2077 |
+ |
shouldThrow(); |
2078 |
+ |
} catch (RejectedExecutionException success) {} |
2079 |
+ |
assertNull(thread.get()); |
2080 |
+ |
|
2081 |
+ |
setRejectedExecutionHandler(p, new ThreadPoolExecutor.DiscardPolicy()); |
2082 |
+ |
p.execute(setThread); |
2083 |
+ |
assertNull(thread.get()); |
2084 |
+ |
|
2085 |
+ |
setRejectedExecutionHandler(p, new ThreadPoolExecutor.CallerRunsPolicy()); |
2086 |
+ |
p.execute(setThread); |
2087 |
+ |
if (p.isShutdown()) |
2088 |
+ |
assertNull(thread.get()); |
2089 |
+ |
else |
2090 |
+ |
assertSame(Thread.currentThread(), thread.get()); |
2091 |
+ |
|
2092 |
+ |
setRejectedExecutionHandler(p, savedHandler); |
2093 |
+ |
|
2094 |
+ |
// check that pool was not perturbed by handlers |
2095 |
+ |
assertEquals(savedTaskCount, p.getTaskCount()); |
2096 |
+ |
assertEquals(savedCompletedTaskCount, p.getCompletedTaskCount()); |
2097 |
+ |
assertEquals(savedQueueSize, p.getQueue().size()); |
2098 |
+ |
} |
2099 |
|
} |