springMVC利用WebArgumentResolver为controller注入当前用户对象参数

MIS系统开发的时候，相信大家都需要获取当前登录用户，用来记录操作员等。

而平时我们极有可能用的这种办法： request.getSession().getAttribute("CURRENT_USER");

比如以前保存资源是这样的：

@RequestMapping(value = "save.json", method = RequestMethod.POST)
@ResponseBody
public String saveResource(HttpServletRequest request, Resource res){

// 这里的res，是从页面表单提交过来的

// 这里的res里面，可能需要记录操作人，那么需要获取到当前用户

CurrentUser currentUser = request.getSession().getAttribute("CURRENT_USER");

}

这里我们需要写：request.getSession().getAttribute("CURRENT_USER");

这没有任何问题，但是能否这一句代码都不写，直接进入这个方法就把CURRENT_USER注入好了？

比如进入下面这个方法时，框架能否注入了当前用户CurrentUser ？yes，请继续看下面。

@RequestMapping(value = "save.json", method = RequestMethod.POST)
@ResponseBody
public String saveResource(CurrentUser user, Resource res){

}

开工：

1 自定义ArgumentResolver实现WebArgumentResolver接口

源码如下：

public class CurrentUserArgumentResolver implements WebArgumentResolver {
    @Override
    public Object resolveArgument(MethodParameter methodParameter, NativeWebRequest webRequest) {
        if(methodParameter.getParameterType() != null
                && methodParameter.getParameterType().equals(CurrentUser.class)){

// 判断controller方法参数有没有写当前用户，如果有，这里返回即可，通常我们从session里面取出来

            HttpServletRequest request = webRequest.getNativeRequest(HttpServletRequest.class);
            Object currentUser = request.getSession().getAttribute("CURRENT_USER");
            return currentUser;
        }
        return UNRESOLVED;
    }
}

2 配置<mvc:annotation-driven>

<mvc:annotation-driven>
        <mvc:argument-resolvers>  
            <bean class="com.xxx.CurrentUserArgumentResolver" />
        </mvc:argument-resolvers>
    </mvc:annotation-driven> 

这里的com.xxx.CurrentUserArgumentResolver就是上面写的CurrentUserArgumentResolver路径

3 测试

@RequestMapping(value = "save.json", method = RequestMethod.POST)
@ResponseBody
public String saveResource(CurrentUser user, Resource res){

// 这里的res，是从页面表单提交过来的

// 看看这里的user，是不是已经有值了？

System.out.println(user);

}

ok，是不是发现这个小招可以写到自己的框架里面呢