util/concurrent/FutureTask.java

/*
 * Written by Doug Lea with assistance from members of JCP JSR-166
 * Expert Group and released to the public domain, as explained at
 * http://creativecommons.org/publicdomain/zero/1.0/
 */

package java.util.concurrent;
import java.util.concurrent.locks.LockSupport;

/**
 * A cancellable asynchronous computation.  This class provides a base
 * implementation of {@link Future}, with methods to start and cancel
 * a computation, query to see if the computation is complete, and
 * retrieve the result of the computation.  The result can only be
 * retrieved when the computation has completed; the {@code get}
 * methods will block if the computation has not yet completed.  Once
 * the computation has completed, the computation cannot be restarted
 * or cancelled (unless the computation is invoked using
 * {@link #runAndReset}).
 *
 * <p>A {@code FutureTask} can be used to wrap a {@link Callable} or
 * {@link Runnable} object.  Because {@code FutureTask} implements
 * {@code Runnable}, a {@code FutureTask} can be submitted to an
 * {@link Executor} for execution.
 *
 * <p>In addition to serving as a standalone class, this class provides
 * {@code protected} functionality that may be useful when creating
 * customized task classes.
 *
 * @since 1.5
 * @author Doug Lea
 * @param <V> The result type returned by this FutureTask's {@code get} methods
 */
public class FutureTask<V> implements RunnableFuture<V> {
    /*
     * Revision notes: This differs from previous versions of this
     * class that relied on AbstractQueuedSynchronizer, mainly to
     * avoid surprising users about retaining interrupt status during
     * cancellation races. Sync control in the current design relies
     * on a "state" field updated via CAS to track completion, along
     * with a simple Treiber stack to hold waiting threads.
     *
     * Style note: As usual, we bypass overhead of using
     * AtomicXFieldUpdaters and instead directly use Unsafe intrinsics.
     */

    /**
     * The run state of this task, initially NEW.  The run state
     * transitions to a terminal state only in methods set,
     * setException, and cancel.  During completion, state may take on
     * transient values of COMPLETING (while outcome is being set) or
     * INTERRUPTING (only while interrupting the runner to satisfy a
     * cancel(true)). Transitions from these intermediate to final
     * states use cheaper ordered/lazy writes because values are unique
     * and cannot be further modified.
     *
     * Possible state transitions:
     * NEW -> COMPLETING -> NORMAL
     * NEW -> COMPLETING -> EXCEPTIONAL
     * NEW -> CANCELLED
     * NEW -> INTERRUPTING -> INTERRUPTED
     */
    private volatile int state;
    private static final int NEW          = 0;
    private static final int COMPLETING   = 1;
    private static final int NORMAL       = 2;
    private static final int EXCEPTIONAL  = 3;
    private static final int CANCELLED    = 4;
    private static final int INTERRUPTING = 5;
    private static final int INTERRUPTED  = 6;

    /** The underlying callable; nulled out after running */
    private Callable<V> callable;
    /** The result to return or exception to throw from get() */
    private Object outcome; // non-volatile, protected by state reads/writes
    /** The thread running the callable; CASed during run() */
    private volatile Thread runner;
    /** Treiber stack of waiting threads */
    private volatile WaitNode waiters;

    /**
     * Returns result or throws exception for completed task.
     *
     * @param s completed state value
     */
    private V report(int s) throws ExecutionException {
        Object x = outcome;
        if (s == NORMAL) {
            @SuppressWarnings("unchecked") V v = (V)x;
            return v;
        }
        if (s >= CANCELLED)
            throw new CancellationException();
        throw new ExecutionException((Throwable)x);
    }

    /**
     * Creates a {@code FutureTask} that will, upon running, execute the
     * given {@code Callable}.
     *
     * @param  callable the callable task
     * @throws NullPointerException if the callable is null
     */
    public FutureTask(Callable<V> callable) {
        if (callable == null)
            throw new NullPointerException();
        this.callable = callable;
        this.state = NEW;       // ensure visibility of callable
    }

