com.wavefront.sdk.common.NamedThreadFactory Maven / Gradle / Ivy

Go to download
package com.wavefront.sdk.common;


import com.wavefront.sdk.common.annotation.NonNull;

import java.util.concurrent.ThreadFactory;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * Simple thread factory to be used with Executors.newScheduledThreadPool that allows
 * assigning name prefixes to all pooled threads to simplify thread identification during
 * troubleshooting.
 *
 * Created by [email protected] on 3/16/17.
 */
public class NamedThreadFactory implements ThreadFactory {
  private final String threadNamePrefix;
  private final AtomicInteger counter = new AtomicInteger();
  private final AtomicBoolean isDaemon = new AtomicBoolean(false);

  public NamedThreadFactory(@NonNull String threadNamePrefix) {
    this.threadNamePrefix = threadNamePrefix;
  }

  public NamedThreadFactory setDaemon(boolean isDaemon) {
    this.isDaemon.set(isDaemon);
    return this;
  }

  @Override
  public Thread newThread(@NonNull Runnable r) {
    Thread toReturn = new Thread(r);
    if (this.isDaemon.get()) {
      toReturn.setDaemon(true);
    }
    toReturn.setName(threadNamePrefix + "-" + counter.getAndIncrement());
    return toReturn;
  }
}