net.snowflake.ingest.utils.ThreadFactoryUtil Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of snowflake-ingest-sdk Show documentation

Snowflake Ingest SDK

There is a newer version: 3.0.1

/*
 * Copyright (c) 2012-2017 Snowflake Computing Inc. All rights reserved.
 */

package net.snowflake.ingest.utils;

import java.util.concurrent.Executors;
import java.util.concurrent.ThreadFactory;
import java.util.concurrent.atomic.AtomicLong;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/** This class provides support functions for writing a thread factory wrapper */
public class ThreadFactoryUtil {
  private static final Logger LOGGER = LoggerFactory.getLogger(ThreadFactoryUtil.class);

  public static ThreadFactory poolThreadFactory(
      final String threadBaseName, final boolean isDaemon) {
    return new ThreadFactory() {
      final AtomicLong count = new AtomicLong(0);

      @Override
      public Thread newThread(Runnable r) {
        final Thread thread = Executors.defaultThreadFactory().newThread(r);
        thread.setName(threadBaseName + "-" + count.incrementAndGet() + "(" + thread.getId() + ")");
        thread.setDaemon(isDaemon);

        final Thread.UncaughtExceptionHandler uncaughtExceptionHandler =
            (t, e) -> LOGGER.error("uncaughtException in thread: " + t, e);

        thread.setUncaughtExceptionHandler(uncaughtExceptionHandler);
        return thread;
      }
    };
  }
}