1 /*
2  * Written by Doug Lea with assistance from members of JCP JSR-166
3  * Expert Group and released to the public domain, as explained at
4  * http://creativecommons.org/publicdomain/zero/1.0/
5  */
6 
7 package java.util.concurrent;
8 
9 import java.io.Serializable;
10 import java.lang.ref.ReferenceQueue;
11 import java.lang.ref.WeakReference;
12 import java.lang.reflect.Constructor;
13 import java.util.Collection;
14 import java.util.List;
15 import java.util.RandomAccess;
16 import java.util.concurrent.locks.ReentrantLock;
17 
18 // BEGIN android-note
19 // removed java 9 code
20 // END android-note
21 
22 /**
23  * Abstract base class for tasks that run within a {@link ForkJoinPool}.
24  * A {@code ForkJoinTask} is a thread-like entity that is much
25  * lighter weight than a normal thread.  Huge numbers of tasks and
26  * subtasks may be hosted by a small number of actual threads in a
27  * ForkJoinPool, at the price of some usage limitations.
28  *
29  * <p>A "main" {@code ForkJoinTask} begins execution when it is
30  * explicitly submitted to a {@link ForkJoinPool}, or, if not already
31  * engaged in a ForkJoin computation, commenced in the {@link
32  * ForkJoinPool#commonPool()} via {@link #fork}, {@link #invoke}, or
33  * related methods.  Once started, it will usually in turn start other
34  * subtasks.  As indicated by the name of this class, many programs
35  * using {@code ForkJoinTask} employ only methods {@link #fork} and
36  * {@link #join}, or derivatives such as {@link
37  * #invokeAll(ForkJoinTask...) invokeAll}.  However, this class also
38  * provides a number of other methods that can come into play in
39  * advanced usages, as well as extension mechanics that allow support
40  * of new forms of fork/join processing.
41  *
42  * <p>A {@code ForkJoinTask} is a lightweight form of {@link Future}.
43  * The efficiency of {@code ForkJoinTask}s stems from a set of
44  * restrictions (that are only partially statically enforceable)
45  * reflecting their main use as computational tasks calculating pure
46  * functions or operating on purely isolated objects.  The primary
47  * coordination mechanisms are {@link #fork}, that arranges
48  * asynchronous execution, and {@link #join}, that doesn't proceed
49  * until the task's result has been computed.  Computations should
50  * ideally avoid {@code synchronized} methods or blocks, and should
51  * minimize other blocking synchronization apart from joining other
52  * tasks or using synchronizers such as Phasers that are advertised to
53  * cooperate with fork/join scheduling. Subdividable tasks should also
54  * not perform blocking I/O, and should ideally access variables that
55  * are completely independent of those accessed by other running
56  * tasks. These guidelines are loosely enforced by not permitting
57  * checked exceptions such as {@code IOExceptions} to be
58  * thrown. However, computations may still encounter unchecked
59  * exceptions, that are rethrown to callers attempting to join
60  * them. These exceptions may additionally include {@link
61  * RejectedExecutionException} stemming from internal resource
62  * exhaustion, such as failure to allocate internal task
63  * queues. Rethrown exceptions behave in the same way as regular
64  * exceptions, but, when possible, contain stack traces (as displayed
65  * for example using {@code ex.printStackTrace()}) of both the thread
66  * that initiated the computation as well as the thread actually
67  * encountering the exception; minimally only the latter.
68  *
69  * <p>It is possible to define and use ForkJoinTasks that may block,
70  * but doing do requires three further considerations: (1) Completion
71  * of few if any <em>other</em> tasks should be dependent on a task
72  * that blocks on external synchronization or I/O. Event-style async
73  * tasks that are never joined (for example, those subclassing {@link
74  * CountedCompleter}) often fall into this category.  (2) To minimize
75  * resource impact, tasks should be small; ideally performing only the
76  * (possibly) blocking action. (3) Unless the {@link
77  * ForkJoinPool.ManagedBlocker} API is used, or the number of possibly
78  * blocked tasks is known to be less than the pool's {@link
79  * ForkJoinPool#getParallelism} level, the pool cannot guarantee that
80  * enough threads will be available to ensure progress or good
81  * performance.
82  *
83  * <p>The primary method for awaiting completion and extracting
84  * results of a task is {@link #join}, but there are several variants:
85  * The {@link Future#get} methods support interruptible and/or timed
86  * waits for completion and report results using {@code Future}
87  * conventions. Method {@link #invoke} is semantically
88  * equivalent to {@code fork(); join()} but always attempts to begin
89  * execution in the current thread. The "<em>quiet</em>" forms of
90  * these methods do not extract results or report exceptions. These
91  * may be useful when a set of tasks are being executed, and you need
92  * to delay processing of results or exceptions until all complete.
93  * Method {@code invokeAll} (available in multiple versions)
94  * performs the most common form of parallel invocation: forking a set
95  * of tasks and joining them all.
96  *
97  * <p>In the most typical usages, a fork-join pair act like a call
98  * (fork) and return (join) from a parallel recursive function. As is
99  * the case with other forms of recursive calls, returns (joins)
100  * should be performed innermost-first. For example, {@code a.fork();
101  * b.fork(); b.join(); a.join();} is likely to be substantially more
102  * efficient than joining {@code a} before {@code b}.
103  *
104  * <p>The execution status of tasks may be queried at several levels
105  * of detail: {@link #isDone} is true if a task completed in any way
106  * (including the case where a task was cancelled without executing);
107  * {@link #isCompletedNormally} is true if a task completed without
108  * cancellation or encountering an exception; {@link #isCancelled} is
109  * true if the task was cancelled (in which case {@link #getException}
110  * returns a {@link java.util.concurrent.CancellationException}); and
111  * {@link #isCompletedAbnormally} is true if a task was either
112  * cancelled or encountered an exception, in which case {@link
113  * #getException} will return either the encountered exception or
114  * {@link java.util.concurrent.CancellationException}.
115  *
116  * <p>The ForkJoinTask class is not usually directly subclassed.
117  * Instead, you subclass one of the abstract classes that support a
118  * particular style of fork/join processing, typically {@link
119  * RecursiveAction} for most computations that do not return results,
120  * {@link RecursiveTask} for those that do, and {@link
121  * CountedCompleter} for those in which completed actions trigger
122  * other actions.  Normally, a concrete ForkJoinTask subclass declares
123  * fields comprising its parameters, established in a constructor, and
124  * then defines a {@code compute} method that somehow uses the control
125  * methods supplied by this base class.
126  *
127  * <p>Method {@link #join} and its variants are appropriate for use
128  * only when completion dependencies are acyclic; that is, the
129  * parallel computation can be described as a directed acyclic graph
130  * (DAG). Otherwise, executions may encounter a form of deadlock as
131  * tasks cyclically wait for each other.  However, this framework
132  * supports other methods and techniques (for example the use of
133  * {@link Phaser}, {@link #helpQuiesce}, and {@link #complete}) that
134  * may be of use in constructing custom subclasses for problems that
135  * are not statically structured as DAGs. To support such usages, a
136  * ForkJoinTask may be atomically <em>tagged</em> with a {@code short}
137  * value using {@link #setForkJoinTaskTag} or {@link
138  * #compareAndSetForkJoinTaskTag} and checked using {@link
139  * #getForkJoinTaskTag}. The ForkJoinTask implementation does not use
140  * these {@code protected} methods or tags for any purpose, but they
141  * may be of use in the construction of specialized subclasses.  For
142  * example, parallel graph traversals can use the supplied methods to
143  * avoid revisiting nodes/tasks that have already been processed.
144  * (Method names for tagging are bulky in part to encourage definition
145  * of methods that reflect their usage patterns.)
146  *
147  * <p>Most base support methods are {@code final}, to prevent
148  * overriding of implementations that are intrinsically tied to the
149  * underlying lightweight task scheduling framework.  Developers
150  * creating new basic styles of fork/join processing should minimally
151  * implement {@code protected} methods {@link #exec}, {@link
152  * #setRawResult}, and {@link #getRawResult}, while also introducing
153  * an abstract computational method that can be implemented in its
154  * subclasses, possibly relying on other {@code protected} methods
155  * provided by this class.
156  *
157  * <p>ForkJoinTasks should perform relatively small amounts of
158  * computation. Large tasks should be split into smaller subtasks,
159  * usually via recursive decomposition. As a very rough rule of thumb,
160  * a task should perform more than 100 and less than 10000 basic
161  * computational steps, and should avoid indefinite looping. If tasks
162  * are too big, then parallelism cannot improve throughput. If too
163  * small, then memory and internal task maintenance overhead may
164  * overwhelm processing.
165  *
166  * <p>This class provides {@code adapt} methods for {@link Runnable}
167  * and {@link Callable}, that may be of use when mixing execution of
168  * {@code ForkJoinTasks} with other kinds of tasks. When all tasks are
169  * of this form, consider using a pool constructed in <em>asyncMode</em>.
170  *
171  * <p>ForkJoinTasks are {@code Serializable}, which enables them to be
172  * used in extensions such as remote execution frameworks. It is
173  * sensible to serialize tasks only before or after, but not during,
174  * execution. Serialization is not relied on during execution itself.
175  *
176  * @since 1.7
177  * @author Doug Lea
178  */
179 public abstract class ForkJoinTask<V> implements Future<V>, Serializable {
180 
181     /*
182      * See the internal documentation of class ForkJoinPool for a
183      * general implementation overview.  ForkJoinTasks are mainly
184      * responsible for maintaining their "status" field amidst relays
185      * to methods in ForkJoinWorkerThread and ForkJoinPool.
186      *
187      * The methods of this class are more-or-less layered into
188      * (1) basic status maintenance
189      * (2) execution and awaiting completion
190      * (3) user-level methods that additionally report results.
191      * This is sometimes hard to see because this file orders exported
192      * methods in a way that flows well in javadocs.
193      */
194 
195     /*
196      * The status field holds run control status bits packed into a
197      * single int to minimize footprint and to ensure atomicity (via
198      * CAS).  Status is initially zero, and takes on nonnegative
199      * values until completed, upon which status (anded with
200      * DONE_MASK) holds value NORMAL, CANCELLED, or EXCEPTIONAL. Tasks
201      * undergoing blocking waits by other threads have the SIGNAL bit
202      * set.  Completion of a stolen task with SIGNAL set awakens any
203      * waiters via notifyAll. Even though suboptimal for some
204      * purposes, we use basic builtin wait/notify to take advantage of
205      * "monitor inflation" in JVMs that we would otherwise need to
206      * emulate to avoid adding further per-task bookkeeping overhead.
207      * We want these monitors to be "fat", i.e., not use biasing or
208      * thin-lock techniques, so use some odd coding idioms that tend
209      * to avoid them, mainly by arranging that every synchronized
210      * block performs a wait, notifyAll or both.
211      *
212      * These control bits occupy only (some of) the upper half (16
213      * bits) of status field. The lower bits are used for user-defined
214      * tags.
215      */
216 
217     /** The run status of this task */
218     volatile int status; // accessed directly by pool and workers
219     static final int DONE_MASK   = 0xf0000000;  // mask out non-completion bits
220     static final int NORMAL      = 0xf0000000;  // must be negative
221     static final int CANCELLED   = 0xc0000000;  // must be < NORMAL
222     static final int EXCEPTIONAL = 0x80000000;  // must be < CANCELLED
223     static final int SIGNAL      = 0x00010000;  // must be >= 1 << 16
224     static final int SMASK       = 0x0000ffff;  // short bits for tags
225 
226     /**
227      * Marks completion and wakes up threads waiting to join this
228      * task.
229      *
230      * @param completion one of NORMAL, CANCELLED, EXCEPTIONAL
231      * @return completion status on exit
232      */
setCompletion(int completion)233     private int setCompletion(int completion) {
234         for (int s;;) {
235             if ((s = status) < 0)
236                 return s;
237             if (U.compareAndSwapInt(this, STATUS, s, s | completion)) {
238                 if ((s >>> 16) != 0)
239                     synchronized (this) { notifyAll(); }
240                 return completion;
241             }
242         }
243     }
244 
245     /**
246      * Primary execution method for stolen tasks. Unless done, calls
247      * exec and records status if completed, but doesn't wait for
248      * completion otherwise.
249      *
250      * @return status on exit from this method
251      */
doExec()252     final int doExec() {
253         int s; boolean completed;
254         if ((s = status) >= 0) {
255             try {
256                 completed = exec();
257             } catch (Throwable rex) {
258                 return setExceptionalCompletion(rex);
259             }
260             if (completed)
261                 s = setCompletion(NORMAL);
262         }
263         return s;
264     }
265 
266     /**
267      * If not done, sets SIGNAL status and performs Object.wait(timeout).
268      * This task may or may not be done on exit. Ignores interrupts.
269      *
270      * @param timeout using Object.wait conventions.
271      */
internalWait(long timeout)272     final void internalWait(long timeout) {
273         int s;
274         if ((s = status) >= 0 && // force completer to issue notify
275             U.compareAndSwapInt(this, STATUS, s, s | SIGNAL)) {
276             synchronized (this) {
277                 if (status >= 0)
278                     try { wait(timeout); } catch (InterruptedException ie) { }
279                 else
280                     notifyAll();
281             }
282         }
283     }
284 
285     /**
286      * Blocks a non-worker-thread until completion.
287      * @return status upon completion
288      */
externalAwaitDone()289     private int externalAwaitDone() {
290         int s = ((this instanceof CountedCompleter) ? // try helping
291                  ForkJoinPool.common.externalHelpComplete(
292                      (CountedCompleter<?>)this, 0) :
293                  ForkJoinPool.common.tryExternalUnpush(this) ? doExec() : 0);
294         if (s >= 0 && (s = status) >= 0) {
295             boolean interrupted = false;
296             do {
297                 if (U.compareAndSwapInt(this, STATUS, s, s | SIGNAL)) {
298                     synchronized (this) {
299                         if (status >= 0) {
300                             try {
301                                 wait(0L);
302                             } catch (InterruptedException ie) {
303                                 interrupted = true;
304                             }
305                         }
306                         else
307                             notifyAll();
308                     }
309                 }
310             } while ((s = status) >= 0);
311             if (interrupted)
312                 Thread.currentThread().interrupt();
313         }
314         return s;
315     }
316 
317     /**
318      * Blocks a non-worker-thread until completion or interruption.
319      */
externalInterruptibleAwaitDone()320     private int externalInterruptibleAwaitDone() throws InterruptedException {
321         int s;
322         if (Thread.interrupted())
323             throw new InterruptedException();
324         if ((s = status) >= 0 &&
325             (s = ((this instanceof CountedCompleter) ?
326                   ForkJoinPool.common.externalHelpComplete(
327                       (CountedCompleter<?>)this, 0) :
328                   ForkJoinPool.common.tryExternalUnpush(this) ? doExec() :
329                   0)) >= 0) {
330             while ((s = status) >= 0) {
331                 if (U.compareAndSwapInt(this, STATUS, s, s | SIGNAL)) {
332                     synchronized (this) {
333                         if (status >= 0)
334                             wait(0L);
335                         else
336                             notifyAll();
337                     }
338                 }
339             }
340         }
341         return s;
342     }
343 
344     /**
345      * Implementation for join, get, quietlyJoin. Directly handles
346      * only cases of already-completed, external wait, and
347      * unfork+exec.  Others are relayed to ForkJoinPool.awaitJoin.
348      *
349      * @return status upon completion
350      */
doJoin()351     private int doJoin() {
352         int s; Thread t; ForkJoinWorkerThread wt; ForkJoinPool.WorkQueue w;
353         return (s = status) < 0 ? s :
354             ((t = Thread.currentThread()) instanceof ForkJoinWorkerThread) ?
355             (w = (wt = (ForkJoinWorkerThread)t).workQueue).
356             tryUnpush(this) && (s = doExec()) < 0 ? s :
357             wt.pool.awaitJoin(w, this, 0L) :
358             externalAwaitDone();
359     }
360 
361     /**
362      * Implementation for invoke, quietlyInvoke.
363      *
364      * @return status upon completion
365      */
doInvoke()366     private int doInvoke() {
367         int s; Thread t; ForkJoinWorkerThread wt;
368         return (s = doExec()) < 0 ? s :
369             ((t = Thread.currentThread()) instanceof ForkJoinWorkerThread) ?
370             (wt = (ForkJoinWorkerThread)t).pool.
371             awaitJoin(wt.workQueue, this, 0L) :
372             externalAwaitDone();
373     }
374 
375     // Exception table support
376 
377     /**
378      * Table of exceptions thrown by tasks, to enable reporting by
379      * callers. Because exceptions are rare, we don't directly keep
380      * them with task objects, but instead use a weak ref table.  Note
381      * that cancellation exceptions don't appear in the table, but are
382      * instead recorded as status values.
383      *
384      * Note: These statics are initialized below in static block.
385      */
386     private static final ExceptionNode[] exceptionTable;
387     private static final ReentrantLock exceptionTableLock;
388     private static final ReferenceQueue<Object> exceptionTableRefQueue;
389 
390     /**
391      * Fixed capacity for exceptionTable.
392      */
393     private static final int EXCEPTION_MAP_CAPACITY = 32;
394 
395     /**
396      * Key-value nodes for exception table.  The chained hash table
397      * uses identity comparisons, full locking, and weak references
398      * for keys. The table has a fixed capacity because it only
399      * maintains task exceptions long enough for joiners to access
400      * them, so should never become very large for sustained
401      * periods. However, since we do not know when the last joiner
402      * completes, we must use weak references and expunge them. We do
403      * so on each operation (hence full locking). Also, some thread in
404      * any ForkJoinPool will call helpExpungeStaleExceptions when its
405      * pool becomes isQuiescent.
406      */
407     static final class ExceptionNode extends WeakReference<ForkJoinTask<?>> {
408         final Throwable ex;
409         ExceptionNode next;
410         final long thrower;  // use id not ref to avoid weak cycles
411         final int hashCode;  // store task hashCode before weak ref disappears
ExceptionNode(ForkJoinTask<?> task, Throwable ex, ExceptionNode next, ReferenceQueue<Object> exceptionTableRefQueue)412         ExceptionNode(ForkJoinTask<?> task, Throwable ex, ExceptionNode next,
413                       ReferenceQueue<Object> exceptionTableRefQueue) {
414             super(task, exceptionTableRefQueue);
415             this.ex = ex;
416             this.next = next;
417             this.thrower = Thread.currentThread().getId();
418             this.hashCode = System.identityHashCode(task);
419         }
420     }
421 
422     /**
423      * Records exception and sets status.
424      *
425      * @return status on exit
426      */
recordExceptionalCompletion(Throwable ex)427     final int recordExceptionalCompletion(Throwable ex) {
428         int s;
429         if ((s = status) >= 0) {
430             int h = System.identityHashCode(this);
431             final ReentrantLock lock = exceptionTableLock;
432             lock.lock();
433             try {
434                 expungeStaleExceptions();
435                 ExceptionNode[] t = exceptionTable;
436                 int i = h & (t.length - 1);
437                 for (ExceptionNode e = t[i]; ; e = e.next) {
438                     if (e == null) {
439                         t[i] = new ExceptionNode(this, ex, t[i],
440                                                  exceptionTableRefQueue);
441                         break;
442                     }
443                     if (e.get() == this) // already present
444                         break;
445                 }
446             } finally {
447                 lock.unlock();
448             }
449             s = setCompletion(EXCEPTIONAL);
450         }
451         return s;
452     }
453 
454     /**
455      * Records exception and possibly propagates.
456      *
457      * @return status on exit
458      */
setExceptionalCompletion(Throwable ex)459     private int setExceptionalCompletion(Throwable ex) {
460         int s = recordExceptionalCompletion(ex);
461         if ((s & DONE_MASK) == EXCEPTIONAL)
462             internalPropagateException(ex);
463         return s;
464     }
465 
466     /**
467      * Hook for exception propagation support for tasks with completers.
468      */
internalPropagateException(Throwable ex)469     void internalPropagateException(Throwable ex) {
470     }
471 
472     /**
473      * Cancels, ignoring any exceptions thrown by cancel. Used during
474      * worker and pool shutdown. Cancel is spec'ed not to throw any
475      * exceptions, but if it does anyway, we have no recourse during
476      * shutdown, so guard against this case.
477      */
cancelIgnoringExceptions(ForkJoinTask<?> t)478     static final void cancelIgnoringExceptions(ForkJoinTask<?> t) {
479         if (t != null && t.status >= 0) {
480             try {
481                 t.cancel(false);
482             } catch (Throwable ignore) {
483             }
484         }
485     }
486 
487     /**
488      * Removes exception node and clears status.
489      */
clearExceptionalCompletion()490     private void clearExceptionalCompletion() {
491         int h = System.identityHashCode(this);
492         final ReentrantLock lock = exceptionTableLock;
493         lock.lock();
494         try {
495             ExceptionNode[] t = exceptionTable;
496             int i = h & (t.length - 1);
497             ExceptionNode e = t[i];
498             ExceptionNode pred = null;
499             while (e != null) {
500                 ExceptionNode next = e.next;
501                 if (e.get() == this) {
502                     if (pred == null)
503                         t[i] = next;
504                     else
505                         pred.next = next;
506                     break;
507                 }
508                 pred = e;
509                 e = next;
510             }
511             expungeStaleExceptions();
512             status = 0;
513         } finally {
514             lock.unlock();
515         }
516     }
517 
518     /**
519      * Returns a rethrowable exception for this task, if available.
520      * To provide accurate stack traces, if the exception was not
521      * thrown by the current thread, we try to create a new exception
522      * of the same type as the one thrown, but with the recorded
523      * exception as its cause. If there is no such constructor, we
524      * instead try to use a no-arg constructor, followed by initCause,
525      * to the same effect. If none of these apply, or any fail due to
526      * other exceptions, we return the recorded exception, which is
527      * still correct, although it may contain a misleading stack
528      * trace.
529      *
530      * @return the exception, or null if none
531      */
getThrowableException()532     private Throwable getThrowableException() {
533         int h = System.identityHashCode(this);
534         ExceptionNode e;
535         final ReentrantLock lock = exceptionTableLock;
536         lock.lock();
537         try {
538             expungeStaleExceptions();
539             ExceptionNode[] t = exceptionTable;
540             e = t[h & (t.length - 1)];
541             while (e != null && e.get() != this)
542                 e = e.next;
543         } finally {
544             lock.unlock();
545         }
546         Throwable ex;
547         if (e == null || (ex = e.ex) == null)
548             return null;
549         if (e.thrower != Thread.currentThread().getId()) {
550             try {
551                 Constructor<?> noArgCtor = null;
552                 // public ctors only
553                 for (Constructor<?> c : ex.getClass().getConstructors()) {
554                     Class<?>[] ps = c.getParameterTypes();
555                     if (ps.length == 0)
556                         noArgCtor = c;
557                     else if (ps.length == 1 && ps[0] == Throwable.class)
558                         return (Throwable)c.newInstance(ex);
559                 }
560                 if (noArgCtor != null) {
561                     Throwable wx = (Throwable)noArgCtor.newInstance();
562                     wx.initCause(ex);
563                     return wx;
564                 }
565             } catch (Exception ignore) {
566             }
567         }
568         return ex;
569     }
570 
571     /**
572      * Polls stale refs and removes them. Call only while holding lock.
573      */
expungeStaleExceptions()574     private static void expungeStaleExceptions() {
575         for (Object x; (x = exceptionTableRefQueue.poll()) != null;) {
576             if (x instanceof ExceptionNode) {
577                 int hashCode = ((ExceptionNode)x).hashCode;
578                 ExceptionNode[] t = exceptionTable;
579                 int i = hashCode & (t.length - 1);
580                 ExceptionNode e = t[i];
581                 ExceptionNode pred = null;
582                 while (e != null) {
583                     ExceptionNode next = e.next;
584                     if (e == x) {
585                         if (pred == null)
586                             t[i] = next;
587                         else
588                             pred.next = next;
589                         break;
590                     }
591                     pred = e;
592                     e = next;
593                 }
594             }
595         }
596     }
597 
598     /**
599      * If lock is available, polls stale refs and removes them.
600      * Called from ForkJoinPool when pools become quiescent.
601      */
helpExpungeStaleExceptions()602     static final void helpExpungeStaleExceptions() {
603         final ReentrantLock lock = exceptionTableLock;
604         if (lock.tryLock()) {
605             try {
606                 expungeStaleExceptions();
607             } finally {
608                 lock.unlock();
609             }
610         }
611     }
612 
613     /**
614      * A version of "sneaky throw" to relay exceptions.
615      */
rethrow(Throwable ex)616     static void rethrow(Throwable ex) {
617         ForkJoinTask.<RuntimeException>uncheckedThrow(ex);
618     }
619 
620     /**
621      * The sneaky part of sneaky throw, relying on generics
622      * limitations to evade compiler complaints about rethrowing
623      * unchecked exceptions.
624      */
625     @SuppressWarnings("unchecked") static <T extends Throwable>
uncheckedThrow(Throwable t)626     void uncheckedThrow(Throwable t) throws T {
627         if (t != null)
628             throw (T)t; // rely on vacuous cast
629         else
630             throw new Error("Unknown Exception");
631     }
632 
633     /**
634      * Throws exception, if any, associated with the given status.
635      */
reportException(int s)636     private void reportException(int s) {
637         if (s == CANCELLED)
638             throw new CancellationException();
639         if (s == EXCEPTIONAL)
640             rethrow(getThrowableException());
641     }
642 
643     // public methods
644 
645     /**
646      * Arranges to asynchronously execute this task in the pool the
647      * current task is running in, if applicable, or using the {@link
648      * ForkJoinPool#commonPool()} if not {@link #inForkJoinPool}.  While
649      * it is not necessarily enforced, it is a usage error to fork a
650      * task more than once unless it has completed and been
651      * reinitialized.  Subsequent modifications to the state of this
652      * task or any data it operates on are not necessarily
653      * consistently observable by any thread other than the one
654      * executing it unless preceded by a call to {@link #join} or
655      * related methods, or a call to {@link #isDone} returning {@code
656      * true}.
657      *
658      * @return {@code this}, to simplify usage
659      */
fork()660     public final ForkJoinTask<V> fork() {
661         Thread t;
662         if ((t = Thread.currentThread()) instanceof ForkJoinWorkerThread)
663             ((ForkJoinWorkerThread)t).workQueue.push(this);
664         else
665             ForkJoinPool.common.externalPush(this);
666         return this;
667     }
668 
669     /**
670      * Returns the result of the computation when it {@link #isDone is
671      * done}.  This method differs from {@link #get()} in that
672      * abnormal completion results in {@code RuntimeException} or
673      * {@code Error}, not {@code ExecutionException}, and that
674      * interrupts of the calling thread do <em>not</em> cause the
675      * method to abruptly return by throwing {@code
676      * InterruptedException}.
677      *
678      * @return the computed result
679      */
join()680     public final V join() {
681         int s;
682         if ((s = doJoin() & DONE_MASK) != NORMAL)
683             reportException(s);
684         return getRawResult();
685     }
686 
687     /**
688      * Commences performing this task, awaits its completion if
689      * necessary, and returns its result, or throws an (unchecked)
690      * {@code RuntimeException} or {@code Error} if the underlying
691      * computation did so.
692      *
693      * @return the computed result
694      */
invoke()695     public final V invoke() {
696         int s;
697         if ((s = doInvoke() & DONE_MASK) != NORMAL)
698             reportException(s);
699         return getRawResult();
700     }
701 
702     /**
703      * Forks the given tasks, returning when {@code isDone} holds for
704      * each task or an (unchecked) exception is encountered, in which
705      * case the exception is rethrown. If more than one task
706      * encounters an exception, then this method throws any one of
707      * these exceptions. If any task encounters an exception, the
708      * other may be cancelled. However, the execution status of
709      * individual tasks is not guaranteed upon exceptional return. The
710      * status of each task may be obtained using {@link
711      * #getException()} and related methods to check if they have been
712      * cancelled, completed normally or exceptionally, or left
713      * unprocessed.
714      *
715      * @param t1 the first task
716      * @param t2 the second task
717      * @throws NullPointerException if any task is null
718      */
invokeAll(ForkJoinTask<?> t1, ForkJoinTask<?> t2)719     public static void invokeAll(ForkJoinTask<?> t1, ForkJoinTask<?> t2) {
720         int s1, s2;
721         t2.fork();
722         if ((s1 = t1.doInvoke() & DONE_MASK) != NORMAL)
723             t1.reportException(s1);
724         if ((s2 = t2.doJoin() & DONE_MASK) != NORMAL)
725             t2.reportException(s2);
726     }
727 
728     /**
729      * Forks the given tasks, returning when {@code isDone} holds for
730      * each task or an (unchecked) exception is encountered, in which
731      * case the exception is rethrown. If more than one task
732      * encounters an exception, then this method throws any one of
733      * these exceptions. If any task encounters an exception, others
734      * may be cancelled. However, the execution status of individual
735      * tasks is not guaranteed upon exceptional return. The status of
736      * each task may be obtained using {@link #getException()} and
737      * related methods to check if they have been cancelled, completed
738      * normally or exceptionally, or left unprocessed.
739      *
740      * @param tasks the tasks
741      * @throws NullPointerException if any task is null
742      */
invokeAll(ForkJoinTask<?>.... tasks)743     public static void invokeAll(ForkJoinTask<?>... tasks) {
744         Throwable ex = null;
745         int last = tasks.length - 1;
746         for (int i = last; i >= 0; --i) {
747             ForkJoinTask<?> t = tasks[i];
748             if (t == null) {
749                 if (ex == null)
750                     ex = new NullPointerException();
751             }
752             else if (i != 0)
753                 t.fork();
754             else if (t.doInvoke() < NORMAL && ex == null)
755                 ex = t.getException();
756         }
757         for (int i = 1; i <= last; ++i) {
758             ForkJoinTask<?> t = tasks[i];
759             if (t != null) {
760                 if (ex != null)
761                     t.cancel(false);
762                 else if (t.doJoin() < NORMAL)
763                     ex = t.getException();
764             }
765         }
766         if (ex != null)
767             rethrow(ex);
768     }
769 
770     /**
771      * Forks all tasks in the specified collection, returning when
772      * {@code isDone} holds for each task or an (unchecked) exception
773      * is encountered, in which case the exception is rethrown. If
774      * more than one task encounters an exception, then this method
775      * throws any one of these exceptions. If any task encounters an
776      * exception, others may be cancelled. However, the execution
777      * status of individual tasks is not guaranteed upon exceptional
778      * return. The status of each task may be obtained using {@link
779      * #getException()} and related methods to check if they have been
780      * cancelled, completed normally or exceptionally, or left
781      * unprocessed.
782      *
783      * @param tasks the collection of tasks
784      * @param <T> the type of the values returned from the tasks
785      * @return the tasks argument, to simplify usage
786      * @throws NullPointerException if tasks or any element are null
787      */
invokeAll(Collection<T> tasks)788     public static <T extends ForkJoinTask<?>> Collection<T> invokeAll(Collection<T> tasks) {
789         if (!(tasks instanceof RandomAccess) || !(tasks instanceof List<?>)) {
790             invokeAll(tasks.toArray(new ForkJoinTask<?>[tasks.size()]));
791             return tasks;
792         }
793         @SuppressWarnings("unchecked")
794         List<? extends ForkJoinTask<?>> ts =
795             (List<? extends ForkJoinTask<?>>) tasks;
796         Throwable ex = null;
797         int last = ts.size() - 1;
798         for (int i = last; i >= 0; --i) {
799             ForkJoinTask<?> t = ts.get(i);
800             if (t == null) {
801                 if (ex == null)
802                     ex = new NullPointerException();
803             }
804             else if (i != 0)
805                 t.fork();
806             else if (t.doInvoke() < NORMAL && ex == null)
807                 ex = t.getException();
808         }
809         for (int i = 1; i <= last; ++i) {
810             ForkJoinTask<?> t = ts.get(i);
811             if (t != null) {
812                 if (ex != null)
813                     t.cancel(false);
814                 else if (t.doJoin() < NORMAL)
815                     ex = t.getException();
816             }
817         }
818         if (ex != null)
819             rethrow(ex);
820         return tasks;
821     }
822 
823     /**
824      * Attempts to cancel execution of this task. This attempt will
825      * fail if the task has already completed or could not be
826      * cancelled for some other reason. If successful, and this task
827      * has not started when {@code cancel} is called, execution of
828      * this task is suppressed. After this method returns
829      * successfully, unless there is an intervening call to {@link
830      * #reinitialize}, subsequent calls to {@link #isCancelled},
831      * {@link #isDone}, and {@code cancel} will return {@code true}
832      * and calls to {@link #join} and related methods will result in
833      * {@code CancellationException}.
834      *
835      * <p>This method may be overridden in subclasses, but if so, must
836      * still ensure that these properties hold. In particular, the
837      * {@code cancel} method itself must not throw exceptions.
838      *
839      * <p>This method is designed to be invoked by <em>other</em>
840      * tasks. To terminate the current task, you can just return or
841      * throw an unchecked exception from its computation method, or
842      * invoke {@link #completeExceptionally(Throwable)}.
843      *
844      * @param mayInterruptIfRunning this value has no effect in the
845      * default implementation because interrupts are not used to
846      * control cancellation.
847      *
848      * @return {@code true} if this task is now cancelled
849      */
cancel(boolean mayInterruptIfRunning)850     public boolean cancel(boolean mayInterruptIfRunning) {
851         return (setCompletion(CANCELLED) & DONE_MASK) == CANCELLED;
852     }
853 
isDone()854     public final boolean isDone() {
855         return status < 0;
856     }
857 
isCancelled()858     public final boolean isCancelled() {
859         return (status & DONE_MASK) == CANCELLED;
860     }
861 
862     /**
863      * Returns {@code true} if this task threw an exception or was cancelled.
864      *
865      * @return {@code true} if this task threw an exception or was cancelled
866      */
isCompletedAbnormally()867     public final boolean isCompletedAbnormally() {
868         return status < NORMAL;
869     }
870 
871     /**
872      * Returns {@code true} if this task completed without throwing an
873      * exception and was not cancelled.
874      *
875      * @return {@code true} if this task completed without throwing an
876      * exception and was not cancelled
877      */
isCompletedNormally()878     public final boolean isCompletedNormally() {
879         return (status & DONE_MASK) == NORMAL;
880     }
881 
882     /**
883      * Returns the exception thrown by the base computation, or a
884      * {@code CancellationException} if cancelled, or {@code null} if
885      * none or if the method has not yet completed.
886      *
887      * @return the exception, or {@code null} if none
888      */
getException()889     public final Throwable getException() {
890         int s = status & DONE_MASK;
891         return ((s >= NORMAL)    ? null :
892                 (s == CANCELLED) ? new CancellationException() :
893                 getThrowableException());
894     }
895 
896     /**
897      * Completes this task abnormally, and if not already aborted or
898      * cancelled, causes it to throw the given exception upon
899      * {@code join} and related operations. This method may be used
900      * to induce exceptions in asynchronous tasks, or to force
901      * completion of tasks that would not otherwise complete.  Its use
902      * in other situations is discouraged.  This method is
903      * overridable, but overridden versions must invoke {@code super}
904      * implementation to maintain guarantees.
905      *
906      * @param ex the exception to throw. If this exception is not a
907      * {@code RuntimeException} or {@code Error}, the actual exception
908      * thrown will be a {@code RuntimeException} with cause {@code ex}.
909      */
completeExceptionally(Throwable ex)910     public void completeExceptionally(Throwable ex) {
911         setExceptionalCompletion((ex instanceof RuntimeException) ||
912                                  (ex instanceof Error) ? ex :
913                                  new RuntimeException(ex));
914     }
915 
916     /**
917      * Completes this task, and if not already aborted or cancelled,
918      * returning the given value as the result of subsequent
919      * invocations of {@code join} and related operations. This method
920      * may be used to provide results for asynchronous tasks, or to
921      * provide alternative handling for tasks that would not otherwise
922      * complete normally. Its use in other situations is
923      * discouraged. This method is overridable, but overridden
924      * versions must invoke {@code super} implementation to maintain
925      * guarantees.
926      *
927      * @param value the result value for this task
928      */
complete(V value)929     public void complete(V value) {
930         try {
931             setRawResult(value);
932         } catch (Throwable rex) {
933             setExceptionalCompletion(rex);
934             return;
935         }
936         setCompletion(NORMAL);
937     }
938 
939     /**
940      * Completes this task normally without setting a value. The most
941      * recent value established by {@link #setRawResult} (or {@code
942      * null} by default) will be returned as the result of subsequent
943      * invocations of {@code join} and related operations.
944      *
945      * @since 1.8
946      */
quietlyComplete()947     public final void quietlyComplete() {
948         setCompletion(NORMAL);
949     }
950 
951     /**
952      * Waits if necessary for the computation to complete, and then
953      * retrieves its result.
954      *
955      * @return the computed result
956      * @throws CancellationException if the computation was cancelled
957      * @throws ExecutionException if the computation threw an
958      * exception
959      * @throws InterruptedException if the current thread is not a
960      * member of a ForkJoinPool and was interrupted while waiting
961      */
get()962     public final V get() throws InterruptedException, ExecutionException {
963         int s = (Thread.currentThread() instanceof ForkJoinWorkerThread) ?
964             doJoin() : externalInterruptibleAwaitDone();
965         if ((s &= DONE_MASK) == CANCELLED)
966             throw new CancellationException();
967         if (s == EXCEPTIONAL)
968             throw new ExecutionException(getThrowableException());
969         return getRawResult();
970     }
971 
972     /**
973      * Waits if necessary for at most the given time for the computation
974      * to complete, and then retrieves its result, if available.
975      *
976      * @param timeout the maximum time to wait
977      * @param unit the time unit of the timeout argument
978      * @return the computed result
979      * @throws CancellationException if the computation was cancelled
980      * @throws ExecutionException if the computation threw an
981      * exception
982      * @throws InterruptedException if the current thread is not a
983      * member of a ForkJoinPool and was interrupted while waiting
984      * @throws TimeoutException if the wait timed out
985      */
get(long timeout, TimeUnit unit)986     public final V get(long timeout, TimeUnit unit)
987         throws InterruptedException, ExecutionException, TimeoutException {
988         int s;
989         long nanos = unit.toNanos(timeout);
990         if (Thread.interrupted())
991             throw new InterruptedException();
992         if ((s = status) >= 0 && nanos > 0L) {
993             long d = System.nanoTime() + nanos;
994             long deadline = (d == 0L) ? 1L : d; // avoid 0
995             Thread t = Thread.currentThread();
996             if (t instanceof ForkJoinWorkerThread) {
997                 ForkJoinWorkerThread wt = (ForkJoinWorkerThread)t;
998                 s = wt.pool.awaitJoin(wt.workQueue, this, deadline);
999             }
1000             else if ((s = ((this instanceof CountedCompleter) ?
1001                            ForkJoinPool.common.externalHelpComplete(
1002                                (CountedCompleter<?>)this, 0) :
1003                            ForkJoinPool.common.tryExternalUnpush(this) ?
1004                            doExec() : 0)) >= 0) {
1005                 long ns, ms; // measure in nanosecs, but wait in millisecs
1006                 while ((s = status) >= 0 &&
1007                        (ns = deadline - System.nanoTime()) > 0L) {
1008                     if ((ms = TimeUnit.NANOSECONDS.toMillis(ns)) > 0L &&
1009                         U.compareAndSwapInt(this, STATUS, s, s | SIGNAL)) {
1010                         synchronized (this) {
1011                             if (status >= 0)
1012                                 wait(ms); // OK to throw InterruptedException
1013                             else
1014                                 notifyAll();
1015                         }
1016                     }
1017                 }
1018             }
1019         }
1020         if (s >= 0)
1021             s = status;
1022         if ((s &= DONE_MASK) != NORMAL) {
1023             if (s == CANCELLED)
1024                 throw new CancellationException();
1025             if (s != EXCEPTIONAL)
1026                 throw new TimeoutException();
1027             throw new ExecutionException(getThrowableException());
1028         }
1029         return getRawResult();
1030     }
1031 
1032     /**
1033      * Joins this task, without returning its result or throwing its
1034      * exception. This method may be useful when processing
1035      * collections of tasks when some have been cancelled or otherwise
1036      * known to have aborted.
1037      */
quietlyJoin()1038     public final void quietlyJoin() {
1039         doJoin();
1040     }
1041 
1042     /**
1043      * Commences performing this task and awaits its completion if
1044      * necessary, without returning its result or throwing its
1045      * exception.
1046      */
quietlyInvoke()1047     public final void quietlyInvoke() {
1048         doInvoke();
1049     }
1050 
1051     /**
1052      * Possibly executes tasks until the pool hosting the current task
1053      * {@linkplain ForkJoinPool#isQuiescent is quiescent}.  This
1054      * method may be of use in designs in which many tasks are forked,
1055      * but none are explicitly joined, instead executing them until
1056      * all are processed.
1057      */
helpQuiesce()1058     public static void helpQuiesce() {
1059         Thread t;
1060         if ((t = Thread.currentThread()) instanceof ForkJoinWorkerThread) {
1061             ForkJoinWorkerThread wt = (ForkJoinWorkerThread)t;
1062             wt.pool.helpQuiescePool(wt.workQueue);
1063         }
1064         else
1065             ForkJoinPool.quiesceCommonPool();
1066     }
1067 
1068     /**
1069      * Resets the internal bookkeeping state of this task, allowing a
1070      * subsequent {@code fork}. This method allows repeated reuse of
1071      * this task, but only if reuse occurs when this task has either
1072      * never been forked, or has been forked, then completed and all
1073      * outstanding joins of this task have also completed. Effects
1074      * under any other usage conditions are not guaranteed.
1075      * This method may be useful when executing
1076      * pre-constructed trees of subtasks in loops.
1077      *
1078      * <p>Upon completion of this method, {@code isDone()} reports
1079      * {@code false}, and {@code getException()} reports {@code
1080      * null}. However, the value returned by {@code getRawResult} is
1081      * unaffected. To clear this value, you can invoke {@code
1082      * setRawResult(null)}.
1083      */
reinitialize()1084     public void reinitialize() {
1085         if ((status & DONE_MASK) == EXCEPTIONAL)
1086             clearExceptionalCompletion();
1087         else
1088             status = 0;
1089     }
1090 
1091     /**
1092      * Returns the pool hosting the current thread, or {@code null}
1093      * if the current thread is executing outside of any ForkJoinPool.
1094      *
1095      * <p>This method returns {@code null} if and only if {@link
1096      * #inForkJoinPool} returns {@code false}.
1097      *
1098      * @return the pool, or {@code null} if none
1099      */
getPool()1100     public static ForkJoinPool getPool() {
1101         Thread t = Thread.currentThread();
1102         return (t instanceof ForkJoinWorkerThread) ?
1103             ((ForkJoinWorkerThread) t).pool : null;
1104     }
1105 
1106     /**
1107      * Returns {@code true} if the current thread is a {@link
1108      * ForkJoinWorkerThread} executing as a ForkJoinPool computation.
1109      *
1110      * @return {@code true} if the current thread is a {@link
1111      * ForkJoinWorkerThread} executing as a ForkJoinPool computation,
1112      * or {@code false} otherwise
1113      */
inForkJoinPool()1114     public static boolean inForkJoinPool() {
1115         return Thread.currentThread() instanceof ForkJoinWorkerThread;
1116     }
1117 
1118     /**
1119      * Tries to unschedule this task for execution. This method will
1120      * typically (but is not guaranteed to) succeed if this task is
1121      * the most recently forked task by the current thread, and has
1122      * not commenced executing in another thread.  This method may be
1123      * useful when arranging alternative local processing of tasks
1124      * that could have been, but were not, stolen.
1125      *
1126      * @return {@code true} if unforked
1127      */
tryUnfork()1128     public boolean tryUnfork() {
1129         Thread t;
1130         return (((t = Thread.currentThread()) instanceof ForkJoinWorkerThread) ?
1131                 ((ForkJoinWorkerThread)t).workQueue.tryUnpush(this) :
1132                 ForkJoinPool.common.tryExternalUnpush(this));
1133     }
1134 
1135     /**
1136      * Returns an estimate of the number of tasks that have been
1137      * forked by the current worker thread but not yet executed. This
1138      * value may be useful for heuristic decisions about whether to
1139      * fork other tasks.
1140      *
1141      * @return the number of tasks
1142      */
getQueuedTaskCount()1143     public static int getQueuedTaskCount() {
1144         Thread t; ForkJoinPool.WorkQueue q;
1145         if ((t = Thread.currentThread()) instanceof ForkJoinWorkerThread)
1146             q = ((ForkJoinWorkerThread)t).workQueue;
1147         else
1148             q = ForkJoinPool.commonSubmitterQueue();
1149         return (q == null) ? 0 : q.queueSize();
1150     }
1151 
1152     /**
1153      * Returns an estimate of how many more locally queued tasks are
1154      * held by the current worker thread than there are other worker
1155      * threads that might steal them, or zero if this thread is not
1156      * operating in a ForkJoinPool. This value may be useful for
1157      * heuristic decisions about whether to fork other tasks. In many
1158      * usages of ForkJoinTasks, at steady state, each worker should
1159      * aim to maintain a small constant surplus (for example, 3) of
1160      * tasks, and to process computations locally if this threshold is
1161      * exceeded.
1162      *
1163      * @return the surplus number of tasks, which may be negative
1164      */
getSurplusQueuedTaskCount()1165     public static int getSurplusQueuedTaskCount() {
1166         return ForkJoinPool.getSurplusQueuedTaskCount();
1167     }
1168 
1169     // Extension methods
1170 
1171     /**
1172      * Returns the result that would be returned by {@link #join}, even
1173      * if this task completed abnormally, or {@code null} if this task
1174      * is not known to have been completed.  This method is designed
1175      * to aid debugging, as well as to support extensions. Its use in
1176      * any other context is discouraged.
1177      *
1178      * @return the result, or {@code null} if not completed
1179      */
getRawResult()1180     public abstract V getRawResult();
1181 
1182     /**
1183      * Forces the given value to be returned as a result.  This method
1184      * is designed to support extensions, and should not in general be
1185      * called otherwise.
1186      *
1187      * @param value the value
1188      */
setRawResult(V value)1189     protected abstract void setRawResult(V value);
1190 
1191     /**
1192      * Immediately performs the base action of this task and returns
1193      * true if, upon return from this method, this task is guaranteed
1194      * to have completed normally. This method may return false
1195      * otherwise, to indicate that this task is not necessarily
1196      * complete (or is not known to be complete), for example in
1197      * asynchronous actions that require explicit invocations of
1198      * completion methods. This method may also throw an (unchecked)
1199      * exception to indicate abnormal exit. This method is designed to
1200      * support extensions, and should not in general be called
1201      * otherwise.
1202      *
1203      * @return {@code true} if this task is known to have completed normally
1204      */
exec()1205     protected abstract boolean exec();
1206 
1207     /**
1208      * Returns, but does not unschedule or execute, a task queued by
1209      * the current thread but not yet executed, if one is immediately
1210      * available. There is no guarantee that this task will actually
1211      * be polled or executed next. Conversely, this method may return
1212      * null even if a task exists but cannot be accessed without
1213      * contention with other threads.  This method is designed
1214      * primarily to support extensions, and is unlikely to be useful
1215      * otherwise.
1216      *
1217      * @return the next task, or {@code null} if none are available
1218      */
peekNextLocalTask()1219     protected static ForkJoinTask<?> peekNextLocalTask() {
1220         Thread t; ForkJoinPool.WorkQueue q;
1221         if ((t = Thread.currentThread()) instanceof ForkJoinWorkerThread)
1222             q = ((ForkJoinWorkerThread)t).workQueue;
1223         else
1224             q = ForkJoinPool.commonSubmitterQueue();
1225         return (q == null) ? null : q.peek();
1226     }
1227 
1228     /**
1229      * Unschedules and returns, without executing, the next task
1230      * queued by the current thread but not yet executed, if the
1231      * current thread is operating in a ForkJoinPool.  This method is
1232      * designed primarily to support extensions, and is unlikely to be
1233      * useful otherwise.
1234      *
1235      * @return the next task, or {@code null} if none are available
1236      */
pollNextLocalTask()1237     protected static ForkJoinTask<?> pollNextLocalTask() {
1238         Thread t;
1239         return ((t = Thread.currentThread()) instanceof ForkJoinWorkerThread) ?
1240             ((ForkJoinWorkerThread)t).workQueue.nextLocalTask() :
1241             null;
1242     }
1243 
1244     /**
1245      * If the current thread is operating in a ForkJoinPool,
1246      * unschedules and returns, without executing, the next task
1247      * queued by the current thread but not yet executed, if one is
1248      * available, or if not available, a task that was forked by some
1249      * other thread, if available. Availability may be transient, so a
1250      * {@code null} result does not necessarily imply quiescence of
1251      * the pool this task is operating in.  This method is designed
1252      * primarily to support extensions, and is unlikely to be useful
1253      * otherwise.
1254      *
1255      * @return a task, or {@code null} if none are available
1256      */
pollTask()1257     protected static ForkJoinTask<?> pollTask() {
1258         Thread t; ForkJoinWorkerThread wt;
1259         return ((t = Thread.currentThread()) instanceof ForkJoinWorkerThread) ?
1260             (wt = (ForkJoinWorkerThread)t).pool.nextTaskFor(wt.workQueue) :
1261             null;
1262     }
1263 
1264     /**
1265      * If the current thread is operating in a ForkJoinPool,
1266      * unschedules and returns, without executing, a task externally
1267      * submitted to the pool, if one is available. Availability may be
1268      * transient, so a {@code null} result does not necessarily imply
1269      * quiescence of the pool.  This method is designed primarily to
1270      * support extensions, and is unlikely to be useful otherwise.
1271      *
1272      * @return a task, or {@code null} if none are available
1273      * @since 9
1274      * @hide
1275      */
1276     // android-changed - hidden
pollSubmission()1277     protected static ForkJoinTask<?> pollSubmission() {
1278         Thread t;
1279         return ((t = Thread.currentThread()) instanceof ForkJoinWorkerThread) ?
1280             ((ForkJoinWorkerThread)t).pool.pollSubmission() : null;
1281     }
1282 
1283     // tag operations
1284 
1285     /**
1286      * Returns the tag for this task.
1287      *
1288      * @return the tag for this task
1289      * @since 1.8
1290      */
getForkJoinTaskTag()1291     public final short getForkJoinTaskTag() {
1292         return (short)status;
1293     }
1294 
1295     /**
1296      * Atomically sets the tag value for this task and returns the old value.
1297      *
1298      * @param newValue the new tag value
1299      * @return the previous value of the tag
1300      * @since 1.8
1301      */
setForkJoinTaskTag(short newValue)1302     public final short setForkJoinTaskTag(short newValue) {
1303         for (int s;;) {
1304             if (U.compareAndSwapInt(this, STATUS, s = status,
1305                                     (s & ~SMASK) | (newValue & SMASK)))
1306                 return (short)s;
1307         }
1308     }
1309 
1310     /**
1311      * Atomically conditionally sets the tag value for this task.
1312      * Among other applications, tags can be used as visit markers
1313      * in tasks operating on graphs, as in methods that check: {@code
1314      * if (task.compareAndSetForkJoinTaskTag((short)0, (short)1))}
1315      * before processing, otherwise exiting because the node has
1316      * already been visited.
1317      *
1318      * @param expect the expected tag value
1319      * @param update the new tag value
1320      * @return {@code true} if successful; i.e., the current value was
1321      * equal to {@code expect} and was changed to {@code update}.
1322      * @since 1.8
1323      */
compareAndSetForkJoinTaskTag(short expect, short update)1324     public final boolean compareAndSetForkJoinTaskTag(short expect, short update) {
1325         for (int s;;) {
1326             if ((short)(s = status) != expect)
1327                 return false;
1328             if (U.compareAndSwapInt(this, STATUS, s,
1329                                     (s & ~SMASK) | (update & SMASK)))
1330                 return true;
1331         }
1332     }
1333 
1334     /**
1335      * Adapter for Runnables. This implements RunnableFuture
1336      * to be compliant with AbstractExecutorService constraints
1337      * when used in ForkJoinPool.
1338      */
1339     static final class AdaptedRunnable<T> extends ForkJoinTask<T>
1340         implements RunnableFuture<T> {
1341         final Runnable runnable;
1342         T result;
AdaptedRunnable(Runnable runnable, T result)1343         AdaptedRunnable(Runnable runnable, T result) {
1344             if (runnable == null) throw new NullPointerException();
1345             this.runnable = runnable;
1346             this.result = result; // OK to set this even before completion
1347         }
getRawResult()1348         public final T getRawResult() { return result; }
setRawResult(T v)1349         public final void setRawResult(T v) { result = v; }
exec()1350         public final boolean exec() { runnable.run(); return true; }
run()1351         public final void run() { invoke(); }
1352         private static final long serialVersionUID = 5232453952276885070L;
1353     }
1354 
1355     /**
1356      * Adapter for Runnables without results.
1357      */
1358     static final class AdaptedRunnableAction extends ForkJoinTask<Void>
1359         implements RunnableFuture<Void> {
1360         final Runnable runnable;
AdaptedRunnableAction(Runnable runnable)1361         AdaptedRunnableAction(Runnable runnable) {
1362             if (runnable == null) throw new NullPointerException();
1363             this.runnable = runnable;
1364         }
getRawResult()1365         public final Void getRawResult() { return null; }
setRawResult(Void v)1366         public final void setRawResult(Void v) { }
exec()1367         public final boolean exec() { runnable.run(); return true; }
run()1368         public final void run() { invoke(); }
1369         private static final long serialVersionUID = 5232453952276885070L;
1370     }
1371 
1372     /**
1373      * Adapter for Runnables in which failure forces worker exception.
1374      */
1375     static final class RunnableExecuteAction extends ForkJoinTask<Void> {
1376         final Runnable runnable;
RunnableExecuteAction(Runnable runnable)1377         RunnableExecuteAction(Runnable runnable) {
1378             if (runnable == null) throw new NullPointerException();
1379             this.runnable = runnable;
1380         }
getRawResult()1381         public final Void getRawResult() { return null; }
setRawResult(Void v)1382         public final void setRawResult(Void v) { }
exec()1383         public final boolean exec() { runnable.run(); return true; }
internalPropagateException(Throwable ex)1384         void internalPropagateException(Throwable ex) {
1385             rethrow(ex); // rethrow outside exec() catches.
1386         }
1387         private static final long serialVersionUID = 5232453952276885070L;
1388     }
1389 
1390     /**
1391      * Adapter for Callables.
1392      */
1393     static final class AdaptedCallable<T> extends ForkJoinTask<T>
1394         implements RunnableFuture<T> {
1395         final Callable<? extends T> callable;
1396         T result;
AdaptedCallable(Callable<? extends T> callable)1397         AdaptedCallable(Callable<? extends T> callable) {
1398             if (callable == null) throw new NullPointerException();
1399             this.callable = callable;
1400         }
getRawResult()1401         public final T getRawResult() { return result; }
setRawResult(T v)1402         public final void setRawResult(T v) { result = v; }
exec()1403         public final boolean exec() {
1404             try {
1405                 result = callable.call();
1406                 return true;
1407             } catch (RuntimeException rex) {
1408                 throw rex;
1409             } catch (Exception ex) {
1410                 throw new RuntimeException(ex);
1411             }
1412         }
run()1413         public final void run() { invoke(); }
1414         private static final long serialVersionUID = 2838392045355241008L;
1415     }
1416 
1417     /**
1418      * Returns a new {@code ForkJoinTask} that performs the {@code run}
1419      * method of the given {@code Runnable} as its action, and returns
1420      * a null result upon {@link #join}.
1421      *
1422      * @param runnable the runnable action
1423      * @return the task
1424      */
adapt(Runnable runnable)1425     public static ForkJoinTask<?> adapt(Runnable runnable) {
1426         return new AdaptedRunnableAction(runnable);
1427     }
1428 
1429     /**
1430      * Returns a new {@code ForkJoinTask} that performs the {@code run}
1431      * method of the given {@code Runnable} as its action, and returns
1432      * the given result upon {@link #join}.
1433      *
1434      * @param runnable the runnable action
1435      * @param result the result upon completion
1436      * @param <T> the type of the result
1437      * @return the task
1438      */
adapt(Runnable runnable, T result)1439     public static <T> ForkJoinTask<T> adapt(Runnable runnable, T result) {
1440         return new AdaptedRunnable<T>(runnable, result);
1441     }
1442 
1443     /**
1444      * Returns a new {@code ForkJoinTask} that performs the {@code call}
1445      * method of the given {@code Callable} as its action, and returns
1446      * its result upon {@link #join}, translating any checked exceptions
1447      * encountered into {@code RuntimeException}.
1448      *
1449      * @param callable the callable action
1450      * @param <T> the type of the callable's result
1451      * @return the task
1452      */
adapt(Callable<? extends T> callable)1453     public static <T> ForkJoinTask<T> adapt(Callable<? extends T> callable) {
1454         return new AdaptedCallable<T>(callable);
1455     }
1456 
1457     // Serialization support
1458 
1459     private static final long serialVersionUID = -7721805057305804111L;
1460 
1461     /**
1462      * Saves this task to a stream (that is, serializes it).
1463      *
1464      * @param s the stream
1465      * @throws java.io.IOException if an I/O error occurs
1466      * @serialData the current run status and the exception thrown
1467      * during execution, or {@code null} if none
1468      */
writeObject(java.io.ObjectOutputStream s)1469     private void writeObject(java.io.ObjectOutputStream s)
1470         throws java.io.IOException {
1471         s.defaultWriteObject();
1472         s.writeObject(getException());
1473     }
1474 
1475     /**
1476      * Reconstitutes this task from a stream (that is, deserializes it).
1477      * @param s the stream
1478      * @throws ClassNotFoundException if the class of a serialized object
1479      *         could not be found
1480      * @throws java.io.IOException if an I/O error occurs
1481      */
readObject(java.io.ObjectInputStream s)1482     private void readObject(java.io.ObjectInputStream s)
1483         throws java.io.IOException, ClassNotFoundException {
1484         s.defaultReadObject();
1485         Object ex = s.readObject();
1486         if (ex != null)
1487             setExceptionalCompletion((Throwable)ex);
1488     }
1489 
1490     // Unsafe mechanics
1491     private static final sun.misc.Unsafe U = sun.misc.Unsafe.getUnsafe();
1492     private static final long STATUS;
1493 
1494     static {
1495         exceptionTableLock = new ReentrantLock();
1496         exceptionTableRefQueue = new ReferenceQueue<Object>();
1497         exceptionTable = new ExceptionNode[EXCEPTION_MAP_CAPACITY];
1498         try {
1499             STATUS = U.objectFieldOffset
1500                 (ForkJoinTask.class.getDeclaredField("status"));
1501         } catch (ReflectiveOperationException e) {
1502             throw new Error(e);
1503         }
1504     }
1505 
1506 }
1507