就在昨天,我做的爱酷学习网出了一个严重的Bug。感谢发现此Bug的朋友“周大圣”,同时也感谢你能花时间为爱酷学习网上传视频!
这个Bug的具体情况是这样的,就是在上传视频后,会将数据提交到服务器,而当提交数据后点击浏览器的后退按扭,却又可以再次提交!这显示是不符合逻辑的,造成了数据重复提交。
好的,下面来讲一下如何在Spring MVC里面解决此问题(其它框架也一样,逻辑一样,思想一样,和具体框架没什么关系)。要解决重复提交,有很多办法,比如说在提交完成后redirect一下,也可以用本文提到的使用token的方法(我不使用redirect是因为那样解决不了ajax提交数据或者移动应用提交数据,另一个原因是现在比较通行的方法是使用token,像python里的django框架也是使用token来解决)。
使用token的逻辑是,给所有的url加一个拦截器,在拦截器里面用java的UUID生成一个随机的UUID并把这个UUID放到session里面,然后在浏览器做数据提交的时候将此UUID提交到服务器。服务器在接收到此UUID后,检查一下该UUID是否已经被提交,如果已经被提交,则不让逻辑继续执行下去…
好的,来点实际代码,也许实际代码才是最好的老师:
注解Token代码:
| @Target(ElementType.METHOD) @Retention(RetentionPolicy.RUNTIME) public @interface Token { boolean save() default false; boolean remove() default false; } |
拦截器TokenInterceptor代码:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
|
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> |
相关代码已经注释,相信你能看懂。
关于这个方法的用法是:在需要生成token的controller上增加@Token(save=true),而在需要检查重复提交的controller上添加@Token(remove=true)就可以了。
另外,你需要在view里在form里增加下面代码:
|
<input
type="hidden"
name="token"
value="${token}"
/>
|
已经完成了,去试试看你的数据还能重复提交了吧。