SpringMVC - controller中获取session

平时使用springMVC

在方法中访问session中经常很自然地调用Servlet API。

用起来非常直观方便，一直没有多考虑什么。

比如这样：

@RequestMapping(value = "/logout")

public String logout(HttpSession session) {

session.removeAttribute("user");

return "/login";

}

但毕竟这样对Servlet API产生了依赖，感觉不够pojo。

于是我试着解决这个问题。

我打算用一个注解，名字就叫"sessionScope"，Target可以是一个Method，也可以是Parameter。

也就是说：

复制代码

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里的东西。

代码如下：

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
复制代码

<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中使用了，比如：

@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;

}

或者在参数上注解：

@RequestMapping(value = "/welcome")

public String welcome(@SessionScope("currentUser")User currentUser) {

return "/main";

}

至于更新session，目前只是用@sessionAttributes配合ModelMap的方式。

嗯，我不是很喜欢这种方式...

或者我可以不用这样做，直接集成Apache Shiro，在controller中直接getSubject()。

把用户信息完全让shiro负责，嗯，这个好。