具体原理很简单,通过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