1：自定义注解JRepeat
package com.huan.study.mybatis.config;import java.lang.annotation.*;
@Retention(RetentionPolicy.RUNTIME)
@Target({ElementType.METHOD})
@Documented
public @interface JRepeat {int lockTime();String lockKey() default "";}
2：自定义切面
package com.huan.study.mybatis.aspect;import com.huan.study.mybatis.config.JRepeat;
import com.huan.study.mybatis.config.RedissonLockClient;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.core.LocalVariableTableParameterNameDiscoverer;
import org.springframework.stereotype.Component;import javax.annotation.Resource;
import java.util.Objects;
import java.util.concurrent.TimeUnit;
@Aspect
@Component
public class RepeatSubmitAspect extends BaseAspect {@Resourceprivate RedissonLockClient redissonLockClient;@Pointcut("@annotation(jRepeat)")public void pointCut(JRepeat jRepeat) {}@Around("pointCut(jRepeat)")public Object repeatSubmit(ProceedingJoinPoint joinPoint,JRepeat jRepeat) throws Throwable {String[] parameterNames = new LocalVariableTableParameterNameDiscoverer().getParameterNames(((MethodSignature) joinPoint.getSignature()).getMethod());if (Objects.nonNull(jRepeat)) {Object[] args = joinPoint.getArgs();StringBuffer lockKeyBuffer = new StringBuffer();String key =getValueBySpEL(jRepeat.lockKey(), parameterNames, args,"RepeatSubmit").get(0);boolean isLocked = false;try {isLocked = redissonLockClient.fairLock(key, TimeUnit.SECONDS, jRepeat.lockTime());if (isLocked) {return joinPoint.proceed();} else {throw new RuntimeException("请勿重复提交");}} finally {if (isLocked) {redissonLockClient.unlock(key);}}}return joinPoint.proceed();}
}
package com.huan.study.mybatis.aspect;import lombok.extern.slf4j.Slf4j;
import org.springframework.expression.EvaluationContext;
import org.springframework.expression.Expression;
import org.springframework.expression.ExpressionParser;
import org.springframework.expression.spel.standard.SpelExpressionParser;
import org.springframework.expression.spel.support.StandardEvaluationContext;import java.util.ArrayList;
import java.util.List;@Slf4j
public class BaseAspect {public List<String> getValueBySpEL(String key, String[] parameterNames, Object[] values, String keyConstant) {List<String> keys = new ArrayList<>();if (!key.contains("#")) {String s = "redis:lock:" + key + keyConstant;log.debug("lockKey:" + s);keys.add(s);return keys;}ExpressionParser parser = new SpelExpressionParser();EvaluationContext context = new StandardEvaluationContext();for (int i = 0; i < parameterNames.length; i++) {context.setVariable(parameterNames[i], values[i]);}Expression expression = parser.parseExpression(key);Object value = expression.getValue(context);if (value != null) {if (value instanceof List) {List value1 = (List) value;for (Object o : value1) {addKeys(keys, o, keyConstant);}} else if (value.getClass().isArray()) {Object[] obj = (Object[]) value;for (Object o : obj) {addKeys(keys, o, keyConstant);}} else {addKeys(keys, value, keyConstant);}}log.info("表达式key={},value={}", key, keys);return keys;}private void addKeys(List<String> keys, Object o, String keyConstant) {keys.add("redis:lock:" + o.toString() + keyConstant);}
}
3：自定义RedissonLockClient
package com.huan.study.mybatis.config;import lombok.extern.slf4j.Slf4j;
import org.redisson.api.RLock;
import org.redisson.api.RedissonClient;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;import javax.annotation.Resource;
import java.util.concurrent.TimeUnit;
@Slf4j
@Component
public class RedissonLockClient {@Autowiredprivate RedissonClient redissonClient;@Resourceprivate RedisTemplate<String, Object> redisTemplate;public RLock getLock(String lockKey) {return redissonClient.getLock(lockKey);}public boolean tryLock(String lockName, long expireSeconds) {return tryLock(lockName, 0, expireSeconds);}public boolean tryLock(String lockName, long waitTime, long expireSeconds) {RLock rLock = getLock(lockName);boolean getLock = false;try {getLock = rLock.tryLock(waitTime, expireSeconds, TimeUnit.SECONDS);if (getLock) {log.info("获取锁成功,lockName={}", lockName);} else {log.info("获取锁失败,lockName={}", lockName);}} catch (InterruptedException e) {log.error("获取式锁异常，lockName=" + lockName, e);getLock = false;}return getLock;}public boolean fairLock(String lockKey, TimeUnit unit, int leaseTime) {RLock fairLock = redissonClient.getFairLock(lockKey);try {boolean existKey = existKey(lockKey);if (existKey) {return false;}return fairLock.tryLock(3, leaseTime, unit);} catch (InterruptedException e) {e.printStackTrace();}return false;}public boolean existKey(String key) {return redisTemplate.hasKey(key);}public RLock lock(String lockKey) {RLock lock = getLock(lockKey);lock.lock();return lock;}public RLock lock(String lockKey, long leaseTime) {RLock lock = getLock(lockKey);lock.lock(leaseTime, TimeUnit.SECONDS);return lock;}public void unlock(String lockName) {try {redissonClient.getLock(lockName).unlock();} catch (Exception e) {log.error("解锁异常，lockName=" + lockName, e);}}
}
4：demo所著相同请求5秒，如果方法执行完成释放
@SneakyThrows@GetMapping("test")@JRepeat(lockKey = "#phone", lockTime = 5)public void test(String phone) {new Thread(() -> {try {System.out.println(new Date() + Thread.currentThread().getName() + " 获取到锁，开始处理任务...");Thread.sleep(10000); System.out.println(new Date() + Thread.currentThread().getName() + " 处理任务完成，释放锁...");} catch (InterruptedException e) {e.printStackTrace();}}).start();System.out.println("===" + new Date());}
5：yml配置redis
spring:redis:database: 0host: 127.0.0.1password: 123456port: 6379
注意：因为是基于AOP切面，如果在方法内，调用的方法上添加该注解会失效
)
)
