spring源码学习之：xml配置文件标签自定义

2022-07-10 14:36:54

Spring框架从2.0版本开始，提供了基于Schema风格的XML扩展机制，允许开发者扩展最基本的spring配置文件（一般是classpath下的spring.xml）。试想一下，如果我们直接在spring.xml中加入一个自定义标签<mytag id="aty"></matag>，会发生什么呢？spring框架启动的时候会报错，因为spring根本不认识我们自定义的& lt;mytag>，这样对spring.xml的校验就会失败，最终结果就是框架不能启动。有什么方法，能够让spring认识并加载解析我们自定义的<mytag>呢？这就是spring提供的xml扩展机制。我们可以在spring.xml中加入自己的标签，之后spring会帮我们解析并纳入自己的管理范围内，这也就是说我们扩展了spring的功能。

现在我们来看下怎么实现这个功能，可以参考spring帮助文档中的extensible-xml.html。我们知道如果在需要在spring.xml中配置数据源，需要进行如下的配置：

<bean id="dataSource" class="org.apache.commons.dbcp.BasicDataSource">

    <property name="driverClassName" value="com.mysql.jdbc.Driver" />

    <property name="url" value="jdbc:mysql://localhost:3309/sampledb" />

    <property name="username" value="root" />

    <property name="password" value="1234" />

</bean>

这种方式配置虽然也比较简单，但是有一个缺点：使用<property>标签不够明显，不如元素属性那么直接。现在我们希望在spring.xml中做如下的配置，就能够完成数据源的配置。

<aty:datasource id="myDataSourcce" url="jdbc:mysql://localhost:3309/demodb" userName="root" password="root" />

这种方式比较直接，配置不容易出错。如果让spring能够解析这个标签，需要4步。

1、提供一个xsd文件，负责对xml的标签<datasource>进行校验

<?xml version="1.0" encoding="UTF-8"?>

<xsd:schema xmlns="http://www.aty.com/schema/aty" xmlns:xsd="http://www.w3.org/2001/XMLSchema"

  xmlns:beans="http://www.springframework.org/schema/beans"

  targetNamespace="http://www.aty.com/schema/aty" elementFormDefault="qualified"

  attributeFormDefault="unqualified">

  <xsd:import namespace="http://www.springframework.org/schema/beans" />

  <xsd:element name="datasource">

    <xsd:complexType>

      <xsd:complexContent>

        <xsd:extension base="beans:identifiedType">

          <xsd:attribute name="url" type="xsd:string" use="required" />

          <xsd:attribute name="userName" type="xsd:string" use="required" />

          <xsd:attribute name="password" type="xsd:string" use="required" />

        </xsd:extension>

      </xsd:complexContent>

    </xsd:complexType>

  </xsd:element>

</xsd:schema>

2、定义一个BeanDefinitionParser负责解析xml,并将必要的信息放入spring中

package net.aty.custom.define;

import net.aty.custom.cfg.DataSourceInfo;

import org.springframework.beans.factory.config.BeanDefinition;

import org.springframework.beans.factory.config.BeanDefinitionHolder;

import org.springframework.beans.factory.support.BeanDefinitionReaderUtils;

import org.springframework.beans.factory.support.RootBeanDefinition;

import org.springframework.beans.factory.xml.BeanDefinitionParser;

import org.springframework.beans.factory.xml.ParserContext;

import org.w3c.dom.Element;

public class DatasourceBeanDefinitionParser implements BeanDefinitionParser

{

  public BeanDefinition parse(Element element, ParserContext context)

  {

    RootBeanDefinition def = new RootBeanDefinition();

    // 设置Bean Class

    def.setBeanClass(DataSourceInfo.class);

    // 注册ID属性

    String id = element.getAttribute("id");

    BeanDefinitionHolder idHolder = new BeanDefinitionHolder(def, id);

    BeanDefinitionReaderUtils.registerBeanDefinition(idHolder,

        context.getRegistry());

    // 注册属性

    String url = element.getAttribute("url");

    String userName = element.getAttribute("userName");

    String password = element.getAttribute("password");

    BeanDefinitionHolder urlHolder = new BeanDefinitionHolder(def, url);

    BeanDefinitionHolder userNameHolder = new BeanDefinitionHolder(def,

        userName);

    BeanDefinitionHolder passwordHolder = new BeanDefinitionHolder(def,

        password);

    BeanDefinitionReaderUtils.registerBeanDefinition(urlHolder,

        context.getRegistry());

    BeanDefinitionReaderUtils.registerBeanDefinition(userNameHolder,

        context.getRegistry());

    BeanDefinitionReaderUtils.registerBeanDefinition(passwordHolder,

        context.getRegistry());

    def.getPropertyValues().addPropertyValue("url", url);

    def.getPropertyValues().addPropertyValue("userName", userName);

    def.getPropertyValues().addPropertyValue("password", password);

    return def;

  }

}

该类的功能：设置相关的BeanClass，解析了对应的xsd文件，并将解析的内容注册到上下文中，同时返回一个BeanDefinition对象（BeanDefinition是Spring的bean定义，提供了bean部分的操作方法，如isSingleton()、isLazyInit() 等)。注意：id属性是一个默认的属性，可以不在xsd文件中描述，但是需要注册它，否则将无法通过getBean方法获取标签定义的bean，也无法被其他bean引用。

3、定义个NamespaceHandler,由sping框架的调用入口。这也是我们自定义xml解析的入口

package net.aty.custom.define;

import org.springframework.beans.factory.xml.NamespaceHandlerSupport;

public class DatasourceNamespaceHandlerSupport extends NamespaceHandlerSupport

{

  @Override

  public void init()

  {

    registerBeanDefinitionParser("datasource",

        new DatasourceBeanDefinitionParser());

  }

}

4、配置schema和handler

Spring没那么聪明，它无法知道我们在什么地方定义了哪些扩展标签，这些标签将被谁解析，怎么解析。这个过程要我们通过一些配置文件来告知Spring知道，它们就是spring.handlers和spring.schemas，它们放在META-INF目录中。 Spring.jar的META-INF目录中也有同名的文件，它们的文件内容基本上是相似的，而Spring在执行过程中，如果发现其他jar文件的 META-INF文件夹中包含有这两个文件，Spring将会合并它们。

spring.handlers内容如下:

http\://www.aty.com/schema/aty=net.aty.custom.define.DatasourceNamespaceHandlerSupport

spring.schemas内容如下:

http\://www.aty.com/schema/aty.xsd=aty.xsd

我的工程目录结构如下图：

测试工程的spring.xml配置如下：

<?xml version="1.0" encoding="UTF-8"?>

<beans xmlns="http://www.springframework.org/schema/beans"

  xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"

  xmlns:aty="http://www.aty.com/schema/aty"

  xsi:schemaLocation="http://www.springframework.org/schema/beans

  http://www.springframework.org/schema/beans/spring-beans-3.1.xsd

  http://www.aty.com/schema/aty

  http://www.aty.com/schema/aty.xsd">

  <aty:datasource id="myDataSourcce" url="jdbc:mysql://localhost:3309/demodb" userName="root" password="root" />

</beans>

测试类代码如下：

import net.aty.custom.cfg.DataSourceInfo;

import org.springframework.context.support.ClassPathXmlApplicationContext;

public class TestMain

{

  private static ClassPathXmlApplicationContext context = new ClassPathXmlApplicationContext(

      "spring.xml");

  public static void main(String[] args)

  {

    DataSourceInfo d = (DataSourceInfo) context.getBean("myDataSourcce");

    System.out.println(d);

  }

}

测试的工程目录结构如下：

码农公寓

相关文章