问题描述
将带有定时任务的项目部署在单台测试环境上,完全没问题。生产上是两台集群服务器,项目部署上去发现定时任务的模块同时在两台机器上各执行了一遍,这将会导致其他意外的发生。
解决方案----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";@Autowiredprivate 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!");}}