AOP实现防止接口重复提交

傷城~ 2021-12-19 04:05 236阅读 0赞

**项目中对于状态变更接口存在重复提交的问题。**

package com.yxx.survey.foundation.aop;
    
    import com.alibaba.fastjson.JSON;
    import com.yxx.survey.foundation.exception.ErrorEnum;
    import com.yxx.survey.foundation.exception.excetion.BusinessException;
    import com.yxx.survey.foundation.redis.RedisService;
    import com.yxx.survey.foundation.util.encryption.HashUtil;
    import lombok.extern.slf4j.Slf4j;
    import org.aspectj.lang.JoinPoint;
    import org.aspectj.lang.annotation.Aspect;
    import org.aspectj.lang.annotation.Before;
    import org.aspectj.lang.annotation.Pointcut;
    import org.springframework.stereotype.Component;
    
    import javax.annotation.Resource;
    
    
    @Aspect
    @Component
    @Slf4j
    public class RepeatSubmitAspect {
        @Resource
        private RedisService redisService;
    
        @Pointcut("@annotation(RepeatSubmitCheck)")
        public void requestPointcut() {
        }
    
        @Before("requestPointcut() && @annotation(repeatSubmitCheck)")
        public void aroundCheck(JoinPoint joinPoint, RepeatSubmitCheck repeatSubmitCheck) {
            String className = joinPoint.getTarget().getClass().getName();
    
            String methodName = joinPoint.getSignature().getName();
    
            Object[] args = joinPoint.getArgs();
            String paramsJsonStr = JSON.toJSONString(args);
    
            String srcStr = className + "_" + methodName + "_" + paramsJsonStr;
            log.info(srcStr);
    
            String signStr = HashUtil.MD5(srcStr);
    
            if (!redisService.setIfNotExists(signStr, "1", repeatSubmitCheck.keepSeconds())) {
                throw BusinessException.with(ErrorEnum.INTERFACE_INVOKE_FREQUENTLY);
            }
        }
    }

package com.yxx.survey.foundation.aop;
    
    import java.lang.annotation.*;
    
    @Target(ElementType.METHOD)
    @Retention(RetentionPolicy.RUNTIME)
    @Documented
    public @interface RepeatSubmitCheck {
        int keepSeconds() default 1;
    }

![1135169-20190422183512099-738632276.png][]

转载于:https://www.cnblogs.com/watson-ljf/p/10751990.html

[1135169-20190422183512099-738632276.png]: /images/20211218/979ae49744ec457280b84ae35092d27c.png