如何使用Spring自定義Xml標簽

Posted on by

前言

在早期基於Xml配置的Spring Mvc項目中，我們往往會使用<context:component-scan basePackage=””>這種自定義標簽來掃描我們在basePackae配置裡的包名下的類，並且會判斷這個類是否要註入到Spring容器中(比如這個類上標記瞭@Component註解就代表需要被Spring註入),如果需要那麼它會幫助我們把這些類一一註入。

正文

在分析這個自定義標簽的解析機制前，我先提前劇透這個自定義標簽是通過哪個強大的類來解析的吧，就是隸屬於spring-context包下的ComponentScanBeanDefinitionParser，這個類在Springboot掃描Bean的過程中也扮演瞭重要角色。

既然知道瞭是這個類解析的，那麼我們可以通過idea強大的搜索功能來搜它的引用之處瞭，這邊就截圖如下：

可以看到這裡面初始化瞭8個帶Parser後綴的各種Parser,從方法registerBeanDefinitionParser看出Spring是通過這個ContextNamespaceHandler來完成對以<context:自定義命名空間開頭的標簽解析器的註冊。我們可以看到Spring內部已經集成瞭幾個常用的NamespaceHandler,截圖如下：

那麼我們自己是否可以自定義一個NamespaceHandler來註冊我們自定義的標簽解析器呢？答案是肯定的。

自定義NameSpaceHandler

final class TestNamespaceHandler extends NamespaceHandlerSupport {

   @Override
   public void init() {

      //註冊parser
      registerBeanDefinitionParser("testBean", new TestBeanDefinitionParser());
      registerBeanDefinitionParser("person", new PersonDefinitionParser());

      //註冊element的 decorater
      registerBeanDefinitionDecorator("set", new PropertyModifyingBeanDefinitionDecorator());
      registerBeanDefinitionDecorator("debug", new DebugBeanDefinitionDecorator());

      //註冊 attr的 decorator
      registerBeanDefinitionDecoratorForAttribute("object-name", new ObjectNameBeanDefinitionDecorator());
   }

到這裡大傢可能會有個疑問，這個NameSpaceHandler是怎麼使用的呢？大傢如果看瞭我之前寫的文章，那就會知道有一種方式可以配置我們自定義的NamespaceHandler.

public class CustomXmlApplicationContext  extends AbstractXmlApplicationContext {

   private static final String CLASSNAME = CustomXmlApplicationContext.class.getSimpleName();
   private static final String FQ_PATH = "org/wonder/frame/customBean";
   private static final String NS_PROPS = format("%s/%s.properties", FQ_PATH, CLASSNAME);

   public CustomXmlApplicationContext(String... configLocations) {
      setConfigLocations(configLocations);
         refresh();
   }

   @Override
   protected void initBeanDefinitionReader(XmlBeanDefinitionReader reader) {
      super.initBeanDefinitionReader(reader);

      //1.指定resolver的 handlerMappingsLocation 就是 NamespaceHandler的 配置文件路徑
      NamespaceHandlerResolver resolver = new DefaultNamespaceHandlerResolver(this.getClassLoader(), NS_PROPS);

      //2.設置resolver
      reader.setNamespaceHandlerResolver(resolver);
      //3.設置驗證模式
      reader.setValidationMode(XmlBeanDefinitionReader.VALIDATION_XSD);
      //4.設置entityResolver
      reader.setEntityResolver(new CustomSchemaResolver());
   }

可以看到我們在初始化BeanDefinitionReader的時候我們可以設置NamespaceHandlerResolver並且配置它的NamespaceHandler文件路徑。那這個NamespaceHandler配置文件應該怎麼寫呢？

http\://www.john.com/resource=org.wonder.frame.customBean.TestNamespaceHandler

就一行配置，但是這裡有兩點要註意：

  • http\://www.john.com/resource要和xsd的targetNamspace一致。
  • http\://www.john.com/resource要和xml配置文件中的自定義namespace一致。

從代碼裡看出來我們解析自定義標簽的時候其實是還需要自定義schema才能完成的。

自定義schema

private static final String CLASSNAME = CustomXmlApplicationContext.class.getSimpleName();
private static final String FQ_PATH = "org/wonder/frame/customBean";
private static final String TEST_XSD = format("%s/%s.xsd", FQ_PATH, CLASSNAME);
private final class CustomSchemaResolver extends PluggableSchemaResolver {

   public CustomSchemaResolver() {
      super(CustomXmlApplicationContext.this.getClassLoader());
   }

   @Override
   public InputSource resolveEntity(String publicId, String systemId) throws IOException {
      InputSource source = super.resolveEntity(publicId, systemId);
      if (source == null) {
         try{
            //todo 指定瞭xsd路徑
            Resource resource = new ClassPathResource(TEST_XSD);
            source = new InputSource(resource.getInputStream());
            source.setPublicId(publicId);
            source.setSystemId(systemId);
            return source;
         }
         catch (FileNotFoundException ex){

         }

      }
      return  null;
   }
}

這裡我們也通過ClassPathResource設置瞭自定義的xsd文件路徑。我們來看看xsd文件長啥樣：

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

<xsd:schema xmlns="http://www.john.com/resource"
         xmlns:xsd="http://www.w3.org/2001/XMLSchema"
         targetNamespace="http://www.john.com/resource"
         elementFormDefault="qualified">

   <xsd:element name="person">
      <xsd:complexType>
         <xsd:attribute name="id" type="xsd:string" use="optional" form="unqualified"/>
         <xsd:attribute name="name" type="xsd:string" use="required" form="unqualified"/>
         <xsd:attribute name="age" type="xsd:integer" use="required" form="unqualified"/>
      </xsd:complexType>
   </xsd:element>

