在spring mvc中自定义参数解析器

HandlerMethodArgumentResolver定义

这是Spring mvc提供的一个参数解析器接口，我们可以将HandlerMethodArgumentResolver理解为是一个参数解析器，我们可以通过写一个类实现HandlerMethodArgumentResolver接口来实现对Controller层中方法参数的修改。

public interface HandlerMethodArgumentResolver {
    boolean supportsParameter(MethodParameter var1);

    @Nullable
    Object resolveArgument(MethodParameter var1, @Nullable ModelAndViewContainer var2, NativeWebRequest var3, @Nullable WebDataBinderFactory var4) throws Exception;
}

接口里有两个方法

supportsParameter

返回值是boolean类型，它的作用是判断Controller层中的参数，是否满足条件，满足条件则执行resolveArgument方法，不满足则跳过。

resolveArgument

它只有在supportsParameter方法返回true的情况下才会被调用。用于处理一些业务，将返回值赋值给Controller层中的这个参数。

使用方法

实现接口

写一个类实现HandlerMethodArgumentResolver接口

/**
 * 有@LoginUser注解的方法参数，注入当前登录用户
 * @author zhangyuxuan
 * @date 2017-03-23 22:02
 */
@Component
public class LoginUserHandlerMethodArgumentResolver implements HandlerMethodArgumentResolver {
    @Autowired
    private ApiUserInfoService apiUserInfoService; 
    
    @Autowired
    private IOrgInfoService orgInfoService;

    @Override
    public boolean supportsParameter(MethodParameter parameter) {
        return parameter.getParameterType().isAssignableFrom(UserInfo.class) && parameter.hasParameterAnnotation(LoginUser.class);
    }

    @Override
    public Object resolveArgument(MethodParameter parameter, ModelAndViewContainer container,
                                  NativeWebRequest request, WebDataBinderFactory factory) throws Exception {
        //获取用户ID
        Object object = request.getAttribute("login_name", RequestAttributes.SCOPE_REQUEST);
        if(object == null){
            return null;
        }

        //获取用户信息
        UserInfo user = apiUserInfoService.queryByLoginName((String)object);
        if(user != null) {
        	OrgInfo orgInfo = orgInfoService.selectOrgInfoById(user.getOrgId());
        	user.setOrgCode(orgInfo.getOrgCode());
        }

        return user;
    }
}

将自定义的处理器注册进Spring mvc

@Configuration
public class WebConfig  extends WebMvcConfigurerAdapter{
	@Autowired
	UserArgumentResolver userArgumentResolver;
	@Override
	public void addArgumentResolvers(List<HandlerMethodArgumentResolver> argumentResolvers) {
		argumentResolvers.add(userArgumentResolver);
	}
}

自定义一个注解（可选）

如果不需要用注解判断，可以不加，直接使用参数类型判断

/**
 * 登录用户信息
 *
 * @author zhangyuxuan
 * @date 2017-03-23 20:39
 */
@Target(ElementType.PARAMETER)
@Retention(RetentionPolicy.RUNTIME)
public @interface LoginUser {

}

在控制类添加参数

这样就可以修改符合条件的参数了

	@PostMapping("/list")
    public R list(@LoginUser UserInfo loginUser, @RequestBody JSONObject data) {
        return R.ok();
    }

如果不用注解判断是否符合，可以不加自定义注解