从-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 管理。

@Configurationpublic 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。

@Slf4jpublic 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的，写法很多，不再赘述。

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

@Slf4jpublic 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})@Documentedpublic @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 {    }}@Slf4jpublic class AnnotationManager {    public AnnotationManager() {        init();    }    public void init(){        log.info("AnnotationManager start");    }}

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

@SpringBootApplication@TestAnnotationpublic 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 start2022-06-21 19:05:07.456  INFO 4598 --- [           main] c.a.testInfra.manager.AiManager          : AiManager start2022-06-21 19:05:07.456  INFO 4598 --- [           main] c.a.testInfra.manager.AnnotationManager  : AnnotationManager start

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