spring boot 最佳实践（八）-- 请求上下文注入

开发Web API时除了用户请求参数以外，还有一些和请求状态相关的信息，比如登陆用户，购物车商品，user-agent，IP等。通常做法是采用HttpSession或request.attribute来存这些对象.然后从Controller拿到HttpRequest一层层的调用。但在无状态web服务中没有session信息，在service中操作request也不利于单元测试和接口解耦。spring MVC为我们提供了自定义的方法参数注入接口，可以在需要的controller中为我们注入需要的请求上下文参数。

我们以登陆用户的注入为例演示实现方法。
如果我们使用spring security，登陆用户的信息被保存到http请求上下文中。所以我们需要实现WebArgumentResolver接口。

public class LoginUserArgumentResolver  implements WebArgumentResolver {

    @Override
    public Object resolveArgument(MethodParameter methodParameter, NativeWebRequest webRequest) throws Exception {
        HttpServletRequest request = webRequest.getNativeRequest(HttpServletRequest.class);
        if (methodParameter.getParameterType() == null || !methodParameter.getParameterType().equals(LoginUser.class)) {
            return UNRESOLVED;
        }
        return request.getAttribute("loginUser");
    }
}

我们需要在spring boot中配置这个参数处理类。

@Configuration
public class AppConfig  implements WebMvcConfigurer {
    @Override
    public void addArgumentResolvers(List<HandlerMethodArgumentResolver> argumentResolvers) {
        argumentResolvers.add(new ServletWebArgumentResolverAdapter(new LoginUserArgumentResolver()));
    }
}

由于WebMvcConfigurer要求注入的接口是HandlerMethodArgumentResolver，所以需要ServletWebArgumentResolverAdapter做封装。如果我们不需要依赖http请求上下文可以直接实现HandlerMethodArgumentResolver接口。

然后我们就可以在controller中愉快的使用LoginUser了。

@RequestMapping(value = "/logout",method = RequestMethod.POST)
public void logout(LoginUser loginUser){
        accountService.logout(loginUser);
}