从-1开始实现一个中间件从-1开始实现一个中间件

别人都写从0开始实现xxx，我先从-1开始就显得更牛逼一些。
今天，先开个头，来教大家怎么实现一个中间件。
新建项目首先，我们新建一个多 module 的项目用于测试。

文章图片

【从-1开始实现一个中间件】项目包含两个模块，test-infra用户中间件模块的开发，demo用于测试。

4.0.0org.springframework.boot spring-boot-starter-parent 2.7.0 com.aixiaoxian.infra aixiaoxian-infra 0.0.1-SNAPSHOTaixiaoxian-infra aixiaoxian-infrapom1.8 demo test-infra maven-source-plugin true package jar-no-fork

开发中间件项目创建 OK 了，接着开始开发一个最最最简单的中间件。
在resources目录下创建META-INFA/spring.factories文件，用于自动装配，别问我啥是自动装配，然后配置一个自动装配类。

org.springframework.boot.autoconfigure.EnableAutoConfiguration=\ com.aixiaoxian.testInfra.config.TestConfiguration

实现 TestConfiguration，最简单的方式，直接使用@Bean注解声明一个 Bean 交给 Spring 管理。

@Configuration public class TestConfiguration implements BeanDefinitionRegistryPostProcessor, EnvironmentAware { private Environment environment; @Bean public TestManager getTestManager() { return new TestManager(); }@Override public void setEnvironment(Environment environment) { this.environment = environment; }}

然后实现真正的中间件逻辑的处理部分TestManager。

@Slf4j public class TestManager {public TestManager() { init(); }public void init(){ log.info("TestManager start"); } }

这样的话，一个最简单的中间件就开发好了，直接把他添加到demo模块中，启动测试即可。

com.aixiaoxian.infra test-infra 0.0.1-SNAPSHOT

换个姿势我们换个姿势去创建 Bean，使用BeanDefinitionRegistryPostProcessor，让 TestConfiguration 去实现它，重写postProcessBeanDefinitionRegistry，注册一个新的 Bean aiManager，这样也是 OK的，写法很多，不再赘述。

@Override public void postProcessBeanDefinitionRegistry(BeanDefinitionRegistry beanDefinitionRegistry) throws BeansException { BeanDefinitionBuilder beanDefinitionBuilder = BeanDefinitionBuilder.genericBeanDefinition(AiManager.class); beanDefinitionBuilder.addConstructorArgValue(this.environment); beanDefinitionRegistry.registerBeanDefinition("aiManager", beanDefinitionBuilder.getBeanDefinition()); }

@Slf4j public class AiManager { private Environment environment; public AiManager(Environment environment) { this.environment = environment; init(); }public void init(){ log.info("AiManager start"); } }

再换个姿势对于自动装配创建 Bean 有了基本的了解，那如果我想声明一个注解给别人用该怎么做？
首先创建一个注解，注意我使用了@Import注解，TestImportSelector 实现TestImportSelector接口。

@Target({ElementType.TYPE}) @Retention(RetentionPolicy.RUNTIME) @Import({TestImportSelector.class}) @Documented public @interface TestAnnotation { }public class TestImportSelector implements DeferredImportSelector { @Override public String[] selectImports(AnnotationMetadata annotationMetadata) { return new String[]{AnnotationConfiguration.class.getName()}; } }

AnnotationConfiguration 写法也很简单了，这样也实现了自动装配，当然了你要是用上面的写法也能达到一样的效果，但是建议这样写，别问，问就是这样。

public class AnnotationConfiguration implements BeanDefinitionRegistryPostProcessor {@Override public void postProcessBeanDefinitionRegistry(BeanDefinitionRegistry beanDefinitionRegistry) throws BeansException { BeanDefinitionBuilder beanDefinitionBuilder = BeanDefinitionBuilder.genericBeanDefinition(AnnotationManager.class); beanDefinitionRegistry.registerBeanDefinition("annotationManager", beanDefinitionBuilder.getBeanDefinition()); }@Override public void postProcessBeanFactory(ConfigurableListableBeanFactory configurableListableBeanFactory) throws BeansException {} }@Slf4j public class AnnotationManager {public AnnotationManager() { init(); }public void init(){ log.info("AnnotationManager start"); } }

最后在demo启动类上打上我们这个注解。

@SpringBootApplication @TestAnnotation public class DemoApplication {public static void main(String[] args) { SpringApplication.run(DemoApplication.class, args); } }

最后我们可以看到输出：

2022-06-21 19:05:07.433INFO 4598 --- [main] c.a.testInfra.manager.TestManager: TestManager start 2022-06-21 19:05:07.456INFO 4598 --- [main] c.a.testInfra.manager.AiManager: AiManager start 2022-06-21 19:05:07.456INFO 4598 --- [main] c.a.testInfra.manager.AnnotationManager: AnnotationManager start

好了，就这样，我猜，没人需要这个源码吧？为了后面的文章，先写个这个铺垫一下，结束。