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

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);  }}