    /**
     * Creates a {@code FutureTask} that will, upon running, execute the
     * given {@code Runnable}, and arrange that {@code get} will return the
     * given result on successful completion.
     *
     * @param runnable the runnable task
     * @param result the result to return on successful completion. If
     * you don't need a particular result, consider using
     * constructions of the form:
     * {@code Future<?> f = new FutureTask<Void>(runnable, null)}
     * @throws NullPointerException if the runnable is null
     */
    public FutureTask(Runnable runnable, V result) {
        this.callable = Executors.callable(runnable, result);
        this.state = NEW;       // ensure visibility of callable
    }

    public boolean isCancelled() {
        return state >= CANCELLED;
    }

    public boolean isDone() {
        return state != NEW;
    }

    public boolean cancel(boolean mayInterruptIfRunning) {
        if (state != NEW)
            return false;
        if (mayInterruptIfRunning) {
            if (!UNSAFE.compareAndSwapInt(this, stateOffset, NEW, INTERRUPTING))
                return false;
            Thread t = runner;
            if (t != null)
                t.interrupt();
            UNSAFE.putOrderedInt(this, stateOffset, INTERRUPTED); // final state
        }
        else if (!UNSAFE.compareAndSwapInt(this, stateOffset, NEW, CANCELLED))
            return false;
        finishCompletion();
        return true;
    }

    /**
     * @throws CancellationException {@inheritDoc}
     */
    public V get() throws InterruptedException, ExecutionException {
        int s = state;
        if (s <= COMPLETING)
            s = awaitDone(false, 0L);
        return report(s);
    }

    /**
     * @throws CancellationException {@inheritDoc}
     */
    public V get(long timeout, TimeUnit unit)
        throws InterruptedException, ExecutionException, TimeoutException {
        if (unit == null)
            throw new NullPointerException();
        int s = state;
        if (s <= COMPLETING &&
            (s = awaitDone(true, unit.toNanos(timeout))) <= COMPLETING)
            throw new TimeoutException();
        return report(s);
    }

    /**
     * Protected method invoked when this task transitions to state
     * {@code isDone} (whether normally or via cancellation). The
     * default implementation does nothing.  Subclasses may override
     * this method to invoke completion callbacks or perform
     * bookkeeping. Note that you can query status inside the
     * implementation of this method to determine whether this task
     * has been cancelled.
     */
    protected void done() { }

    /**
     * Sets the result of this future to the given value unless
     * this future has already been set or has been cancelled.
     *
     * <p>This method is invoked internally by the {@link #run} method
     * upon successful completion of the computation.
     *
     * @param v the value
     */
    protected void set(V v) {
        if (UNSAFE.compareAndSwapInt(this, stateOffset, NEW, COMPLETING)) {
            outcome = v;
            UNSAFE.putOrderedInt(this, stateOffset, NORMAL); // final state
            finishCompletion();
        }
    }

    /**
     * Causes this future to report an {@link ExecutionException}
     * with the given throwable as its cause, unless this future has
     * already been set or has been cancelled.
     *
     * <p>This method is invoked internally by the {@link #run} method
     * upon failure of the computation.
     *
     * @param t the cause of failure
     */
    protected void setException(Throwable t) {
        if (UNSAFE.compareAndSwapInt(this, stateOffset, NEW, COMPLETING)) {
            outcome = t;
            UNSAFE.putOrderedInt(this, stateOffset, EXCEPTIONAL); // final state
            finishCompletion();
        }
    }

    public void run() {
        if (state != NEW ||
            !UNSAFE.compareAndSwapObject(this, runnerOffset,
                                         null, Thread.currentThread()))
            return;
        try {
            Callable<V> c = callable;
            if (c != null && state == NEW) {
                V result;
                boolean ran;
                try {
                    result = c.call();
                    ran = true;
                } catch (Throwable ex) {
                    result = null;
                    ran = false;
                    setException(ex);
                }
                if (ran)
                    set(result);
            }
        } finally {
            // runner must be non-null until state is settled to
            // prevent concurrent calls to run()
            runner = null;
            // state must be re-read after nulling runner to prevent
            // leaked interrupts
            int s = state;
            if (s >= INTERRUPTING)
                handlePossibleCancellationInterrupt(s);
        }
    }

