Spring注解方式防止重复提交原理详解

2023-09-05 07:29:04 54

Srping注解方式防止重复提交原理分析，供大家参考，具体内容如下

方法一：Springmvc使用Token

使用token的逻辑是，给所有的url加一个拦截器，在拦截器里面用java的UUID生成一个随机的UUID并把这个UUID放到session里面，然后在浏览器做数据提交的时候将此UUID提交到服务器。服务器在接收到此UUID后，检查一下该UUID是否已经被提交，如果已经被提交，则不让逻辑继续执行下去…**

1首先要定义一个annotation：用@Retention和@Target标注接口

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public@interfaceToken{
booleansave()defaultfalse;
booleanremove()defaultfalse;
}

2定义拦截器TokenInterceptor：

publicclassTokenInterceptorextendsHandlerInterceptorAdapter{

@Override
publicbooleanpreHandle(HttpServletRequestrequest,HttpServletResponseresponse,Objecthandler)throwsException{
if(handlerinstanceofHandlerMethod){
HandlerMethodhandlerMethod=(HandlerMethod)handler;
Methodmethod=handlerMethod.getMethod();
Tokenannotation=method.getAnnotation(Token.class);
if(annotation!=null){
booleanneedSaveSession=annotation.save();
if(needSaveSession){
request.getSession(false).setAttribute("token",UUID.randomUUID().toString());
}
booleanneedRemoveSession=annotation.remove();
if(needRemoveSession){
if(isRepeatSubmit(request)){
returnfalse;
}
request.getSession(false).removeAttribute("token");
}
}
returntrue;
}else{
returnsuper.preHandle(request,response,handler);
}
}

privatebooleanisRepeatSubmit(HttpServletRequestrequest){
StringserverToken=(String)request.getSession(false).getAttribute("token");
if(serverToken==null){
returntrue;
}
StringclinetToken=request.getParameter("token");
if(clinetToken==null){
returntrue;
}
if(!serverToken.equals(clinetToken)){
returntrue;
}
returnfalse;
}
}

SpringMVC的配置文件里加入：











@RequestMapping("/add.jspf")
@Token(save=true)
publicStringadd(){
//省略
returnTPL_BASE+"index";
}

@RequestMapping("/save.jspf")
@Token(remove=true)
publicvoidsave(){
//省略
}

用法：

在Controller类的用于定向到添加/修改操作的方法上增加自定义的注解类@Token(save=true)

在Controller类的用于表单提交保存的的方法上增加@Token(remove=true)

在表单中增加用于存储token，每次需要报token值传入到后台类，用于从缓存对比是否是重复提交操作

方法二：springboot中用注解方式

每次操作，生成的key存放于缓存中，比如用google的Gruava或者Redis做缓存

定义Annotation类

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Inherited
public@interfaceLocalLock{

/**
*@authorfly
*/
Stringkey()default"";

/**
*过期时间TODO由于用的guava暂时就忽略这属性吧集成redis需要用到
*
*@authorfly
*/
intexpire()default5;
}

设置拦截类

@Aspect
@Configuration
publicclassLockMethodInterceptor{

privatestaticfinalCacheCACHES=CacheBuilder.newBuilder()
//最大缓存100个
.maximumSize(1000)
//设置写缓存后5秒钟过期
.expireAfterWrite(5,TimeUnit.SECONDS)
.build();

@Around("execution(public**(..))&&@annotation(com.demo.testduplicate.Test1.LocalLock)")
publicObjectinterceptor(ProceedingJoinPointpjp){
MethodSignaturesignature=(MethodSignature)pjp.getSignature();
Methodmethod=signature.getMethod();
LocalLocklocalLock=method.getAnnotation(LocalLock.class);
Stringkey=getKey(localLock.key(),pjp.getArgs());
if(!StringUtils.isEmpty(key)){
if(CACHES.getIfPresent(key)!=null){
thrownewRuntimeException("请勿重复请求");
}
//如果是第一次请求,就将key当前对象压入缓存中
CACHES.put(key,key);
}
try{
returnpjp.proceed();
}catch(Throwablethrowable){
thrownewRuntimeException("服务器异常");
}finally{
//TODO为了演示效果,这里就不调用CACHES.invalidate(key);代码了
}
}

/**
*key的生成策略,如果想灵活可以写成接口与实现类的方式（TODO后续讲解）
*
*@paramkeyExpress表达式
*@paramargs参数
*@return生成的key
*/
privateStringgetKey(StringkeyExpress,Object[]args){
for(inti=0;i
Controller类引用
@RestController
@RequestMapping("/books")
publicclassBookController{

@LocalLock(key="book:arg[0]")
@GetMapping
publicStringsave(@RequestParamStringtoken){
return"success-"+token;
}
}
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持毛票票。

Spring注解方式防止重复提交原理详解

热门推荐

随机推荐