redis分布式鎖解決集群服務器定時任務重復執行問題

問題描述

將帶有定時任務的項目部署在單臺測試環境上，完全沒問題。生產上是兩臺集群服務器，項目部署上去發現定時任務的模塊同時在兩臺機器上各執行了一遍，這將會導致其他意外的發生。

解決方案----redis分布式鎖

使用redis分布式鎖，為定時任務唯一指定的key加鎖，并設置鎖超時時間。當觸發定時任務時，一臺服務的任務進入切面，通過setNX(key,value)方法為唯一的key加鎖，如果當前key不存在，將放入緩存，并返回true,通過expire(key,second)設置鎖超時時間，結束后跳出執行定時任務方法。第二臺服務任務進入時，設置鎖的時候發現該鎖已存在于緩存，并返回false，不跳轉到執行定時任務方法。

核心代碼

1.分布式鎖切面

@Aspect
@Slf4j
@Component
public?class?CacheLockAspect?{
?private?static?final?String?LOCK_VALUE?=?"locked";
?@Autowired
?private?RedisConnection?connection;
?@Around("execution(*?*.*(..))?&&?@annotation(com.common.annotation.CacheLock)")
?public?void?cacheLockPoint(ProceedingJoinPoint?pjp)?{
?Method?cacheMethod?=?null;
?for?(Method?method?:?pjp.getTarget().getClass().getMethods())?{
?if?(null!=method.getAnnotation(CacheLock.class)){
?cacheMethod?=?method;
?break;
?}
?}
?try?{
?String?lockKey?=?cacheMethod.getAnnotation(CacheLock.class).lockedPrefix();
?long?timeOut?=?cacheMethod.getAnnotation(CacheLock.class).expireTime();
?if(null?==?lockKey){
?throw?new?ManagerException(ErrorMsgEnum.LOCK_NAME_EMPTY);
?}
?if?(connection.setNX(lockKey.getBytes(),LOCK_VALUE.getBytes()))?{
?connection.expire(lockKey.getBytes(),timeOut);
?log.info("method:{}獲取鎖:{},開始運行！",cacheMethod,lockKey);
?pjp.proceed();
?return;
?}
?log.info("method:{}未獲取鎖:{},運行失敗！",cacheMethod,lockKey);
?}?catch?(Throwable?e)?{
?log.error("method:{},運行錯誤！",cacheMethod,e);
?throw?new?ManagerException(ErrorMsgEnum.LOCK_JOB_ERROR,e);
?}
?}
}

2.手寫方法級注解

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
public?@interface?CacheLock?{
?String?lockedPrefix()?default?"";?//redis?鎖key的前綴
?long?expireTime()?default?10;?//key在redis里存在的時間，1000S
}

3.定時任務服務

@Slf4j
@Service
public?class?TimeTaskService?{
?/**
?*?執行定時任務
?**/
?@Scheduled(cron?=?"0?0?1?*?*??")
?@CacheLock(lockedPrefix?=?"TimeTaskService",expireTime=30)
?public?void?executeTask()?{
?System.out.println("hello?world!");
?}
}