    /**
     * Executes the computation without setting its result, and then
     * resets this future to initial state, failing to do so if the
     * computation encounters an exception or is cancelled.  This is
     * designed for use with tasks that intrinsically execute more
     * than once.
     *
     * @return true if successfully run and reset
     */
    protected boolean runAndReset() {
        if (state != NEW ||
            !UNSAFE.compareAndSwapObject(this, runnerOffset,
                                         null, Thread.currentThread()))
            return false;
        boolean ran = false;
        int s = state;
        try {
            Callable<V> c = callable;
            if (c != null && s == NEW) {
                try {
                    c.call(); // don't set result
                    ran = true;
                } catch (Throwable ex) {
                    setException(ex);
                }
            }
        } finally {
            // runner must be non-null until state is settled to
            // prevent concurrent calls to run()
            runner = null;
            // state must be re-read after nulling runner to prevent
            // leaked interrupts
            s = state;
            if (s >= INTERRUPTING)
                handlePossibleCancellationInterrupt(s);
        }
        return ran && s == NEW;
    }

    /**
     * Ensures that any interrupt from a possible cancel(true) is only
     * delivered to a task while in run or runAndReset.
     */
    private void handlePossibleCancellationInterrupt(int s) {
        // It is possible for our interrupter to stall before getting a
        // chance to interrupt us.  Let's spin-wait patiently.
        if (s == INTERRUPTING)
            while (state == INTERRUPTING)
                Thread.yield(); // wait out pending interrupt

        // assert state == INTERRUPTED;

        // We want to clear any interrupt we may have received from
        // cancel(true).  However, it is permissible to use interrupts
        // as an independent mechanism for a task to communicate with
        // its caller, and there is no way to clear only the
        // cancellation interrupt.
        //
        // Thread.interrupted();
    }

    /**
     * Simple linked list nodes to record waiting threads in a Treiber
     * stack.  See other classes such as Phaser and SynchronousQueue
     * for more detailed explanation.
     */
    static final class WaitNode {
        volatile Thread thread;
        volatile WaitNode next;
        WaitNode() { thread = Thread.currentThread(); }
    }

    /**
     * Removes and signals all waiting threads, invokes done(), and
     * nulls out callable.
     */
    private void finishCompletion() {
        // assert state > COMPLETING;
        for (WaitNode q; (q = waiters) != null;) {
            if (UNSAFE.compareAndSwapObject(this, waitersOffset, q, null)) {
                for (;;) {
                    Thread t = q.thread;
                    if (t != null) {
                        q.thread = null;
                        LockSupport.unpark(t);
                    }
                    WaitNode next = q.next;
                    if (next == null)
                        break;
                    q.next = null; // unlink to help gc
                    q = next;
                }
                break;
            }
        }

        done();

        callable = null;        // to reduce footprint
    }

    /**
     * Awaits completion or aborts on interrupt or timeout.
     *
     * @param timed true if use timed waits
     * @param nanos time to wait, if timed
     * @return state upon completion
     */
    private int awaitDone(boolean timed, long nanos)
        throws InterruptedException {
        long last = timed ? System.nanoTime() : 0L;
        WaitNode q = null;
        boolean queued = false;
        for (;;) {
            if (Thread.interrupted()) {
                removeWaiter(q);
                throw new InterruptedException();
            }

            int s = state;
            if (s > COMPLETING) {
                if (q != null)
                    q.thread = null;
                return s;
            }
            else if (s == COMPLETING) // cannot time out yet
                Thread.yield();
            else if (q == null)
                q = new WaitNode();
            else if (!queued)
                queued = UNSAFE.compareAndSwapObject(this, waitersOffset,
                                                     q.next = waiters, q);
            else if (timed) {
                long now = System.nanoTime();
                if ((nanos -= (now - last)) <= 0L) {
                    removeWaiter(q);
                    return state;
                }
                last = now;
                LockSupport.parkNanos(this, nanos);
            }
            else
                LockSupport.park(this);
        }
    }

