Spring IOC容器源码解析系列，建议大家按顺序阅读，欢迎讨论

(spring源码均为4.1.6.RELEASE版本)

前言

在spring的xml配置文件中，当我们想使用spring-context或者spring-aop的配置时，都是有前缀的标签。如：

1<context:component-scan/>
2<aop:config/>

而普通的<bean>标签则是没有前缀的，为什么会有这样的区别呢？

因为spring的xml文件默认的名称空间(namespace)是http://www.springframework.org/schema/beans，只支持bean，import，alias等基本标签，而要支持context,aop,mvc等其他名称空间,就需要在xml文件中单独声明。

1<beans xmlns="http://www.springframework.org/schema/beans"
2xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
3xmlns:context="http://www.springframework.org/schema/context"
4xmlns:aop="http://www.springframework.org/schema/aop"
5xsi:schemaLocation="
6http://www.springframework.org/schema/beans
7http://www.springframework.org/schema/beans/spring-beans-3.0.xsd
8http://www.springframework.org/schema/context
9http://www.springframework.org/schema/context/spring-context-3.0.xsd
10http://www.springframework.org/schema/aop 
11http://www.springframework.org/schema/aop/spring-aop-3.0.xsd">

那么如果想自定义xml标签，用来支持一些特殊的功能或者集成一个框架到spring中去，应该怎么去实现呢？

就是通过NamespaceHandler接口。来看他的定义：

Base interface used by the DefaultBeanDefinitionDocumentReader for handling custom namespaces in a Spring XML configuration file.

DefaultBeanDefinitionDocumentReader中处理xml配置文件中自定义命名空间的基本接口

接口中定义了三个方法，一般常用的只有两个

1public interface NamespaceHandler {
2
3	// 初始化操作，一般都是注册子标签的解析器BeanDefinitionParser
4	void init();
5
6	// 解析标签的具体操作
7	BeanDefinition parse(Element element, ParserContext parserContext);
8}

下面通过一个demo来看如何使用自定义xml标签

自定义xml标签

我们想实现一个类似于<bean>标签功能的自定义标签，像这样的：

<sample:entity id="sampleBean" class="com.lcifn.spring.demo.namespace.SampleBean"/>

id就是bean的名称，class是bean的全路径包名。然后通过ApplicationContext.getBean("sampleBean")就可以获取bean的实例对象。

首先要写一个sample.xsd文件，用来定义xml的组织结构。

1<?xml version="1.0" encoding="UTF-8"?>
2<xsd:schema xmlns="http://www.lcifn.com/sample"
3		xmlns:xsd="http://www.w3.org/2001/XMLSchema"
4		targetNamespace="http://www.lcifn.com/sample"
5		elementFormDefault="qualified"
6		attributeFormDefault="unqualified">
7    <xsd:element name="entity">
8        <xsd:complexType>
9    		<xsd:attribute name="id" type="xsd:string" />
10    		<xsd:attribute name="class" type="xsd:string" />
11    	</xsd:complexType>
12    </xsd:element>
13</xsd:schema>

这里声明了命名空间为http://www.lcifn.com/sample，并定义了元素entity,以及它的两个属性id和class。

然后在classpath下创建一个META-INF文件夹，并创建spring.handlers和spring.schemas两个文件，这两个文件是模仿spring自己的jar包下的写法，打开spring-beans或者spring-context的jar包，都会有一个META-INF的文件夹，其中就有spring.handlers和spring.schemas两个文件。

spring.schemas用来定义xml文件schemaLocation声明的xsd和本地xsd文件的关系，比如spring中context的schemaLocation如下

1xsi:schemaLocation="
2http://www.springframework.org/schema/context
3http://www.springframework.org/schema/context/spring-context-3.0.xsd"

通常解析xml是会去请求http://www.springframework.org/schema/context/spring-context-3.0.xsd这个远程文件，但往往程序运行时未必能访问网络，如果可以访问本地的xsd文件就可以解决这个问题。于是就有了spring.schemas文件。比如spring-context的jar包下的spring.schemas定义了context的xsd文件关系。

http\://www.springframework.org/schema/context/spring-context-4.1.xsd=org/springframework/context/config/spring-context-4.1.xsd

这里的value值时classpath下本地xsd文件的包路径名。

刚刚写的sample.xsd我放在了com.lcifn.spring.demo.xsd包路径下，因此我们的spring.schemas如下：

http\://www.lcifn.com/sample.xsd=com/lcifn/spring/demo/xsd/sample.xsd

spring.handlers是spring定义的xml命名空间和其对应的处理器的映射，比如context的：

http\://www.springframework.org/schema/context=org.springframework.context.config.ContextNamespaceHandler

