import java.lang.annotation.*;
/**
?* 防重
?* @date 2020/8/12
?* @return
?*/
//標識該注解用于方法上
@Target({ElementType.METHOD})
//申明該注解為運行時注解，編譯后改注解不會被遺棄
@Retention(RetentionPolicy.RUNTIME)
//javadoc工具記錄
@Documented
public @interface PreventSubmit?
{
}

import com.qianxian.common.exception.AppException;
import com.qianxian.common.util.TokenUtil;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.redisson.api.RLock;
import org.redisson.api.RedissonClient;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestAttributes;
import org.springframework.web.context.request.RequestContextHolder;
import javax.servlet.http.HttpServletRequest;
import java.util.Arrays;
import java.util.concurrent.TimeUnit;
/**
?* 防重復提交
?* @date 2020/8/12
?* @return
?*/
@Component
@Aspect
@Slf4j
public class PreventSubmitAspect {
? ? /**
? ? ?* 放重redis前綴
? ? ?*/
? ? private static String API_PREVENT_SUBMIT = "api:preventSubmit:";
? ? /**
? ? ?* 放重分布式鎖前綴
? ? ?*/
? ? private static String API_LOCK_PREVENT_SUBMIT = "api:preventSubmit:lock:";
? ? /**
? ? ?* 失效時間
? ? ?*/
? ? private static Integer INVALID_NUMBER = 3;
? ? /**
? ? ?* redis
? ? ?*/
? ? @Autowired
? ? private StringRedisTemplate stringRedisTemplate;
? ? /**
? ? ?* 分布式鎖
? ? ?*/
? ? @Autowired
? ? private RedissonClient redissonClient;
? ? /**
? ? ?* 防重
? ? ?* @date 2020/8/12
? ? ?* @return
? ? ?*/
? ? @Around("@annotation(com.qianxian.user.annotation.PreventSubmit)")
? ? public Object preventSubmitAspect(ProceedingJoinPoint joinPoint) throws Throwable {
? ? ? ? RLock lock = null;
? ? ? ? try {
? ? ? ? ? ? //獲取目標方法的參數(shù)
? ? ? ? ? ? Object[] args = joinPoint.getArgs();
? ? ? ? ? ? //獲取當前request請求
? ? ? ? ? ? RequestAttributes requestAttributes = RequestContextHolder.getRequestAttributes();
? ? ? ? ? ? HttpServletRequest request = (HttpServletRequest) requestAttributes.resolveReference(RequestAttributes.REFERENCE_REQUEST);
? ? ? ? ? ? //獲取請求地址
? ? ? ? ? ? String requestUri = request.getRequestURI();
? ? ? ? ? ? //獲取用戶ID
? ? ? ? ? ? Long userId = null;
? ? ? ? ? ? try {
? ? ? ? ? ? ? ? userId = TokenUtil.getUserId(request);
? ? ? ? ? ? }catch (Exception e){}
? ? ? ? ? ? //拼接鎖前綴，采用同一方法，同一用戶,同一接口
? ? ? ? ? ? String temp = requestUri.concat(Arrays.asList(args).toString()) + (userId != null ? userId : "");
? ? ? ? ? ? temp = temp.replaceAll("/","");
? ? ? ? ? ? //拼接rediskey
? ? ? ? ? ? String lockPrefix = API_LOCK_PREVENT_SUBMIT.concat(temp);
? ? ? ? ? ? String redisPrefix = API_PREVENT_SUBMIT.concat(temp);
? ? ? ? ? ? /**
? ? ? ? ? ? ?* 對同一方法同一用戶同一參數(shù)加鎖,即使獲取不到用戶ID,每個用戶請求數(shù)據(jù)也會不一致，不會造成接口堵塞
? ? ? ? ? ? ?*/
? ? ? ? ? ? lock = this.redissonClient.getLock(lockPrefix);
? ? ? ? ? ? lock.lock();
? ? ? ? ? ? String flag = this.stringRedisTemplate.opsForValue().get(redisPrefix);
? ? ? ? ? ? if(StringUtils.isNotEmpty(flag)){
? ? ? ? ? ? ? ? throw new AppException("您當前的操作太頻繁了,請稍后再試!");
? ? ? ? ? ? }
? ? ? ? ? ? //存入redis,設置失效時間
? ? ? ? ? ? this.stringRedisTemplate.opsForValue()
? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?.set(redisPrefix,redisPrefix,INVALID_NUMBER, TimeUnit.SECONDS);
? ? ? ? ? ? //執(zhí)行目標方法
? ? ? ? ? ? Object result = joinPoint.proceed(args);
? ? ? ? ? ? return result;
? ? ? ? }finally {
? ? ? ? ? ? if(lock != null){
? ? ? ? ? ? ? ? lock.unlock();
? ? ? ? ? ? }
? ? ? ? }
? ? }
}