添加参数解析器

添加参数解析器

在项目开发中，发现在controller层实现多个方法时，要传入的参数各不相同，可能需要对这些参数进行某些操作（比如通过request获取token，然后根据token获取对应的user对象），为了减少代码冗余，可以重写WebMvcConfigurerAdapter中的addArgumentResolvers(List<HandlerMethodArgumentResolver> argumentResolvers);方法来给argumentResolvers添加自定义的参数解析器。

• 自定义参数解析器UserArgumentResolver，要实现HandlerMethodArgumentResolver接口，并实现它的两个方法。

@Service
public class UserArgumentResolver implements HandlerMethodArgumentResolver {

	@Autowired
	MiaoshaUserService userService;
	//判断传入参数是否为需要处理类型
	public boolean supportsParameter(MethodParameter parameter) {
		Class<?> clazz = parameter.getParameterType();
		return clazz==MiaoshaUser.class;
	}
	//对参数进行解析操作、这段代码被重用
	public Object resolveArgument(MethodParameter parameter, ModelAndViewContainer mavContainer,
			NativeWebRequest webRequest, WebDataBinderFactory binderFactory) throws Exception {
		HttpServletRequest request = webRequest.getNativeRequest(HttpServletRequest.class);
		HttpServletResponse response = webRequest.getNativeResponse(HttpServletResponse.class);
		
		//有两种方式(移动端、PC端)，一种以参数传、一种以cookie传
		String paramToken = request.getParameter(MiaoshaUserService.COOKI_NAME_TOKEN);
		String cookieToken = getCookieValue(request, MiaoshaUserService.COOKI_NAME_TOKEN);
		if(StringUtils.isEmpty(cookieToken) && StringUtils.isEmpty(paramToken)) {
			return null;
		}
		String token = StringUtils.isEmpty(paramToken)?cookieToken:paramToken;
		return userService.getByToken(response, token);
	}

	private String getCookieValue(HttpServletRequest request, String cookiName) {
		Cookie[]  cookies = request.getCookies();
		if(cookies==null||cookies.length<=0){
			return null;
		}
		for(Cookie cookie : cookies) {
			if(cookie.getName().equals(cookiName)) {
				return cookie.getValue();
			}
		}
		return null;
	}

}

• 将自定义的参数解析器添加到WebMvcConfigurerAdapter中，记得要@Configuration注解

@Configuration
public class WebConfig  extends WebMvcConfigurerAdapter{
	
	@Autowired
	UserArgumentResolver userArgumentResolver;
	
	@Override
	public void addArgumentResolvers(List<HandlerMethodArgumentResolver> argumentResolvers) {
		//添加到参数解析器队列中
		argumentResolvers.add(userArgumentResolver);
	}
}

• 在controller层中的简单应用

@Controller
@RequestMapping("/goods")
public class GoodsController {

    @RequestMapping("/to_list")
    public String list(Model model,MiaoshaUser user) {
    	//直接通过参数解析器获得需要的user
    	//而不用通过request中的token获取后再在redis中获取对应的user返回
    	model.addAttribute("user", user);
        return "goods_list";
    }
    
}

理解：
由于前端传递过来的数据都是先通过Controller层进行处理，
比如从request中的cookie获取token，根据token获取redis中的用户信息，
可以将这个操作独立出来，作为一个参数解析器（HandlerMethodArgumentResolver），然后每次controller层有传入相关参数的方法，都会先通过这个解析器进行解析返回所指定的参数，然后直接使用就行了。这个过程是service部分，也就是相当于比控制层更早的处理到数据然后返回给控制层。