ContextNamespaceHandler继承了NamespaceHandlerSupport类，用来对context命名空间的xml标签进行解析，并与spring ioc容器进行交互(如注册BeanDefinition等)。因此我们也需要定义一个NamespaceHandlerSupport的子类SampleNamespaceHandler,并在spring.handlers中与sample的命名空间进行关联。

SampleNamespaceHandler.java

1package com.lcifn.spring.demo.namespace;
2
3import org.springframework.beans.factory.xml.NamespaceHandlerSupport;
4
5public class SampleNamespaceHandler extends NamespaceHandlerSupport{
6
7	public void init() {
8		registerBeanDefinitionParser("entity", new EntityBeanDefinitionParse());
9	}
10
11}

实现init方法，注册entity标签的解析器EntityBeanDefinitionParse。

EntityBeanDefinitionParse.java

1package com.lcifn.spring.demo.namespace;
2
3import org.springframework.beans.factory.config.BeanDefinition;
4import org.springframework.beans.factory.config.BeanDefinitionHolder;
5import org.springframework.beans.factory.support.BeanDefinitionReaderUtils;
6import org.springframework.beans.factory.support.RootBeanDefinition;
7import org.springframework.beans.factory.xml.BeanDefinitionParser;
8import org.springframework.beans.factory.xml.ParserContext;
9import org.w3c.dom.Element;
10
11public class EntityBeanDefinitionParse implements BeanDefinitionParser{
12
13	public BeanDefinition parse(Element element, ParserContext parserContext) {
14		String id = element.getAttribute("id");
15		String beanClassName = element.getAttribute("class");
16		BeanDefinition definition = new RootBeanDefinition();
17		definition.setBeanClassName(beanClassName);
18		BeanDefinitionReaderUtils.registerBeanDefinition(new BeanDefinitionHolder(definition, id), parserContext.getRegistry());
19		return definition;
20	}
21}

EntityBeanDefinitionParse实现spring的BeanDefinitionParser接口，在parse方法里获取entity的id和class属性，并实例化一个RootBeanDefinition，然后注册到spring容器中。

定义一个SampleBean类，只有一个print方法

1package com.lcifn.spring.demo.namespace;
2
3public class SampleBean {
4
5	public void print(){
6		System.out.println("sample bean works");
7	}
8}

定义namespace-handler.xml配置文件，要声明xmlns:sample，以及在xsi:schemaLocation中指定sample的xsd

1<?xml version="1.0" encoding="UTF-8"?>
2<beans xmlns="http://www.springframework.org/schema/beans"
3	xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" 
4	xmlns:sample="http://www.lcifn.com/sample"
5	xsi:schemaLocation="
6	http://www.springframework.org/schema/beans
7	http://www.springframework.org/schema/beans/spring-beans-3.0.xsd
8	http://www.lcifn.com/sample
9	http://www.lcifn.com/sample.xsd">
10	
11	<sample:entity id="sampleBean" class="com.lcifn.spring.demo.namespace.SampleBean"/>
12</beans>

再写一个Test类

1package com.lcifn.spring.demo.test;
2
3import org.springframework.context.ApplicationContext;
4import org.springframework.context.support.ClassPathXmlApplicationContext;
5
6import com.lcifn.spring.demo.namespace.SampleBean;
7
8public class SampleNamespaceHandlerTest {
9
10	public static void main(String[] args) {
11		ApplicationContext context = new ClassPathXmlApplicationContext("namespace-handler.xml");
12		SampleBean sampleBean = context.getBean("sampleBean",SampleBean.class);
13		sampleBean.print();
14	}
15}

运行后显示：

sample bean works

表示我们自定义的entity被spring解析并加载到容器中了。

源码分析

在spring源码-IOC容器(二)-Bean的定位解析注册中讲解了spring是如何解析xml文件的，但主要就bean标签的解析进行了详解。这里我们来分析自定义标签的解析过程。

1.加载spring.schemas

spring在创建DOM解析的DocumentBuilder时，设置了EntityResolver的实现类ResourceEntityResolver

1XmlBeanDefinitionReader.getEntityResolver()
2
3ResourceLoader resourceLoader = getResourceLoader();
4if (resourceLoader != null) {
5	this.entityResolver = new ResourceEntityResolver(resourceLoader);
6}
7
8DefaultDocumentLoader.createDocumentBuilder()
9
10DocumentBuilder docBuilder = factory.newDocumentBuilder();
11if (entityResolver != null) {
12	docBuilder.setEntityResolver(entityResolver);
13}

而ResourceEntityResolver中的resolveEntity方法先调用其父类DelegatingEntityResolver的方法