   <xsd:element name="testBean">
      <xsd:complexType>
         <xsd:attribute name="id" type="xsd:string" use="required" form="unqualified"/>
         <xsd:attribute name="name" type="xsd:string" use="required" form="unqualified"/>
         <xsd:attribute name="age" type="xsd:integer" use="required" form="unqualified"/>
      </xsd:complexType>
   </xsd:element>

   <xsd:element name="set">
      <xsd:complexType>
         <xsd:attribute name="name" type="xsd:string" use="required" form="unqualified"/>
         <xsd:attribute name="age" type="xsd:integer" use="required" form="unqualified"/>
      </xsd:complexType>
   </xsd:element>

   <xsd:element name="debug"/>
   <xsd:attribute name="object-name" type="xsd:string"/>
</xsd:schema>

Parser

我們先來分析下TestBeanDefinitionParser和PersonDefinitionParser這兩者有啥區別：

TestBeanDefinitionParser直接實現瞭BeanDefinitionParser接口，內部直接定義一個RootBeanDefinition並且註冊。

private static class TestBeanDefinitionParser implements BeanDefinitionParser {

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

      RootBeanDefinition definition = new RootBeanDefinition();
      definition.setBeanClass(CustomBean.class);

      MutablePropertyValues mpvs = new MutablePropertyValues();
      mpvs.add("name", element.getAttribute("name"));
      mpvs.add("age", element.getAttribute("age"));

      //1.設置beanDefinition的 屬性 propertyValues
      definition.setPropertyValues(mpvs);

      //2.獲取到beanDefinition的 registry
      parserContext.getRegistry().registerBeanDefinition(element.getAttribute("id"), definition);
      return null;
   }
}

PersonDefinitionParser繼承自AbstractSingleBeanDefinitionParser抽象類，內部使用BeanDefinitionBuilder構造器來完成BeanDefinition的創建。

private static final class PersonDefinitionParser extends AbstractSingleBeanDefinitionParser {

   @Override
   protected Class<?> getBeanClass(Element element) {
      return CustomBean.class;
   }

   @Override
   protected void doParse(Element element, BeanDefinitionBuilder builder) {
      builder.addPropertyValue("name", element.getAttribute("name"));
      builder.addPropertyValue("age", element.getAttribute("age"));
   }
}

Decorator

我們看到在NameSpaceHandler中我們除瞭parser外還可以定義自定義元素的decorator和自定義attribute的decorator,那這兩個decorator是用來幹嘛的呢？我們先來看下上述代碼中的PropertyModifyingBeanDefinitionDecorator。

private static class PropertyModifyingBeanDefinitionDecorator implements BeanDefinitionDecorator {

   @Override
   public BeanDefinitionHolder decorate(Node node, BeanDefinitionHolder definition, ParserContext parserContext) {
      Element element = (Element) node;
      //1.獲取BeanDefinition
      BeanDefinition def = definition.getBeanDefinition();

      MutablePropertyValues mpvs = (def.getPropertyValues() == null) ? new MutablePropertyValues() : def.getPropertyValues();
      mpvs.add("name", element.getAttribute("name"));
      mpvs.add("age", element.getAttribute("age"));

      ((AbstractBeanDefinition) def).setPropertyValues(mpvs);
      return definition;
   }
}

從decorate方法內部看出這個decorator是用來給我們的BeanDefinition來添加屬性的。這樣一來我們就可以在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:test="http://www.john.com/resource"
      xmlns:util="http://www.springframework.org/schema/util"
      xsi:schemaLocation="http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans-2.0.xsd
       http://www.springframework.org/schema/util http://www.springframework.org/schema/util/spring-util-2.0.xsd
       http://www.john.com/resource http://www.john.com/resource/org/wonder/frame/customBean/CustomXmlApplicationContext.xsd"
   default-lazy-init="true">
   <test:testBean id="testBean" name="Rob Harrop" age="23"/>
   <bean id="customisedTestBean" class="org.wonder.frame.customBean.CustomBean">
      <!-- 自定義標簽加 自定義屬性 -->
      <test:set name="John wonder" age="36"/>
   </bean>

</beans>

我們看到testBean這個自定義標簽定義瞭兩個屬性name和age。之後我們在使用這個testBean的時候就可以獲取到它的name和age屬性瞭。

CustomBean bean = (CustomBean) beanFactory.getBean("testBean");
System.out.println("name is:" +bean.getName() +" and age is:"+ bean.getAge());

那麼ObjectNameBeanDefinitionDecorator這個attribute的Decorator是幹嘛的呢？看如下示例

<!--為bean設置自定義Attr-->
<bean id="decorateWithAttribute" class="org.springframework.tests.sample.beans.TestBean" test:object-name="foo"/>

我們可以為這個Bean添加自定義Attribute，那麼添加瞭這個Attribute我們怎麼使用呢？看如下示例：

BeanDefinition beanDefinition = this.beanFactory.getBeanDefinition("decorateWithAttribute");
assertEquals("foo", beanDefinition.getAttribute("objectName"));

我們通過BeanDefinition的getAttribute就能獲取到這個attribute值。

從Spring源碼得知BeanDefinition擴展瞭AttributeAccessor接口，這個接口是用於附加和訪問Bean元數據的通用的接口。直接實現這個接口的是AttributeAccessorSupport類。這個類裡定義瞭名為attributes 的LinkedHashMap。

總結

Spring通過自定義標簽和自定義屬性實現瞭很多擴展功能，很多我們常用的Spring配置內部都是通過它來完成的。

以上就是如何使用Spring自定義Xml標簽的詳細內容，更多關於使用Spring自定義Xml標簽的資料請關註WalkonNet其它相關文章！

推薦閱讀：