浅谈springboot项目中定时任务如何优雅退出

@SpringBootApplication
/**指定mapper对应包的路径*/
@MapperScan("com.youlanw.kz.dao")
/**开启计划任务*/
@EnableScheduling
/**开启异常重试机制*/
@EnableRetry
public class YlkzTaskApplication {

  public static ConfigurableApplicationContext context;
  
  public static void main(String[] args) {
    context = SpringApplication.run(YlkzTaskApplication.class, args);
    /**
     * 捕捉命令实现优雅退出
     */
    MySignalHandler.install("TERM");  //捕捉kill命令
    MySignalHandler.install("INT");   //捕捉ctrl+c命令
  }
}

import org.slf4j.LoggerFactory;
import sun.misc.Signal;
import sun.misc.SignalHandler;
/**
 * @description: 定时任务控制类(实现优雅退出)
 * @method:
 * @author: mamengmeng
 * @date: 10:51 2018/8/13
 */
public class MySignalHandler implements SignalHandler {

  private final static org.slf4j.Logger logger = LoggerFactory.getLogger(MySignalHandler.class);

  private SignalHandler oldHandler;
  /**
   * 定时任务总开关-状态：true：打开 false：关闭
   */
  public static boolean base_flag = true;

  @Override
  public void handle(Signal signal) {
    signalAction(signal);
  }

  public static SignalHandler install(String signalName) {
    Signal diagSignal = new Signal(signalName);
    MySignalHandler instance = new MySignalHandler();
    instance.oldHandler = Signal.handle(diagSignal, instance);
    return instance;
  }

  public void signalAction(Signal signal) {
    try {
      //关闭总开关
      this.base_flag = false;
      logger.info("\n执行优雅退出操作\n等待运行中任务执行完毕…………");
      Thread.sleep(3000);
      StringBuffer stringBuffer = new StringBuffer("a");
      //此处为相关的业务代码，只要还有一个定时任务在执行，那么就等待线程任务执行完毕。
      while (BaseApplyTask.apply_flag || BaseResumeTask.resume_flag || CorpDemandTask.demand_flag || RecommendResumeTask.resume_flag || BaseCodeTask.code_flag || RecommendoneTask.resume_flag ||ResumeByZcbTask.zpbresume_flag) {
        //等待线程任务执行完毕
        stringBuffer.append("");
      }
      //获取到的上下文对象关闭相应的程序。
      YlkzTaskApplication.context.close();
      logger.info("\n================\n程序已安全退出！\n================");
      oldHandler.handle(signal);
    } catch (Exception e) {
      logger.error("handle|Signal handler" + "failed, reason "
          + e.getMessage());
      e.printStackTrace();
    }
  }
}

@Component
public class BaseCodeTask {
  private final static Logger logger = LoggerFactory.getLogger(BaseCodeTask.class);

  @Autowired
  private ResumeService resumeService;

  public static boolean code_flag = true;      //简历任务执行状态 true:执行中 false：执行完毕
  private static final Integer LIMIT = 500;
  private final static long time = 60 * 1000;    //一分钟
  /**
   * @param
   * @description: 同步简历信息(定时任务)
   * 任务执行间隔时间：6秒
   * 待同步数据为空，则5分钟后执行下一次
   * @method: sendResume
   * @author: zhengmingjie
   * @date: 16:17 2018/8/3
   * @return: void
   */
  @Scheduled(initialDelay = 1000, fixedDelay = time / 10)
  @Async
  public void sendResume() throws Exception {
    List<Resume> list = null;
    try {
      //总开关状态：true：打开 false：关闭
      if (!MySignalHandler.base_flag)
        return;
      this.code_flag = true;
      logger.info("\n======定时任务：初始化基本数据======\n开始执行\n");
      //以下是业务代码。相关的定时任务批处理
      resumeService.initializationMap();
      resumeService.setCodeDictionary();
      resumeService.setCityInfo();
      resumeService.setCodePostInfo();
      logger.info("\n======定时任务：初始化基本数据======\n结束\n");
    } catch (Exception e) {
      e.printStackTrace();
    } finally {
      this.code_flag = false;
    }
  }
}

package com.fiberhome.ms.cus.cashform.ui;

import java.util.Date;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.Trigger;
import org.springframework.scheduling.TriggerContext;
import org.springframework.scheduling.annotation.SchedulingConfigurer;
import org.springframework.scheduling.config.ScheduledTaskRegistrar;
import org.springframework.scheduling.support.CronTrigger;
import org.springframework.stereotype.Component;
@Component
public class DynamicScheduledTask implements SchedulingConfigurer {
@Autowired
private ScheduleExport scheduleExport;

// private static String DEFAULT_CRON = "0/10 * * * * ?";
private String cron = "";

public String getCron() {
return cron;
}

public void setCron(String cron) {
this.cron = cron;
}

@Override
public void configureTasks(ScheduledTaskRegistrar taskRegistrar) {
// TODO Auto-generated method stub
taskRegistrar.addTriggerTask(new Runnable() {

@Override
public void run() {
// TODO Auto-generated method stub
try {
scheduleExport.scheduleTaskExport();//异步定时生成文件
System.out.println("Msg:定时生成文件成功");
} catch (Exception e) {
// TODO: handle exception
e.printStackTrace();
System.out.println("Error:定时生成文件错误");
}
}
}, new Trigger() {
@Override
public Date nextExecutionTime(TriggerContext triggerContext) {
// TODO Auto-generated method stub
if ("".equals(cron)|| cron == null)
return null;
CronTrigger trigger = new CronTrigger(cron);// 定时任务触发，可修改定时任务的执行周期
Date nextExecDate = trigger.nextExecutionTime(triggerContext);
return nextExecDate;
}
});
System.out.println("can?");
}
}

package com.fiberhome.ms.cus.cashform.ui.util;

import java.util.Date;
import java.util.concurrent.ScheduledFuture;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.Trigger;
import org.springframework.scheduling.TriggerContext;
import org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler;
import org.springframework.scheduling.support.CronTrigger;
import org.springframework.stereotype.Component;

import com.fiberhome.ms.cus.cashform.ui.ScheduleExport;

@Component
public class DynamicScheduleTaskSecond {
@Autowired
private ThreadPoolTaskScheduler threadPoolTaskScheduler;
@Autowired
private ScheduleExport scheduleExport;
private ScheduledFuture<?> future;

private String cron = "";

public String getCron() {
return cron;
}

public void setCron(String cron) {
this.cron = cron;
stopCron();
future = threadPoolTaskScheduler.schedule(new Runnable() {

@Override
public void run() {
// TODO Auto-generated method stub
try {
scheduleExport.scheduleTaskExport();// 异步定时生成文件
System.out.println("Msg:定时生成文件成功");
} catch (Exception e) {
// TODO: handle exception
e.printStackTrace();
System.out.println("Error:定时生成文件错误");
}
}
}, new Trigger() {
@Override
public Date nextExecutionTime(TriggerContext triggerContext) {
// TODO Auto-generated method stub
if ("".equals(cron) || cron == null)
return null;
CronTrigger trigger = new CronTrigger(cron);// 定时任务触发，可修改定时任务的执行周期
Date nextExecDate = trigger.nextExecutionTime(triggerContext);
return nextExecDate;
}
});
}

public void stopCron() {
if (future != null) {
future.cancel(true);//取消任务调度
}
}
}