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

本文涉及的产品
注册配置 MSE Nacos/ZooKeeper,118元/月
性能测试 PTS,5000VUM额度
Serverless 应用引擎免费试用套餐包,4320000 CU,有效期3个月
简介: 别人都写从0开始实现xxx,我先从-1开始就显得更牛逼一些。今天,先开个头,来教大家怎么实现一个中间件。

新建项目


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

fccb9f580ee655f7d67f03902942888a.jpg项目包含两个模块,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

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


相关文章
|
消息中间件 JSON 中间件
你用Go写过中间件吗?带你用Go实现【操作日志中间件】
管理后台所有修改,添加,删除的操作都要记录;操作日志的统计不影响主程序的性能
557 6
你用Go写过中间件吗?带你用Go实现【操作日志中间件】
|
中间件 uml
阿里中间件seata源码剖析六:TCC模式中2阶段提交实现
阿里中间件seata源码剖析六:TCC模式中2阶段提交实现
380 3
阿里中间件seata源码剖析六:TCC模式中2阶段提交实现
|
中间件 程序员 Go
你用Go写过中间件吗?带你用Gin实现【用户角色权限管理中间件】
管理后台有超管权限,超管拥有所有权限;普通管理员可以设置角色,角色单选;角色可以赋予多个权限,权限多选;这样我们就实现了对普通管理员的角色和权限的灵活管理
341 7
你用Go写过中间件吗?带你用Gin实现【用户角色权限管理中间件】
|
中间件 Python
Python编程:Django中间件实现登陆验证
Python编程:Django中间件实现登陆验证
191 6
Python编程:Django中间件实现登陆验证
|
消息中间件 设计模式 缓存
天天造轮子第七天 - 中间件实现 - Compose 的 N 种姿势
天天造轮子第七天 - 中间件实现 - Compose 的 N 种姿势
224 5
|
JavaScript 中间件
两分钟搞懂从函数组合到中间件实现
很多JS框架中都会用到中间件,形成一个洋葱圈结构,这样可以很好的应付业务逻辑中的对于切面描述的需要。 经典案例比如Redux 中间件和Koa 中间件
204 3
|
存储 分布式计算 监控
阿里云互联网中间件:让企业实现业务云化持续创新|学习笔记
快速学习 阿里云互联网中间件:让企业实现业务云化持续创新
215 7
|
编解码 运维 监控
电商直播平台如何借助容器与中间件实现研发效率提升100%?
经过实际场景验证及用户的综合评估,电商直播平台借助全面的云原生容器化能力和中间件产品能力,大幅提升开发部署运维效率达50%~100%,极大地提升了用户体验,为业务持续发展打下了坚实的基础。
7249 3
电商直播平台如何借助容器与中间件实现研发效率提升100%?
|
存储 JavaScript NoSQL
SpringBoot2 整合MinIO中间件,实现文件便捷管理
MinIO是一个非常轻量的服务,可以很简单的和其他应用的结合,类似 NodeJS, Redis 或者 MySQL。
770 6
SpringBoot2 整合MinIO中间件,实现文件便捷管理
|
消息中间件 Prometheus Cloud Native
阿里云原生中间件首次实现自研、开源、商用“三位一体”,技术飞轮效应显现
阿里云在探索中一直存在的苦恼,是内部的自研体系、商业化的产品技术与开源的项目,三方的技术路线一直没有机会融为一体。然而,就在今年阿里云提出了“三位一体”理念,即将“自研技术”、“开源项目”、“商业产品”形成统一的技术体系,最大化技术的价值。
阿里云原生中间件首次实现自研、开源、商用“三位一体”,技术飞轮效应显现