ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/jsr166/jsr166/src/main/java/util/concurrent/FutureTask.java
Revision: 1.98
Committed: Fri Dec 2 15:47:22 2011 UTC (12 years, 6 months ago) by jsr166
Branch: MAIN
Changes since 1.97: +3 -4 lines
Log Message:
avoid introducing locals just for warning suppression

File Contents

# User Rev Content
1 tim 1.1 /*
2 dl 1.2 * Written by Doug Lea with assistance from members of JCP JSR-166
3 dl 1.23 * Expert Group and released to the public domain, as explained at
4 jsr166 1.59 * http://creativecommons.org/publicdomain/zero/1.0/
5 tim 1.1 */
6    
7     package java.util.concurrent;
8 dl 1.62 import java.util.concurrent.locks.LockSupport;
9 dl 1.13
10 tim 1.1 /**
11 dl 1.8 * A cancellable asynchronous computation. This class provides a base
12     * implementation of {@link Future}, with methods to start and cancel
13     * a computation, query to see if the computation is complete, and
14 dl 1.4 * retrieve the result of the computation. The result can only be
15 jsr166 1.64 * retrieved when the computation has completed; the {@code get}
16     * methods will block if the computation has not yet completed. Once
17 dl 1.8 * the computation has completed, the computation cannot be restarted
18 jsr166 1.64 * or cancelled (unless the computation is invoked using
19     * {@link #runAndReset}).
20 tim 1.1 *
21 jsr166 1.64 * <p>A {@code FutureTask} can be used to wrap a {@link Callable} or
22     * {@link Runnable} object. Because {@code FutureTask} implements
23     * {@code Runnable}, a {@code FutureTask} can be submitted to an
24     * {@link Executor} for execution.
25 tim 1.1 *
26 dl 1.14 * <p>In addition to serving as a standalone class, this class provides
27 jsr166 1.64 * {@code protected} functionality that may be useful when creating
28 dl 1.14 * customized task classes.
29     *
30 tim 1.1 * @since 1.5
31 dl 1.4 * @author Doug Lea
32 jsr166 1.64 * @param <V> The result type returned by this FutureTask's {@code get} methods
33 tim 1.1 */
34 peierls 1.39 public class FutureTask<V> implements RunnableFuture<V> {
35 dl 1.62 /*
36     * Revision notes: This differs from previous versions of this
37     * class that relied on AbstractQueuedSynchronizer, mainly to
38     * avoid surprising users about retaining interrupt status during
39     * cancellation races. Sync control in the current design relies
40     * on a "state" field updated via CAS to track completion, along
41     * with a simple Treiber stack to hold waiting threads.
42     *
43     * Style note: As usual, we bypass overhead of using
44     * AtomicXFieldUpdaters and instead directly use Unsafe intrinsics.
45     */
46    
47     /**
48 jsr166 1.73 * The run state of this task, initially NEW. The run state
49 dl 1.78 * transitions to a terminal state only in methods set,
50     * setException, and cancel. During completion, state may take on
51     * transient values of COMPLETING (while outcome is being set) or
52     * INTERRUPTING (only while interrupting the runner to satisfy a
53     * cancel(true)). Transitions from these intermediate to final
54     * states use cheaper ordered/lazy writes because values are unique
55     * and cannot be further modified.
56 jsr166 1.69 *
57     * Possible state transitions:
58 jsr166 1.73 * NEW -> COMPLETING -> NORMAL
59     * NEW -> COMPLETING -> EXCEPTIONAL
60     * NEW -> CANCELLED
61     * NEW -> INTERRUPTING -> INTERRUPTED
62 dl 1.62 */
63     private volatile int state;
64 jsr166 1.73 private static final int NEW = 0;
65 jsr166 1.70 private static final int COMPLETING = 1;
66     private static final int NORMAL = 2;
67     private static final int EXCEPTIONAL = 3;
68     private static final int CANCELLED = 4;
69     private static final int INTERRUPTING = 5;
70     private static final int INTERRUPTED = 6;
71 dl 1.62
72 dl 1.78 /** The underlying callable; nulled out after running */
73 dl 1.77 private Callable<V> callable;
74 dl 1.62 /** The result to return or exception to throw from get() */
75     private Object outcome; // non-volatile, protected by state reads/writes
76     /** The thread running the callable; CASed during run() */
77     private volatile Thread runner;
78     /** Treiber stack of waiting threads */
79     private volatile WaitNode waiters;
80    
81     /**
82 jsr166 1.64 * Returns result or throws exception for completed task.
83     *
84 dl 1.62 * @param s completed state value
85     */
86 jsr166 1.98 @SuppressWarnings("unchecked")
87 dl 1.62 private V report(int s) throws ExecutionException {
88     Object x = outcome;
89 jsr166 1.98 if (s == NORMAL)
90     return (V)x;
91 jsr166 1.69 if (s >= CANCELLED)
92 dl 1.62 throw new CancellationException();
93     throw new ExecutionException((Throwable)x);
94     }
95 dl 1.11
96 tim 1.1 /**
97 jsr166 1.64 * Creates a {@code FutureTask} that will, upon running, execute the
98     * given {@code Callable}.
99 tim 1.1 *
100     * @param callable the callable task
101 jsr166 1.79 * @throws NullPointerException if the callable is null
102 tim 1.1 */
103     public FutureTask(Callable<V> callable) {
104 dl 1.9 if (callable == null)
105     throw new NullPointerException();
106 dl 1.62 this.callable = callable;
107 jsr166 1.85 this.state = NEW; // ensure visibility of callable
108 tim 1.1 }
109    
110     /**
111 jsr166 1.64 * Creates a {@code FutureTask} that will, upon running, execute the
112     * given {@code Runnable}, and arrange that {@code get} will return the
113 tim 1.1 * given result on successful completion.
114     *
115 jsr166 1.54 * @param runnable the runnable task
116 tim 1.1 * @param result the result to return on successful completion. If
117 dl 1.9 * you don't need a particular result, consider using
118 dl 1.16 * constructions of the form:
119 jsr166 1.58 * {@code Future<?> f = new FutureTask<Void>(runnable, null)}
120 jsr166 1.79 * @throws NullPointerException if the runnable is null
121 tim 1.1 */
122 dl 1.15 public FutureTask(Runnable runnable, V result) {
123 dl 1.62 this.callable = Executors.callable(runnable, result);
124 jsr166 1.85 this.state = NEW; // ensure visibility of callable
125 dl 1.20 }
126    
127     public boolean isCancelled() {
128 jsr166 1.69 return state >= CANCELLED;
129 dl 1.20 }
130 jsr166 1.35
131 dl 1.20 public boolean isDone() {
132 jsr166 1.73 return state != NEW;
133 dl 1.13 }
134    
135     public boolean cancel(boolean mayInterruptIfRunning) {
136 dl 1.78 if (state != NEW)
137     return false;
138     if (mayInterruptIfRunning) {
139     if (!UNSAFE.compareAndSwapInt(this, stateOffset, NEW, INTERRUPTING))
140     return false;
141     Thread t = runner;
142     if (t != null)
143     t.interrupt();
144     UNSAFE.putOrderedInt(this, stateOffset, INTERRUPTED); // final state
145     }
146     else if (!UNSAFE.compareAndSwapInt(this, stateOffset, NEW, CANCELLED))
147     return false;
148     finishCompletion();
149     return true;
150 dl 1.13 }
151 jsr166 1.35
152 jsr166 1.43 /**
153     * @throws CancellationException {@inheritDoc}
154     */
155 dl 1.2 public V get() throws InterruptedException, ExecutionException {
156 jsr166 1.64 int s = state;
157 jsr166 1.91 if (s <= COMPLETING)
158     s = awaitDone(false, 0L);
159     return report(s);
160 tim 1.1 }
161    
162 jsr166 1.43 /**
163     * @throws CancellationException {@inheritDoc}
164     */
165 dl 1.2 public V get(long timeout, TimeUnit unit)
166 tim 1.1 throws InterruptedException, ExecutionException, TimeoutException {
167 jsr166 1.82 if (unit == null)
168     throw new NullPointerException();
169 jsr166 1.64 int s = state;
170     if (s <= COMPLETING &&
171 jsr166 1.82 (s = awaitDone(true, unit.toNanos(timeout))) <= COMPLETING)
172 dl 1.62 throw new TimeoutException();
173     return report(s);
174 tim 1.1 }
175    
176     /**
177 dl 1.20 * Protected method invoked when this task transitions to state
178 jsr166 1.64 * {@code isDone} (whether normally or via cancellation). The
179 dl 1.20 * default implementation does nothing. Subclasses may override
180     * this method to invoke completion callbacks or perform
181     * bookkeeping. Note that you can query status inside the
182     * implementation of this method to determine whether this task
183     * has been cancelled.
184     */
185     protected void done() { }
186    
187     /**
188 jsr166 1.64 * Sets the result of this future to the given value unless
189 dl 1.29 * this future has already been set or has been cancelled.
190 jsr166 1.64 *
191     * <p>This method is invoked internally by the {@link #run} method
192 dl 1.40 * upon successful completion of the computation.
193 jsr166 1.64 *
194 tim 1.1 * @param v the value
195 jsr166 1.35 */
196 dl 1.2 protected void set(V v) {
197 dl 1.78 if (UNSAFE.compareAndSwapInt(this, stateOffset, NEW, COMPLETING)) {
198     outcome = v;
199     UNSAFE.putOrderedInt(this, stateOffset, NORMAL); // final state
200     finishCompletion();
201     }
202 tim 1.1 }
203    
204     /**
205 jsr166 1.64 * Causes this future to report an {@link ExecutionException}
206     * with the given throwable as its cause, unless this future has
207 dl 1.24 * already been set or has been cancelled.
208 jsr166 1.64 *
209     * <p>This method is invoked internally by the {@link #run} method
210 dl 1.40 * upon failure of the computation.
211 jsr166 1.64 *
212 jsr166 1.41 * @param t the cause of failure
213 jsr166 1.35 */
214 dl 1.2 protected void setException(Throwable t) {
215 dl 1.78 if (UNSAFE.compareAndSwapInt(this, stateOffset, NEW, COMPLETING)) {
216     outcome = t;
217     UNSAFE.putOrderedInt(this, stateOffset, EXCEPTIONAL); // final state
218     finishCompletion();
219     }
220 tim 1.1 }
221 jsr166 1.35
222 dl 1.24 public void run() {
223 jsr166 1.87 if (state != NEW ||
224     !UNSAFE.compareAndSwapObject(this, runnerOffset,
225     null, Thread.currentThread()))
226     return;
227     try {
228 dl 1.77 Callable<V> c = callable;
229     if (c != null && state == NEW) {
230 jsr166 1.87 V result;
231 jsr166 1.92 boolean ran;
232 dl 1.77 try {
233     result = c.call();
234 jsr166 1.92 ran = true;
235 dl 1.77 } catch (Throwable ex) {
236 jsr166 1.92 result = null;
237     ran = false;
238 dl 1.77 setException(ex);
239     }
240 jsr166 1.92 if (ran)
241     set(result);
242 dl 1.62 }
243 jsr166 1.87 } finally {
244 jsr166 1.93 // runner must be non-null until state is settled to
245     // prevent concurrent calls to run()
246 jsr166 1.68 runner = null;
247 jsr166 1.93 // state must be re-read after nulling runner to prevent
248     // leaked interrupts
249 jsr166 1.86 int s = state;
250     if (s >= INTERRUPTING)
251     handlePossibleCancellationInterrupt(s);
252 dl 1.62 }
253 dl 1.24 }
254    
255     /**
256 dl 1.30 * Executes the computation without setting its result, and then
257 jsr166 1.64 * resets this future to initial state, failing to do so if the
258 dl 1.24 * computation encounters an exception or is cancelled. This is
259     * designed for use with tasks that intrinsically execute more
260     * than once.
261 jsr166 1.64 *
262 dl 1.24 * @return true if successfully run and reset
263     */
264     protected boolean runAndReset() {
265 jsr166 1.87 if (state != NEW ||
266     !UNSAFE.compareAndSwapObject(this, runnerOffset,
267     null, Thread.currentThread()))
268     return false;
269 jsr166 1.92 boolean ran = false;
270     int s = state;
271 jsr166 1.87 try {
272 dl 1.77 Callable<V> c = callable;
273 jsr166 1.92 if (c != null && s == NEW) {
274 dl 1.77 try {
275     c.call(); // don't set result
276 jsr166 1.92 ran = true;
277 dl 1.77 } catch (Throwable ex) {
278     setException(ex);
279     }
280 jsr166 1.68 }
281 jsr166 1.87 } finally {
282 jsr166 1.93 // runner must be non-null until state is settled to
283     // prevent concurrent calls to run()
284 jsr166 1.68 runner = null;
285 jsr166 1.93 // state must be re-read after nulling runner to prevent
286     // leaked interrupts
287 jsr166 1.92 s = state;
288 jsr166 1.87 if (s >= INTERRUPTING)
289     handlePossibleCancellationInterrupt(s);
290 dl 1.62 }
291 jsr166 1.92 return ran && s == NEW;
292 dl 1.14 }
293 dl 1.3
294 dl 1.14 /**
295 jsr166 1.95 * Ensures that any interrupt from a possible cancel(true) is only
296     * delivered to a task while in run or runAndReset.
297 jsr166 1.86 */
298     private void handlePossibleCancellationInterrupt(int s) {
299     // It is possible for our interrupter to stall before getting a
300     // chance to interrupt us. Let's spin-wait patiently.
301 jsr166 1.96 if (s == INTERRUPTING)
302     while (state == INTERRUPTING)
303 jsr166 1.86 Thread.yield(); // wait out pending interrupt
304 jsr166 1.96
305 jsr166 1.89 // assert state == INTERRUPTED;
306 jsr166 1.94
307 jsr166 1.95 // We want to clear any interrupt we may have received from
308     // cancel(true). However, it is permissible to use interrupts
309     // as an independent mechanism for a task to communicate with
310     // its caller, and there is no way to clear only the
311     // cancellation interrupt.
312     //
313     // Thread.interrupted();
314 jsr166 1.86 }
315    
316     /**
317 dl 1.62 * Simple linked list nodes to record waiting threads in a Treiber
318 jsr166 1.64 * stack. See other classes such as Phaser and SynchronousQueue
319 dl 1.62 * for more detailed explanation.
320 dl 1.20 */
321 dl 1.62 static final class WaitNode {
322     volatile Thread thread;
323 dl 1.76 volatile WaitNode next;
324     WaitNode() { thread = Thread.currentThread(); }
325 dl 1.62 }
326 dl 1.42
327 dl 1.62 /**
328 jsr166 1.85 * Removes and signals all waiting threads, invokes done(), and
329     * nulls out callable.
330 dl 1.62 */
331 dl 1.78 private void finishCompletion() {
332 jsr166 1.90 // assert state > COMPLETING;
333 jsr166 1.81 for (WaitNode q; (q = waiters) != null;) {
334 dl 1.62 if (UNSAFE.compareAndSwapObject(this, waitersOffset, q, null)) {
335     for (;;) {
336     Thread t = q.thread;
337     if (t != null) {
338     q.thread = null;
339     LockSupport.unpark(t);
340     }
341     WaitNode next = q.next;
342     if (next == null)
343 dl 1.78 break;
344 dl 1.62 q.next = null; // unlink to help gc
345     q = next;
346     }
347 dl 1.78 break;
348 dl 1.62 }
349 dl 1.24 }
350 jsr166 1.85
351 dl 1.78 done();
352 jsr166 1.85
353     callable = null; // to reduce footprint
354 dl 1.62 }
355 dl 1.24
356 dl 1.62 /**
357 jsr166 1.64 * Awaits completion or aborts on interrupt or timeout.
358     *
359 dl 1.62 * @param timed true if use timed waits
360 jsr166 1.64 * @param nanos time to wait, if timed
361 dl 1.62 * @return state upon completion
362     */
363     private int awaitDone(boolean timed, long nanos)
364     throws InterruptedException {
365 jsr166 1.63 long last = timed ? System.nanoTime() : 0L;
366 dl 1.62 WaitNode q = null;
367     boolean queued = false;
368 jsr166 1.64 for (;;) {
369 dl 1.62 if (Thread.interrupted()) {
370     removeWaiter(q);
371     throw new InterruptedException();
372     }
373 jsr166 1.64
374     int s = state;
375     if (s > COMPLETING) {
376 dl 1.62 if (q != null)
377     q.thread = null;
378     return s;
379     }
380 dl 1.97 else if (s == COMPLETING) // cannot time out yet
381     Thread.yield();
382 dl 1.62 else if (q == null)
383     q = new WaitNode();
384     else if (!queued)
385     queued = UNSAFE.compareAndSwapObject(this, waitersOffset,
386     q.next = waiters, q);
387     else if (timed) {
388     long now = System.nanoTime();
389     if ((nanos -= (now - last)) <= 0L) {
390     removeWaiter(q);
391     return state;
392 dl 1.50 }
393 dl 1.62 last = now;
394     LockSupport.parkNanos(this, nanos);
395 dl 1.50 }
396 dl 1.62 else
397     LockSupport.park(this);
398 dl 1.24 }
399 dl 1.62 }
400 dl 1.24
401 dl 1.62 /**
402 jsr166 1.64 * Tries to unlink a timed-out or interrupted wait node to avoid
403     * accumulating garbage. Internal nodes are simply unspliced
404 dl 1.62 * without CAS since it is harmless if they are traversed anyway
405 jsr166 1.81 * by releasers. To avoid effects of unsplicing from already
406     * removed nodes, the list is retraversed in case of an apparent
407     * race. This is slow when there are a lot of nodes, but we don't
408     * expect lists to be long enough to outweigh higher-overhead
409     * schemes.
410 dl 1.62 */
411     private void removeWaiter(WaitNode node) {
412     if (node != null) {
413     node.thread = null;
414 jsr166 1.81 retry:
415     for (;;) { // restart on removeWaiter race
416     for (WaitNode pred = null, q = waiters, s; q != null; q = s) {
417     s = q.next;
418     if (q.thread != null)
419     pred = q;
420     else if (pred != null) {
421     pred.next = s;
422     if (pred.thread == null) // check for race
423     continue retry;
424     }
425     else if (!UNSAFE.compareAndSwapObject(this, waitersOffset,
426     q, s))
427     continue retry;
428 jsr166 1.55 }
429 jsr166 1.81 break;
430 jsr166 1.56 }
431 dl 1.14 }
432 dl 1.62 }
433 dl 1.14
434 dl 1.62 // Unsafe mechanics
435     private static final sun.misc.Unsafe UNSAFE;
436     private static final long stateOffset;
437     private static final long runnerOffset;
438     private static final long waitersOffset;
439     static {
440     try {
441     UNSAFE = sun.misc.Unsafe.getUnsafe();
442     Class<?> k = FutureTask.class;
443     stateOffset = UNSAFE.objectFieldOffset
444     (k.getDeclaredField("state"));
445     runnerOffset = UNSAFE.objectFieldOffset
446     (k.getDeclaredField("runner"));
447     waitersOffset = UNSAFE.objectFieldOffset
448     (k.getDeclaredField("waiters"));
449     } catch (Exception e) {
450     throw new Error(e);
451 dl 1.14 }
452 dl 1.15 }
453 dl 1.62
454 dl 1.15 }