ThreadPools.java

package core.framework.impl.async;

import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.ThreadFactory;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * @author neo
 */
public final class ThreadPools {
    public static ExecutorService cachedThreadPool(int poolSize, String prefix) {
        ThreadPoolExecutor threadPool = new ThreadPoolExecutor(poolSize, poolSize, 60, TimeUnit.SECONDS, new LinkedBlockingQueue<>(), new ThreadFactoryImpl(prefix));
        threadPool.allowCoreThreadTimeOut(true);
        return threadPool;
    }

    public static ExecutorService fixedThreadPool(int poolSize, String prefix) {
        return Executors.newFixedThreadPool(poolSize, new ThreadFactoryImpl(prefix));
    }

    public static ScheduledExecutorService singleThreadScheduler(String prefix) {
        return Executors.newSingleThreadScheduledExecutor(new ThreadFactoryImpl(prefix));
    }

    static class ThreadFactoryImpl implements ThreadFactory {
        private final AtomicInteger count = new AtomicInteger(1);
        private final String prefix;

        ThreadFactoryImpl(String prefix) {
            this.prefix = prefix;
        }

        @Override
        public Thread newThread(Runnable runnable) {
            return new Thread(runnable, prefix + count.getAndIncrement());
        }
    }
}