新建项目
首先,我们新建一个多 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
好了,就这样,我猜,没人需要这个源码吧?为了后面的文章,先写个这个铺垫一下,结束。