View Javadoc
1   /*
2    * Licensed to the Apache Software Foundation (ASF) under one or more
3    * contributor license agreements.  See the NOTICE file distributed with
4    * this work for additional information regarding copyright ownership.
5    * The ASF licenses this file to You under the Apache License, Version 2.0
6    * (the "License"); you may not use this file except in compliance with
7    * the License.  You may obtain a copy of the License at
8    *
9    *      http://www.apache.org/licenses/LICENSE-2.0
10   *
11   * Unless required by applicable law or agreed to in writing, software
12   * distributed under the License is distributed on an "AS IS" BASIS,
13   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14   * See the License for the specific language governing permissions and
15   * limitations under the License.
16   */
17  package org.apache.commons.lang3.concurrent;
18  
19  import java.lang.Thread.UncaughtExceptionHandler;
20  import java.util.Objects;
21  import java.util.concurrent.ExecutorService;
22  import java.util.concurrent.Executors;
23  import java.util.concurrent.ThreadFactory;
24  import java.util.concurrent.atomic.AtomicLong;
25  
26  /**
27   * An implementation of the {@link ThreadFactory} interface that provides some
28   * configuration options for the threads it creates.
29   * <p>
30   * A {@link ThreadFactory} is used for instance by an {@link ExecutorService} to
31   * create the threads it uses for executing tasks. In many cases users do not
32   * have to care about a {@link ThreadFactory} because the default one used by an
33   * {@link ExecutorService} will do. However, if there are special requirements
34   * for the threads, a custom {@link ThreadFactory} has to be created.
35   * </p>
36   * <p>
37   * This class provides some frequently needed configuration options for the
38   * threads it creates. These are the following:
39   * </p>
40   * <ul>
41   * <li>A name pattern for the threads created by this factory can be specified.
42   * This is often useful if an application uses multiple executor services for
43   * different purposes. If the names of the threads used by these services have
44   * meaningful names, log output or exception traces can be much easier to read.
45   * Naming patterns are <em>format strings</em> as used by the {@code
46   * String.format()} method. The string can contain the place holder {@code %d}
47   * which will be replaced by the number of the current thread ({@code
48   * ThreadFactoryImpl} keeps a counter of the threads it has already created).
49   * For instance, the naming pattern {@code "My %d. worker thread"} will result
50   * in thread names like {@code "My 1. worker thread"}, {@code
51   * "My 2. worker thread"} and so on.</li>
52   * <li>A flag whether the threads created by this factory should be daemon
53   * threads. This can impact the exit behavior of the current Java application
54   * because the JVM shuts down if there are only daemon threads running.</li>
55   * <li>The priority of the thread. Here an integer value can be provided. The
56   * {@link Thread} class defines constants for valid ranges of priority
57   * values.</li>
58   * <li>The {@link UncaughtExceptionHandler} for the thread. This handler is
59   * called if an uncaught exception occurs within the thread.</li>
60   * </ul>
61   * <p>
62   * {@link BasicThreadFactory} wraps another thread factory which actually
63   * creates new threads. The configuration options are set on the threads created
64   * by the wrapped thread factory. On construction time the factory to be wrapped
65   * can be specified. If none is provided, a default {@link ThreadFactory} is
66   * used.
67   * </p>
68   * <p>
69   * Instances of {@link BasicThreadFactory} are not created directly, but the
70   * nested {@link Builder} class is used for this purpose. Using the builder only
71   * the configuration options an application is interested in need to be set. The
72   * following example shows how a {@link BasicThreadFactory} is created and
73   * installed in an {@link ExecutorService}:
74   * </p>
75   *
76   * <pre>
77   * // Create a factory that produces daemon threads with a naming pattern and
78   * // a priority
79   * BasicThreadFactory factory = new BasicThreadFactory.Builder()
80   *     .namingPattern(&quot;workerthread-%d&quot;)
81   *     .daemon(true)
82   *     .priority(Thread.MAX_PRIORITY)
83   *     .build();
84   * // Create an executor service for single-threaded execution
85   * ExecutorService exec = Executors.newSingleThreadExecutor(factory);
86   * </pre>
87   *
88   * @since 3.0
89   */
90  public class BasicThreadFactory implements ThreadFactory {
91  
92      /**
93       * A <em>builder</em> class for creating instances of {@code
94       * BasicThreadFactory}.
95       * <p>
96       * Using this builder class instances of {@link BasicThreadFactory} can be
97       * created and initialized. The class provides methods that correspond to
98       * the configuration options supported by {@link BasicThreadFactory}. Method
99       * chaining is supported. Refer to the documentation of {@code
100      * BasicThreadFactory} for a usage example.
101      * </p>
102      */
103     public static class Builder implements org.apache.commons.lang3.builder.Builder<BasicThreadFactory> {
104 
105         /** The wrapped factory. */
106         private ThreadFactory factory;
107 
108         /** The uncaught exception handler. */
109         private Thread.UncaughtExceptionHandler exceptionHandler;
110 
111         /** The naming pattern. */
112         private String namingPattern;
113 
114         /** The priority. */
115         private Integer priority;
116 
117         /** The daemon flag. */
118         private Boolean daemon;
119 
120         /**
121          * Constructs a new instance.
122          */
123         public Builder() {
124             // empty
125         }
126 
127         /**
128          * Creates a new {@link BasicThreadFactory} with all configuration
129          * options that have been specified by calling methods on this builder.
130          * After creating the factory {@link #reset()} is called.
131          *
132          * @return the new {@link BasicThreadFactory}
133          */
134         @Override
135         public BasicThreadFactory build() {
136             final BasicThreadFactory factory = new BasicThreadFactory(this);
137             reset();
138             return factory;
139         }
140 
141         /**
142          * Sets the daemon flag for the new {@link BasicThreadFactory}. If this
143          * flag is set to <b>true</b> the new thread factory will create daemon
144          * threads.
145          *
146          * @param daemon the value of the daemon flag
147          * @return a reference to this {@link Builder}
148          */
149         public Builder daemon(final boolean daemon) {
150             this.daemon = Boolean.valueOf(daemon);
151             return this;
152         }
153 
154         /**
155          * Sets the naming pattern to be used by the new {@code
156          * BasicThreadFactory}.
157          *
158          * @param namingPattern the naming pattern (must not be <b>null</b>)
159          * @return a reference to this {@link Builder}
160          * @throws NullPointerException if the naming pattern is <b>null</b>
161          */
162         public Builder namingPattern(final String namingPattern) {
163             this.namingPattern = Objects.requireNonNull(namingPattern, "pattern");
164             return this;
165         }
166 
167         /**
168          * Sets the priority for the threads created by the new {@code
169          * BasicThreadFactory}.
170          *
171          * @param priority the priority
172          * @return a reference to this {@link Builder}
173          */
174         public Builder priority(final int priority) {
175             this.priority = Integer.valueOf(priority);
176             return this;
177         }
178 
179         /**
180          * Resets this builder. All configuration options are set to default
181          * values. Note: If the {@link #build()} method was called, it is not
182          * necessary to call {@code reset()} explicitly because this is done
183          * automatically.
184          */
185         public void reset() {
186             factory = null;
187             exceptionHandler = null;
188             namingPattern = null;
189             priority = null;
190             daemon = null;
191         }
192 
193         /**
194          * Sets the uncaught exception handler for the threads created by the
195          * new {@link BasicThreadFactory}.
196          *
197          * @param exceptionHandler the {@link UncaughtExceptionHandler} (must not be
198          * <b>null</b>)
199          * @return a reference to this {@link Builder}
200          * @throws NullPointerException if the exception handler is <b>null</b>
201          */
202         public Builder uncaughtExceptionHandler(
203                 final Thread.UncaughtExceptionHandler exceptionHandler) {
204             this.exceptionHandler = Objects.requireNonNull(exceptionHandler, "handler");
205             return this;
206         }
207 
208         /**
209          * Sets the {@link ThreadFactory} to be wrapped by the new {@code
210          * BasicThreadFactory}.
211          *
212          * @param factory the wrapped {@link ThreadFactory} (must not be
213          * <b>null</b>)
214          * @return a reference to this {@link Builder}
215          * @throws NullPointerException if the passed in {@link ThreadFactory}
216          * is <b>null</b>
217          */
218         public Builder wrappedFactory(final ThreadFactory factory) {
219             this.factory = Objects.requireNonNull(factory, "factory");
220             return this;
221         }
222     }
223 
224     /** A counter for the threads created by this factory. */
225     private final AtomicLong threadCounter;
226 
227     /** Stores the wrapped factory. */
228     private final ThreadFactory wrappedFactory;
229 
230     /** Stores the uncaught exception handler. */
231     private final Thread.UncaughtExceptionHandler uncaughtExceptionHandler;
232 
233     /** Stores the naming pattern for newly created threads. */
234     private final String namingPattern;
235 
236     /** Stores the priority. */
237     private final Integer priority;
238 
239     /** Stores the daemon status flag. */
240     private final Boolean daemon;
241 
242     /**
243      * Creates a new instance of {@link ThreadFactory} and configures it
244      * from the specified {@link Builder} object.
245      *
246      * @param builder the {@link Builder} object
247      */
248     private BasicThreadFactory(final Builder builder) {
249         wrappedFactory = builder.factory != null ? builder.factory : Executors.defaultThreadFactory();
250         namingPattern = builder.namingPattern;
251         priority = builder.priority;
252         daemon = builder.daemon;
253         uncaughtExceptionHandler = builder.exceptionHandler;
254         threadCounter = new AtomicLong();
255     }
256 
257     /**
258      * Gets the daemon flag. This flag determines whether newly created
259      * threads should be daemon threads. If <b>true</b>, this factory object
260      * calls {@code setDaemon(true)} on the newly created threads. Result can be
261      * <b>null</b> if no daemon flag was provided at creation time.
262      *
263      * @return the daemon flag
264      */
265     public final Boolean getDaemonFlag() {
266         return daemon;
267     }
268 
269     /**
270      * Gets the naming pattern for naming newly created threads. Result can
271      * be <b>null</b> if no naming pattern was provided.
272      *
273      * @return the naming pattern
274      */
275     public final String getNamingPattern() {
276         return namingPattern;
277     }
278 
279     /**
280      * Gets the priority of the threads created by this factory. Result can
281      * be <b>null</b> if no priority was specified.
282      *
283      * @return the priority for newly created threads
284      */
285     public final Integer getPriority() {
286         return priority;
287     }
288 
289     /**
290      * Gets the number of threads this factory has already created. This
291      * class maintains an internal counter that is incremented each time the
292      * {@link #newThread(Runnable)} method is invoked.
293      *
294      * @return the number of threads created by this factory
295      */
296     public long getThreadCount() {
297         return threadCounter.get();
298     }
299 
300     /**
301      * Gets the {@link UncaughtExceptionHandler} for the threads created by
302      * this factory. Result can be <b>null</b> if no handler was provided.
303      *
304      * @return the {@link UncaughtExceptionHandler}
305      */
306     public final Thread.UncaughtExceptionHandler getUncaughtExceptionHandler() {
307         return uncaughtExceptionHandler;
308     }
309 
310     /**
311      * Gets the wrapped {@link ThreadFactory}. This factory is used for
312      * actually creating threads. This method never returns <b>null</b>. If no
313      * {@link ThreadFactory} was passed when this object was created, a default
314      * thread factory is returned.
315      *
316      * @return the wrapped {@link ThreadFactory}
317      */
318     public final ThreadFactory getWrappedFactory() {
319         return wrappedFactory;
320     }
321 
322     /**
323      * Initializes the specified thread. This method is called by
324      * {@link #newThread(Runnable)} after a new thread has been obtained from
325      * the wrapped thread factory. It initializes the thread according to the
326      * options set for this factory.
327      *
328      * @param thread the thread to be initialized
329      */
330     private void initializeThread(final Thread thread) {
331         if (getNamingPattern() != null) {
332             final Long count = Long.valueOf(threadCounter.incrementAndGet());
333             thread.setName(String.format(getNamingPattern(), count));
334         }
335         if (getUncaughtExceptionHandler() != null) {
336             thread.setUncaughtExceptionHandler(getUncaughtExceptionHandler());
337         }
338         if (getPriority() != null) {
339             thread.setPriority(getPriority().intValue());
340         }
341         if (getDaemonFlag() != null) {
342             thread.setDaemon(getDaemonFlag().booleanValue());
343         }
344     }
345 
346     /**
347      * Creates a new thread. This implementation delegates to the wrapped
348      * factory for creating the thread. Then, on the newly created thread the
349      * corresponding configuration options are set.
350      *
351      * @param runnable the {@link Runnable} to be executed by the new thread
352      * @return the newly created thread
353      */
354     @Override
355     public Thread newThread(final Runnable runnable) {
356         final Thread thread = getWrappedFactory().newThread(runnable);
357         initializeThread(thread);
358         return thread;
359     }
360 }