zhouxy108/hutool
1
0
Fork 0
mirror of https://gitee.com/chinabugotech/hutool.git synced 2025-07-21 15:09:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

backuo

Browse Source
This commit is contained in:
Looly
2022-03-15 00:25:27 +08:00
parent f6d19d19ea
commit a2c5115831
4 changed files with 1 additions and 277 deletions
Download Patch File Download Diff File Expand all files Collapse all files

149
hutool-extra/src/main/java/cn/hutool/extra/spring/SpringCronUtil.java
View File

@@ -1,149 +0,0 @@
package cn.hutool.extra.spring;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.map.MapUtil;
import cn.hutool.core.util.IdUtil;
import org.springframework.scheduling.TaskScheduler;
import org.springframework.scheduling.support.CronTrigger;
import org.springframework.stereotype.Component;
import javax.annotation.Resource;
import java.io.Serializable;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ScheduledFuture;
/**
* Spring 动态定时任务封装
* <ol>
* <li>创建定时任务</li>
* <li>修改定时任务</li>
* <li>取消定时任务</li>
* <li>高级操作</li>
* </ol>
* 参考：<a href="https://docs.spring.io/spring-framework/docs/current/reference/html/integration.html#scheduling">Spring doc</a>
*
* @author JC
* @date 03/13
*/
@Component
public class SpringCronUtil {
/**
* 任务调度器
*/
private static TaskScheduler taskScheduler;
/**
* ID 与 Future 绑定
*/
private static final Map<Serializable, ScheduledFuture<?>> TASK_FUTURE = MapUtil.newConcurrentHashMap();
/**
* ID 与 Runnable 绑定
*/
private static final Map<Serializable, Runnable> TASK_RUNNABLE = MapUtil.newConcurrentHashMap();
/**
* 加入定时任务
*
* @param task 任务
* @param expression 定时任务执行时间的cron表达式
* @return 定时任务ID
*/
public static String schedule(Runnable task, String expression) {
String id = IdUtil.fastUUID();
return schedule(id, task, expression);
}
/**
* 加入定时任务
*
* @param id 定时任务ID
* @param expression 定时任务执行时间的cron表达式
* @param task 任务
* @return 定时任务ID
*/
public static String schedule(Serializable id, Runnable task, String expression) {
ScheduledFuture<?> schedule = taskScheduler.schedule(task, new CronTrigger(expression));
TASK_FUTURE.put(id, schedule);
TASK_RUNNABLE.put(id, task);
return id.toString();
}
/**
* 修改定时任务
*
* @param id 定时任务ID
* @param expression 定时任务执行时间的cron表达式
* @return 是否修改成功，{@code false}表示未找到对应ID的任务
*/
public static boolean update(Serializable id, String expression) {
if (!TASK_FUTURE.containsKey(id)) {
return false;
}
ScheduledFuture<?> future = TASK_FUTURE.get(id);
if (future == null) {
return false;
}
future.cancel(true);
schedule(id, TASK_RUNNABLE.get(id), expression);
return true;
}
/**
* 移除任务
*
* @param schedulerId 任务ID
* @return 是否移除成功，{@code false}表示未找到对应ID的任务
*/
public static boolean cancel(Serializable schedulerId) {
if (!TASK_FUTURE.containsKey(schedulerId)) {
return false;
}
ScheduledFuture<?> future = TASK_FUTURE.get(schedulerId);
boolean cancel = future.cancel(false);
if (cancel) {
TASK_FUTURE.remove(schedulerId);
TASK_RUNNABLE.remove(schedulerId);
}
return cancel;
}
@Resource
public void setTaskScheduler(TaskScheduler taskScheduler) {
SpringCronUtil.taskScheduler = taskScheduler;
}
/**
* @return 获得Scheduler对象
*/
public static TaskScheduler getScheduler() {
return taskScheduler;
}
/**
* 获得当前运行的所有任务
*
* @return 所有任务
*/
public static List<Serializable> getAllTask() {
if (CollUtil.isNotEmpty(TASK_FUTURE.keySet())) {
return new ArrayList<>(TASK_FUTURE.keySet());
}
return new ArrayList<>();
}
/**
* 取消所有的任务
*/
public static void destroy() {
for (ScheduledFuture<?> future : TASK_FUTURE.values()) {
if (future != null) {
future.cancel(true);
}
}
TASK_FUTURE.clear();
TASK_RUNNABLE.clear();
}
}

31
hutool-extra/src/main/java/cn/hutool/extra/spring/config/SpringCronConfig.java
View File

@@ -1,31 +0,0 @@
package cn.hutool.extra.spring.config;
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.scheduling.TaskScheduler;
import org.springframework.scheduling.annotation.EnableScheduling;
import org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler;
/**
* 可自行配置任务线程池, 修改默认参数
*
* @author JC
* @date 03/13
*/
@Configuration
@EnableScheduling
public class SpringCronConfig {
@Bean
@ConditionalOnMissingBean(value = TaskScheduler.class)
public TaskScheduler taskScheduler() {
ThreadPoolTaskScheduler scheduler = new ThreadPoolTaskScheduler();
// 任务线程池初始化
scheduler.setThreadNamePrefix("TaskScheduler-");
scheduler.setPoolSize(Runtime.getRuntime().availableProcessors() / 3 + 1);
// 保证能立刻丢弃运行中的任务
scheduler.setRemoveOnCancelPolicy(true);
return scheduler;
}
}

4
hutool-extra/src/main/resources/META-INF/spring.factories
View File

@@ -1,5 +1,3 @@
# Auto Configure
org.springframework.boot.autoconfigure.EnableAutoConfiguration=\
cn.hutool.extra.spring.SpringUtil,\
cn.hutool.extra.spring.config.SpringCronConfig,\
cn.hutool.extra.spring.SpringCronUtil
cn.hutool.extra.spring.SpringUtil