springmvc如何实现自定义注解，如@RequestParam

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

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

定义注解：

@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 resolvers) {
        resolvers.add(new CurrentUserUserHandlerMethodArgumentResolver());
    }
}

测试

@RestController
public class UserController {

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

curl localhost:8080/userInfo?userId=1 

返回结果：{"username":"1","password":"1"}

根据此简单例子可以做类似的annotion