您的位置:首页 > 编程语言 > Java开发

Spring MVC防止数据重复提交

2014-09-29 13:59 295 查看

注解Token代码:

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

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface Token {

boolean save() default false;

boolean remove() default false;
}


拦截器TokenInterceptor代码:

import java.lang.reflect.Method;
import java.util.UUID;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.springframework.web.method.HandlerMethod;
import org.springframework.web.servlet.handler.HandlerInterceptorAdapter;

public class TokenInterceptor extends HandlerInterceptorAdapter {

@Override
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
if (handler instanceof HandlerMethod) {
HandlerMethod handlerMethod = (HandlerMethod) handler;
Method method = handlerMethod.getMethod();
Token annotation = method.getAnnotation(Token.class);
if (annotation != null) {
boolean needSaveSession = annotation.save();
if (needSaveSession) {
request.getSession(false).setAttribute("token", UUID.randomUUID().toString());
}
boolean needRemoveSession = annotation.remove();
if (needRemoveSession) {
if (isRepeatSubmit(request)) {
return false;
}
request.getSession(false).removeAttribute("token");
}
}
return true;
} else {
return super.preHandle(request, response, handler);
}
}

private boolean isRepeatSubmit(HttpServletRequest request) {
String serverToken = (String) request.getSession(false).getAttribute("token");
if (serverToken == null) {
return true;
}
String clinetToken = request.getParameter("token");
if (clinetToken == null) {
return true;
}
if (!serverToken.equals(clinetToken)) {
return true;
}
return false;
}
}

然后在Spring MVC的配置文件里加入:

<!-- 拦截器配置 -->
<mvc:interceptors>
<!-- 配置Shiro拦截器,实现注册用户的注入 -->
<mvc:interceptor>
<mvc:mapping path="/**"/>
<bean class="com.storezhang.video.shiro.ShiroInterceptor"/>
</mvc:interceptor>
<!-- 配置Token拦截器,防止用户重复提交数据 -->
<mvc:interceptor>
<mvc:mapping path="/**"/>
<bean class="com.storezhang.web.spring.TokenInterceptor"/>
</mvc:interceptor>
</mvc:interceptors>

相关的Controller:

详细页面的展示,顺便把Token中的save为true传入页面

@RequestMapping("/showDetail.do")
@Token(save=true)
public ModelAndView showDeatil(HttpServletRequest request, Model model){
//相关的方法
}
保存方法:

<span style="white-space:pre">	</span>@RequestMapping("/save")
@Token(remove=true)
@ResponseBody
public Integer save(HttpSession session,HttpServletRequest request) throws ParseException{
//相关代码
}


注:

相关代码已经注释,相信你能看懂。
关于这个方法的用法是:在需要生成token的controller上增加@Token(save=true),而在需要检查重复提交的controller上添加@Token(remove=true)就可以了。
另外,你需要在view里在form里增加下面代码:

<input type="hidden"name="token"value="${token}"/>


内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: