spring-aop防止重复提交网络请求

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/u011649691/article/details/82702456

具体原理很简单,通过spring-aop的环绕通知,请求开始时将请求参数转换校验是否已存在,已存在则报错,否则存储,请求完成后删除。

具体代码如下:
1、注释@interface

package com.yuntu.commons.intelligent.annotation;

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

/**
 * Created by niuzy on 2018-09-13.
 */
@Target({ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
public @interface NotDuplicate {
}

2、spring切面和环绕通知

import com.yuntu.commons.ServiceException;
import com.yuntu.commons.intelligent.ExceptionConstants;
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.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;

import java.lang.reflect.Method;
import java.util.Set;
import java.util.concurrent.ConcurrentSkipListSet;

/**
 * Created by niuzy on 2018-09-13.
 */
@Aspect
@Component
public class NotDuplicateAop {
    private Logger LOG = LoggerFactory.getLogger(NotDuplicateAop.class);
    private static final Set<String> KEY =  new ConcurrentSkipListSet<>();

    @Pointcut("@annotation(com.yuntu.commons.intelligent.annotation.NotDuplicate)")
    public void duplicate() {
    }

    /**
     * 对方法拦截后进行参数验证
     * @param pjp
     * @return
     * @throws Throwable
     */
    @Around("duplicate()")
    public Object duplicate(ProceedingJoinPoint pjp) throws Throwable {
        MethodSignature msig = (MethodSignature) pjp.getSignature();
        Method currentMethod = pjp.getTarget().getClass().getMethod(msig.getName(), msig.getParameterTypes());
        //拼接签名
        StringBuilder sb = new StringBuilder(currentMethod.toString());
        Object[] args = pjp.getArgs();
        for (Object object : args) {
            if(object != null){
                sb.append(object.getClass().toString());
                sb.append(object.toString());
            }
        }
        String sign = sb.toString();
        boolean success = KEY.add(sign);
        if(!success){
            throw new ServiceException(ExceptionConstants.ILLEGAL_REQUEST_EXCEPTION,"该方法正在执行,不能重复请求");
        }
        try {
            return pjp.proceed();
        } finally {
            KEY.remove(sign);
        }

    }
}

3、使用:
在需要的方法上添加@NotDuplicate

猜你喜欢

转载自blog.csdn.net/u011649691/article/details/82702456