基于FeignClient的Controller自动生成注入机制

Posted 2023-03-11

参考技术A

目前公司后端服务项目结构为，xxx-api、xxx-service 其中 api模块定义了此微服务模块对外暴露的接口，service模块实现了api模块定义的接口并创建了对应供调用的Controller。所以实际来说Controller就是一个代理类，用来代理实现Service模块真正对外的HTTP请求响应。那么我们是不是可以通过生成Controller代理类然后注入到容器的方式来代替手写Controller来自动实现这些代理类了？

能不能实现了？显然是可以的，实现的首要条件是：1）请求规则如何定义、2）如何注入Controller到Spring IOC容器中，下面我们来一一解答。

请求规则如何定义？

我们在使用微服务调用时，实际都是通过Feign来进行实际使用的，所以请求路径实际上来说在基于Feign的接口上都已经通过注解标明了信息。如：

如何注入Controller到Spring IOC容器中？

这里我们首先要了解Spring MVC的运行机制，实际的请求处理是通过DispatcherServlet映射给注册到HandlerMapping中的Handler进行处理的，这个时候我们可以发现通过注册Handler到HandlerMapping 即可实现这个注入逻辑了。在Spring Boot中默认会注册七个HandlerMapping，我们只需要关注 RequestMappingHandlerMapping 和 SimpleUrlHandlerMapping，更详细的内容可自行查看 WebMvcAutoConfiguration 类。因为接口定义的关系这里我们只能通过注册 Handler 到 RequestMappingHandlerMapping 来实现注册逻辑。

如何注册 Handler 到 RequestMappingHandlerMapping？

这里一个核心的问题就是Spring MVC默认是通过获取类和方法的注解信息创建 RequestMappingInfo 然后将 RequestMappingInfo、Handler实例以及方法注册到 HandlerMapping 中，他实际的过程是全自动的，通过获取IOC容器中注解了 @Controller 和 @RequestMapping 的Bean，直接解析然后注册进去(AbstractHandlerMethodMaping#detectHandlerMethods)，我们要通过动态代理生成对应Bean，那么就没有了对应的类层级的 @RequestMapping 注解了，所以通过手动创建 RequestMappingInfo 然后手动调用方法注册的方式进行注册了(AbstractHandlerMethodMaping#registerHandlerMethod)。

方案1，手动生成信息注册的方案：

方案2，通过继承 RequestMappingInfoHandlerMapping 的方式来实现：

这两个方案都要注意扫描范围，不然会将引入的模块也生成了Handler。
方案2比较简单，花了一点点时间实现了方案2，目前已应用在项目中， 源码见此地址 。