@RequestParam等参数绑定注解是怎么实现的？自定义参数绑定注解的妙用

我欲只争朝夕

发布于 2023-11-1 11:47

浏览

0收藏

介绍

SpringMVC参数绑定的注解有很多，如@RequestParam，@RequestBody，@PathVariable，@RequestHeader，@CookieValue等。这些注解的实现方式很类似，都是有一个对应的解析器，解析完返回一个对象，放在方法的参数上。对参数绑定注解不熟悉的看推荐阅读

如@RequestParam的解析器为RequestParamMethodArgumentResolver，@RequestBody的解析器为PathVariableMethodArgumentResolver等，这些解析器的原理超级简单。我先写一个自己的解析器，你大概就能猜到这些解析器的工作原理了。假如说现在有一个场景，前端每次从前面传入一个userId，而后端呢，每次都去查数据库，然后拿到用户的信息。如果有很多个controller，每个controller上来都是一样的逻辑，去查数据库，然后拿用户信息，这样的代码就很烂。如何精简呢？答案就是自定义注解实现参数绑定

如下代码的版本为Spring Boot 2.0.4.RELEASE

定义注解

@Target(ElementType.PARAMETER)
@Retention(RetentionPolicy.RUNTIME)
public @interface CurrentUser {

}

实现解析器

public class CurrentUserUserHandlerMethodArgumentResolver implements HandlerMethodArgumentResolver {

    /** 用于判定是否需要处理该参数分解，返回true为需要，并会去调用下面的方法resolveArgument。*/
    @Override
    public boolean supportsParameter(MethodParameter parameter){
        return parameter.getParameterType().isAssignableFrom(User.class) && parameter.hasParameterAnnotation(CurrentUser.class);
    }

    /** 真正用于处理参数分解的方法，返回的Object就是controller方法上的形参对象。*/
    @Override
    public Object resolveArgument(MethodParameter parameter, ModelAndViewContainer container,
                                  NativeWebRequest request, WebDataBinderFactory factory) throws Exception {

        String userId = request.getParameter("userId");

        // 为了方便不模拟查数据库的过程，直接new一个username和password都是userId的对象
        return new User(userId, userId);
    }
}

加入配置

@Configuration
public class WebConfig implements WebMvcConfigurer {

    @Override
    public void addArgumentResolvers(List<HandlerMethodArgumentResolver> resolvers){
        resolvers.add(new CurrentUserUserHandlerMethodArgumentResolver());
    }
}

测试controller

@RestController
public class UserController {

    @RequestMapping(value = "userInfo", method = RequestMethod.GET)
    public Map<String, String> userInfo(@CurrentUser User user){
        Map<String, String> result = new HashMap<>();
        result.put("username", user.getUsername());
        result.put("password", user.getPassword());
        return result;
    }
}