springmvc 下使用自定义注解获取登录信息-CSDN博客

使用原因

大家的项目中在controller层是怎样获取登录用户的信息呢？

User loginUser=LoginUtil.getLoginUser();

   
   1
2

我想有些同学是通过这样获得的，如果这样实现的话，
恭喜你，你的Controller层已经丧失了单元测试的能力。

因为执行这个controller所需要的参数并没有完全通过参数列表来获得，而依赖于全局环境（web环境）
这样做使得springmvc为环境解耦所设计的整个架构变得毫无意义。

实现

那怎么样写才算是高大上呢？先贴出最后效果

@RequestMapping(...)
public 返回值 方法名(@logined User user){
}

   
   1
2
3
4

相信大家一眼就能看出他的优点。下面说一下如何实现

自定义注解 @logined

@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.PARAMETER)
public @interface Logined {
}

   
   1
2
3
4
5

实现HandlerMethodArgumentResolver接口

public class LoginedArgumentResolver implements HandlerMethodArgumentResolver{

/**
 * 检查解析器是否支持解析该参数
 */
@Override
public boolean supportsParameter(MethodParameter parameter) {
    if(
            //如果该参数注解有@Logined
            parameter.getParameterAnnotation(Logined.class)!=null&&
            //如果该参数的类型为User
            parameter.getParameterType()==User.class
            ){
        //支持解析该参数
        return true;
    }
    return false;
}

@Override
public Object resolveArgument(MethodParameter parameter,
        ModelAndViewContainer mavContainer, NativeWebRequest webRequest,
        WebDataBinderFactory binderFactory) throws Exception {
        HttpServletRequest request= (HttpServletRequest) webRequest.getNativeRequest();
    //这里暂时把User对象放在session中
        User user=(User)request.getSession().getAttribute("loginUser");
        return user;
    }
}

   
   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

在springmvc的配置文件中配置

<mvc:annotation-driven>
    <mvc:argument-resolvers>
        <bean class="org.apollo.springmvc.argResolver.LoginedArgumentResolver"></bean>
    </mvc:argument-resolvers>
</mvc:annotation-driven>

   
   1
2
3
4
5
6