    /**
     * Tries to unlink a timed-out or interrupted wait node to avoid
     * accumulating garbage.  Internal nodes are simply unspliced
     * without CAS since it is harmless if they are traversed anyway
     * by releasers.  To avoid effects of unsplicing from already
     * removed nodes, the list is retraversed in case of an apparent
     * race.  This is slow when there are a lot of nodes, but we don't
     * expect lists to be long enough to outweigh higher-overhead
     * schemes.
     */
    private void removeWaiter(WaitNode node) {
        if (node != null) {
            node.thread = null;
            retry:
            for (;;) {          // restart on removeWaiter race
                for (WaitNode pred = null, q = waiters, s; q != null; q = s) {
                    s = q.next;
                    if (q.thread != null)
                        pred = q;
                    else if (pred != null) {
                        pred.next = s;
                        if (pred.thread == null) // check for race
                            continue retry;
                    }
                    else if (!UNSAFE.compareAndSwapObject(this, waitersOffset,
                                                          q, s))
                        continue retry;
                }
                break;
            }
        }
    }

    // Unsafe mechanics
    private static final sun.misc.Unsafe UNSAFE;
    private static final long stateOffset;
    private static final long runnerOffset;
    private static final long waitersOffset;
    static {
        try {
            UNSAFE = sun.misc.Unsafe.getUnsafe();
            Class<?> k = FutureTask.class;
            stateOffset = UNSAFE.objectFieldOffset
                (k.getDeclaredField("state"));
            runnerOffset = UNSAFE.objectFieldOffset
                (k.getDeclaredField("runner"));
            waitersOffset = UNSAFE.objectFieldOffset
                (k.getDeclaredField("waiters"));
        } catch (Exception e) {
            throw new Error(e);
        }
    }

}
Revision:	1.97
Committed:	Wed Nov 16 18:24:14 2011 UTC (12 years, 6 months ago) by dl
Branch:	MAIN
Changes since 1.96:	+2 -0 lines
Log Message:	Don't time out if detectably completing
#	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			private V report(int s) throws ExecutionException {
87			Object x = outcome;
88	jsr166	1.88	if (s == NORMAL) {
89			@SuppressWarnings("unchecked") V v = (V)x;
90			return v;
91			}
92	jsr166	1.69	if (s >= CANCELLED)
93	dl	1.62	throw new CancellationException();
94			throw new ExecutionException((Throwable)x);
95			}
96	dl	1.11
97	tim	1.1	/**
98	jsr166	1.64	* Creates a {@code FutureTask} that will, upon running, execute the
99			* given {@code Callable}.
100	tim	1.1	*
101			* @param callable the callable task
102	jsr166	1.79	* @throws NullPointerException if the callable is null
103	tim	1.1	*/
104			public FutureTask(Callable<V> callable) {
105	dl	1.9	if (callable == null)
106			throw new NullPointerException();
107	dl	1.62	this.callable = callable;
108	jsr166	1.85	this.state = NEW; // ensure visibility of callable
109	tim	1.1	}
110
111			/**
112	jsr166	1.64	* Creates a {@code FutureTask} that will, upon running, execute the
113			* given {@code Runnable}, and arrange that {@code get} will return the
114	tim	1.1	* given result on successful completion.
115			*
116	jsr166	1.54	* @param runnable the runnable task
117	tim	1.1	* @param result the result to return on successful completion. If
118	dl	1.9	* you don't need a particular result, consider using
119	dl	1.16	* constructions of the form:
120	jsr166	1.58	* {@code Future<?> f = new FutureTask<Void>(runnable, null)}
121	jsr166	1.79	* @throws NullPointerException if the runnable is null
122	tim	1.1	*/
123	dl	1.15	public FutureTask(Runnable runnable, V result) {
124	dl	1.62	this.callable = Executors.callable(runnable, result);
125	jsr166	1.85	this.state = NEW; // ensure visibility of callable
126	dl	1.20	}
127
128			public boolean isCancelled() {
129	jsr166	1.69	return state >= CANCELLED;
130	dl	1.20	}
131	jsr166	1.35
132	dl	1.20	public boolean isDone() {
133	jsr166	1.73	return state != NEW;
134	dl	1.13	}
135
136			public boolean cancel(boolean mayInterruptIfRunning) {
137	dl	1.78	if (state != NEW)
138			return false;
139			if (mayInterruptIfRunning) {
140			if (!UNSAFE.compareAndSwapInt(this, stateOffset, NEW, INTERRUPTING))
141			return false;
142			Thread t = runner;
143			if (t != null)
144			t.interrupt();
145			UNSAFE.putOrderedInt(this, stateOffset, INTERRUPTED); // final state
146			}
147			else if (!UNSAFE.compareAndSwapInt(this, stateOffset, NEW, CANCELLED))
148			return false;
149			finishCompletion();
150			return true;
151	dl	1.13	}
152	jsr166	1.35
153	jsr166	1.43	/**
154			* @throws CancellationException {@inheritDoc}
155			*/
156	dl	1.2	public V get() throws InterruptedException, ExecutionException {
157	jsr166	1.64	int s = state;
158	jsr166	1.91	if (s <= COMPLETING)
159			s = awaitDone(false, 0L);
160			return report(s);
161	tim	1.1	}
162
163	jsr166	1.43	/**
164			* @throws CancellationException {@inheritDoc}
165			*/
166	dl	1.2	public V get(long timeout, TimeUnit unit)
167	tim	1.1	throws InterruptedException, ExecutionException, TimeoutException {
168	jsr166	1.82	if (unit == null)
169			throw new NullPointerException();
170	jsr166	1.64	int s = state;
171			if (s <= COMPLETING &&
172	jsr166	1.82	(s = awaitDone(true, unit.toNanos(timeout))) <= COMPLETING)
173	dl	1.62	throw new TimeoutException();
174			return report(s);
175	tim	1.1	}
176
177			/**
178	dl	1.20	* Protected method invoked when this task transitions to state
179	jsr166	1.64	* {@code isDone} (whether normally or via cancellation). The
180	dl	1.20	* default implementation does nothing. Subclasses may override
181			* this method to invoke completion callbacks or perform
182			* bookkeeping. Note that you can query status inside the
183			* implementation of this method to determine whether this task
184			* has been cancelled.
185			*/
186			protected void done() { }
187
188			/**
189	jsr166	1.64	* Sets the result of this future to the given value unless
190	dl	1.29	* this future has already been set or has been cancelled.
191	jsr166	1.64	*
192			* <p>This method is invoked internally by the {@link #run} method
193	dl	1.40	* upon successful completion of the computation.
194	jsr166	1.64	*
195	tim	1.1	* @param v the value
196	jsr166	1.35	*/
197	dl	1.2	protected void set(V v) {
198	dl	1.78	if (UNSAFE.compareAndSwapInt(this, stateOffset, NEW, COMPLETING)) {
199			outcome = v;
200			UNSAFE.putOrderedInt(this, stateOffset, NORMAL); // final state
201			finishCompletion();
202			}
203	tim	1.1	}
204
205			/**
206	jsr166	1.64	* Causes this future to report an {@link ExecutionException}
207			* with the given throwable as its cause, unless this future has
208	dl	1.24	* already been set or has been cancelled.
209	jsr166	1.64	*
210			* <p>This method is invoked internally by the {@link #run} method
211	dl	1.40	* upon failure of the computation.
212	jsr166	1.64	*
213	jsr166	1.41	* @param t the cause of failure
214	jsr166	1.35	*/
215	dl	1.2	protected void setException(Throwable t) {
216	dl	1.78	if (UNSAFE.compareAndSwapInt(this, stateOffset, NEW, COMPLETING)) {
217			outcome = t;
218			UNSAFE.putOrderedInt(this, stateOffset, EXCEPTIONAL); // final state
219			finishCompletion();
220			}
221	tim	1.1	}
222	jsr166	1.35
223	dl	1.24	public void run() {
224	jsr166	1.87	if (state != NEW \|\|
225			!UNSAFE.compareAndSwapObject(this, runnerOffset,
226			null, Thread.currentThread()))
227			return;
228			try {
229	dl	1.77	Callable<V> c = callable;
230			if (c != null && state == NEW) {
231	jsr166	1.87	V result;
232	jsr166	1.92	boolean ran;
233	dl	1.77	try {
234			result = c.call();
235	jsr166	1.92	ran = true;
236	dl	1.77	} catch (Throwable ex) {
237	jsr166	1.92	result = null;
238			ran = false;
239	dl	1.77	setException(ex);
240			}
241	jsr166	1.92	if (ran)
242			set(result);
243	dl	1.62	}
244	jsr166	1.87	} finally {
245	jsr166	1.93	// runner must be non-null until state is settled to
246			// prevent concurrent calls to run()
247	jsr166	1.68	runner = null;
248	jsr166	1.93	// state must be re-read after nulling runner to prevent
249			// leaked interrupts
250	jsr166	1.86	int s = state;
251			if (s >= INTERRUPTING)
252			handlePossibleCancellationInterrupt(s);
253	dl	1.62	}
254	dl	1.24	}
255
256			/**
257	dl	1.30	* Executes the computation without setting its result, and then
258	jsr166	1.64	* resets this future to initial state, failing to do so if the
259	dl	1.24	* computation encounters an exception or is cancelled. This is
260			* designed for use with tasks that intrinsically execute more
261			* than once.
262	jsr166	1.64	*
263	dl	1.24	* @return true if successfully run and reset
264			*/
265			protected boolean runAndReset() {
266	jsr166	1.87	if (state != NEW \|\|
267			!UNSAFE.compareAndSwapObject(this, runnerOffset,
268			null, Thread.currentThread()))
269			return false;
270	jsr166	1.92	boolean ran = false;
271			int s = state;
272	jsr166	1.87	try {
273	dl	1.77	Callable<V> c = callable;
274	jsr166	1.92	if (c != null && s == NEW) {
275	dl	1.77	try {
276			c.call(); // don't set result
277	jsr166	1.92	ran = true;
278	dl	1.77	} catch (Throwable ex) {
279			setException(ex);
280			}
281	jsr166	1.68	}
282	jsr166	1.87	} finally {
283	jsr166	1.93	// runner must be non-null until state is settled to
284			// prevent concurrent calls to run()
285	jsr166	1.68	runner = null;
286	jsr166	1.93	// state must be re-read after nulling runner to prevent
287			// leaked interrupts
288	jsr166	1.92	s = state;
289	jsr166	1.87	if (s >= INTERRUPTING)
290			handlePossibleCancellationInterrupt(s);
291	dl	1.62	}
292	jsr166	1.92	return ran && s == NEW;
293	dl	1.14	}
294	dl	1.3
295	dl	1.14	/**
296	jsr166	1.95	* Ensures that any interrupt from a possible cancel(true) is only
297			* delivered to a task while in run or runAndReset.
298	jsr166	1.86	*/
299			private void handlePossibleCancellationInterrupt(int s) {
300			// It is possible for our interrupter to stall before getting a
301			// chance to interrupt us. Let's spin-wait patiently.
302	jsr166	1.96	if (s == INTERRUPTING)
303			while (state == INTERRUPTING)
304	jsr166	1.86	Thread.yield(); // wait out pending interrupt
305	jsr166	1.96
306	jsr166	1.89	// assert state == INTERRUPTED;
307	jsr166	1.94
308	jsr166	1.95	// We want to clear any interrupt we may have received from
309			// cancel(true). However, it is permissible to use interrupts
310			// as an independent mechanism for a task to communicate with
311			// its caller, and there is no way to clear only the
312			// cancellation interrupt.
313			//
314			// Thread.interrupted();
315	jsr166	1.86	}
316
317			/**
318	dl	1.62	* Simple linked list nodes to record waiting threads in a Treiber
319	jsr166	1.64	* stack. See other classes such as Phaser and SynchronousQueue
320	dl	1.62	* for more detailed explanation.
321	dl	1.20	*/
322	dl	1.62	static final class WaitNode {
323			volatile Thread thread;
324	dl	1.76	volatile WaitNode next;
325			WaitNode() { thread = Thread.currentThread(); }
326	dl	1.62	}
327	dl	1.42
328	dl	1.62	/**
329	jsr166	1.85	* Removes and signals all waiting threads, invokes done(), and
330			* nulls out callable.
331	dl	1.62	*/
332	dl	1.78	private void finishCompletion() {
333	jsr166	1.90	// assert state > COMPLETING;
334	jsr166	1.81	for (WaitNode q; (q = waiters) != null;) {
335	dl	1.62	if (UNSAFE.compareAndSwapObject(this, waitersOffset, q, null)) {
336			for (;;) {
337			Thread t = q.thread;
338			if (t != null) {
339			q.thread = null;
340			LockSupport.unpark(t);
341			}
342			WaitNode next = q.next;
343			if (next == null)
344	dl	1.78	break;
345	dl	1.62	q.next = null; // unlink to help gc
346			q = next;
347			}
348	dl	1.78	break;
349	dl	1.62	}
350	dl	1.24	}
351	jsr166	1.85
352	dl	1.78	done();
353	jsr166	1.85
354			callable = null; // to reduce footprint
355	dl	1.62	}
356	dl	1.24
357	dl	1.62	/**
358	jsr166	1.64	* Awaits completion or aborts on interrupt or timeout.
359			*
360	dl	1.62	* @param timed true if use timed waits
361	jsr166	1.64	* @param nanos time to wait, if timed
362	dl	1.62	* @return state upon completion
363			*/
364			private int awaitDone(boolean timed, long nanos)
365			throws InterruptedException {
366	jsr166	1.63	long last = timed ? System.nanoTime() : 0L;
367	dl	1.62	WaitNode q = null;
368			boolean queued = false;
369	jsr166	1.64	for (;;) {
370	dl	1.62	if (Thread.interrupted()) {
371			removeWaiter(q);
372			throw new InterruptedException();
373			}
374	jsr166	1.64
375			int s = state;
376			if (s > COMPLETING) {
377	dl	1.62	if (q != null)
378			q.thread = null;
379			return s;
380			}
381	dl	1.97	else if (s == COMPLETING) // cannot time out yet
382			Thread.yield();
383	dl	1.62	else if (q == null)
384			q = new WaitNode();
385			else if (!queued)
386			queued = UNSAFE.compareAndSwapObject(this, waitersOffset,
387			q.next = waiters, q);
388			else if (timed) {
389			long now = System.nanoTime();
390			if ((nanos -= (now - last)) <= 0L) {
391			removeWaiter(q);
392			return state;
393	dl	1.50	}
394	dl	1.62	last = now;
395			LockSupport.parkNanos(this, nanos);
396	dl	1.50	}
397	dl	1.62	else
398			LockSupport.park(this);
399	dl	1.24	}
400	dl	1.62	}
401	dl	1.24
402	dl	1.62	/**
403	jsr166	1.64	* Tries to unlink a timed-out or interrupted wait node to avoid
404			* accumulating garbage. Internal nodes are simply unspliced
405	dl	1.62	* without CAS since it is harmless if they are traversed anyway
406	jsr166	1.81	* by releasers. To avoid effects of unsplicing from already
407			* removed nodes, the list is retraversed in case of an apparent
408			* race. This is slow when there are a lot of nodes, but we don't
409			* expect lists to be long enough to outweigh higher-overhead
410			* schemes.
411	dl	1.62	*/
412			private void removeWaiter(WaitNode node) {
413			if (node != null) {
414			node.thread = null;
415	jsr166	1.81	retry:
416			for (;;) { // restart on removeWaiter race
417			for (WaitNode pred = null, q = waiters, s; q != null; q = s) {
418			s = q.next;
419			if (q.thread != null)
420			pred = q;
421			else if (pred != null) {
422			pred.next = s;
423			if (pred.thread == null) // check for race
424			continue retry;
425			}
426			else if (!UNSAFE.compareAndSwapObject(this, waitersOffset,
427			q, s))
428			continue retry;
429	jsr166	1.55	}
430	jsr166	1.81	break;
431	jsr166	1.56	}
432	dl	1.14	}
433	dl	1.62	}
434	dl	1.14
435	dl	1.62	// Unsafe mechanics
436			private static final sun.misc.Unsafe UNSAFE;
437			private static final long stateOffset;
438			private static final long runnerOffset;
439			private static final long waitersOffset;
440			static {
441			try {
442			UNSAFE = sun.misc.Unsafe.getUnsafe();
443			Class<?> k = FutureTask.class;
444			stateOffset = UNSAFE.objectFieldOffset
445			(k.getDeclaredField("state"));
446			runnerOffset = UNSAFE.objectFieldOffset
447			(k.getDeclaredField("runner"));
448			waitersOffset = UNSAFE.objectFieldOffset
449			(k.getDeclaredField("waiters"));
450			} catch (Exception e) {
451			throw new Error(e);
452	dl	1.14	}
453	dl	1.15	}
454	dl	1.62
455	dl	1.15	}