数据库脚本采用定时任务线程池ThreadPoolTaskScheduler来实现定时任务。动态定时任务就是可以配置的，而不是写死在代码中。所以我们要将其写入到数据库中，然后暴露接口就可以进行配置比如创建、启动、结束任务。
DROP TABLE IF EXISTS `tb_task`;
CREATE TABLE `tb_task` (
`task_id` varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL COMMENT '定时任务id',
`task_name` varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT NULL COMMENT '定时任务名称',
`task_desc` varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT NULL COMMENT '定时任务描述',
`task_exp` varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT NULL COMMENT '定时任务Cron表达式',
`task_status` int(0) NULL DEFAULT NULL COMMENT '定时任务状态，0停用 1启用',
`task_class` varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT NULL COMMENT '定时任务的Runnable任务类完整路径',
`update_time` datetime(0) NULL DEFAULT NULL COMMENT '更新时间',
`create_time` datetime(0) NULL DEFAULT NULL COMMENT '创建时间',
PRIMARY KEY (`task_id`) USING BTREE
) ENGINE = InnoDB CHARACTER SET = utf8 COLLATE = utf8_general_ci COMMENT = '动态定时任务表' ROW_FORMAT = Compact;
INSERT INTO `tb_task` VALUES ('1', '测试打印helloworld', '这是一个描述', '0/10 * * * * ? ', 0, 'com.example.demo.Task.HelloTask', '2022-05-16 09:59:03', NULL);
INSERT INTO `tb_task` VALUES ('2', '这是一个含参数的task', '这是一个描述', '0/10 * * * * ? ', 0, 'com.example.demo.Task.parameterTask', '2022-05-16 10:41:10', NULL);
实体类
package com.example.demo.entity;
import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableName;
import com.baomidou.mybatisplus.annotation.IdType;
import java.time.LocalDateTime;
import com.baomidou.mybatisplus.annotation.TableId;
import java.io.Serializable;
import java.util.Date;
import lombok.Data;
import lombok.EqualsAndHashCode;
import lombok.experimental.Accessors;
@Data
@EqualsAndHashCode(callSuper = false)
@Accessors(chain = true)
@TableName("tb_task")
public class TbTask implements Serializable {
private static final long serialVersionUID = 1L;
@TableId(value = "task_id", type = IdType.NONE)
private String taskId;
@TableField("task_name")
private String taskName;
@TableField("task_desc")
private String taskDesc;
@TableField("task_exp")
private String taskExp;
@TableField("task_status")
private Integer taskStatus;
@TableField("task_class")
private String taskClass;
@TableField("update_time")
private Date updateTime;
@TableField("create_time")
private Date createTime;
}
定时任务实现类
package com.example.demo.component;
import com.example.demo.entity.TbTask;
import com.example.demo.mapper.TbTaskMapper;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler;
import org.springframework.scheduling.support.CronTrigger;
import org.springframework.stereotype.Component;
import java.util.Date;
import java.util.List;
import java.util.Objects;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ScheduledFuture;
@Slf4j
@Component
public class TaskScheduler {
//数据库的任务
public static ConcurrentHashMap tasks = new ConcurrentHashMap<>(10);
//正在运行的任务
public static ConcurrentHashMap runTasks = new ConcurrentHashMap<>(10);
//线程池任务调度
private final ThreadPoolTaskScheduler threadPoolTaskScheduler = new ThreadPoolTaskScheduler();
@Autowired
private TbTaskMapper tbTaskMapper;
@Autowired
public TaskScheduler() {
this.threadPoolTaskScheduler.setPoolSize(10);
this.threadPoolTaskScheduler.setThreadNamePrefix("task-thread-");
this.threadPoolTaskScheduler.setWaitForTasksToCompleteOnShutdown(true);
this.threadPoolTaskScheduler.initialize();
}
public void getAllTbTask() {
//查询所有，并put到tasks
TaskScheduler.tasks.clear();
List list = tbTaskMapper.selectList(null);
list.forEach((task) -> TaskScheduler.tasks.put(task.getTaskId(), task));
}
public void start(String taskId) {
try {
//如果为空，重新获取
if (TaskScheduler.tasks.size() <= 0) {
this.getAllTbTask();
}
TbTask tbTask = TaskScheduler.tasks.get(taskId);
if (tbTask == null) {
tbTask = tbTaskMapper.selectById(taskId);
}
if (tbTask == null)
throw new RuntimeException("task不存在");
//获取并实例化Runnable任务类
Class clazz = Class.forName(tbTask.getTaskClass());
Runnable runnable = (Runnable) clazz.newInstance();
//Cron表达式
CronTrigger cron = new CronTrigger(tbTask.getTaskExp());
//执行，并put到runTasks
TaskScheduler.runTasks.put(taskId, Objects.requireNonNull(this.threadPoolTaskScheduler.schedule(runnable, cron)));
this.updateTaskStatus(taskId, 1);
log.info("{}，任务启动！", taskId);
} catch (ClassNotFoundException | InstantiationException | IllegalAccessException e) {
log.error("{}，任务启动失败...", taskId);
e.printStackTrace();
}
}
public void start(String taskId,Runnable runnable) {
try {
//如果为空，重新获取
if (TaskScheduler.tasks.size() <= 0) {
this.getAllTbTask();
}
TbTask tbTask = TaskScheduler.tasks.get(taskId);
if (tbTask == null) {
tbTask = tbTaskMapper.selectById(taskId);
}
if (tbTask == null)
throw new RuntimeException("task不存在");
//Cron表达式
CronTrigger cron = new CronTrigger(tbTask.getTaskExp());
//执行，并put到runTasks
TaskScheduler.runTasks.put(taskId, Objects.requireNonNull(this.threadPoolTaskScheduler.schedule(runnable, cron)));
this.updateTaskStatus(taskId, 1);
log.info("{}，任务启动！", taskId);
} catch (RuntimeException e) {
log.error("{}，任务启动失败...", taskId);
e.printStackTrace();
}
}
public void stop(String taskId) {
// 停止任务
TaskScheduler.runTasks.get(taskId).cancel(true);
TaskScheduler.runTasks.remove(taskId);
this.updateTaskStatus(taskId, 0);
log.info("{}，任务停止...", taskId);
}
public void updateTaskStatus(String taskId, int status) {
TbTask task = tbTaskMapper.selectById(taskId);
task.setTaskStatus(status);
task.setUpdateTime(new Date());
tbTaskMapper.updateById(task);
}
}
controller以及postman调用
由于只是demo，有些未完善。
package com.example.demo.controller;
import com.example.demo.Task.parameterTask;
import com.example.demo.component.TaskScheduler;
import com.example.demo.component.classGenerate;
import com.example.demo.entity.TbTask;
import com.example.demo.entity.VO.ResultVo;
import com.example.demo.service.ITbTaskService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import java.util.HashMap;
@RestController
@RequestMapping("/demo/task")
public class TaskController {
@Autowired
TaskScheduler taskScheduler;
@Autowired
ITbTaskService tbTaskService;
@PostMapping
public ResultVo addTask(@RequestBody TbTask tbTask){
tbTaskService.save(tbTask);
String taskId = tbTask.getTaskId();
taskScheduler.start(taskId);
return ResultVo.SUCCESS("添加并启动成功").data("taskId",taskId);
}
@PostMapping("/{name}")
public ResultVo addTask2(@RequestBody TbTask tbTask,@PathVariable("name") String name){
tbTaskService.save(tbTask);
String taskId = tbTask.getTaskId();
parameterTask parameterTask = new parameterTask();
parameterTask.setName(name);
taskScheduler.start(taskId,parameterTask);
return ResultVo.SUCCESS("添加并启动成功").data("taskId",taskId);
}
@GetMapping("/start/{taskId}")
public ResultVo start(@PathVariable("taskId") String taskId){
taskScheduler.start(taskId);
return ResultVo.SUCCESS("启动成功").data("taskId",taskId);
}
@GetMapping("/stop/{taskId}")
public ResultVo stop(@PathVariable("taskId") String taskId){
taskScheduler.stop(taskId);
return ResultVo.SUCCESS("停止任务").data("taskId",taskId);
}
@PostMapping("/generate")
public ResultVo generateJava(@RequestBody HashMap map) throws Exception {
String path = map.get("path");
String javaCode = map.get("javaCode");
classGenerate.createStudentByFile(path,javaCode);
return ResultVo.SUCCESS().message("代码生成成功");
}
}
下面是postman测试
package com.example.demo.component;
import org.springframework.stereotype.Component;
import javax.tools.JavaCompiler;
import javax.tools.JavaFileObject;
import javax.tools.StandardJavaFileManager;
import javax.tools.ToolProvider;
import java.io.File;
import java.io.FileWriter;
import java.net.URL;
import java.net.URLClassLoader;
import java.util.ArrayList;
import java.util.List;
public class classGenerate {
public static void createStudentByFile(String path,String JavaCode) throws Exception{
// String studentString = "package org.myself.pojo;public class Test{private String studentId;public String getStudentId(){return this.studentId;}public void setStudentId(String studentId){this.studentId = studentId;}}";
String fileName = System.getProperty("user.dir") + "/src/main/java"+path;
File file = new File(fileName);
FileWriter fileWriter = new FileWriter(file);
fileWriter.write(JavaCode);
fileWriter.flush();
fileWriter.close();
JavaCompiler compiler = ToolProvider.getSystemJavaCompiler();
StandardJavaFileManager manager = compiler.getStandardFileManager(null,null,null);
Iterable javaFileObjects = manager.getJavaFileObjects(fileName);
String dest = System.getProperty("user.dir") + "/target/classes";
//options就是指定编译输入目录，与我们命令行写javac -d C://是一样的
List options = new ArrayList();
options.add("-d");
options.add(dest);
JavaCompiler.CompilationTask task = compiler.getTask(null,manager,null,options,null,javaFileObjects);
task.call();
manager.close();
URL[] urls = new URL[]{new URL("file:/" + System.getProperty("user.dir") + "/target/classes")};
//加载class时要告诉你的classloader去哪个位置加载class文件
ClassLoader classLoader = new URLClassLoader(urls);
// Object student = classLoader.loadClass("com.example.demo.Task.HelloTask2").newInstance();
}
}
测试
@PostMapping("/generate")
public ResultVo generateJava(@RequestBody HashMap map) throws Exception {
String path = map.get("path");
String javaCode = map.get("javaCode");
classGenerate.createStudentByFile(path,javaCode);
return ResultVo.SUCCESS().message("代码生成成功");
}
postman
由于是传入代码，所以遇到换行要使用n，""转义使用",否则会报错，在postman测试中json参数需要在一行中。
注意：原先我上述代码中使用了@Slf4j,但是这样的话class编译的时候会报错，没办法找到。可能是lombok的插件的原因吧。此外编译后，会有一些警告，但是不影响编译完成。
警告报错如下：
E:projecthello_worldsrcmainjavacomexampledemoTaskHelloTask2.java:4: 警告: Can't initialize javac processor due to (most likely) a class loader problem: java.lang.NoClassDefFoundError: com/sun/tools/javac/processing/JavacProcessingEnvironment
public class HelloTask2 implements Runnable{
^
at lombok.javac.apt.LombokProcessor.init(LombokProcessor.java:85)
at lombok.core.AnnotationProcessor$JavacDescriptor.want(AnnotationProcessor.java:87)
at lombok.core.AnnotationProcessor.init(AnnotationProcessor.java:140)
at lombok.launch.AnnotationProcessorHider$AnnotationProcessor.init(AnnotationProcessor.java:69)
at com.sun.tools.javac.processing.JavacProcessingEnvironment$ProcessorState.(JavacProcessingEnvironment.java:500)
at com.sun.tools.javac.processing.JavacProcessingEnvironment$DiscoveredProcessors$ProcessorStateIterator.next(JavacProcessingEnvironment.java:597)
at com.sun.tools.javac.processing.JavacProcessingEnvironment.discoverAndRunProcs(JavacProcessingEnvironment.java:690)
at com.sun.tools.javac.processing.JavacProcessingEnvironment.access$1800(JavacProcessingEnvironment.java:91)
at com.sun.tools.javac.processing.JavacProcessingEnvironment$Round.run(JavacProcessingEnvironment.java:1035)
at com.sun.tools.javac.processing.JavacProcessingEnvironment.doProcessing(JavacProcessingEnvironment.java:1176)
at com.sun.tools.javac.main.JavaCompiler.processAnnotations(JavaCompiler.java:1170)
at com.sun.tools.javac.main.JavaCompiler.compile(JavaCompiler.java:856)
at com.sun.tools.javac.main.Main.compile(Main.java:523)
at com.sun.tools.javac.api.JavacTaskImpl.doCall(JavacTaskImpl.java:129)
at com.sun.tools.javac.api.JavacTaskImpl.call(JavacTaskImpl.java:138)
at com.example.demo.component.classGenerate.createStudentByFile(classGenerate.java:47)
at com.example.demo.controller.TaskController.generateJava(TaskController.java:79)
at sun.reflect.NativeMethodAccessorImpl.invoke0(Native Method)
at sun.reflect.NativeMethodAccessorImpl.invoke(NativeMethodAccessorImpl.java:62)
at sun.reflect.DelegatingMethodAccessorImpl.invoke(DelegatingMethodAccessorImpl.java:43)
at java.lang.reflect.Method.invoke(Method.java:498)
at org.springframework.web.method.support.InvocableHandlerMethod.doInvoke(InvocableHandlerMethod.java:209)
at org.springframework.web.method.support.InvocableHandlerMethod.invokeForRequest(InvocableHandlerMethod.java:136)
at org.springframework.web.servlet.mvc.method.annotation.ServletInvocableHandlerMethod.invokeAndHandle(ServletInvocableHandlerMethod.java:102)
at org.springframework.web.servlet.mvc.method.annotation.RequestMappingHandlerAdapter.invokeHandlerMethod(RequestMappingHandlerAdapter.java:877)
at org.springframework.web.servlet.mvc.method.annotation.RequestMappingHandlerAdapter.handleInternal(RequestMappingHandlerAdapter.java:783)
at org.springframework.web.servlet.mvc.method.AbstractHandlerMethodAdapter.handle(AbstractHandlerMethodAdapter.java:87)
at org.springframework.web.servlet.DispatcherServlet.doDispatch(DispatcherServlet.java:991)
at org.springframework.web.servlet.DispatcherServlet.doService(DispatcherServlet.java:925)
at org.springframework.web.servlet.FrameworkServlet.processRequest(FrameworkServlet.java:974)
at org.springframework.web.servlet.FrameworkServlet.doPost(FrameworkServlet.java:877)
at javax.servlet.http.HttpServlet.service(HttpServlet.java:661)
at org.springframework.web.servlet.FrameworkServlet.service(FrameworkServlet.java:851)
at javax.servlet.http.HttpServlet.service(HttpServlet.java:742)
at org.apache.catalina.core.ApplicationFilterChain.internalDoFilter(ApplicationFilterChain.java:231)
at org.apache.catalina.core.ApplicationFilterChain.doFilter(ApplicationFilterChain.java:166)
at org.apache.tomcat.websocket.server.WsFilter.doFilter(WsFilter.java:52)
at org.apache.catalina.core.ApplicationFilterChain.internalDoFilter(ApplicationFilterChain.java:193)
at org.apache.catalina.core.ApplicationFilterChain.doFilter(ApplicationFilterChain.java:166)
at org.springframework.web.filter.RequestContextFilter.doFilterInternal(RequestContextFilter.java:99)
at org.springframework.web.filter.OncePerRequestFilter.doFilter(OncePerRequestFilter.java:107)
at org.apache.catalina.core.ApplicationFilterChain.internalDoFilter(ApplicationFilterChain.java:193)
at org.apache.catalina.core.ApplicationFilterChain.doFilter(ApplicationFilterChain.java:166)
at org.springframework.web.filter.HttpPutFormContentFilter.doFilterInternal(HttpPutFormContentFilter.java:109)
at org.springframework.web.filter.OncePerRequestFilter.doFilter(OncePerRequestFilter.java:107)
at org.apache.catalina.core.ApplicationFilterChain.internalDoFilter(ApplicationFilterChain.java:193)
at org.apache.catalina.core.ApplicationFilterChain.doFilter(ApplicationFilterChain.java:166)
at org.springframework.web.filter.HiddenHttpMethodFilter.doFilterInternal(HiddenHttpMethodFilter.java:81)
at org.springframework.web.filter.OncePerRequestFilter.doFilter(OncePerRequestFilter.java:107)
at org.apache.catalina.core.ApplicationFilterChain.internalDoFilter(ApplicationFilterChain.java:193)
at org.apache.catalina.core.ApplicationFilterChain.doFilter(ApplicationFilterChain.java:166)
at org.springframework.web.filter.CharacterEncodingFilter.doFilterInternal(CharacterEncodingFilter.java:200)
at org.springframework.web.filter.OncePerRequestFilter.doFilter(OncePerRequestFilter.java:107)
at org.apache.catalina.core.ApplicationFilterChain.internalDoFilter(ApplicationFilterChain.java:193)
at org.apache.catalina.core.ApplicationFilterChain.doFilter(ApplicationFilterChain.java:166)
at org.apache.catalina.core.StandardWrapperValve.invoke(StandardWrapperValve.java:198)
at org.apache.catalina.core.StandardContextValve.invoke(StandardContextValve.java:96)
at org.apache.catalina.authenticator.AuthenticatorBase.invoke(AuthenticatorBase.java:496)
at org.apache.catalina.core.StandardHostValve.invoke(StandardHostValve.java:140)
at org.apache.catalina.valves.ErrorReportValve.invoke(ErrorReportValve.java:81)
at org.apache.catalina.core.StandardEngineValve.invoke(StandardEngineValve.java:87)
at org.apache.catalina.connector.CoyoteAdapter.service(CoyoteAdapter.java:342)
at org.apache.coyote.http11.Http11Processor.service(Http11Processor.java:803)
at org.apache.coyote.AbstractProcessorLight.process(AbstractProcessorLight.java:66)
at org.apache.coyote.AbstractProtocol$ConnectionHandler.process(AbstractProtocol.java:790)
at org.apache.tomcat.util.net.NioEndpoint$SocketProcessor.doRun(NioEndpoint.java:1459)
at org.apache.tomcat.util.net.SocketProcessorBase.run(SocketProcessorBase.java:49)
at java.util.concurrent.ThreadPoolExecutor.runWorker(ThreadPoolExecutor.java:1149)
at java.util.concurrent.ThreadPoolExecutor$Worker.run(ThreadPoolExecutor.java:624)
at org.apache.tomcat.util.threads.TaskThread$WrappingRunnable.run(TaskThread.java:61)
at java.lang.Thread.run(Thread.java:748)
Caused by: java.lang.ClassNotFoundException: com.sun.tools.javac.processing.JavacProcessingEnvironment
at java.lang.ClassLoader.findClass(ClassLoader.java:523)
at java.lang.ClassLoader.loadClass(ClassLoader.java:418)
at lombok.launch.ShadowClassLoader.loadClass(ShadowClassLoader.java:422)
at java.lang.ClassLoader.loadClass(ClassLoader.java:351)
... 71 more
1 个警告
但是这个警告不影响！！
解决办法：
在SDKs中添加tools.jar ，路径一般在jdk/lib/ 下面
由于我们前面创建的是一个Runnable对象，所以我们可以对他进行创建定时任务
仓库链接
