import java.lang.annotation.*;
/**
?* 防重
?* @date 2020/8/12
?* @return
?*/
//標(biāo)識(shí)該注解用于方法上
@Target({ElementType.METHOD})
//申明該注解為運(yùn)行時(shí)注解，編譯后改注解不會(huì)被遺棄
@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;
/**
?* 防重復(fù)提交
?* @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:";
? ? /**
? ? ?* 失效時(shí)間
? ? ?*/
? ? 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 {
? ? ? ? ? ? //獲取目標(biāo)方法的參數(shù)
? ? ? ? ? ? Object[] args = joinPoint.getArgs();
? ? ? ? ? ? //獲取當(dāng)前request請(qǐng)求
? ? ? ? ? ? RequestAttributes requestAttributes = RequestContextHolder.getRequestAttributes();
? ? ? ? ? ? HttpServletRequest request = (HttpServletRequest) requestAttributes.resolveReference(RequestAttributes.REFERENCE_REQUEST);
? ? ? ? ? ? //獲取請(qǐng)求地址
? ? ? ? ? ? String requestUri = request.getRequestURI();
? ? ? ? ? ? //獲取用戶(hù)ID
? ? ? ? ? ? Long userId = null;
? ? ? ? ? ? try {
? ? ? ? ? ? ? ? userId = TokenUtil.getUserId(request);
? ? ? ? ? ? }catch (Exception e){}
? ? ? ? ? ? //拼接鎖前綴，采用同一方法，同一用戶(hù),同一接口
? ? ? ? ? ? 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);
? ? ? ? ? ? /**
? ? ? ? ? ? ?* 對(duì)同一方法同一用戶(hù)同一參數(shù)加鎖,即使獲取不到用戶(hù)ID,每個(gè)用戶(hù)請(qǐng)求數(shù)據(jù)也會(huì)不一致，不會(huì)造成接口堵塞
? ? ? ? ? ? ?*/
? ? ? ? ? ? lock = this.redissonClient.getLock(lockPrefix);
? ? ? ? ? ? lock.lock();
? ? ? ? ? ? String flag = this.stringRedisTemplate.opsForValue().get(redisPrefix);
? ? ? ? ? ? if(StringUtils.isNotEmpty(flag)){
? ? ? ? ? ? ? ? throw new AppException("您當(dāng)前的操作太頻繁了,請(qǐng)稍后再試!");
? ? ? ? ? ? }
? ? ? ? ? ? //存入redis,設(shè)置失效時(shí)間
? ? ? ? ? ? this.stringRedisTemplate.opsForValue()
? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?.set(redisPrefix,redisPrefix,INVALID_NUMBER, TimeUnit.SECONDS);
? ? ? ? ? ? //執(zhí)行目標(biāo)方法
? ? ? ? ? ? Object result = joinPoint.proceed(args);
? ? ? ? ? ? return result;
? ? ? ? }finally {
? ? ? ? ? ? if(lock != null){
? ? ? ? ? ? ? ? lock.unlock();
? ? ? ? ? ? }
? ? ? ? }
? ? }
}