1 /*
2  * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License").
5  * You may not use this file except in compliance with the License.
6  * A copy of the License is located at
7  *
8  *  http://aws.amazon.com/apache2.0
9  *
10  * or in the "license" file accompanying this file. This file is distributed
11  * on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
12  * express or implied. See the License for the specific language governing
13  * permissions and limitations under the License.
14  */
15
16 package software.amazon.awssdk.utils;
17
18 import java.util.concurrent.ThreadFactory;
19 import java.util.concurrent.atomic.AtomicInteger;
20 import software.amazon.awssdk.annotations.SdkProtectedApi;
21
22 /**
23  * An decorator for {@link ThreadFactory} that allows naming threads based on a format.
24  */
25 @SdkProtectedApi
26 public class NamedThreadFactory implements ThreadFactory {
27     private final ThreadFactory delegate;
28     private final String namePrefix;
29     private final AtomicInteger threadCount = new AtomicInteger(0);
30
31     public NamedThreadFactory(ThreadFactory delegate, String namePrefix) {
32         this.delegate = Validate.notNull(delegate, "delegate must not be null");
33         this.namePrefix = Validate.notBlank(namePrefix, "namePrefix must not be blank");
34     }
35
36     @Override
37     public Thread newThread(Runnable runnable) {
38         Thread thread = delegate.newThread(runnable);
39         thread.setName(namePrefix + "-" + threadCount.getAndIncrement());
40         return thread;
41     }
42 }
43