11
22

发表于 • • 被 98 人看爆

问题描述
将带有定时任务的项目部署在单台测试环境上，完全没问题。生产上是两台集群服务器，项目部署上去发现定时任务的模块同时在两台机器上各执行了一遍，这将会导致其他意外的发生。

解决方案----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!");
}

}

分享到：