平时使用springMVC,在方法中访问session中经常很自然地调用Servlet API。用起来非常直观方便,一直没有多考虑什么。
比如这样:
1
2
3
4
5
|
@RequestMapping (value = "/logout" )
public String logout(HttpSession session) {
session.removeAttribute( "user" );
return "/login" ;
}
|
但毕竟这样对Servlet API产生了依赖,感觉不够pojo。
于是我试着解决这个问题。
我打算用一个注解,名字就叫"sessionScope",Target可以是一个Method,也可以是Parameter。
也就是说:
1
2
3
4
5
6
7
8
9
10
11
12
|
import java.lang.annotation.Documented;
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
@Target ({ ElementType.PARAMETER,ElementType.METHOD })
@Retention (RetentionPolicy.RUNTIME)
@Documented
public @interface SessionScope {
String value();
}
|
然后我要注册一个ArgumentResolver,专门解决被注解的东东,把他们统统替换成session里的东西。
代码如下:
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
43
44
45
46
|
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.core.MethodParameter;
import org.springframework.web.bind.support.WebDataBinderFactory;
import org.springframework.web.context.request.NativeWebRequest;
import org.springframework.web.context.request.RequestAttributes;
import org.springframework.web.method.support.HandlerMethodArgumentResolver;
import org.springframework.web.method.support.ModelAndViewContainer;
public class SessionScopeMethodArgumentResolver implements
HandlerMethodArgumentResolver {
@Override
public boolean supportsParameter(MethodParameter parameter) {
//让方法和参数,两种target通过
if (parameter.hasParameterAnnotation(SessionScope. class )) return true ;
else if (parameter.getMethodAnnotation(SessionScope. class ) != null ) return true ;
return false ;
}
@Override
public Object resolveArgument(MethodParameter parameter,
ModelAndViewContainer mavContainer, NativeWebRequest webRequest,
WebDataBinderFactory binderFactory) throws Exception {
String annoVal = null ;
if (parameter.getParameterAnnotation(SessionScope. class )!= null ){
logger.debug( "param anno val::::" +parameter.getParameterAnnotation(SessionScope. class ).value());
annoVal = parameter.getParameterAnnotation(SessionScope. class ).value();
} else if (parameter.getMethodAnnotation(SessionScope. class )!= null ){
logger.debug( "method anno val::::" +parameter.getMethodAnnotation(SessionScope. class ).value());
annoVal = parameter.getMethodAnnotation(SessionScope. class )!= null ?
StringUtils.defaultString(parameter.getMethodAnnotation(SessionScope. class ).value())
:StringUtils.EMPTY;
}
if (webRequest.getAttribute(annoVal,RequestAttributes.SCOPE_SESSION) != null ){
return webRequest.getAttribute(annoVal,RequestAttributes.SCOPE_SESSION);
}
else
return null ;
}
final Logger logger = LoggerFactory.getLogger(SessionScopeMethodArgumentResolver. class );
}
|
supportParameter判断对象是否被注解,被注解则进行resolve。
resolve时获取注解值,注解值为session key,用webRequest从session scope中取值。
另外需要将此配置放到org.springframework.web.servlet.mvc.method.annotation.RequestMappingHandlerAdapter的customArgumentResolvers列表中,可以使用bean标签配置,也可以直接使用mvc标签。
即:
namespace为:xmlns:mvc="http://www.springframework.org/schema/mvc"
schema location为:http://www.springframework.org/schema/mvc
http://www.springframework.org/schema/mvc/spring-mvc-3.1.xsd
1
2
3
4
5
6
|
< mvc:annotation-driven >
< mvc:argument-resolvers >
< bean class = "pac.common.SessionScopeMethodArgumentResolver" />
</ mvc:argument-resolvers >
</ mvc:annotation-driven >
< mvc:default-servlet-handler />
|
话说mvc:annotation-driven和mvc:default-servlet-handler的顺序不能颠倒,该不会只有我出现这种情况吧- -..
现在可以在controller中使用了,比如:
1
2
3
4
5
6
7
8
9
10
11
|
@RequestMapping (value = "/index" )
@SessionScope ( "currentUser" )
public ModelAndView index(User currentUser) {
ModelAndView mav;
if (currentUser== null || currentUser.getId()== null )
mav = new ModelAndView( "/login" );
else {
mav = new ModelAndView( "/index" );
}
return mav;
}
|
或者在参数上注解:
1
2
3
4
|
@RequestMapping (value = "/welcome" )
public String welcome( @SessionScope ( "currentUser" )User currentUser) {
return "/main" ;
}
|
至于更新session,目前只是用@sessionAttributes配合ModelMap的方式。
或者我可以不用这样做,直接集成Apache Shiro,在controller中直接getSubject()。
把用户信息完全让shiro负责,嗯,这个好。
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持服务器之家。
原文链接:http://www.cnblogs.com/crime-king/p/3851693.html