ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/jsr166/jsr166/src/jsr166y/ForkJoinPool.java
Revision: 1.55
Committed: Sun Apr 18 13:59:57 2010 UTC (14 years, 1 month ago) by dl
Branch: MAIN
Changes since 1.54: +1 -1 lines
Log Message:
Typo

File Contents

# User Rev Content
1 dl 1.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/licenses/publicdomain
5     */
6    
7     package jsr166y;
8 jsr166 1.22
9 dl 1.1 import java.util.concurrent.*;
10 jsr166 1.22
11     import java.util.ArrayList;
12     import java.util.Arrays;
13     import java.util.Collection;
14     import java.util.Collections;
15     import java.util.List;
16     import java.util.concurrent.locks.LockSupport;
17     import java.util.concurrent.locks.ReentrantLock;
18     import java.util.concurrent.atomic.AtomicInteger;
19 dl 1.53 import java.util.concurrent.CountDownLatch;
20 dl 1.1
21     /**
22 jsr166 1.29 * An {@link ExecutorService} for running {@link ForkJoinTask}s.
23 jsr166 1.39 * A {@code ForkJoinPool} provides the entry point for submissions
24     * from non-{@code ForkJoinTask}s, as well as management and
25 jsr166 1.48 * monitoring operations.
26 dl 1.1 *
27 dl 1.42 * <p>A {@code ForkJoinPool} differs from other kinds of {@link
28     * ExecutorService} mainly by virtue of employing
29     * <em>work-stealing</em>: all threads in the pool attempt to find and
30     * execute subtasks created by other active tasks (eventually blocking
31     * waiting for work if none exist). This enables efficient processing
32     * when most tasks spawn other subtasks (as do most {@code
33     * ForkJoinTask}s). A {@code ForkJoinPool} may also be used for mixed
34     * execution of some plain {@code Runnable}- or {@code Callable}-
35     * based activities along with {@code ForkJoinTask}s. When setting
36     * {@linkplain #setAsyncMode async mode}, a {@code ForkJoinPool} may
37     * also be appropriate for use with fine-grained tasks of any form
38     * that are never joined. Otherwise, other {@code ExecutorService}
39     * implementations are typically more appropriate choices.
40 dl 1.1 *
41 dl 1.42 * <p>A {@code ForkJoinPool} is constructed with a given target
42     * parallelism level; by default, equal to the number of available
43     * processors. Unless configured otherwise via {@link
44     * #setMaintainsParallelism}, the pool attempts to maintain this
45     * number of active (or available) threads by dynamically adding,
46     * suspending, or resuming internal worker threads, even if some tasks
47 dl 1.44 * are stalled waiting to join others. However, no such adjustments
48     * are performed in the face of blocked IO or other unmanaged
49 jsr166 1.39 * synchronization. The nested {@link ManagedBlocker} interface
50     * enables extension of the kinds of synchronization accommodated.
51     * The target parallelism level may also be changed dynamically
52 dl 1.42 * ({@link #setParallelism}). The total number of threads may be
53     * limited using method {@link #setMaximumPoolSize}, in which case it
54     * may become possible for the activities of a pool to stall due to
55 dl 1.53 * the lack of available threads to process new tasks. When the pool
56     * is executing tasks, these and other configuration setting methods
57     * may only gradually affect actual pool sizes. It is normally best
58     * practice to invoke these methods only when the pool is known to be
59     * quiescent.
60 dl 1.1 *
61     * <p>In addition to execution and lifecycle control methods, this
62     * class provides status check methods (for example
63 jsr166 1.29 * {@link #getStealCount}) that are intended to aid in developing,
64 dl 1.1 * tuning, and monitoring fork/join applications. Also, method
65 jsr166 1.29 * {@link #toString} returns indications of pool state in a
66 dl 1.2 * convenient form for informal monitoring.
67 dl 1.1 *
68 dl 1.42 * <p><b>Sample Usage.</b> Normally a single {@code ForkJoinPool} is
69     * used for all parallel task execution in a program or subsystem.
70     * Otherwise, use would not usually outweigh the construction and
71     * bookkeeping overhead of creating a large set of threads. For
72 jsr166 1.43 * example, a common pool could be used for the {@code SortTasks}
73 dl 1.42 * illustrated in {@link RecursiveAction}. Because {@code
74     * ForkJoinPool} uses threads in {@linkplain java.lang.Thread#isDaemon
75 jsr166 1.43 * daemon} mode, there is typically no need to explicitly {@link
76 dl 1.42 * #shutdown} such a pool upon program exit.
77     *
78     * <pre>
79     * static final ForkJoinPool mainPool = new ForkJoinPool();
80     * ...
81     * public void sort(long[] array) {
82     * mainPool.invoke(new SortTask(array, 0, array.length));
83     * }
84     * </pre>
85     *
86 dl 1.1 * <p><b>Implementation notes</b>: This implementation restricts the
87 dl 1.2 * maximum number of running threads to 32767. Attempts to create
88 jsr166 1.48 * pools with greater than the maximum number result in
89 jsr166 1.39 * {@code IllegalArgumentException}.
90 jsr166 1.16 *
91 jsr166 1.48 * <p>This implementation rejects submitted tasks (that is, by throwing
92     * {@link RejectedExecutionException}) only when the pool is shut down.
93     *
94 jsr166 1.16 * @since 1.7
95     * @author Doug Lea
96 dl 1.1 */
97 dl 1.2 public class ForkJoinPool extends AbstractExecutorService {
98 dl 1.1
99     /*
100 dl 1.53 * Implementation Overview
101     *
102     * This class provides the central bookkeeping and control for a
103     * set of worker threads: Submissions from non-FJ threads enter
104     * into a submission queue. Workers take these tasks and typically
105     * split them into subtasks that may be stolen by other workers.
106     * The main work-stealing mechanics implemented in class
107     * ForkJoinWorkerThread give first priority to processing tasks
108     * from their own queues (LIFO or FIFO, depending on mode), then
109     * to randomized FIFO steals of tasks in other worker queues, and
110     * lastly to new submissions. These mechanics do not consider
111     * affinities, loads, cache localities, etc, so rarely provide the
112     * best possible performance on a given machine, but portably
113     * provide good throughput by averaging over these factors.
114     * (Further, even if we did try to use such information, we do not
115     * usually have a basis for exploiting it. For example, some sets
116     * of tasks profit from cache affinities, but others are harmed by
117     * cache pollution effects.)
118     *
119     * The main throughput advantages of work-stealing stem from
120     * decentralized control -- workers mostly steal tasks from each
121     * other. We do not want to negate this by creating bottlenecks
122     * implementing the management responsibilities of this class. So
123     * we use a collection of techniques that avoid, reduce, or cope
124     * well with contention. These entail several instances of
125     * bit-packing into CASable fields to maintain only the minimally
126     * required atomicity. To enable such packing, we restrict maximum
127     * parallelism to (1<<15)-1 (enabling twice this to fit into a 16
128     * bit field), which is far in excess of normal operating range.
129     * Even though updates to some of these bookkeeping fields do
130     * sometimes contend with each other, they don't normally
131     * cache-contend with updates to others enough to warrant memory
132     * padding or isolation. So they are all held as fields of
133     * ForkJoinPool objects. The main capabilities are as follows:
134     *
135     * 1. Creating and removing workers. Workers are recorded in the
136     * "workers" array. This is an array as opposed to some other data
137     * structure to support index-based random steals by workers.
138     * Updates to the array recording new workers and unrecording
139     * terminated ones are protected from each other by a lock
140     * (workerLock) but the array is otherwise concurrently readable,
141     * and accessed directly by workers. To simplify index-based
142     * operations, the array size is always a power of two, and all
143     * readers must tolerate null slots. Currently, all but the first
144     * worker thread creation is on-demand, triggered by task
145     * submissions, replacement of terminated workers, and/or
146     * compensation for blocked workers. However, all other support
147     * code is set up to work with other policies.
148     *
149     * 2. Bookkeeping for dynamically adding and removing workers. We
150     * maintain a given level of parallelism (or, if
151     * maintainsParallelism is false, at least avoid starvation). When
152     * some workers are known to be blocked (on joins or via
153     * ManagedBlocker), we may create or resume others to take their
154     * place until they unblock (see below). Implementing this
155     * requires counts of the number of "running" threads (i.e., those
156     * that are neither blocked nor artifically suspended) as well as
157     * the total number. These two values are packed into one field,
158     * "workerCounts" because we need accurate snapshots when deciding
159     * to create, resume or suspend. To support these decisions,
160     * updates must be prospective (not retrospective). For example,
161     * the running count is decremented before blocking by a thread
162     * about to block, but incremented by the thread about to unblock
163     * it. (In a few cases, these prospective updates may need to be
164     * rolled back, for example when deciding to create a new worker
165     * but the thread factory fails or returns null. In these cases,
166     * we are no worse off wrt other decisions than we would be
167     * otherwise.) Updates to the workerCounts field sometimes
168     * transiently encounter a fair amount of contention when join
169     * dependencies are such that many threads block or unblock at
170     * about the same time. We alleviate this by sometimes bundling
171     * updates (for example blocking one thread on join and resuming a
172     * spare cancel each other out), and in most other cases
173     * performing an alternative action (like releasing waiters and
174     * finding spares; see below) as a more productive form of
175     * backoff.
176     *
177     * 3. Maintaining global run state. The run state of the pool
178     * consists of a runLevel (SHUTDOWN, TERMINATING, etc) similar to
179     * those in other Executor implementations, as well as a count of
180     * "active" workers -- those that are, or soon will be, or
181     * recently were executing tasks. The runLevel and active count
182     * are packed together in order to correctly trigger shutdown and
183     * termination. Without care, active counts can be subject to very
184     * high contention. We substantially reduce this contention by
185     * relaxing update rules. A worker must claim active status
186     * prospectively, by activating if it sees that a submitted or
187     * stealable task exists (it may find after activating that the
188     * task no longer exists). It stays active while processing this
189     * task (if it exists) and any other local subtasks it produces,
190     * until it cannot find any other tasks. It then tries
191     * inactivating (see method preStep), but upon update contention
192     * instead scans for more tasks, later retrying inactivation if it
193     * doesn't find any.
194     *
195     * 4. Managing idle workers waiting for tasks. We cannot let
196     * workers spin indefinitely scanning for tasks when none are
197     * available. On the other hand, we must quickly prod them into
198     * action when new tasks are submitted or generated. We
199     * park/unpark these idle workers using an event-count scheme.
200     * Field eventCount is incremented upon events that may enable
201     * workers that previously could not find a task to now find one:
202     * Submission of a new task to the pool, or another worker pushing
203     * a task onto a previously empty queue. (We also use this
204     * mechanism for termination and reconfiguration actions that
205     * require wakeups of idle workers). Each worker maintains its
206     * last known event count, and blocks when a scan for work did not
207     * find a task AND its lastEventCount matches the current
208     * eventCount. Waiting idle workers are recorded in a variant of
209     * Treiber stack headed by field eventWaiters which, when nonzero,
210     * encodes the thread index and count awaited for by the worker
211     * thread most recently calling eventSync. This thread in turn has
212     * a record (field nextEventWaiter) for the next waiting worker.
213     * In addition to allowing simpler decisions about need for
214     * wakeup, the event count bits in eventWaiters serve the role of
215     * tags to avoid ABA errors in Treiber stacks. To reduce delays
216     * in task diffusion, workers not otherwise occupied may invoke
217     * method releaseWaiters, that removes and signals (unparks)
218     * workers not waiting on current count. To minimize task
219     * production stalls associate with signalling, any worker pushing
220     * a task on an empty queue invokes the weaker method signalWork,
221     * that only releases idle workers until it detects interference
222     * by other threads trying to release, and lets them take
223     * over. The net effect is a tree-like diffusion of signals, where
224     * released threads and possibly others) help with unparks. To
225     * further reduce contention effects a bit, failed CASes to
226     * increment field eventCount are tolerated without retries.
227     * Conceptually they are merged into the same event, which is OK
228     * when their only purpose is to enable workers to scan for work.
229     *
230     * 5. Managing suspension of extra workers. When a worker is about
231     * to block waiting for a join (or via ManagedBlockers), we may
232     * create a new thread to maintain parallelism level, or at least
233     * avoid starvation (see below). Usually, extra threads are needed
234     * for only very short periods, yet join dependencies are such
235     * that we sometimes need them in bursts. Rather than create new
236     * threads each time this happens, we suspend no-longer-needed
237     * extra ones as "spares". For most purposes, we don't distinguish
238     * "extra" spare threads from normal "core" threads: On each call
239     * to preStep (the only point at which we can do this) a worker
240     * checks to see if there are now too many running workers, and if
241     * so, suspends itself. Methods preJoin and doBlock look for
242     * suspended threads to resume before considering creating a new
243     * replacement. We don't need a special data structure to maintain
244     * spares; simply scanning the workers array looking for
245     * worker.isSuspended() is fine because the calling thread is
246     * otherwise not doing anything useful anyway; we are at least as
247     * happy if after locating a spare, the caller doesn't actually
248     * block because the join is ready before we try to adjust and
249     * compensate. Note that this is intrinsically racy. One thread
250     * may become a spare at about the same time as another is
251     * needlessly being created. We counteract this and related slop
252     * in part by requiring resumed spares to immediately recheck (in
253     * preStep) to see whether they they should re-suspend. The only
254     * effective difference between "extra" and "core" threads is that
255     * we allow the "extra" ones to time out and die if they are not
256     * resumed within a keep-alive interval of a few seconds. This is
257     * implemented mainly within ForkJoinWorkerThread, but requires
258     * some coordination (isTrimmed() -- meaning killed while
259     * suspended) to correctly maintain pool counts.
260     *
261     * 6. Deciding when to create new workers. The main dynamic
262     * control in this class is deciding when to create extra threads,
263     * in methods preJoin and doBlock. We always need to create one
264     * when the number of running threads becomes zero. But because
265     * blocked joins are typically dependent, we don't necessarily
266     * need or want one-to-one replacement. Using a one-to-one
267     * compensation rule often leads to enough useless overhead
268     * creating, suspending, resuming, and/or killing threads to
269     * signficantly degrade throughput. We use a rule reflecting the
270     * idea that, the more spare threads you already have, the more
271     * evidence you need to create another one; where "evidence" is
272     * expressed as the current deficit -- target minus running
273     * threads. To reduce flickering and drift around target values,
274     * the relation is quadratic: adding a spare if (dc*dc)>=(sc*pc)
275     * (where dc is deficit, sc is number of spare threads and pc is
276     * target parallelism.) This effectively reduces churn at the
277     * price of systematically undershooting target parallelism when
278     * many threads are blocked. However, biasing toward undeshooting
279     * partially compensates for the above mechanics to suspend extra
280     * threads, that normally lead to overshoot because we can only
281     * suspend workers in-between top-level actions. It also better
282     * copes with the fact that some of the methods in this class tend
283     * to never become compiled (but are interpreted), so some
284     * components of the entire set of controls might execute many
285     * times faster than others. And similarly for cases where the
286     * apparent lack of work is just due to GC stalls and other
287     * transient system activity.
288     *
289     * 7. Maintaining other configuration parameters and monitoring
290     * statistics. Updates to fields controlling parallelism level,
291     * max size, etc can only meaningfully take effect for individual
292     * threads upon their next top-level actions; i.e., between
293     * stealing/running tasks/submission, which are separated by calls
294     * to preStep. Memory ordering for these (assumed infrequent)
295     * reconfiguration calls is ensured by using reads and writes to
296     * volatile field workerCounts (that must be read in preStep anyway)
297     * as "fences" -- user-level reads are preceded by reads of
298     * workCounts, and writes are followed by no-op CAS to
299     * workerCounts. The values reported by other management and
300     * monitoring methods are either computed on demand, or are kept
301     * in fields that are only updated when threads are otherwise
302     * idle.
303     *
304     * Beware that there is a lot of representation-level coupling
305     * among classes ForkJoinPool, ForkJoinWorkerThread, and
306     * ForkJoinTask. For example, direct access to "workers" array by
307     * workers, and direct access to ForkJoinTask.status by both
308     * ForkJoinPool and ForkJoinWorkerThread. There is little point
309     * trying to reduce this, since any associated future changes in
310     * representations will need to be accompanied by algorithmic
311     * changes anyway.
312     *
313     * Style notes: There are lots of inline assignments (of form
314     * "while ((local = field) != 0)") which are usually the simplest
315     * way to ensure read orderings. Also several occurrences of the
316     * unusual "do {} while(!cas...)" which is the simplest way to
317     * force an update of a CAS'ed variable. There are also a few
318     * other coding oddities that help some methods perform reasonably
319     * even when interpreted (not compiled).
320     *
321     * The order of declarations in this file is: (1) statics (2)
322     * fields (along with constants used when unpacking some of them)
323     * (3) internal control methods (4) callbacks and other support
324     * for ForkJoinTask and ForkJoinWorkerThread classes, (5) exported
325     * methods (plus a few little helpers).
326 dl 1.1 */
327    
328     /**
329 jsr166 1.35 * Factory for creating new {@link ForkJoinWorkerThread}s.
330     * A {@code ForkJoinWorkerThreadFactory} must be defined and used
331     * for {@code ForkJoinWorkerThread} subclasses that extend base
332     * functionality or initialize threads with different contexts.
333 dl 1.1 */
334     public static interface ForkJoinWorkerThreadFactory {
335     /**
336     * Returns a new worker thread operating in the given pool.
337     *
338     * @param pool the pool this thread works in
339 jsr166 1.48 * @throws NullPointerException if the pool is null
340 dl 1.1 */
341     public ForkJoinWorkerThread newThread(ForkJoinPool pool);
342     }
343    
344     /**
345 jsr166 1.17 * Default ForkJoinWorkerThreadFactory implementation; creates a
346 dl 1.1 * new ForkJoinWorkerThread.
347     */
348 dl 1.2 static class DefaultForkJoinWorkerThreadFactory
349 dl 1.1 implements ForkJoinWorkerThreadFactory {
350     public ForkJoinWorkerThread newThread(ForkJoinPool pool) {
351 dl 1.53 return new ForkJoinWorkerThread(pool);
352 dl 1.1 }
353     }
354    
355     /**
356 dl 1.2 * Creates a new ForkJoinWorkerThread. This factory is used unless
357     * overridden in ForkJoinPool constructors.
358 dl 1.1 */
359 dl 1.2 public static final ForkJoinWorkerThreadFactory
360 dl 1.1 defaultForkJoinWorkerThreadFactory =
361     new DefaultForkJoinWorkerThreadFactory();
362    
363     /**
364     * Permission required for callers of methods that may start or
365     * kill threads.
366     */
367     private static final RuntimePermission modifyThreadPermission =
368     new RuntimePermission("modifyThread");
369    
370     /**
371     * If there is a security manager, makes sure caller has
372     * permission to modify threads.
373     */
374     private static void checkPermission() {
375     SecurityManager security = System.getSecurityManager();
376     if (security != null)
377     security.checkPermission(modifyThreadPermission);
378     }
379    
380     /**
381     * Generator for assigning sequence numbers as pool names.
382     */
383     private static final AtomicInteger poolNumberGenerator =
384     new AtomicInteger();
385    
386     /**
387 dl 1.53 * Absolute bound for parallelism level. Twice this number must
388     * fit into a 16bit field to enable word-packing for some counts.
389     */
390     private static final int MAX_THREADS = 0x7fff;
391    
392     /**
393     * Array holding all worker threads in the pool. Array size must
394     * be a power of two. Updates and replacements are protected by
395     * workerLock, but the array is always kept in a consistent enough
396     * state to be randomly accessed without locking by workers
397     * performing work-stealing, as well as other traversal-based
398     * methods in this class. All readers must tolerate that some
399     * array slots may be null.
400 dl 1.1 */
401     volatile ForkJoinWorkerThread[] workers;
402    
403     /**
404 dl 1.53 * Queue for external submissions.
405 dl 1.1 */
406 dl 1.53 private final LinkedTransferQueue<ForkJoinTask<?>> submissionQueue;
407 dl 1.1
408     /**
409 dl 1.53 * Lock protecting updates to workers array.
410 dl 1.1 */
411 dl 1.53 private final ReentrantLock workerLock;
412 dl 1.1
413     /**
414 dl 1.53 * Latch released upon termination.
415 dl 1.1 */
416 dl 1.53 private final CountDownLatch terminationLatch;
417 dl 1.1
418     /**
419     * Creation factory for worker threads.
420     */
421     private final ForkJoinWorkerThreadFactory factory;
422    
423     /**
424 dl 1.53 * Sum of per-thread steal counts, updated only when threads are
425     * idle or terminating.
426 dl 1.1 */
427 dl 1.53 private volatile long stealCount;
428 dl 1.1
429     /**
430 dl 1.53 * Encoded record of top of treiber stack of threads waiting for
431     * events. The top 32 bits contain the count being waited for. The
432     * bottom word contains one plus the pool index of waiting worker
433     * thread.
434 dl 1.1 */
435 dl 1.53 private volatile long eventWaiters;
436    
437     private static final int EVENT_COUNT_SHIFT = 32;
438     private static final long WAITER_INDEX_MASK = (1L << EVENT_COUNT_SHIFT)-1L;
439 dl 1.1
440     /**
441 dl 1.53 * A counter for events that may wake up worker threads:
442     * - Submission of a new task to the pool
443     * - A worker pushing a task on an empty queue
444     * - termination and reconfiguration
445 dl 1.1 */
446 dl 1.53 private volatile int eventCount;
447    
448     /**
449     * Lifecycle control. The low word contains the number of workers
450     * that are (probably) executing tasks. This value is atomically
451     * incremented before a worker gets a task to run, and decremented
452     * when worker has no tasks and cannot find any. Bits 16-18
453     * contain runLevel value. When all are zero, the pool is
454     * running. Level transitions are monotonic (running -> shutdown
455     * -> terminating -> terminated) so each transition adds a bit.
456     * These are bundled together to ensure consistent read for
457     * termination checks (i.e., that runLevel is at least SHUTDOWN
458     * and active threads is zero).
459     */
460     private volatile int runState;
461    
462     // Note: The order among run level values matters.
463     private static final int RUNLEVEL_SHIFT = 16;
464     private static final int SHUTDOWN = 1 << RUNLEVEL_SHIFT;
465     private static final int TERMINATING = 1 << (RUNLEVEL_SHIFT + 1);
466     private static final int TERMINATED = 1 << (RUNLEVEL_SHIFT + 2);
467     private static final int ACTIVE_COUNT_MASK = (1 << RUNLEVEL_SHIFT) - 1;
468     private static final int ONE_ACTIVE = 1; // active update delta
469 dl 1.1
470     /**
471 dl 1.53 * Holds number of total (i.e., created and not yet terminated)
472     * and running (i.e., not blocked on joins or other managed sync)
473     * threads, packed together to ensure consistent snapshot when
474     * making decisions about creating and suspending spare
475     * threads. Updated only by CAS. Note that adding a new worker
476     * requires incrementing both counts, since workers start off in
477     * running state. This field is also used for memory-fencing
478     * configuration parameters.
479     */
480     private volatile int workerCounts;
481    
482     private static final int TOTAL_COUNT_SHIFT = 16;
483     private static final int RUNNING_COUNT_MASK = (1 << TOTAL_COUNT_SHIFT) - 1;
484     private static final int ONE_RUNNING = 1;
485     private static final int ONE_TOTAL = 1 << TOTAL_COUNT_SHIFT;
486    
487     /*
488     * Fields parallelism. maxPoolSize, locallyFifo,
489     * maintainsParallelism, and ueh are non-volatile, but external
490     * reads/writes use workerCount fences to ensure visability.
491 dl 1.1 */
492    
493     /**
494 dl 1.53 * The target parallelism level.
495 dl 1.1 */
496 dl 1.53 private int parallelism;
497 dl 1.1
498     /**
499 dl 1.53 * The maximum allowed pool size.
500 dl 1.1 */
501 dl 1.53 private int maxPoolSize;
502 dl 1.1
503     /**
504 dl 1.53 * True if use local fifo, not default lifo, for local polling
505     * Replicated by ForkJoinWorkerThreads
506 dl 1.1 */
507 dl 1.53 private boolean locallyFifo;
508 dl 1.1
509     /**
510 dl 1.53 * Controls whether to add spares to maintain parallelism
511 dl 1.1 */
512 dl 1.53 private boolean maintainsParallelism;
513 dl 1.1
514     /**
515 dl 1.53 * The uncaught exception handler used when any worker
516     * abruptly terminates
517 dl 1.6 */
518 dl 1.53 private Thread.UncaughtExceptionHandler ueh;
519 dl 1.6
520     /**
521 dl 1.53 * Pool number, just for assigning useful names to worker threads
522 dl 1.1 */
523 dl 1.53 private final int poolNumber;
524 dl 1.1
525 dl 1.53 // utilities for updating fields
526 dl 1.1
527     /**
528 dl 1.53 * Adds delta to running count. Used mainly by ForkJoinTask.
529 jsr166 1.17 *
530 dl 1.1 * @param delta the number to add
531     */
532     final void updateRunningCount(int delta) {
533 dl 1.53 int wc;
534     do {} while (!UNSAFE.compareAndSwapInt(this, workerCountsOffset,
535     wc = workerCounts,
536     wc + delta));
537 dl 1.1 }
538    
539     /**
540 dl 1.53 * Write fence for user modifications of pool parameters
541     * (parallelism. etc). Note that it doesn't matter if CAS fails.
542 dl 1.1 */
543 dl 1.53 private void workerCountWriteFence() {
544     int wc;
545     UNSAFE.compareAndSwapInt(this, workerCountsOffset,
546     wc = workerCounts, wc);
547 dl 1.1 }
548    
549     /**
550 dl 1.53 * Read fence for external reads of pool parameters
551     * (parallelism. maxPoolSize, etc).
552     */
553     private void workerCountReadFence() {
554     int ignore = workerCounts;
555     }
556 dl 1.1
557     /**
558 jsr166 1.17 * Tries incrementing active count; fails on contention.
559 dl 1.53 * Called by workers before executing tasks.
560 jsr166 1.17 *
561 jsr166 1.16 * @return true on success
562 dl 1.1 */
563 dl 1.4 final boolean tryIncrementActiveCount() {
564 dl 1.53 int c;
565     return UNSAFE.compareAndSwapInt(this, runStateOffset,
566     c = runState, c + ONE_ACTIVE);
567 dl 1.1 }
568    
569     /**
570 jsr166 1.16 * Tries decrementing active count; fails on contention.
571 dl 1.53 * Called when workers cannot find tasks to run.
572     */
573     final boolean tryDecrementActiveCount() {
574     int c;
575     return UNSAFE.compareAndSwapInt(this, runStateOffset,
576     c = runState, c - ONE_ACTIVE);
577     }
578    
579     /**
580     * Advances to at least the given level. Returns true if not
581     * already in at least the given level.
582     */
583     private boolean advanceRunLevel(int level) {
584     for (;;) {
585     int s = runState;
586     if ((s & level) != 0)
587     return false;
588     if (UNSAFE.compareAndSwapInt(this, runStateOffset, s, s | level))
589     return true;
590     }
591     }
592    
593     // workers array maintenance
594    
595     /**
596     * Records and returns a workers array index for new worker.
597     */
598     private int recordWorker(ForkJoinWorkerThread w) {
599     // Try using slot totalCount-1. If not available, scan and/or resize
600     int k = (workerCounts >>> TOTAL_COUNT_SHIFT) - 1;
601     final ReentrantLock lock = this.workerLock;
602     lock.lock();
603     try {
604     ForkJoinWorkerThread[] ws = workers;
605     int len = ws.length;
606     if (k < 0 || k >= len || ws[k] != null) {
607     for (k = 0; k < len && ws[k] != null; ++k)
608     ;
609     if (k == len)
610     ws = Arrays.copyOf(ws, len << 1);
611     }
612     ws[k] = w;
613     workers = ws; // volatile array write ensures slot visibility
614     } finally {
615     lock.unlock();
616     }
617     return k;
618     }
619    
620     /**
621     * Nulls out record of worker in workers array
622     */
623     private void forgetWorker(ForkJoinWorkerThread w) {
624     int idx = w.poolIndex;
625     // Locking helps method recordWorker avoid unecessary expansion
626     final ReentrantLock lock = this.workerLock;
627     lock.lock();
628     try {
629     ForkJoinWorkerThread[] ws = workers;
630     if (idx >= 0 && idx < ws.length && ws[idx] == w) // verify
631     ws[idx] = null;
632     } finally {
633     lock.unlock();
634     }
635     }
636    
637     // adding and removing workers
638    
639     /**
640     * Tries to create and add new worker. Assumes that worker counts
641     * are already updated to accommodate the worker, so adjusts on
642     * failure.
643     *
644     * @return new worker or null if creation failed
645     */
646     private ForkJoinWorkerThread addWorker() {
647     ForkJoinWorkerThread w = null;
648     try {
649     w = factory.newThread(this);
650     } finally { // Adjust on either null or exceptional factory return
651     if (w == null) {
652     onWorkerCreationFailure();
653     return null;
654     }
655     }
656     w.start(recordWorker(w), locallyFifo, ueh);
657     return w;
658     }
659    
660     /**
661     * Adjusts counts upon failure to create worker
662     */
663     private void onWorkerCreationFailure() {
664     int c;
665     do {} while (!UNSAFE.compareAndSwapInt(this, workerCountsOffset,
666     c = workerCounts,
667     c - (ONE_RUNNING|ONE_TOTAL)));
668     tryTerminate(false); // in case of failure during shutdown
669     }
670    
671     /**
672     * Create enough total workers to establish target parallelism,
673     * giving up if terminating or addWorker fails
674     */
675     private void ensureEnoughTotalWorkers() {
676     int wc;
677     while (runState < TERMINATING &&
678     ((wc = workerCounts) >>> TOTAL_COUNT_SHIFT) < parallelism) {
679     if ((UNSAFE.compareAndSwapInt(this, workerCountsOffset,
680     wc, wc + (ONE_RUNNING|ONE_TOTAL)) &&
681     addWorker() == null))
682     break;
683     }
684     }
685    
686     /**
687     * Final callback from terminating worker. Removes record of
688     * worker from array, and adjusts counts. If pool is shutting
689     * down, tries to complete terminatation, else possibly replaces
690     * the worker.
691     *
692     * @param w the worker
693     */
694     final void workerTerminated(ForkJoinWorkerThread w) {
695     if (w.active) { // force inactive
696     w.active = false;
697     do {} while (!tryDecrementActiveCount());
698     }
699     forgetWorker(w);
700    
701     // decrement total count, and if was running, running count
702     int unit = w.isTrimmed()? ONE_TOTAL : (ONE_RUNNING|ONE_TOTAL);
703     int wc;
704     do {} while (!UNSAFE.compareAndSwapInt(this, workerCountsOffset,
705     wc = workerCounts, wc - unit));
706    
707     accumulateStealCount(w); // collect final count
708     if (!tryTerminate(false))
709     ensureEnoughTotalWorkers();
710     }
711    
712     // Waiting for and signalling events
713    
714     /**
715     * Ensures eventCount on exit is different (mod 2^32) than on
716     * entry. CAS failures are OK -- any change in count suffices.
717     */
718     private void advanceEventCount() {
719     int c;
720     UNSAFE.compareAndSwapInt(this, eventCountOffset, c = eventCount, c+1);
721     }
722    
723     /**
724     * Releases workers blocked on a count not equal to current count.
725     */
726     final void releaseWaiters() {
727     long top;
728     int id;
729     while ((id = (int)((top = eventWaiters) & WAITER_INDEX_MASK)) > 0 &&
730     (int)(top >>> EVENT_COUNT_SHIFT) != eventCount) {
731     ForkJoinWorkerThread[] ws = workers;
732     ForkJoinWorkerThread w;
733     if (ws.length >= id && (w = ws[id - 1]) != null &&
734     UNSAFE.compareAndSwapLong(this, eventWaitersOffset,
735     top, w.nextWaiter))
736     LockSupport.unpark(w);
737     }
738     }
739    
740     /**
741     * Advances eventCount and releases waiters until interference by
742     * other releasing threads is detected.
743     */
744     final void signalWork() {
745     int ec;
746     UNSAFE.compareAndSwapInt(this, eventCountOffset, ec=eventCount, ec+1);
747     outer:for (;;) {
748     long top = eventWaiters;
749     ec = eventCount;
750     for (;;) {
751     ForkJoinWorkerThread[] ws; ForkJoinWorkerThread w;
752     int id = (int)(top & WAITER_INDEX_MASK);
753     if (id <= 0 || (int)(top >>> EVENT_COUNT_SHIFT) == ec)
754     return;
755     if ((ws = workers).length < id || (w = ws[id - 1]) == null ||
756     !UNSAFE.compareAndSwapLong(this, eventWaitersOffset,
757     top, top = w.nextWaiter))
758     continue outer; // possibly stale; reread
759     LockSupport.unpark(w);
760     if (top != eventWaiters) // let someone else take over
761     return;
762     }
763     }
764     }
765    
766     /**
767     * If worker is inactive, blocks until terminating or event count
768     * advances from last value held by worker; in any case helps
769     * release others.
770     *
771     * @param w the calling worker thread
772     */
773     private void eventSync(ForkJoinWorkerThread w) {
774     if (!w.active) {
775     int prev = w.lastEventCount;
776     long nextTop = (((long)prev << EVENT_COUNT_SHIFT) |
777     ((long)(w.poolIndex + 1)));
778     long top;
779     while ((runState < SHUTDOWN || !tryTerminate(false)) &&
780     (((int)(top = eventWaiters) & WAITER_INDEX_MASK) == 0 ||
781     (int)(top >>> EVENT_COUNT_SHIFT) == prev) &&
782     eventCount == prev) {
783     if (UNSAFE.compareAndSwapLong(this, eventWaitersOffset,
784     w.nextWaiter = top, nextTop)) {
785     accumulateStealCount(w); // transfer steals while idle
786     Thread.interrupted(); // clear/ignore interrupt
787     while (eventCount == prev)
788     w.doPark();
789     break;
790     }
791     }
792     w.lastEventCount = eventCount;
793     }
794     releaseWaiters();
795     }
796    
797     /**
798     * Callback from workers invoked upon each top-level action (i.e.,
799     * stealing a task or taking a submission and running
800     * it). Performs one or both of the following:
801     *
802     * * If the worker cannot find work, updates its active status to
803     * inactive and updates activeCount unless there is contention, in
804     * which case it may try again (either in this or a subsequent
805     * call). Additionally, awaits the next task event and/or helps
806     * wake up other releasable waiters.
807 jsr166 1.17 *
808 dl 1.53 * * If there are too many running threads, suspends this worker
809     * (first forcing inactivation if necessary). If it is not
810     * resumed before a keepAlive elapses, the worker may be "trimmed"
811     * -- killed while suspended within suspendAsSpare. Otherwise,
812     * upon resume it rechecks to make sure that it is still needed.
813     *
814     * @param w the worker
815     * @param worked false if the worker scanned for work but didn't
816     * find any (in which case it may block waiting for work).
817     */
818     final void preStep(ForkJoinWorkerThread w, boolean worked) {
819     boolean active = w.active;
820     boolean inactivate = !worked & active;
821     for (;;) {
822     if (inactivate) {
823     int c = runState;
824     if (UNSAFE.compareAndSwapInt(this, runStateOffset,
825     c, c - ONE_ACTIVE))
826     inactivate = active = w.active = false;
827     }
828     int wc = workerCounts;
829     if ((wc & RUNNING_COUNT_MASK) <= parallelism) {
830     if (!worked)
831     eventSync(w);
832     return;
833     }
834     if (!(inactivate |= active) && // must inactivate to suspend
835     UNSAFE.compareAndSwapInt(this, workerCountsOffset,
836     wc, wc - ONE_RUNNING) &&
837     !w.suspendAsSpare()) // false if trimmed
838     return;
839     }
840     }
841    
842     /**
843     * Adjusts counts and creates or resumes compensating threads for
844     * a worker about to block on task joinMe, returning early if
845     * joinMe becomes ready. First tries resuming an existing spare
846     * (which usually also avoids any count adjustment), but must then
847     * decrement running count to determine whether a new thread is
848     * needed. See above for fuller explanation.
849     */
850     final void preJoin(ForkJoinTask<?> joinMe) {
851     boolean dec = false; // true when running count decremented
852     for (;;) {
853     releaseWaiters(); // help other threads progress
854    
855     if (joinMe.status < 0) // surround spare search with done checks
856     return;
857     ForkJoinWorkerThread spare = null;
858     for (ForkJoinWorkerThread w : workers) {
859     if (w != null && w.isSuspended()) {
860     spare = w;
861     break;
862     }
863     }
864     if (joinMe.status < 0)
865     return;
866    
867     if (spare != null && spare.tryUnsuspend()) {
868     if (dec || joinMe.requestSignal() < 0) {
869     int c;
870     do {} while (!UNSAFE.compareAndSwapInt(this,
871     workerCountsOffset,
872     c = workerCounts,
873     c + ONE_RUNNING));
874     } // else no net count change
875     LockSupport.unpark(spare);
876     return;
877     }
878    
879     int wc = workerCounts; // decrement running count
880     if (!dec && (wc & RUNNING_COUNT_MASK) != 0 &&
881     (dec = UNSAFE.compareAndSwapInt(this, workerCountsOffset,
882     wc, wc -= ONE_RUNNING)) &&
883     joinMe.requestSignal() < 0) { // cannot block
884     int c; // back out
885     do {} while (!UNSAFE.compareAndSwapInt(this,
886     workerCountsOffset,
887     c = workerCounts,
888     c + ONE_RUNNING));
889     return;
890     }
891    
892     if (dec) {
893     int tc = wc >>> TOTAL_COUNT_SHIFT;
894     int pc = parallelism;
895     int dc = pc - (wc & RUNNING_COUNT_MASK); // deficit count
896     if ((dc < pc && (dc <= 0 || (dc * dc < (tc - pc) * pc) ||
897     !maintainsParallelism)) ||
898     tc >= maxPoolSize) // cannot add
899     return;
900     if (spare == null &&
901     UNSAFE.compareAndSwapInt(this, workerCountsOffset, wc,
902     wc + (ONE_RUNNING|ONE_TOTAL))) {
903     addWorker();
904     return;
905     }
906     }
907     }
908     }
909    
910     /**
911     * Same idea as preJoin but with too many differing details to
912     * integrate: There are no task-based signal counts, and only one
913     * way to do the actual blocking. So for simplicity it is directly
914     * incorporated into this method.
915     */
916     final void doBlock(ManagedBlocker blocker, boolean maintainPar)
917     throws InterruptedException {
918     maintainPar &= maintainsParallelism; // override
919     boolean dec = false;
920     boolean done = false;
921     for (;;) {
922     releaseWaiters();
923     if (done = blocker.isReleasable())
924     break;
925     ForkJoinWorkerThread spare = null;
926     for (ForkJoinWorkerThread w : workers) {
927     if (w != null && w.isSuspended()) {
928     spare = w;
929     break;
930     }
931     }
932     if (done = blocker.isReleasable())
933     break;
934     if (spare != null && spare.tryUnsuspend()) {
935     if (dec) {
936     int c;
937     do {} while (!UNSAFE.compareAndSwapInt(this,
938     workerCountsOffset,
939     c = workerCounts,
940     c + ONE_RUNNING));
941     }
942     LockSupport.unpark(spare);
943     break;
944     }
945     int wc = workerCounts;
946     if (!dec && (wc & RUNNING_COUNT_MASK) != 0)
947     dec = UNSAFE.compareAndSwapInt(this, workerCountsOffset,
948     wc, wc -= ONE_RUNNING);
949     if (dec) {
950     int tc = wc >>> TOTAL_COUNT_SHIFT;
951     int pc = parallelism;
952     int dc = pc - (wc & RUNNING_COUNT_MASK);
953     if ((dc < pc && (dc <= 0 || (dc * dc < (tc - pc) * pc) ||
954     !maintainPar)) ||
955     tc >= maxPoolSize)
956     break;
957     if (spare == null &&
958     UNSAFE.compareAndSwapInt(this, workerCountsOffset, wc,
959     wc + (ONE_RUNNING|ONE_TOTAL))){
960     addWorker();
961     break;
962     }
963     }
964     }
965    
966     try {
967     if (!done)
968     do {} while (!blocker.isReleasable() && !blocker.block());
969     } finally {
970     if (dec) {
971     int c;
972     do {} while (!UNSAFE.compareAndSwapInt(this,
973     workerCountsOffset,
974     c = workerCounts,
975     c + ONE_RUNNING));
976     }
977     }
978     }
979    
980     /**
981 dl 1.54 * Unless there are not enough other running threads, adjusts
982     * counts for a a worker in performing helpJoin that cannot find
983     * any work, so that this worker can now block.
984     *
985     * @return true if worker may block
986     */
987     final boolean preBlockHelpingJoin(ForkJoinTask<?> joinMe) {
988     while (joinMe.status >= 0) {
989     releaseWaiters(); // help other threads progress
990    
991     // if a spare exists, resume it to maintain parallelism level
992     if ((workerCounts & RUNNING_COUNT_MASK) <= parallelism) {
993     ForkJoinWorkerThread spare = null;
994     for (ForkJoinWorkerThread w : workers) {
995     if (w != null && w.isSuspended()) {
996     spare = w;
997     break;
998     }
999     }
1000     if (joinMe.status < 0)
1001     break;
1002     if (spare != null) {
1003     if (spare.tryUnsuspend()) {
1004     boolean canBlock = true;
1005     if (joinMe.requestSignal() < 0) {
1006     canBlock = false; // already done
1007     int c;
1008     do {} while (!UNSAFE.compareAndSwapInt
1009     (this, workerCountsOffset,
1010     c = workerCounts, c + ONE_RUNNING));
1011     }
1012     LockSupport.unpark(spare);
1013     return canBlock;
1014     }
1015     continue; // recheck -- another spare may exist
1016     }
1017     }
1018    
1019     int wc = workerCounts; // reread to shorten CAS window
1020     int rc = wc & RUNNING_COUNT_MASK;
1021     if (rc <= 2) // keep this and at most one other thread alive
1022     break;
1023    
1024     if (UNSAFE.compareAndSwapInt(this, workerCountsOffset,
1025     wc, wc - ONE_RUNNING)) {
1026     if (joinMe.requestSignal() >= 0)
1027     return true;
1028     int c; // back out
1029     do {} while (!UNSAFE.compareAndSwapInt
1030     (this, workerCountsOffset,
1031     c = workerCounts, c + ONE_RUNNING));
1032     break;
1033     }
1034     }
1035     return false;
1036     }
1037    
1038     /**
1039 dl 1.53 * Possibly initiates and/or completes termination.
1040     *
1041     * @param now if true, unconditionally terminate, else only
1042     * if shutdown and empty queue and no active workers
1043     * @return true if now terminating or terminated
1044 dl 1.1 */
1045 dl 1.53 private boolean tryTerminate(boolean now) {
1046     if (now)
1047     advanceRunLevel(SHUTDOWN); // ensure at least SHUTDOWN
1048     else if (runState < SHUTDOWN ||
1049     !submissionQueue.isEmpty() ||
1050     (runState & ACTIVE_COUNT_MASK) != 0)
1051 dl 1.4 return false;
1052 dl 1.53
1053     if (advanceRunLevel(TERMINATING))
1054     startTerminating();
1055    
1056     // Finish now if all threads terminated; else in some subsequent call
1057     if ((workerCounts >>> TOTAL_COUNT_SHIFT) == 0) {
1058     advanceRunLevel(TERMINATED);
1059     terminationLatch.countDown();
1060     }
1061 dl 1.4 return true;
1062 dl 1.1 }
1063    
1064     /**
1065 dl 1.53 * Actions on transition to TERMINATING
1066     */
1067     private void startTerminating() {
1068     // Clear out and cancel submissions, ignoring exceptions
1069     ForkJoinTask<?> task;
1070     while ((task = submissionQueue.poll()) != null) {
1071     try {
1072     task.cancel(false);
1073     } catch (Throwable ignore) {
1074     }
1075     }
1076     // Propagate run level
1077     for (ForkJoinWorkerThread w : workers) {
1078     if (w != null)
1079     w.shutdown(); // also resumes suspended workers
1080     }
1081     // Ensure no straggling local tasks
1082     for (ForkJoinWorkerThread w : workers) {
1083     if (w != null)
1084     w.cancelTasks();
1085     }
1086     // Wake up idle workers
1087     advanceEventCount();
1088     releaseWaiters();
1089     // Unstick pending joins
1090     for (ForkJoinWorkerThread w : workers) {
1091     if (w != null && !w.isTerminated()) {
1092     try {
1093     w.interrupt();
1094     } catch (SecurityException ignore) {
1095     }
1096     }
1097     }
1098     }
1099    
1100     // misc support for ForkJoinWorkerThread
1101    
1102     /**
1103     * Returns pool number
1104 dl 1.1 */
1105 dl 1.53 final int getPoolNumber() {
1106     return poolNumber;
1107 dl 1.1 }
1108    
1109     /**
1110 dl 1.53 * Accumulates steal count from a worker, clearing
1111     * the worker's value
1112 dl 1.1 */
1113 dl 1.53 final void accumulateStealCount(ForkJoinWorkerThread w) {
1114     int sc = w.stealCount;
1115     if (sc != 0) {
1116     long c;
1117     w.stealCount = 0;
1118     do {} while (!UNSAFE.compareAndSwapLong(this, stealCountOffset,
1119     c = stealCount, c + sc));
1120 dl 1.1 }
1121     }
1122    
1123     /**
1124 dl 1.53 * Returns the approximate (non-atomic) number of idle threads per
1125     * active thread.
1126     */
1127     final int idlePerActive() {
1128     int ac = runState; // no mask -- artifically boosts during shutdown
1129     int pc = parallelism; // use targeted parallelism, not rc
1130     // Use exact results for small values, saturate past 4
1131     return pc <= ac? 0 : pc >>> 1 <= ac? 1 : pc >>> 2 <= ac? 3 : pc >>> 3;
1132     }
1133    
1134     // Public and protected methods
1135 dl 1.1
1136     // Constructors
1137    
1138     /**
1139 dl 1.42 * Creates a {@code ForkJoinPool} with parallelism equal to {@link
1140     * java.lang.Runtime#availableProcessors}, and using the {@linkplain
1141     * #defaultForkJoinWorkerThreadFactory default thread factory}.
1142 jsr166 1.17 *
1143 dl 1.1 * @throws SecurityException if a security manager exists and
1144     * the caller is not permitted to modify threads
1145     * because it does not hold {@link
1146 jsr166 1.17 * java.lang.RuntimePermission}{@code ("modifyThread")}
1147 dl 1.1 */
1148     public ForkJoinPool() {
1149     this(Runtime.getRuntime().availableProcessors(),
1150     defaultForkJoinWorkerThreadFactory);
1151     }
1152    
1153     /**
1154 dl 1.42 * Creates a {@code ForkJoinPool} with the indicated parallelism
1155     * level and using the {@linkplain
1156     * #defaultForkJoinWorkerThreadFactory default thread factory}.
1157 jsr166 1.17 *
1158 dl 1.42 * @param parallelism the parallelism level
1159 dl 1.1 * @throws IllegalArgumentException if parallelism less than or
1160 jsr166 1.47 * equal to zero, or greater than implementation limit
1161 dl 1.1 * @throws SecurityException if a security manager exists and
1162     * the caller is not permitted to modify threads
1163     * because it does not hold {@link
1164 jsr166 1.17 * java.lang.RuntimePermission}{@code ("modifyThread")}
1165 dl 1.1 */
1166     public ForkJoinPool(int parallelism) {
1167     this(parallelism, defaultForkJoinWorkerThreadFactory);
1168     }
1169    
1170     /**
1171 dl 1.42 * Creates a {@code ForkJoinPool} with parallelism equal to {@link
1172     * java.lang.Runtime#availableProcessors}, and using the given
1173     * thread factory.
1174 jsr166 1.17 *
1175 dl 1.1 * @param factory the factory for creating new threads
1176 jsr166 1.48 * @throws NullPointerException if the factory is null
1177 dl 1.1 * @throws SecurityException if a security manager exists and
1178     * the caller is not permitted to modify threads
1179     * because it does not hold {@link
1180 jsr166 1.17 * java.lang.RuntimePermission}{@code ("modifyThread")}
1181 dl 1.1 */
1182     public ForkJoinPool(ForkJoinWorkerThreadFactory factory) {
1183     this(Runtime.getRuntime().availableProcessors(), factory);
1184     }
1185    
1186     /**
1187 jsr166 1.41 * Creates a {@code ForkJoinPool} with the given parallelism and
1188     * thread factory.
1189 dl 1.1 *
1190 dl 1.42 * @param parallelism the parallelism level
1191 dl 1.1 * @param factory the factory for creating new threads
1192     * @throws IllegalArgumentException if parallelism less than or
1193 jsr166 1.47 * equal to zero, or greater than implementation limit
1194 jsr166 1.48 * @throws NullPointerException if the factory is null
1195 dl 1.1 * @throws SecurityException if a security manager exists and
1196     * the caller is not permitted to modify threads
1197     * because it does not hold {@link
1198 jsr166 1.17 * java.lang.RuntimePermission}{@code ("modifyThread")}
1199 dl 1.1 */
1200     public ForkJoinPool(int parallelism, ForkJoinWorkerThreadFactory factory) {
1201 dl 1.53 checkPermission();
1202     if (factory == null)
1203     throw new NullPointerException();
1204 dl 1.1 if (parallelism <= 0 || parallelism > MAX_THREADS)
1205     throw new IllegalArgumentException();
1206 dl 1.53 this.poolNumber = poolNumberGenerator.incrementAndGet();
1207     int arraySize = initialArraySizeFor(parallelism);
1208     this.parallelism = parallelism;
1209 dl 1.1 this.factory = factory;
1210     this.maxPoolSize = MAX_THREADS;
1211     this.maintainsParallelism = true;
1212 dl 1.53 this.workers = new ForkJoinWorkerThread[arraySize];
1213     this.submissionQueue = new LinkedTransferQueue<ForkJoinTask<?>>();
1214 dl 1.1 this.workerLock = new ReentrantLock();
1215 dl 1.53 this.terminationLatch = new CountDownLatch(1);
1216     // Start first worker; remaining workers added upon first submission
1217     workerCounts = ONE_RUNNING | ONE_TOTAL;
1218     addWorker();
1219 dl 1.1 }
1220    
1221     /**
1222 dl 1.53 * Returns initial power of two size for workers array.
1223     * @param pc the initial parallelism level
1224     */
1225     private static int initialArraySizeFor(int pc) {
1226     // See Hackers Delight, sec 3.2. We know MAX_THREADS < (1 >>> 16)
1227     int size = pc < MAX_THREADS ? pc + 1 : MAX_THREADS;
1228     size |= size >>> 1;
1229     size |= size >>> 2;
1230     size |= size >>> 4;
1231     size |= size >>> 8;
1232     return size + 1;
1233 dl 1.1 }
1234    
1235     // Execution methods
1236    
1237     /**
1238     * Common code for execute, invoke and submit
1239     */
1240     private <T> void doSubmit(ForkJoinTask<T> task) {
1241 dl 1.23 if (task == null)
1242     throw new NullPointerException();
1243 dl 1.53 if (runState >= SHUTDOWN)
1244 dl 1.1 throw new RejectedExecutionException();
1245     submissionQueue.offer(task);
1246 dl 1.53 advanceEventCount();
1247     releaseWaiters();
1248     if ((workerCounts >>> TOTAL_COUNT_SHIFT) < parallelism)
1249     ensureEnoughTotalWorkers();
1250 dl 1.1 }
1251    
1252     /**
1253 jsr166 1.17 * Performs the given task, returning its result upon completion.
1254     *
1255 dl 1.1 * @param task the task
1256     * @return the task's result
1257 jsr166 1.48 * @throws NullPointerException if the task is null
1258     * @throws RejectedExecutionException if the task cannot be
1259     * scheduled for execution
1260 dl 1.1 */
1261     public <T> T invoke(ForkJoinTask<T> task) {
1262     doSubmit(task);
1263     return task.join();
1264     }
1265    
1266     /**
1267     * Arranges for (asynchronous) execution of the given task.
1268 jsr166 1.17 *
1269 dl 1.1 * @param task the task
1270 jsr166 1.48 * @throws NullPointerException if the task is null
1271     * @throws RejectedExecutionException if the task cannot be
1272     * scheduled for execution
1273 dl 1.1 */
1274 dl 1.37 public void execute(ForkJoinTask<?> task) {
1275 dl 1.1 doSubmit(task);
1276     }
1277    
1278     // AbstractExecutorService methods
1279    
1280 jsr166 1.48 /**
1281     * @throws NullPointerException if the task is null
1282     * @throws RejectedExecutionException if the task cannot be
1283     * scheduled for execution
1284     */
1285 dl 1.1 public void execute(Runnable task) {
1286 dl 1.23 ForkJoinTask<?> job;
1287 jsr166 1.26 if (task instanceof ForkJoinTask<?>) // avoid re-wrap
1288     job = (ForkJoinTask<?>) task;
1289 dl 1.23 else
1290 dl 1.33 job = ForkJoinTask.adapt(task, null);
1291 dl 1.23 doSubmit(job);
1292 dl 1.1 }
1293    
1294 jsr166 1.48 /**
1295     * @throws NullPointerException if the task is null
1296     * @throws RejectedExecutionException if the task cannot be
1297     * scheduled for execution
1298     */
1299 dl 1.1 public <T> ForkJoinTask<T> submit(Callable<T> task) {
1300 dl 1.33 ForkJoinTask<T> job = ForkJoinTask.adapt(task);
1301 dl 1.1 doSubmit(job);
1302     return job;
1303     }
1304    
1305 jsr166 1.48 /**
1306     * @throws NullPointerException if the task is null
1307     * @throws RejectedExecutionException if the task cannot be
1308     * scheduled for execution
1309     */
1310 dl 1.1 public <T> ForkJoinTask<T> submit(Runnable task, T result) {
1311 dl 1.33 ForkJoinTask<T> job = ForkJoinTask.adapt(task, result);
1312 dl 1.1 doSubmit(job);
1313     return job;
1314     }
1315    
1316 jsr166 1.48 /**
1317     * @throws NullPointerException if the task is null
1318     * @throws RejectedExecutionException if the task cannot be
1319     * scheduled for execution
1320     */
1321 dl 1.1 public ForkJoinTask<?> submit(Runnable task) {
1322 dl 1.23 ForkJoinTask<?> job;
1323 jsr166 1.26 if (task instanceof ForkJoinTask<?>) // avoid re-wrap
1324     job = (ForkJoinTask<?>) task;
1325 dl 1.23 else
1326 dl 1.33 job = ForkJoinTask.adapt(task, null);
1327 dl 1.1 doSubmit(job);
1328     return job;
1329     }
1330    
1331     /**
1332 dl 1.23 * Submits a ForkJoinTask for execution.
1333     *
1334     * @param task the task to submit
1335     * @return the task
1336 jsr166 1.48 * @throws NullPointerException if the task is null
1337 dl 1.23 * @throws RejectedExecutionException if the task cannot be
1338     * scheduled for execution
1339     */
1340     public <T> ForkJoinTask<T> submit(ForkJoinTask<T> task) {
1341     doSubmit(task);
1342     return task;
1343     }
1344    
1345 jsr166 1.48 /**
1346     * @throws NullPointerException {@inheritDoc}
1347     * @throws RejectedExecutionException {@inheritDoc}
1348     */
1349 dl 1.1 public <T> List<Future<T>> invokeAll(Collection<? extends Callable<T>> tasks) {
1350 jsr166 1.20 ArrayList<ForkJoinTask<T>> forkJoinTasks =
1351 dl 1.1 new ArrayList<ForkJoinTask<T>>(tasks.size());
1352 jsr166 1.20 for (Callable<T> task : tasks)
1353 dl 1.33 forkJoinTasks.add(ForkJoinTask.adapt(task));
1354 jsr166 1.20 invoke(new InvokeAll<T>(forkJoinTasks));
1355    
1356     @SuppressWarnings({"unchecked", "rawtypes"})
1357 dl 1.54 List<Future<T>> futures = (List<Future<T>>) (List) forkJoinTasks;
1358 jsr166 1.20 return futures;
1359 dl 1.1 }
1360    
1361     static final class InvokeAll<T> extends RecursiveAction {
1362     final ArrayList<ForkJoinTask<T>> tasks;
1363     InvokeAll(ArrayList<ForkJoinTask<T>> tasks) { this.tasks = tasks; }
1364     public void compute() {
1365 jsr166 1.17 try { invokeAll(tasks); }
1366     catch (Exception ignore) {}
1367 dl 1.1 }
1368 jsr166 1.18 private static final long serialVersionUID = -7914297376763021607L;
1369 dl 1.1 }
1370    
1371     /**
1372 jsr166 1.17 * Returns the factory used for constructing new workers.
1373 dl 1.1 *
1374     * @return the factory used for constructing new workers
1375     */
1376     public ForkJoinWorkerThreadFactory getFactory() {
1377     return factory;
1378     }
1379    
1380     /**
1381 dl 1.2 * Returns the handler for internal worker threads that terminate
1382     * due to unrecoverable errors encountered while executing tasks.
1383 jsr166 1.17 *
1384 jsr166 1.28 * @return the handler, or {@code null} if none
1385 dl 1.2 */
1386     public Thread.UncaughtExceptionHandler getUncaughtExceptionHandler() {
1387 dl 1.53 workerCountReadFence();
1388     return ueh;
1389 dl 1.2 }
1390    
1391     /**
1392     * Sets the handler for internal worker threads that terminate due
1393     * to unrecoverable errors encountered while executing tasks.
1394     * Unless set, the current default or ThreadGroup handler is used
1395     * as handler.
1396     *
1397     * @param h the new handler
1398 jsr166 1.28 * @return the old handler, or {@code null} if none
1399 dl 1.2 * @throws SecurityException if a security manager exists and
1400     * the caller is not permitted to modify threads
1401     * because it does not hold {@link
1402 jsr166 1.17 * java.lang.RuntimePermission}{@code ("modifyThread")}
1403 dl 1.2 */
1404     public Thread.UncaughtExceptionHandler
1405     setUncaughtExceptionHandler(Thread.UncaughtExceptionHandler h) {
1406     checkPermission();
1407 dl 1.53 workerCountReadFence();
1408     Thread.UncaughtExceptionHandler old = ueh;
1409     if (h != old) {
1410 dl 1.2 ueh = h;
1411 dl 1.53 workerCountWriteFence();
1412     for (ForkJoinWorkerThread w : workers) {
1413     if (w != null)
1414     w.setUncaughtExceptionHandler(h);
1415 dl 1.2 }
1416     }
1417     return old;
1418     }
1419    
1420     /**
1421 jsr166 1.16 * Sets the target parallelism level of this pool.
1422 jsr166 1.17 *
1423 dl 1.1 * @param parallelism the target parallelism
1424     * @throws IllegalArgumentException if parallelism less than or
1425 jsr166 1.16 * equal to zero or greater than maximum size bounds
1426 dl 1.1 * @throws SecurityException if a security manager exists and
1427     * the caller is not permitted to modify threads
1428     * because it does not hold {@link
1429 jsr166 1.17 * java.lang.RuntimePermission}{@code ("modifyThread")}
1430 dl 1.1 */
1431     public void setParallelism(int parallelism) {
1432     checkPermission();
1433     if (parallelism <= 0 || parallelism > maxPoolSize)
1434     throw new IllegalArgumentException();
1435 dl 1.53 workerCountReadFence();
1436     int pc = this.parallelism;
1437     if (pc != parallelism) {
1438     this.parallelism = parallelism;
1439     workerCountWriteFence();
1440     // Release spares. If too many, some will die after re-suspend
1441     for (ForkJoinWorkerThread w : workers) {
1442     if (w != null && w.tryUnsuspend()) {
1443     updateRunningCount(1);
1444     LockSupport.unpark(w);
1445 dl 1.50 }
1446 dl 1.1 }
1447 dl 1.53 ensureEnoughTotalWorkers();
1448     advanceEventCount();
1449     releaseWaiters(); // force config recheck by existing workers
1450 dl 1.1 }
1451     }
1452    
1453     /**
1454 dl 1.42 * Returns the targeted parallelism level of this pool.
1455 dl 1.1 *
1456 dl 1.42 * @return the targeted parallelism level of this pool
1457 dl 1.1 */
1458     public int getParallelism() {
1459 dl 1.53 // workerCountReadFence(); // inlined below
1460     int ignore = workerCounts;
1461 dl 1.1 return parallelism;
1462     }
1463    
1464     /**
1465     * Returns the number of worker threads that have started but not
1466     * yet terminated. This result returned by this method may differ
1467 jsr166 1.29 * from {@link #getParallelism} when threads are created to
1468 dl 1.1 * maintain parallelism when others are cooperatively blocked.
1469     *
1470     * @return the number of worker threads
1471     */
1472     public int getPoolSize() {
1473 dl 1.53 return workerCounts >>> TOTAL_COUNT_SHIFT;
1474 dl 1.1 }
1475    
1476     /**
1477     * Returns the maximum number of threads allowed to exist in the
1478 dl 1.45 * pool. Unless set using {@link #setMaximumPoolSize}, the
1479 dl 1.42 * maximum is an implementation-defined value designed only to
1480     * prevent runaway growth.
1481 jsr166 1.17 *
1482 dl 1.1 * @return the maximum
1483     */
1484     public int getMaximumPoolSize() {
1485 dl 1.53 workerCountReadFence();
1486 dl 1.1 return maxPoolSize;
1487     }
1488    
1489     /**
1490     * Sets the maximum number of threads allowed to exist in the
1491 dl 1.45 * pool. The given value should normally be greater than or equal
1492     * to the {@link #getParallelism parallelism} level. Setting this
1493     * value has no effect on current pool size. It controls
1494 dl 1.54 * construction of new threads. The use of this method may cause
1495     * tasks that intrinsically require extra threads for dependent
1496     * computations to indefinitely stall. If you are instead trying
1497 dl 1.55 * to minimize internal thread creation, consider setting {@link
1498 dl 1.54 * #setMaintainsParallelism} as false.
1499 jsr166 1.17 *
1500 jsr166 1.36 * @throws IllegalArgumentException if negative or greater than
1501 jsr166 1.16 * internal implementation limit
1502 dl 1.1 */
1503     public void setMaximumPoolSize(int newMax) {
1504     if (newMax < 0 || newMax > MAX_THREADS)
1505     throw new IllegalArgumentException();
1506     maxPoolSize = newMax;
1507 dl 1.53 workerCountWriteFence();
1508 dl 1.1 }
1509    
1510     /**
1511 jsr166 1.28 * Returns {@code true} if this pool dynamically maintains its
1512     * target parallelism level. If false, new threads are added only
1513     * to avoid possible starvation. This setting is by default true.
1514 jsr166 1.17 *
1515 jsr166 1.28 * @return {@code true} if maintains parallelism
1516 dl 1.1 */
1517     public boolean getMaintainsParallelism() {
1518 dl 1.53 workerCountReadFence();
1519 dl 1.1 return maintainsParallelism;
1520     }
1521    
1522     /**
1523     * Sets whether this pool dynamically maintains its target
1524     * parallelism level. If false, new threads are added only to
1525     * avoid possible starvation.
1526 jsr166 1.17 *
1527 jsr166 1.28 * @param enable {@code true} to maintain parallelism
1528 dl 1.1 */
1529     public void setMaintainsParallelism(boolean enable) {
1530     maintainsParallelism = enable;
1531 dl 1.53 workerCountWriteFence();
1532 dl 1.1 }
1533    
1534     /**
1535 dl 1.6 * Establishes local first-in-first-out scheduling mode for forked
1536     * tasks that are never joined. This mode may be more appropriate
1537     * than default locally stack-based mode in applications in which
1538     * worker threads only process asynchronous tasks. This method is
1539 jsr166 1.29 * designed to be invoked only when the pool is quiescent, and
1540 dl 1.6 * typically only before any tasks are submitted. The effects of
1541 jsr166 1.16 * invocations at other times may be unpredictable.
1542 dl 1.6 *
1543 jsr166 1.29 * @param async if {@code true}, use locally FIFO scheduling
1544 jsr166 1.16 * @return the previous mode
1545 jsr166 1.29 * @see #getAsyncMode
1546 dl 1.6 */
1547     public boolean setAsyncMode(boolean async) {
1548 dl 1.53 workerCountReadFence();
1549 dl 1.6 boolean oldMode = locallyFifo;
1550 dl 1.53 if (oldMode != async) {
1551     locallyFifo = async;
1552     workerCountWriteFence();
1553     for (ForkJoinWorkerThread w : workers) {
1554     if (w != null)
1555     w.setAsyncMode(async);
1556 dl 1.6 }
1557     }
1558     return oldMode;
1559     }
1560    
1561     /**
1562 jsr166 1.28 * Returns {@code true} if this pool uses local first-in-first-out
1563 jsr166 1.16 * scheduling mode for forked tasks that are never joined.
1564 dl 1.6 *
1565 jsr166 1.28 * @return {@code true} if this pool uses async mode
1566 jsr166 1.29 * @see #setAsyncMode
1567 dl 1.6 */
1568     public boolean getAsyncMode() {
1569 dl 1.53 workerCountReadFence();
1570 dl 1.6 return locallyFifo;
1571     }
1572    
1573     /**
1574 dl 1.2 * Returns an estimate of the number of worker threads that are
1575     * not blocked waiting to join tasks or for other managed
1576 dl 1.53 * synchronization. This method may overestimate the
1577     * number of running threads.
1578 dl 1.1 *
1579     * @return the number of worker threads
1580     */
1581     public int getRunningThreadCount() {
1582 dl 1.53 return workerCounts & RUNNING_COUNT_MASK;
1583 dl 1.1 }
1584    
1585     /**
1586 dl 1.2 * Returns an estimate of the number of threads that are currently
1587 dl 1.1 * stealing or executing tasks. This method may overestimate the
1588     * number of active threads.
1589 jsr166 1.17 *
1590 jsr166 1.16 * @return the number of active threads
1591 dl 1.1 */
1592     public int getActiveThreadCount() {
1593 dl 1.53 return runState & ACTIVE_COUNT_MASK;
1594 dl 1.1 }
1595    
1596     /**
1597 jsr166 1.28 * Returns {@code true} if all worker threads are currently idle.
1598     * An idle worker is one that cannot obtain a task to execute
1599     * because none are available to steal from other threads, and
1600     * there are no pending submissions to the pool. This method is
1601     * conservative; it might not return {@code true} immediately upon
1602     * idleness of all threads, but will eventually become true if
1603     * threads remain inactive.
1604 jsr166 1.17 *
1605 jsr166 1.28 * @return {@code true} if all threads are currently idle
1606 dl 1.1 */
1607     public boolean isQuiescent() {
1608 dl 1.53 return (runState & ACTIVE_COUNT_MASK) == 0;
1609 dl 1.1 }
1610    
1611     /**
1612     * Returns an estimate of the total number of tasks stolen from
1613     * one thread's work queue by another. The reported value
1614     * underestimates the actual total number of steals when the pool
1615     * is not quiescent. This value may be useful for monitoring and
1616 jsr166 1.17 * tuning fork/join programs: in general, steal counts should be
1617 dl 1.1 * high enough to keep threads busy, but low enough to avoid
1618     * overhead and contention across threads.
1619 jsr166 1.17 *
1620 jsr166 1.16 * @return the number of steals
1621 dl 1.1 */
1622     public long getStealCount() {
1623 dl 1.53 return stealCount;
1624 dl 1.1 }
1625    
1626     /**
1627 dl 1.2 * Returns an estimate of the total number of tasks currently held
1628     * in queues by worker threads (but not including tasks submitted
1629     * to the pool that have not begun executing). This value is only
1630     * an approximation, obtained by iterating across all threads in
1631     * the pool. This method may be useful for tuning task
1632     * granularities.
1633 jsr166 1.17 *
1634 jsr166 1.16 * @return the number of queued tasks
1635 dl 1.1 */
1636     public long getQueuedTaskCount() {
1637     long count = 0;
1638 dl 1.53 for (ForkJoinWorkerThread w : workers) {
1639     if (w != null)
1640     count += w.getQueueSize();
1641 dl 1.1 }
1642     return count;
1643     }
1644    
1645     /**
1646 jsr166 1.40 * Returns an estimate of the number of tasks submitted to this
1647     * pool that have not yet begun executing. This method takes time
1648 dl 1.1 * proportional to the number of submissions.
1649 jsr166 1.17 *
1650 jsr166 1.16 * @return the number of queued submissions
1651 dl 1.1 */
1652     public int getQueuedSubmissionCount() {
1653     return submissionQueue.size();
1654     }
1655    
1656     /**
1657 jsr166 1.28 * Returns {@code true} if there are any tasks submitted to this
1658     * pool that have not yet begun executing.
1659 jsr166 1.17 *
1660 jsr166 1.16 * @return {@code true} if there are any queued submissions
1661 dl 1.1 */
1662     public boolean hasQueuedSubmissions() {
1663     return !submissionQueue.isEmpty();
1664     }
1665    
1666     /**
1667     * Removes and returns the next unexecuted submission if one is
1668     * available. This method may be useful in extensions to this
1669     * class that re-assign work in systems with multiple pools.
1670 jsr166 1.17 *
1671 jsr166 1.28 * @return the next submission, or {@code null} if none
1672 dl 1.1 */
1673     protected ForkJoinTask<?> pollSubmission() {
1674     return submissionQueue.poll();
1675     }
1676    
1677     /**
1678 dl 1.6 * Removes all available unexecuted submitted and forked tasks
1679     * from scheduling queues and adds them to the given collection,
1680     * without altering their execution status. These may include
1681 jsr166 1.41 * artificially generated or wrapped tasks. This method is
1682     * designed to be invoked only when the pool is known to be
1683 dl 1.6 * quiescent. Invocations at other times may not remove all
1684     * tasks. A failure encountered while attempting to add elements
1685 jsr166 1.16 * to collection {@code c} may result in elements being in
1686 dl 1.6 * neither, either or both collections when the associated
1687     * exception is thrown. The behavior of this operation is
1688     * undefined if the specified collection is modified while the
1689     * operation is in progress.
1690 jsr166 1.17 *
1691 dl 1.6 * @param c the collection to transfer elements into
1692     * @return the number of elements transferred
1693     */
1694 dl 1.30 protected int drainTasksTo(Collection<? super ForkJoinTask<?>> c) {
1695 dl 1.6 int n = submissionQueue.drainTo(c);
1696 dl 1.53 for (ForkJoinWorkerThread w : workers) {
1697     if (w != null)
1698     n += w.drainTasksTo(c);
1699 dl 1.6 }
1700     return n;
1701     }
1702    
1703     /**
1704 dl 1.1 * Returns a string identifying this pool, as well as its state,
1705     * including indications of run state, parallelism level, and
1706     * worker and task counts.
1707     *
1708     * @return a string identifying this pool, as well as its state
1709     */
1710     public String toString() {
1711     long st = getStealCount();
1712     long qt = getQueuedTaskCount();
1713     long qs = getQueuedSubmissionCount();
1714 dl 1.53 int wc = workerCounts;
1715     int tc = wc >>> TOTAL_COUNT_SHIFT;
1716     int rc = wc & RUNNING_COUNT_MASK;
1717     int pc = parallelism;
1718     int rs = runState;
1719     int ac = rs & ACTIVE_COUNT_MASK;
1720 dl 1.1 return super.toString() +
1721 dl 1.53 "[" + runLevelToString(rs) +
1722     ", parallelism = " + pc +
1723     ", size = " + tc +
1724     ", active = " + ac +
1725     ", running = " + rc +
1726 dl 1.1 ", steals = " + st +
1727     ", tasks = " + qt +
1728     ", submissions = " + qs +
1729     "]";
1730     }
1731    
1732 dl 1.53 private static String runLevelToString(int s) {
1733     return ((s & TERMINATED) != 0 ? "Terminated" :
1734     ((s & TERMINATING) != 0 ? "Terminating" :
1735     ((s & SHUTDOWN) != 0 ? "Shutting down" :
1736     "Running")));
1737 dl 1.1 }
1738    
1739     /**
1740     * Initiates an orderly shutdown in which previously submitted
1741     * tasks are executed, but no new tasks will be accepted.
1742     * Invocation has no additional effect if already shut down.
1743     * Tasks that are in the process of being submitted concurrently
1744     * during the course of this method may or may not be rejected.
1745 jsr166 1.17 *
1746 dl 1.1 * @throws SecurityException if a security manager exists and
1747     * the caller is not permitted to modify threads
1748     * because it does not hold {@link
1749 jsr166 1.17 * java.lang.RuntimePermission}{@code ("modifyThread")}
1750 dl 1.1 */
1751     public void shutdown() {
1752     checkPermission();
1753 dl 1.53 advanceRunLevel(SHUTDOWN);
1754     tryTerminate(false);
1755 dl 1.1 }
1756    
1757     /**
1758 dl 1.42 * Attempts to cancel and/or stop all tasks, and reject all
1759     * subsequently submitted tasks. Tasks that are in the process of
1760     * being submitted or executed concurrently during the course of
1761     * this method may or may not be rejected. This method cancels
1762     * both existing and unexecuted tasks, in order to permit
1763     * termination in the presence of task dependencies. So the method
1764     * always returns an empty list (unlike the case for some other
1765     * Executors).
1766 jsr166 1.17 *
1767 dl 1.1 * @return an empty list
1768     * @throws SecurityException if a security manager exists and
1769     * the caller is not permitted to modify threads
1770     * because it does not hold {@link
1771 jsr166 1.17 * java.lang.RuntimePermission}{@code ("modifyThread")}
1772 dl 1.1 */
1773     public List<Runnable> shutdownNow() {
1774     checkPermission();
1775 dl 1.53 tryTerminate(true);
1776 dl 1.1 return Collections.emptyList();
1777     }
1778    
1779     /**
1780 jsr166 1.16 * Returns {@code true} if all tasks have completed following shut down.
1781 dl 1.1 *
1782 jsr166 1.16 * @return {@code true} if all tasks have completed following shut down
1783 dl 1.1 */
1784     public boolean isTerminated() {
1785 dl 1.53 return runState >= TERMINATED;
1786 dl 1.1 }
1787    
1788     /**
1789 jsr166 1.16 * Returns {@code true} if the process of termination has
1790 dl 1.42 * commenced but not yet completed. This method may be useful for
1791     * debugging. A return of {@code true} reported a sufficient
1792     * period after shutdown may indicate that submitted tasks have
1793     * ignored or suppressed interruption, causing this executor not
1794     * to properly terminate.
1795 dl 1.1 *
1796 dl 1.42 * @return {@code true} if terminating but not yet terminated
1797 dl 1.1 */
1798     public boolean isTerminating() {
1799 dl 1.53 return (runState & (TERMINATING|TERMINATED)) == TERMINATING;
1800 dl 1.1 }
1801    
1802     /**
1803 jsr166 1.16 * Returns {@code true} if this pool has been shut down.
1804 dl 1.1 *
1805 jsr166 1.16 * @return {@code true} if this pool has been shut down
1806 dl 1.1 */
1807     public boolean isShutdown() {
1808 dl 1.53 return runState >= SHUTDOWN;
1809 dl 1.42 }
1810    
1811     /**
1812 dl 1.1 * Blocks until all tasks have completed execution after a shutdown
1813     * request, or the timeout occurs, or the current thread is
1814     * interrupted, whichever happens first.
1815     *
1816     * @param timeout the maximum time to wait
1817     * @param unit the time unit of the timeout argument
1818 jsr166 1.16 * @return {@code true} if this executor terminated and
1819     * {@code false} if the timeout elapsed before termination
1820 dl 1.1 * @throws InterruptedException if interrupted while waiting
1821     */
1822     public boolean awaitTermination(long timeout, TimeUnit unit)
1823     throws InterruptedException {
1824 dl 1.53 return terminationLatch.await(timeout, unit);
1825 dl 1.1 }
1826    
1827     /**
1828     * Interface for extending managed parallelism for tasks running
1829 jsr166 1.35 * in {@link ForkJoinPool}s.
1830     *
1831     * <p>A {@code ManagedBlocker} provides two methods.
1832 jsr166 1.28 * Method {@code isReleasable} must return {@code true} if
1833     * blocking is not necessary. Method {@code block} blocks the
1834     * current thread if necessary (perhaps internally invoking
1835 jsr166 1.40 * {@code isReleasable} before actually blocking).
1836 jsr166 1.17 *
1837 dl 1.1 * <p>For example, here is a ManagedBlocker based on a
1838     * ReentrantLock:
1839 jsr166 1.17 * <pre> {@code
1840     * class ManagedLocker implements ManagedBlocker {
1841     * final ReentrantLock lock;
1842     * boolean hasLock = false;
1843     * ManagedLocker(ReentrantLock lock) { this.lock = lock; }
1844     * public boolean block() {
1845     * if (!hasLock)
1846     * lock.lock();
1847     * return true;
1848     * }
1849     * public boolean isReleasable() {
1850     * return hasLock || (hasLock = lock.tryLock());
1851 dl 1.1 * }
1852 jsr166 1.17 * }}</pre>
1853 dl 1.1 */
1854     public static interface ManagedBlocker {
1855     /**
1856     * Possibly blocks the current thread, for example waiting for
1857     * a lock or condition.
1858 jsr166 1.17 *
1859 jsr166 1.28 * @return {@code true} if no additional blocking is necessary
1860     * (i.e., if isReleasable would return true)
1861 dl 1.1 * @throws InterruptedException if interrupted while waiting
1862 jsr166 1.17 * (the method is not required to do so, but is allowed to)
1863 dl 1.1 */
1864     boolean block() throws InterruptedException;
1865    
1866     /**
1867 jsr166 1.28 * Returns {@code true} if blocking is unnecessary.
1868 dl 1.1 */
1869     boolean isReleasable();
1870     }
1871    
1872     /**
1873     * Blocks in accord with the given blocker. If the current thread
1874 jsr166 1.38 * is a {@link ForkJoinWorkerThread}, this method possibly
1875     * arranges for a spare thread to be activated if necessary to
1876     * ensure parallelism while the current thread is blocked.
1877 dl 1.1 *
1878 jsr166 1.38 * <p>If {@code maintainParallelism} is {@code true} and the pool
1879     * supports it ({@link #getMaintainsParallelism}), this method
1880     * attempts to maintain the pool's nominal parallelism. Otherwise
1881     * it activates a thread only if necessary to avoid complete
1882     * starvation. This option may be preferable when blockages use
1883     * timeouts, or are almost always brief.
1884     *
1885     * <p>If the caller is not a {@link ForkJoinTask}, this method is
1886     * behaviorally equivalent to
1887 jsr166 1.17 * <pre> {@code
1888     * while (!blocker.isReleasable())
1889     * if (blocker.block())
1890     * return;
1891     * }</pre>
1892 jsr166 1.38 *
1893     * If the caller is a {@code ForkJoinTask}, then the pool may
1894     * first be expanded to ensure parallelism, and later adjusted.
1895 dl 1.1 *
1896     * @param blocker the blocker
1897 jsr166 1.28 * @param maintainParallelism if {@code true} and supported by
1898     * this pool, attempt to maintain the pool's nominal parallelism;
1899     * otherwise activate a thread only if necessary to avoid
1900     * complete starvation.
1901 jsr166 1.16 * @throws InterruptedException if blocker.block did so
1902 dl 1.1 */
1903     public static void managedBlock(ManagedBlocker blocker,
1904     boolean maintainParallelism)
1905     throws InterruptedException {
1906     Thread t = Thread.currentThread();
1907 dl 1.53 if (t instanceof ForkJoinWorkerThread)
1908     ((ForkJoinWorkerThread) t).pool.
1909     doBlock(blocker, maintainParallelism);
1910     else
1911     awaitBlocker(blocker);
1912 dl 1.1 }
1913    
1914 dl 1.53 /**
1915     * Performs Non-FJ blocking
1916     */
1917 dl 1.1 private static void awaitBlocker(ManagedBlocker blocker)
1918     throws InterruptedException {
1919 jsr166 1.17 do {} while (!blocker.isReleasable() && !blocker.block());
1920 dl 1.1 }
1921    
1922 dl 1.33 // AbstractExecutorService overrides. These rely on undocumented
1923     // fact that ForkJoinTask.adapt returns ForkJoinTasks that also
1924     // implement RunnableFuture.
1925 dl 1.2
1926     protected <T> RunnableFuture<T> newTaskFor(Runnable runnable, T value) {
1927 jsr166 1.34 return (RunnableFuture<T>) ForkJoinTask.adapt(runnable, value);
1928 dl 1.2 }
1929    
1930     protected <T> RunnableFuture<T> newTaskFor(Callable<T> callable) {
1931 jsr166 1.34 return (RunnableFuture<T>) ForkJoinTask.adapt(callable);
1932 dl 1.2 }
1933    
1934 jsr166 1.27 // Unsafe mechanics
1935 dl 1.1
1936 jsr166 1.21 private static final sun.misc.Unsafe UNSAFE = getUnsafe();
1937 dl 1.53 private static final long workerCountsOffset =
1938     objectFieldOffset("workerCounts", ForkJoinPool.class);
1939     private static final long runStateOffset =
1940     objectFieldOffset("runState", ForkJoinPool.class);
1941 jsr166 1.25 private static final long eventCountOffset =
1942 jsr166 1.27 objectFieldOffset("eventCount", ForkJoinPool.class);
1943 dl 1.53 private static final long eventWaitersOffset =
1944     objectFieldOffset("eventWaiters",ForkJoinPool.class);
1945     private static final long stealCountOffset =
1946     objectFieldOffset("stealCount",ForkJoinPool.class);
1947    
1948 jsr166 1.27
1949     private static long objectFieldOffset(String field, Class<?> klazz) {
1950     try {
1951     return UNSAFE.objectFieldOffset(klazz.getDeclaredField(field));
1952     } catch (NoSuchFieldException e) {
1953     // Convert Exception to corresponding Error
1954     NoSuchFieldError error = new NoSuchFieldError(field);
1955     error.initCause(e);
1956     throw error;
1957     }
1958     }
1959    
1960     /**
1961     * Returns a sun.misc.Unsafe. Suitable for use in a 3rd party package.
1962     * Replace with a simple call to Unsafe.getUnsafe when integrating
1963     * into a jdk.
1964     *
1965     * @return a sun.misc.Unsafe
1966     */
1967     private static sun.misc.Unsafe getUnsafe() {
1968     try {
1969     return sun.misc.Unsafe.getUnsafe();
1970     } catch (SecurityException se) {
1971     try {
1972     return java.security.AccessController.doPrivileged
1973     (new java.security
1974     .PrivilegedExceptionAction<sun.misc.Unsafe>() {
1975     public sun.misc.Unsafe run() throws Exception {
1976     java.lang.reflect.Field f = sun.misc
1977     .Unsafe.class.getDeclaredField("theUnsafe");
1978     f.setAccessible(true);
1979     return (sun.misc.Unsafe) f.get(null);
1980     }});
1981     } catch (java.security.PrivilegedActionException e) {
1982     throw new RuntimeException("Could not initialize intrinsics",
1983     e.getCause());
1984     }
1985     }
1986     }
1987 dl 1.1 }