在 springboot 项目中执行定时任务 - 代码实现

最编程 2024-10-06 17:55:28

...

待执行定时任务实体类

import lombok.Data;

@Data
public class CampaignRecordEntity {
    private Long id;

    private String pushTime;

    private Long campaignId;

    private String type;

    private String proxy;
}

接收执行时间参数并持久化待执行任务

// todo 下面这里要记录本次任务的参数内容，并为本次任务请求生成一个任务ID 这里假设任务ID 001
String pushTime = "2024-09-29 18:00:00";// 参数传入的任务执行时刻
String taskId = "001";
Timer timer = new Timer(uuid);
TimerTask task = new TimerTask() {
    @Override
    public void run() {
        try {
        	// todo 下面可以写你待执行任务需要执行的具体内容
        	// 执行完要删除掉任务记录，以免每次重启都查询到任务记录并生成不必要的任务线程
            int deleteRecordNum = browserPushMsgMapper.deleteCampaignRecordByCampaignId(taskId );
            if (deleteRecordNum > 0) {
                log.info("delete campaign record success.");
            }
        } catch (Exception ex) {
            ex.printStackTrace();
        }
    }
};
timer.schedule(task, DateUtils.StringToDate(pushTime, "yyyy-MM-dd HH:mm:ss"));
uuidTimerTaskContainer.put(uuid, task);
CampaignRecordEntity entity = new CampaignRecordEntity();
entity.setCampaignId(e.getId());
entity.setPushTime(pushTime);
int num = mapper.insertCampaignRecord(entity, "Remo");
if (num > 0) {
    log.info("insert campaign record success.");
}

服务启动时待执行任务查询生成

由于服务重启会导致重启前生成的任务线程被杀死，因此重启后需要重新生成任务线程：

@Slf4j
@Component
@Order(100)
public class SendMsgStartTask implements CommandLineRunner {
    @Resource
    private Mapper mapper;
    
    @Override
    public void run(String... args) throws Exception {
        List<CampaignRecordEntity> records = browserPushMsgMapper.queryCampaignRecords();
        if (records != null && records.size() > 0) {
            for (CampaignRecordEntity entity : records) {
                try {
                    Timer timer = new Timer(UuidUtils.getUuidString(20));
                    String taskId = entity.getId();
                    TimerTask task = new TimerTask() {
                        @Override
                        public void run() {
                           	// todo 执行到执行时刻需要执行的内容
                           	// 执行完后要清楚待执行任务
                            int deleteRecordNum = mapper.deleteCampaignRecordByCampaignId(taskId);
                            if (deleteRecordNum > 0) {
                                log.info("delete campaign record success.");
                            }
                        }
                    };
                    timer.schedule(task, DateUtils.StringToDate(DateUtils.millisecondToDate(recordVo.getTime(), "yyyy-MM-dd HH:mm:ss"), "yyyy-MM-dd HH:mm:ss"));
                } catch (Exception ex) {
                    ex.printStackTrace();
                }
            }
        }
    }
}

上一篇：排序算法 - 合并排序、计数排序

下一篇： css 简单网页布局 - 浮动 (a)