就在昨天,我做的爱酷学习网出了一个严重的Bug。感谢发现此Bug的朋友“周大圣”,同时也感谢你能花时间为爱酷学习网上传视频!
这个Bug的具体情况是这样的,就是在上传视频后,会将数据提交到服务器,而当提交数据后点击浏览器的后退按扭,却又可以再次提交!这显示是不符合逻辑的,造成了数据重复提交。
好的,下面来讲一下如何在Spring MVC里面解决此问题(其它框架也一样,逻辑一样,思想一样,和具体框架没什么关系)。要解决重复提交,有很多办法,比如说在提交完成后redirect一下,也可以用本文提到的使用token的方法(我不使用redirect是因为那样解决不了ajax提交数据或者移动应用提交数据,另一个原因是现在比较通行的方法是使用token,像python里的django框架也是使用token来解决)。
使用token的逻辑是,给所有的url加一个拦截器,在拦截器里面用java的UUID生成一个随机的UUID并把这个UUID放到session里面,然后在浏览器做数据提交的时候将此UUID提交到服务器。服务器在接收到此UUID后,检查一下该UUID是否已经被提交,如果已经被提交,则不让逻辑继续执行下去…
好的,来点实际代码,也许实际代码才是最好的老师:
注解Token代码:
1
2
3
4
5
6
7
8
|
@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的配置文件里加入:
1
2
3
4
5
6
7
8
9
10
11
12
13
|
<!-- 拦截器配置 -->
<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里增加下面代码:
1
|
<input
type
=
"hidden"
name
=
"token"
value
=
"${token}"
/>
|
已经完成了,去试试看你的数据还能重复提交了吧。