springboot實現多實例crontab搶佔定時任務

github:https://github.com/jiasion/eslog

wechat:minghui-666

利用redisson實現多實例搶佔定時任務

pom.xml

<dependency>
            <groupId>org.redisson</groupId>
            <artifactId>redisson</artifactId>
            <version>3.12.0</version>
</dependency>

Kernel.java - 重寫多線程調度

package com.brand.log.scheduler;

import org.springframework.context.annotation.Configuration;
import org.springframework.scheduling.annotation.SchedulingConfigurer;
import org.springframework.scheduling.config.ScheduledTaskRegistrar;

import java.util.concurrent.Executors;

@Configuration
public class Kernel implements SchedulingConfigurer {

    @Override
    public void configureTasks(ScheduledTaskRegistrar taskRegistrar) {
        //設定一個長度10的定時任務線程池
        taskRegistrar.setScheduler(Executors.newScheduledThreadPool(4));
    }
}

RedissonManager.java  -  分佈式鎖的實現

package com.brand.log.util;

import lombok.extern.slf4j.Slf4j;
import org.redisson.Redisson;
import org.redisson.config.Config;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;

@Component
@Slf4j
public class RedissonManager {


    @Value("${spring.redis.host}")
    private String host;

    @Value("${spring.redis.port}")
    private int port;

    private Redisson redisson = null;
    private Config config = new Config();

    @PostConstruct
    private void init() {
        try {
            config.useSingleServer().setAddress("redis://" + host + ":" + port);
            log.info("redisson address {} {}", host, port);
            redisson = (Redisson) Redisson.create(config);
            log.info("Redisson 初始化完成");
        }
        catch (Exception e) {
            log.error("init Redisson error ", e);
        }
    }

    public Redisson getRedisson() {
        return redisson;
    }
}

CronSynData.java

package com.brand.log.scheduler;
import com.brand.log.util.DateFormatV1;
import com.brand.log.util.RedisUtil;
import com.brand.log.util.RedissonManager;
import lombok.extern.slf4j.Slf4j;
import org.redisson.Redisson;
import org.redisson.api.RLock;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;
import java.util.concurrent.TimeUnit;

@Component
@Slf4j
public class CronSynData {

    @Autowired
    RedissonManager redissonManager;

    @Autowired
    RedisUtil redisUtil;

    @Autowired
    DateFormatV1 dateFormatV1;

    private String lokFlag = ".handleKernel";

    private Redisson redisson = null;

    /*
    * java定時腳本掛靠實例
    * 多實例會有重複調用問題 + 使用Redisson實現分佈式鎖
    * 業務邏輯必須加鎖 + 且需要保證 tryLock 等待時間小於cron的最小間隔執行時間
    * */
    @Scheduled(cron = "*/10 * * * * *")
    public void handleKernel() {
        redisson = redissonManager.getRedisson();
        if (redisson != null) {
            RLock lock = redisson.getLock(this.getClass().getName() + lokFlag);
            Boolean stat = false;
            try {
                // 嘗試加鎖，立即返回，最多等待5s自動解鎖
                stat = lock.tryLock(0, 5, TimeUnit.SECONDS);
                if (stat) {
                    log.info("{} 取鎖成功！{}",this.getClass().getName(), Thread.currentThread().getName());
                    redisUtil.checkCount("log:limit_", dateFormatV1.getDate("HH", "GMT+8"), 60*10, 1000);
                } else {
                    log.info("{}沒有獲取到鎖:{}", this.getClass().getName(), Thread.currentThread().getName());
                }
            } catch (InterruptedException e) {
                log.error("Redisson 獲取分佈式鎖異常", e);
                if (!stat){
                    return;
                }
                lock.unlock();
            }
        }

    }

}

kibana - 6個實例