SpringBoot接口幂等

import com.paascloud.core.lock.RedisDistributedLock; import com.paascloud.provider.config.annotation.Ide; import com.paascloud.provider.config.exception.BusinessException; import com.paascloud.wrapper.Wrapper; import lombok.extern.slf4j.Slf4j; 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.beans.factory.annotation.Autowired; import org.springframework.stereotype.Component; import org.springframework.web.context.request.RequestContextHolder; import org.springframework.web.context.request.ServletRequestAttributes; import javax.servlet.http.HttpServletRequest; import java.lang.reflect.Method; import java.util.UUID; @Slf4j @Aspect @Component public class ReqSubmitAspect { @Autowired private RedisDistributedLock redisDistributedLock; @Pointcut("@annotation(com.paascloud.provider.config.annotation.Ide)") public void idePointCut() { } @Around("idePointCut()") public Object around(ProceedingJoinPoint joinPoint) throws Throwable { MethodSignature signature = (MethodSignature) joinPoint.getSignature(); Method method = signature.getMethod(); Ide ide = method.getAnnotation(Ide.class); int lockSeconds = ide.lockTime(); ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes(); HttpServletRequest request = attributes.getRequest(); String token = request.getHeader("token"); String requestURI = request.getRequestURI(); String key = getIdeKey(token, requestURI); // 获取锁 boolean lock = redisDistributedLock.lock(key, lockSeconds); log.info("tryLock key = [{}]", key); if (lock) { log.info("tryLock success, key = [{}]", key); // 获取锁成功 Object result; try { // 执行进程 result = joinPoint.proceed(); } finally { // 解锁 redisDistributedLock.releaseLock(key); log.info("releaseLock success, key = [{}]", key); } return result; } else { // 获取锁失败，认为是重复提交的请求 log.info("tryLock fail, key = [{}]", key); throw new BusinessException(Wrapper.ERROR_CODE, "重复请求，请稍后再试!"); } } private String getIdeKey(String token, String requestURI) { return "cms:identity:" + token + requestURI; } }

 

 

import java.lang.annotation.*; /** * @version 1.00 * @Date in 18:32 2020/6/19 * identity 接口幂等注解 */ @Target(ElementType.METHOD) @Retention(RetentionPolicy.RUNTIME) @Documented public @interface Ide { /** * 设置请求锁定时间,超时后自动释放锁 * * @return */ int lockTime() default 10; } import lombok.extern.slf4j.Slf4j; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.data.redis.core.RedisCallback; import org.springframework.data.redis.core.RedisTemplate; import org.springframework.stereotype.Component; import org.springframework.util.StringUtils; import redis.clients.jedis.Jedis; import redis.clients.jedis.JedisCluster; import java.util.ArrayList; import java.util.List; import java.util.UUID; /** * redis分布式锁实现 * * @author jiangjunwei * * */ @Slf4j @Component public class RedisDistributedLock extends AbstractDistributedLock { @Autowired private RedisTemplate<String, Object> redisTemplate; private ThreadLocal<String> lockFlag = new ThreadLocal<>(); private static final String UNLOCK_LUA; /* * 通过lua脚本释放锁,来达到释放锁的原子操作 */ static { UNLOCK_LUA = "if redis.call(\"get\",KEYS[1]) == ARGV[1] " + "then " + " return redis.call(\"del\",KEYS[1]) " + "else " + " return 0 " + "end "; } public RedisDistributedLock(RedisTemplate<String, Object> redisTemplate) { super(); this.redisTemplate = redisTemplate; } @Override public boolean lock(String key, long expire, int retryTimes, long sleepMillis) { // log.info("======lock====== "+key ); boolean result = setRedis(key, expire); // 如果获取锁失败，按照传入的重试次数进行重试 while ((!result) && retryTimes-- > 0) { try { // log.info("get redisDistributeLock failed, retrying..." + retryTimes); Thread.sleep(sleepMillis); } catch (InterruptedException e) { log.info("Interrupted!", e); Thread.currentThread().interrupt(); } result = setRedis(key, expire); } return result; } private boolean setRedis(final String key, final long expire) { try { String status = redisTemplate.execute((RedisCallback<String>) connection -> { Object nativeConnection = connection.getNativeConnection(); String uuid = UUID.randomUUID().toString(); lockFlag.set(uuid); String result = null; // 集群模式和单机模式虽然执行脚本的方法一样，但是没有共同的接口，所以只能分开执行 // 集群模式 if (nativeConnection instanceof JedisCluster) { result = ((JedisCluster) nativeConnection).set(key, uuid, "NX", "PX", expire); } // 单机模式 else if (nativeConnection instanceof Jedis) { result = ((Jedis) nativeConnection).set(key, uuid, "NX", "PX", expire); } return result; }); return !StringUtils.isEmpty(status); } catch (Exception e) { log.error("set redisDistributeLock occured an exception", e); } return false; } @Override public boolean releaseLock(String key) { // 释放锁的时候，有可能因为持锁之后方法执行时间大于锁的有效期，此时有可能已经被另外一个线程持有锁，所以不能直接删除 try { // log.info("======unlock====== "+key ); final List<String> keys = new ArrayList<>(); keys.add(key); final List<String> args = new ArrayList<>(); args.add(lockFlag.get()); // 使用lua脚本删除redis中匹 配value的key，可以避免由于方法执行时间过长而redis锁自动过期失效的时候误删其他线程的锁 // spring自带的执行脚本方法中，集群模式直接抛出不支持执行脚本的异常，所以只能拿到原redis的connection来执行脚本 Long result = redisTemplate.execute((RedisCallback<Long>) connection -> { Object nativeConnection = connection.getNativeConnection(); // 集群模式和单机模式虽然执行脚本的方法一样，但是没有共同的接口，所以只能分开执行 // 集群模式 if (nativeConnection instanceof JedisCluster) { return (Long) ((JedisCluster) nativeConnection).eval(UNLOCK_LUA, keys, args); } // 单机模式 else if (nativeConnection instanceof Jedis) { return (Long) ((Jedis) nativeConnection).eval(UNLOCK_LUA, keys, args); } return 0L; }); return result != null && result > 0; } catch (Exception e) { log.error("release redisDistributeLock occured an exception", e); } finally { lockFlag.remove(); } return false; } }