前言
项目中遇到这么一个使用场景,用户的登录信息给予token保存,在需要有登录信息的地方,每次都要去获取用户Id,但每次在请求方法中去获取用户信息,代码重复,冗余,很low于是想到了用@ModelAttribute 这个属性
使用场景
不用@ModelAttribute 时候在需要用户信息的请求中每次需要单独获取用户信息
1 String token = request.getAttribute("token").toString(); 2 User LoginUser = tokenService.decodeToken(token);
代码重复每次都需要单独去写,
于是我想到了去优化一下代码,在需要使用户信息的controller中写一个公共方法,每次直接获取就可以了
1private User gerUserInfo(HttpServletRequest request){ 2 String token = request.getAttribute("token").toString(); 3 User LoginUser = tokenService.decodeToken(token); 4 return LoginUser; 5 }
这样写代码是简化了一些,但是没什么特别大的改观,还是要在每个需求用户信息的请求Controller中调用此方法取获取用用户信息,如果多个Controller需要获取用户信息的话还需要重复写
也是想到继承,写一个公共的controller叫BaseController,每次在需要用户信息的controller中继承BaseController 然后在调用就可以了
1@RestController 2public class BaseController { 3 @Autowired 4 private TokenService tokenService; 5 6 7 private User gerUserInfo(HttpServletRequest request){ 8 String token = request.getAttribute("token").toString(); 9 User LoginUser = tokenService.decodeToken(token); 10 return LoginUser; 11 } 12}
这样看上去似乎比之前两种做法都简单便捷很多,在需要使用用户信息的controller中直接继承调用就可以啦,但是并没有根本解决我们的问题,我们还是需要写重复代码,在每个controller单独获取用户信息,这是最优嘛?并不是!!!
其实呢springboot提供@ModelAttribute这个注解属性使用这个通过参数注入就可获取啦
我们把上面的稍微调整一下如:
1@RestController 2public class BaseController { 3 @Autowired 4 private TokenService tokenService; 5 6 7 @ModelAttribute 8 public void userInfo(ModelMap modelMap, HttpServletRequest request) { 9 String token = request.getAttribute("token").toString(); 10 User LoginUser = tokenService.decodeToken(token); 11 12 modelMap.addAttribute("LoginUser", LoginUser); 13 modelMap.addAttribute("userId", LoginUser.getUserId()); 14 15 } 16} 17
然后在需要使用用户信息的controller中进行参数映射就行啦
1@ApiOperation(value = "用户快过期优惠卷信息",tags = "优惠卷接口") 2 @GetMapping("/expiredCoupon") 3 public List<Coupon> userExpiredCoupon(@ModelAttribute("userId") @ApiParam(hidden = true) String userId){ 4 return couponService.getUserExpiredCoupon(userId); 5 }
1@GetMapping("/info") 2 @ApiOperation("获取用户信息") 3 public User getUseInfo(@ModelAttribute("LoginUser") User user) { 4 return user; 5 }
这样用户信息通过形参直接注入到controller中,我们直接在请求中使用就可以啦
@ModelAttribute详解
- 被
@ModelAttribute注释的方法会在此controller每个方法执行前被执行 标注在方法上面的注解,将方法返回的对象存储在model中,该方法在这个控制器其他映射方法执行之前调用 @ModelAttribute注释一个方法的参数 从model中获取参数@ModelAttribute("LoginUser") User user参数user的值来源于BaseControlleruserInfo()方法中的model属性
具体更详细使用参考 @ModelAttribute注解的使用总结
