springboot的定时任务防止重复执行，数据库加锁实现

今天有个新需求

需求就是：我们有个定时任务，开发环境机器有2台，如果发布到linux上，那就每个任务就会执行两次。那如果发到生产，那就更多机器了，执行更多次了，那么 我们就要加锁去搞定

想法

想法就是去拿锁，拿到锁就执行定时任务，拿不到就当没事发生。

执行

    @Scheduled(cron = "0 0/5 * * * ? ")
    public void scheduleFixedRateTask() {
        String lockKey = "ecm_retry_transfer_to_dm_task";
        int updatedRows = taskService.getLock(lockKey);
        log.info("updateRows:{}", updatedRows);
        if (updatedRows > 0) {
            try {
                //do something
            } catch (Exception e) {
                e.printStackTrace();
            } finally {
                taskService.unlock(lockKey);
            }
        }
    }

步骤

1、创建表：

create table `ecm_schedule_task`
(
    `lock_key`       varchar(255) not null,
    `lock_status`    int          not null default 0,
    `lock_timestamp` timestamp null default null,
    primary key (`lock_key`)
);
insert into ecm_schedule_task
(lock_key, lock_status)
values ('ecm_retry_transfer_to_dm_task', 0);

2、代码实现，我这里用的orm框架是jpa，如果你们用的是mybatis，那就换成对应的格式代码就可以

sql的意思是：其中lock_key是锁的唯一标识符，lock_status表示锁的状态（0表示未锁定，1表示已锁定），lock_timestamp是锁定时间戳，用于确定锁的有效性。

    //这段代码尝试将lock_status设置为1（锁定状态），但只有在当前状态为未锁定或锁已经过期（例如，超过10分钟）时才会成功。
    @Modifying
    @Transactional
    @Query(value = "UPDATE ecm_schedule_task SET lock_status = 1, lock_timestamp = NOW() 
    WHERE lock_key = ?1 AND (lock_status = 0 OR lock_timestamp < NOW() - INTERVAL 5 MINUTE)",nativeQuery = true)
    int getLock(String taskName);

   //完成任务后，更新锁记录以释放锁。
    @Modifying
    @Transactional
    @Query(value = "UPDATE ecm_schedule_task SET lock_status = 0 WHERE lock_key = ?1",nativeQuery = true)
    void unlock(String lockKey);

验证

我们可以打日志，看到每五分钟执行，只有一台机器拿到锁。完美！

innovation

1、其实我们用redis和redission去做是最好的，直接setnx就搞定
2、我这个是老项目，所以只能用mysql去做
3、也可以用xxl-job去处理
4、也可以用乐观锁，加一个字段version，这样update的时候where加上version的版本，这样修改到的就是拿到了锁，修改不到的就拿不到锁。
5、欢迎大家指教！