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

别人都写从0开始实现xxx，我先从-1开始就显得更牛逼一些。

今天，先开个头，来教大家怎么实现一个中间件。

新建项目

首先，我们新建一个多 module 的项目用于测试。

项目包含两个模块，test-infra用户中间件模块的开发，demo用于测试。

<?xml version="1.0" encoding="UTF-8"?> <project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"          xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd">     <modelVersion>4.0.0</modelVersion>     <parent>         <groupId>org.springframework.boot</groupId>         <artifactId>spring-boot-starter-parent</artifactId>         <version>2.7.0</version>         <relativePath/> <!-- lookup parent from repository -->     </parent>     <groupId>com.aixiaoxian.infra</groupId>     <artifactId>aixiaoxian-infra</artifactId>     <version>0.0.1-SNAPSHOT</version>      <name>aixiaoxian-infra</name>     <description>aixiaoxian-infra</description>     <packaging>pom</packaging>      <properties>         <java.version>1.8</java.version>     </properties>      <modules>         <module>demo</module>         <module>test-infra</module>     </modules>      <dependencies>      </dependencies>      <build>         <plugins>             <!-- Source -->             <plugin>                 <artifactId>maven-source-plugin</artifactId>                 <inherited>true</inherited>                 <executions>                     <execution>                         <phase>package</phase>                         <goals>                             <goal>jar-no-fork</goal>                         </goals>                     </execution>                 </executions>             </plugin>         </plugins>     </build>  </project> 

开发中间件

项目创建 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模块中，启动测试即可。

 <dependency>    <groupId>com.aixiaoxian.infra</groupId>    <artifactId>test-infra</artifactId>    <version>0.0.1-SNAPSHOT</version>  </dependency> 

换个姿势

我们换个姿势去创建 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.433  INFO 4598 --- [           main] c.a.testInfra.manager.TestManager        : TestManager start 2022-06-21 19:05:07.456  INFO 4598 --- [           main] c.a.testInfra.manager.AiManager          : AiManager start 2022-06-21 19:05:07.456  INFO 4598 --- [           main] c.a.testInfra.manager.AnnotationManager  : AnnotationManager start 

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