package com.usi.util;
import java.util.Collection;
import java.util.List;
import java.util.concurrent.*;
/**
* @description:线程池工具类
* @author:
* @create: 2021-06-24 15:37
**/
public class ThreadPoolUtils {
public static final int CORE_POOL_SIZE = Runtime.getRuntime().availableProcessors() + 1;
public static final int MAX_POOL_SIZE = Runtime.getRuntime().availableProcessors() * 2;
public static final int KEEP_ALIVE_TIME = 1000;
public static final int BLOCK_QUEUE_SIZE = 1000;
private volatile static ThreadPoolExecutor threadPool ;
private volatile static ScheduledExecutorService scheduleExec;
/**
* 获取线程池对象
*
* @return
*/
public static ThreadPoolExecutor getThreadPoolExecutor() {
if (threadPool != null) {
return threadPool;
} else {
synchronized (ThreadPoolUtils.class) {
if (threadPool == null) {
threadPool = new ThreadPoolExecutor(CORE_POOL_SIZE, MAX_POOL_SIZE, KEEP_ALIVE_TIME, TimeUnit.SECONDS,
new LinkedBlockingQueue<>(BLOCK_QUEUE_SIZE), new ThreadPoolExecutor.CallerRunsPolicy());
}
}
return threadPool;
}
}
/**
* 获取调度线程池对象
*
* @return
*/
public static ScheduledExecutorService getScheduledExecutorService() {
if (scheduleExec != null) {
return scheduleExec;
} else {
synchronized (ThreadPoolUtils.class) {
if (scheduleExec == null) {
scheduleExec =scheduleExec = Executors.newScheduledThreadPool(CORE_POOL_SIZE);
}
}
return scheduleExec;
}
}
/**
* 在未来某个时间执行给定的命令
* 该命令可能在新的线程、已入池的线程或者正调用的线程中执行,这由 Executor 实现决定。
*
* @param command 命令
*/
public static void execute(final Runnable command) {
getThreadPoolExecutor().execute(command);
}
/**
* 在未来某个时间执行给定的命令链表
* 该命令可能在新的线程、已入池的线程或者正调用的线程中执行,这由 Executor 实现决定。
*
* @param commands 命令链表
*/
public static void execute(final List commands) {
for (Runnable command : commands) {
getThreadPoolExecutor().execute(command);
}
}
/**
* 待以前提交的任务执行完毕后关闭线程池
* 启动一次顺序关闭,执行以前提交的任务,但不接受新任务。
* 如果已经关闭,则调用没有作用。
*/
public static void shutDown() {
getThreadPoolExecutor().shutdown();
}
/**
* 试图停止所有正在执行的活动任务
* 试图停止所有正在执行的活动任务,暂停处理正在等待的任务,并返回等待执行的任务列表。
* 无法保证能够停止正在处理的活动执行任务,但是会尽力尝试。
*
* @return 等待执行的任务的列表
*/
public static List shutDownNow() {
return getThreadPoolExecutor().shutdownNow();
}
/**
* 判断线程池是否已关闭
*
* @return {@code true}: 是
{@code false}: 否
*/
public static boolean isShutDown() {
return getThreadPoolExecutor().isShutdown();
}
/**
* 关闭线程池后判断所有任务是否都已完成
* 注意,除非首先调用 shutdown 或 shutdownNow,否则 isTerminated 永不为 true。
*
* @return {@code true}: 是
{@code false}: 否
*/
public static boolean isTerminated() {
return getThreadPoolExecutor().isTerminated();
}
/**
* 请求关闭、发生超时或者当前线程中断
* 无论哪一个首先发生之后,都将导致阻塞,直到所有任务完成执行。
*
* @param timeout 最长等待时间
* @param unit 时间单位
* @return {@code true}: 请求成功
{@code false}: 请求超时
* @throws InterruptedException 终端异常
*/
public static boolean awaitTermination(final long timeout, final TimeUnit unit) throws InterruptedException {
return getThreadPoolExecutor().awaitTermination(timeout, unit);
}
/**
* 提交一个Callable任务用于执行
* 如果想立即阻塞任务的等待,则可以使用{@code result = threadPool.submit(aCallable).get();}形式的构造。
*
* @param task 任务
* @param 泛型
* @return 表示任务等待完成的Future, 该Future的{@code get}方法在成功完成时将会返回该任务的结果。
*/
public static Future submit(final Callable task) {
return getThreadPoolExecutor().submit(task);
}
/**
* 提交一个Runnable任务用于执行
*
* @param task 任务
* @param result 返回的结果
* @param 泛型
* @return 表示任务等待完成的Future, 该Future的{@code get}方法在成功完成时将会返回该任务的结果。
*/
public static Future submit(final Runnable task, final T result) {
return getThreadPoolExecutor().submit(task, result);
}
/**
* 提交一个Runnable任务用于执行
*
* @param task 任务
* @return 表示任务等待完成的Future, 该Future的{@code get}方法在成功完成时将会返回null结果。
*/
public static Future> submit(final Runnable task) {
return getThreadPoolExecutor().submit(task);
}
/**
* 执行给定的任务
* 当所有任务完成时,返回保持任务状态和结果的Future列表。
* 返回列表的所有元素的{@link Future#isDone}为{@code true}。
* 注意,可以正常地或通过抛出异常来终止已完成任务。
* 如果正在进行此操作时修改了给定的 collection,则此方法的结果是不确定的。
*
* @param tasks 任务集合
* @param 泛型
* @return 表示任务的 Future 列表,列表顺序与给定任务列表的迭代器所生成的顺序相同,每个任务都已完成。
* @throws InterruptedException 如果等待时发生中断,在这种情况下取消尚未完成的任务。
*/
public static List> invokeAll(final Collection extends Callable> tasks) throws InterruptedException {
return getThreadPoolExecutor().invokeAll(tasks);
}
/**
* 执行给定的任务
* 当所有任务完成或超时期满时(无论哪个首先发生),返回保持任务状态和结果的Future列表。
* 返回列表的所有元素的{@link Future#isDone}为{@code true}。
* 一旦返回后,即取消尚未完成的任务。
* 注意,可以正常地或通过抛出异常来终止已完成任务。
* 如果此操作正在进行时修改了给定的 collection,则此方法的结果是不确定的。
*
* @param tasks 任务集合
* @param timeout 最长等待时间
* @param unit 时间单位
* @param 泛型
* @return 表示任务的 Future 列表,列表顺序与给定任务列表的迭代器所生成的顺序相同。如果操作未超时,则已完成所有任务。如果确实超时了,则某些任务尚未完成。
* @throws InterruptedException 如果等待时发生中断,在这种情况下取消尚未完成的任务
*/
public static List> invokeAll(final Collection extends Callable> tasks, final long timeout, final TimeUnit unit) throws
InterruptedException {
return getThreadPoolExecutor().invokeAll(tasks, timeout, unit);
}
/**
* 执行给定的任务
* 如果某个任务已成功完成(也就是未抛出异常),则返回其结果。
* 一旦正常或异常返回后,则取消尚未完成的任务。
* 如果此操作正在进行时修改了给定的collection,则此方法的结果是不确定的。
*
* @param tasks 任务集合
* @param 泛型
* @return 某个任务返回的结果
* @throws InterruptedException 如果等待时发生中断
* @throws ExecutionException 如果没有任务成功完成
*/
public static T invokeAny(final Collection extends Callable> tasks) throws InterruptedException, ExecutionException {
return getThreadPoolExecutor().invokeAny(tasks);
}
/**
* 执行给定的任务
* 如果在给定的超时期满前某个任务已成功完成(也就是未抛出异常),则返回其结果。
* 一旦正常或异常返回后,则取消尚未完成的任务。
* 如果此操作正在进行时修改了给定的collection,则此方法的结果是不确定的。
*
* @param tasks 任务集合
* @param timeout 最长等待时间
* @param unit 时间单位
* @param 泛型
* @return 某个任务返回的结果
* @throws InterruptedException 如果等待时发生中断
* @throws ExecutionException 如果没有任务成功完成
* @throws TimeoutException 如果在所有任务成功完成之前给定的超时期满
*/
public static T invokeAny(final Collection extends Callable> tasks, final long timeout, final TimeUnit unit) throws
InterruptedException, ExecutionException, TimeoutException {
return getThreadPoolExecutor().invokeAny(tasks, timeout, unit);
}
/**
* 延迟执行Runnable命令
*
* @param command 命令
* @param delay 延迟时间
* @param unit 单位
* @return 表示挂起任务完成的ScheduledFuture,并且其{@code get()}方法在完成后将返回{@code null}
*/
public static ScheduledFuture> schedule(final Runnable command, final long delay, final TimeUnit unit) {
return getScheduledExecutorService().schedule(command, delay, unit);
}
/**
* 延迟执行Callable命令
*
* @param callable 命令
* @param delay 延迟时间
* @param unit 时间单位
* @param 泛型
* @return 可用于提取结果或取消的ScheduledFuture
*/
public static ScheduledFuture schedule(final Callable callable, final long delay, final TimeUnit unit) {
return getScheduledExecutorService().schedule(callable, delay, unit);
}
/**
* 延迟并循环执行命令
*
* @param command 命令
* @param initialDelay 首次执行的延迟时间
* @param period 连续执行之间的周期
* @param unit 时间单位
* @return 表示挂起任务完成的ScheduledFuture,并且其{@code get()}方法在取消后将抛出异常
*/
public static ScheduledFuture> scheduleWithFixedRate(final Runnable command, final long initialDelay,
final long period, final TimeUnit unit) {
return getScheduledExecutorService().scheduleAtFixedRate(command, initialDelay, period, unit);
}
/**
* 延迟并以固定休息时间循环执行命令
*
* @param command 命令
* @param initialDelay 首次执行的延迟时间
* @param delay 每一次执行终止和下一次执行开始之间的延迟
* @param unit 时间单位
* @return 表示挂起任务完成的ScheduledFuture,并且其{@code get()}方法在取消后将抛出异常
*/
public static ScheduledFuture> scheduleWithFixedDelay(final Runnable command, final long initialDelay,
final long delay, final TimeUnit unit) {
return getScheduledExecutorService().scheduleWithFixedDelay(command, initialDelay, delay, unit);
}
}