1DelegatingEntityResolver.resolveEntity()
2
3if (systemId != null) {
4	if (systemId.endsWith(DTD_SUFFIX)) {
5		return this.dtdResolver.resolveEntity(publicId, systemId);
6	}
7	else if (systemId.endsWith(XSD_SUFFIX)) {
8		return this.schemaResolver.resolveEntity(publicId, systemId);
9	}
10}

现在的spring都使用xsd文件，而这里的schemaResolver在构造函数中默认设置为PluggableSchemaResolver

this.schemaResolver = new PluggableSchemaResolver(classLoader);

而PluggableSchemaResolver的resolveEntity方法中调用getSchemaMappings()返回xsd文件的路径

1PluggableSchemaResolver.resolveEntity()
2
3if (systemId != null) {
4	String resourceLocation = getSchemaMappings().get(systemId);
5	if (resourceLocation != null) {
6		Resource resource = new ClassPathResource(resourceLocation, this.classLoader);
7		try {
8			InputSource source = new InputSource(resource.getInputStream());
9			source.setPublicId(publicId);
10			source.setSystemId(systemId);
11			if (logger.isDebugEnabled()) {
12				logger.debug("Found XML schema [" + systemId + "] in classpath: " + resourceLocation);
13			}
14			return source;
15		}
16		catch (FileNotFoundException ex) {
17			if (logger.isDebugEnabled()) {
18				logger.debug("Couldn't find XML schema [" + systemId + "]: " + resource, ex);
19			}
20		}
21	}
22}

第一次调用getSchemaMappings()时会去加载所有META-INF/spring.schemas配置，其中的schemaMappingsLocation默认为DEFAULT_SCHEMA_MAPPINGS_LOCATION

1PluggableSchemaResolver	
2
3public static final String DEFAULT_SCHEMA_MAPPINGS_LOCATION = "META-INF/spring.schemas";
4
5private Map<String, String> getSchemaMappings() {
6	if (this.schemaMappings == null) {
7		synchronized (this) {
8			if (this.schemaMappings == null) {
9				if (logger.isDebugEnabled()) {
10					logger.debug("Loading schema mappings from [" + this.schemaMappingsLocation + "]");
11				}
12				try {
13					Properties mappings =
14							PropertiesLoaderUtils.loadAllProperties(this.schemaMappingsLocation, this.classLoader);
15					if (logger.isDebugEnabled()) {
16						logger.debug("Loaded schema mappings: " + mappings);
17					}
18					Map<String, String> schemaMappings = new ConcurrentHashMap<String, String>(mappings.size());
19					CollectionUtils.mergePropertiesIntoMap(mappings, schemaMappings);
20					this.schemaMappings = schemaMappings;
21				}
22				catch (IOException ex) {
23					throw new IllegalStateException(
24							"Unable to load schema mappings from location [" + this.schemaMappingsLocation + "]", ex);
25				}
26			}
27		}
28	}
29	return this.schemaMappings;
30}

至此spring.schemas在xml解析前就已经配置到DocumentBuilder中

2.加载spring.handlers

在DefaultBeanDefinitionDocumentReader类parseBeanDefinitions方法中，解析每一个element，也就是每一个标签时，会先判断element是否属于默认的命名空间。

1DefaultBeanDefinitionDocumentReader
2
3protected void parseBeanDefinitions(Element root, BeanDefinitionParserDelegate delegate) {
4	if (delegate.isDefaultNamespace(root)) {
5		NodeList nl = root.getChildNodes();
6		for (int i = 0; i < nl.getLength(); i++) {
7			Node node = nl.item(i);
8			if (node instanceof Element) {
9				Element ele = (Element) node;
10				if (delegate.isDefaultNamespace(ele)) {
11					parseDefaultElement(ele, delegate);
12				}
13				else {
14					delegate.parseCustomElement(ele);
15				}
16			}
17		}
18	}
19	else {
20		delegate.parseCustomElement(root);
21	}
22}

如果不是默认的命名空间，就调用BeanDefinitionParserDelegate的parseCustomElement方法，来解析出自定义命名空间的处理器，即NamespaceHandler的实现类。

1BeanDefinitionParserDelegate.java
2
3public BeanDefinition parseCustomElement(Element ele, BeanDefinition containingBd) {
4	String namespaceUri = getNamespaceURI(ele);
5	NamespaceHandler handler = this.readerContext.getNamespaceHandlerResolver().resolve(namespaceUri);
6	if (handler == null) {
7		error("Unable to locate Spring NamespaceHandler for XML schema namespace [" + namespaceUri + "]", ele);
8		return null;
9	}
10	return handler.parse(ele, new ParserContext(this.readerContext, this, containingBd));
11}

spring默认的NamespaceHandlerResolver为DefaultNamespaceHandlerResolver，而在resolve方法中调用getHandlerMappings方法，然后匹配传入的namespaceUri来获取NamespaceHandler的实例或是其className。

