redis分布式鎖解決集群服務(wù)器定時任務(wù)重復(fù)執(zhí)行問題

問題描述

將帶有定時任務(wù)的項目部署在單臺測試環(huán)境上，完全沒問題。生產(chǎn)上是兩臺集群服務(wù)器，項目部署上去發(fā)現(xiàn)定時任務(wù)的模塊同時在兩臺機器上各執(zhí)行了一遍，這將會導(dǎo)致其他意外的發(fā)生。

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

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

核心代碼

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.定時任務(wù)服務(wù)

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