聊聊自定义SPI如何使用自定义标签注入到spring容器中

前言

之前我们聊过自定义的SPI如何与spring进行整合，今天我们就来聊下如何通过自定义标签将spi对象注入到spring容器中

实现套路

1、自定义xsd

示例：

ps： 如果对xsd不熟悉的朋友，可以参考如下链接

https://www.w3school.com.cn/schema/index.asp

2、自定义解析BeanDefinitionParser解析器

示例：

public class AnnotationBeanDefinitionParser implements BeanDefinitionParser {

    private final Class beanClass;

    public AnnotationBeanDefinitionParser(Class beanClass) {
        this.beanClass = beanClass;
    }

    @Override
    public BeanDefinition parse(Element element, ParserContext parserContext) {

        String packageToScan = element.getAttribute("basePackages");
        String[] packagesToScan = trimArrayElements(commaDelimitedListToStringArray(packageToScan));

        RootBeanDefinition beanDefinition = new RootBeanDefinition();
        beanDefinition.setBeanClass(beanClass);
        beanDefinition.setLazyInit(false);
        beanDefinition.getConstructorArgumentValues().addIndexedArgumentValue(0,packagesToScan);
        String beanName = BeanUtils.generateBeanName(element,"id",parserContext,beanClass.getName());
        parserContext.getRegistry().registerBeanDefinition(beanName,beanDefinition);

        return beanDefinition;
    }


}

3、定义NamespaceHandler实现类处理自定义标签的处理器

示例：

public class SpiNamespaceHandler extends NamespaceHandlerSupport {

    @Override
    public void init() {
        registerBeanDefinitionParser("scan", new AnnotationBeanDefinitionParser(SpiAnnotationPostProcessor.class));
    }
}

4、将写入处理器、标签的位置写入spring.handlers、spring.schemas中

示例：

spring.handlers

http\://lybgeek.github.com/schema/spi=com.github.lybgeek.spring.schema.SpiNamespaceHandler

spring.schemas

http\://lybgeek.github.com/schema/spi/spi.xsd=META-INF/spi/spi.xsd

注： spring.handlers、spring.schemas需放置在resource/META-INF目录底下

示例演示

1、配置xml

2、在启动类上导入xml

@SpringBootApplication
@ImportResource(locations = "classpath:/spi.xml")
public class SpiTestXmlApplication {


    public static void main(String[] args) throws Exception{
        SpringApplication.run(SpiTestXmlApplication.class);
    }


}

3、验证SPI是否注入spring容器

 @Override
    public void run(ApplicationArguments args) throws Exception {

        applicationContext.getBeansOfType(SpringSqlDialect.class)
                .forEach((beanName,bean) -> System.out.println(beanName + "-->" + bean));
    }

控制台输入如下

springMysqlDialect-->com.github.lybgeek.dialect.mysql.SpringMysqlDialect@73041b7d
mysql-hello-->com.github.lybgeek.dialect.mysql.SpringMysqlDialect@574059d5
springOracleDialect-->com.github.lybgeek.dialect.oracle.SpringOracleDialect@4a50d04a

说明已经导入到spring容器中

总结

自从spring3+开始引入注解驱动后，在新项目基本上很少会使用xml，但如果是一些老旧的项目，大家如果想实现自定义标签注入到spring，就可以使用本文的方式。

套路就是如下

• 1、自定义xsd
• 2、自定义解析BeanDefinitionParser解析器
• 3、定义NamespaceHandler实现类处理自定义标签的处理器
• 4、将写入处理器、标签的位置写入spring.handlers、spring.schemas中

本文的实现也是相对简单，如果想深入使用，推荐看看dubbo自定义spring标签

demo链接

https://github.com/lyb-geek/springboot-learning/tree/master/springboot-spi-enhance/springboot-spi-framework-spring