1DefaultNamespaceHandlerResolver.java
2
3public NamespaceHandler resolve(String namespaceUri) {
4	Map<String, Object> handlerMappings = getHandlerMappings();
5	Object handlerOrClassName = handlerMappings.get(namespaceUri);
6	if (handlerOrClassName == null) {
7		return null;
8	}
9	else if (handlerOrClassName instanceof NamespaceHandler) {
10		return (NamespaceHandler) handlerOrClassName;
11	}
12	else {
13		String className = (String) handlerOrClassName;
14		try {
15			Class<?> handlerClass = ClassUtils.forName(className, this.classLoader);
16			if (!NamespaceHandler.class.isAssignableFrom(handlerClass)) {
17				throw new FatalBeanException("Class [" + className + "] for namespace [" + namespaceUri +
18						"] does not implement the [" + NamespaceHandler.class.getName() + "] interface");
19			}
20			NamespaceHandler namespaceHandler = (NamespaceHandler) BeanUtils.instantiateClass(handlerClass);
21			namespaceHandler.init();
22			handlerMappings.put(namespaceUri, namespaceHandler);
23			return namespaceHandler;
24		}
25		catch (ClassNotFoundException ex) {
26			throw new FatalBeanException("NamespaceHandler class [" + className + "] for namespace [" +
27					namespaceUri + "] not found", ex);
28		}
29		catch (LinkageError err) {
30			throw new FatalBeanException("Invalid NamespaceHandler class [" + className + "] for namespace [" +
31					namespaceUri + "]: problem with handler class file or dependent class", err);
32		}
33	}
34}

getHandlerMappings方法第一次调用时，加载META-INF/spring.handlers配置

1DefaultNamespaceHandlerResolver.java
2
3private Map<String, Object> getHandlerMappings() {
4	if (this.handlerMappings == null) {
5		synchronized (this) {
6			if (this.handlerMappings == null) {
7				try {
8					Properties mappings =
9							PropertiesLoaderUtils.loadAllProperties(this.handlerMappingsLocation, this.classLoader);
10					if (logger.isDebugEnabled()) {
11						logger.debug("Loaded NamespaceHandler mappings: " + mappings);
12					}
13					Map<String, Object> handlerMappings = new ConcurrentHashMap<String, Object>(mappings.size());
14					CollectionUtils.mergePropertiesIntoMap(mappings, handlerMappings);
15					this.handlerMappings = handlerMappings;
16				}
17				catch (IOException ex) {
18					throw new IllegalStateException(
19							"Unable to load NamespaceHandler mappings from location [" + this.handlerMappingsLocation + "]", ex);
20				}
21			}
22		}
23	}
24	return this.handlerMappings;
25}

至此就解析到所有的spring.handlers并拿到自定义的NamespaceHandler实现类的实例。

3.NamespaceHandlerSupport

在BeanDefinitionParserDelegate的parseCustomElement方法中，调用NamespaceHandler的parse方法，并传入要解析的element和上下文信息(其中包含BeanFactory的实例)。

spring为我们封装了一个支持类NamespaceHandlerSupport，对parse方法进行了处理，拿到子标签的名称(比如上面sample:entity的entity)，然后匹配到自定义的BeanDefinitionParser实现类(比如上面的EntityBeanDefinitionParse)，调用其parse方法进行真正的解析。

1NamespaceHandlerSupport.java
2
3public BeanDefinition parse(Element element, ParserContext parserContext) {
4	return findParserForElement(element, parserContext).parse(element, parserContext);
5}
6
7private BeanDefinitionParser findParserForElement(Element element, ParserContext parserContext) {
8	String localName = parserContext.getDelegate().getLocalName(element);
9	BeanDefinitionParser parser = this.parsers.get(localName);
10	if (parser == null) {
11		parserContext.getReaderContext().fatal(
12				"Cannot locate BeanDefinitionParser for element [" + localName + "]", element);
13	}
14	return parser;
15}

其中的this.parsers就是BeanDefinitionParser的一个Map集，通过在init方法中注册进来。比如上面SampleNamespaceHandler的init方法

1public void init() {
2	registerBeanDefinitionParser("entity", new EntityBeanDefinitionParse());
3}

而registerBeanDefinitionParser就是将自定义的BeanDefinitionParser实现扔到parsers中去

1protected final void registerBeanDefinitionParser(String elementName, BeanDefinitionParser parser) {
2	this.parsers.put(elementName, parser);
3}

至此整个自定义xml的流程就讲解完了。

spring源码

前言

自定义xml标签

源码分析

1.加载spring.schemas

2.加载spring.handlers

3.NamespaceHandlerSupport

评论区

相关推荐