陈建华的博客
专注web开发
SpringMVC自定义注解 - controller中获取session
2017-05-22 13:40:47   阅读875次

平时使用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) {
        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负责,嗯,这个好。




-----------------------------------------------------
转载请注明来源此处
原地址:#

-----网友评论----
暂无评论
-----发表评论----
微网聚博客乐园 ©2014 blog.mn886.net 鲁ICP备14012923号   网站导航