从-1开始实现一个中间件
艾小仙
共 9948字，需浏览 20分钟
·
2022-06-26 10:23
别人都写从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
好了，就这样，我猜，没人需要这个源码吧？为了后面的文章，先写个这个铺垫一下，结束。
评论