扩展Spring Scheduler的执行逻辑
Spring Scheduler因为是单线程执行，所以容易出现部分任务耗时太长，虽然可以使用多线程的方式来提升效率，但是也担心定时任务占用太多的资源，所以还是尽可能地优化定时任务的耗时，那我们首先就可以对耗时进行监控
Spring Scheduler使用ScheduledAnnotationBeanPostProcessor#createRunnable来创建定时任务的执行Runnable（实际类型是ScheduledMethodRunnable），那么我们就可以对该ScheduledMethodRunnable进行扩展，增加任务耗时，设置traceId，方便进行日志分析等
扩展ScheduledMethodRunnable
- 设置traceId
- 监控耗时
@Slf4j
public class TenmaoScheduledMethodRunnable extends ScheduledMethodRunnable {
public TenmaoScheduledMethodRunnable(Object target, Method method) {
super(target, method);
}
public TenmaoScheduledMethodRunnable(Object target, String methodName) throws NoSuchMethodException {
super(target, methodName);
}
@Override
public void run() {
//设置traceId，方便进行日志跟踪
MDC.put("traceId", UUID.randomUUID().toString().replaceAll("-", ""));
log.info("{}: start to run", getMethod());
long startMillis = System.currentTimeMillis();
try {
super.run();
} finally {
//监控耗时
long endMillis = System.currentTimeMillis();
log.info("{}: cost {} millis", getMethod(), endMillis - startMillis);
MDC.remove("traceId");
}
}
}扩展ScheduledAnnotationBeanPostProcessor
让ScheduledAnnotationBeanPostProcessor使用TenmaoScheduledMethodRunnable作为定时任务真正的执行逻辑
@Component
public class TenmaoScheduledAnnotationBeanPostProcessor extends ScheduledAnnotationBeanPostProcessor {
@Override
protected Runnable createRunnable(Object target, Method method) {
//这两行代码从ScheduledAnnotationBeanPostProcessor#createRunnable复制过来
Assert.isTrue(method.getParameterCount() == 0, "Only no-arg methods may be annotated with @Scheduled");
Method invocableMethod = AopUtils.selectInvocableMethod(method, target.getClass());
//把ScheduledMethodRunnable替换为TenmaoScheduledMethodRunnable
return new TenmaoScheduledMethodRunnable(target, invocableMethod);
}
}注意
- 如果没有在
TenmaoScheduledAnnotationBeanPostProcessor增加@Component注解，那么@EnableScheduling一定要去掉
思考
- 是不是也可以使用Aop的方式实现这一点呢？ 貌似有一点麻烦，因为很多定时任务是private，并不会被代理
参考
本文参与 腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。
如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读
目录
相关产品与服务
Elasticsearch Service
腾讯云 Elasticsearch Service（ES）是云端全托管海量数据检索分析服务，拥有高性能自研内核，集成X-Pack。ES 支持通过自治索引、存算分离、集群巡检等特性轻松管理集群，也支持免运维、自动弹性、按需使用的 Serverless 模式。使用 ES 您可以高效构建信息检索、日志分析、运维监控等服务，它独特的向量检索还可助您构建基于语义、图像的AI深度应用。
腾讯云开发者
