擴展Spring的兩種方式

 

 

Spring的擴展方式

  1. 自定義xml的方式,現實中的案例 dubbo框架,apollo框架
  2. 通過實現註解,實現ImportBeanDefinitionRegistrar接口實現. 現實中的案例 JPA,Feign,dubbo

下面我會通過兩個小demo的是形式實現上述兩種方案.

通過自定義Xml的方式擴展Spring

需要用到的知識點:

  1. xsd  xml schema definition 是xml文件的定義文件.個人覺得可以理解爲xml的meta,裏面定義了xml的規範,spring在解析xml的時候也會通過這個文件驗證xml文件是否有效(可以模仿 dubbo,applo他們的xsd文件)
  2. spring.handlers,spring.schemas. spring在啓動的時候回掃描meta-info目錄下面的這個文件,通過handlers文件可以獲取解析自定xml的handler 類
  3. 繼承NamesHandlerSupport類

源碼:

自定義xml文件,xsd 文件, spring.handlers,spring.schemas文件

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:user="http://www.ivan.com/schema/user"
       xsi:schemaLocation="http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans.xsd
       http://www.ivan.com/schema/user http://www.ivan.com/schema/user.xsd">

    <user:config username="ivan" password="123456"/>

</beans>

xsd: 

<?xml version="1.0" encoding="UTF-8" standalone="no"?>
<xsd:schema xmlns:xsd="http://www.w3.org/2001/XMLSchema"
            xmlns:beans="http://www.springframework.org/schema/beans"
            xmlns:tool="http://www.springframework.org/schema/tool"
            xmlns="http://www.ivan.com/schema/user"
            targetNamespace="http://www.ivan.com/schema/user">

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

    <xsd:annotation>
        <xsd:documentation>
            <![CDATA[ Namespace support for the  User  ]]></xsd:documentation>
    </xsd:annotation>

    <xsd:element name="config">
        <xsd:annotation>
            <xsd:documentation>
                <![CDATA[ Apollo configuration section to integrate with Spring.]]>
            </xsd:documentation>
        </xsd:annotation>
        <xsd:complexType>
            <xsd:attribute name="username" type="xsd:string" use="required">
                <xsd:annotation>
                    <xsd:documentation>
                        <![CDATA[
                           The username is user's name.
                        ]]>
                    </xsd:documentation>
                </xsd:annotation>
            </xsd:attribute>
            <xsd:attribute name="password" type="xsd:int" use="required">
                <xsd:annotation>
                    <xsd:documentation>
                        <![CDATA[
                            The password is user's password.
                        ]]>
                    </xsd:documentation>
                </xsd:annotation>
            </xsd:attribute>
        </xsd:complexType>
    </xsd:element>


</xsd:schema>

spring.handlers

http\://www.ivan.com/schema/user=com.example.learn.CustomizationXmlHandler

spring.schemas

http\://www.ivan.com/schema/user.xsd=META-INF/user.xsd

 啓動類:

package com.example.learn;

import org.springframework.context.support.ClassPathXmlApplicationContext;

public class DemoApplication {

    public static void main(String[] args) {
        ClassPathXmlApplicationContext context = new ClassPathXmlApplicationContext(new String[]{"customization.xml"});
        User u = context.getBean(User.class);
        System.out.println(u.getUsername());
        System.out.println(u.getPassword());
    }

}

繼承NamesHandlerSupport類: 

package com.example.learn;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.xml.NamespaceHandlerSupport;


public class CustomizationXmlHandler extends NamespaceHandlerSupport {

    private static final Logger LOGGER = LoggerFactory.getLogger(CustomizationXmlHandler.class);

    @Override
    public void init() {
        registerBeanDefinitionParser("config", new ConfigParser());
    }
}

Paser類:

package com.example.learn;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.config.BeanDefinition;
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 ConfigParser implements BeanDefinitionParser {

    private static final Logger LOGGER = LoggerFactory.getLogger(ConfigParser.class);

    @Override
    public BeanDefinition parse(Element element, ParserContext parserContext) {
        BeanDefinition beanDefinition = new RootBeanDefinition();
        beanDefinition.setBeanClassName(User.class.getName());
        String username = element.getAttribute("username");
        int password = Integer.valueOf(element.getAttribute("password"));
        beanDefinition.getPropertyValues().add("username", username);
        beanDefinition.getPropertyValues().add("password", password);
        parserContext.getRegistry().registerBeanDefinition(User.class.getName(), beanDefinition);
        return beanDefinition;
    }
}

User:

package com.example.learn;

import lombok.Getter;
import lombok.Setter;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

@Setter
@Getter
public class User {

    private static final Logger LOGGER = LoggerFactory.getLogger(User.class);

    private String username;

    private int password;
}

 

上述是通過xml擴展spring,原理應該是這樣當如果要做到像dubbo,apollo這樣的還是框架在解析與註冊definition還是要複雜許多的.

通過註解和實現ImportBeanDefinitionRegistrar來擴展spring

需要了解的知識點

  1. 瞭解ImportBeanDefinitionResgistrar接口,改接口需配合@Configuration,@ImportSelector,@Import三個註解使用.能夠動態的註冊bean
  2. ClassPathScanningCandidateComponentProvider 類,掃描我們需要用註冊到spring容器的beanDefinition.該類可設置一些條件對掃描類進行過濾
  3. jdk 反射 Proxy.newInstance()方法
  4. 註解的一些常識

源碼:

啓動類:

package com.example.demo;

import com.example.demo.remote.UserService;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.context.annotation.AnnotationConfigApplicationContext;

@SpringBootApplication
@EnableHttpRemoteCall(scanBasePackage = "com.example.demo.remote")
public class SpringExtentionApplication {

    public static void main(String[] args) {

        AnnotationConfigApplicationContext annotationConfigApplicationContext =
                new AnnotationConfigApplicationContext(SpringExtentionApplication.class);
        annotationConfigApplicationContext.start();
        UserService userService = annotationConfigApplicationContext.getBean(UserService.class);
        userService.findUserById("r123");
    }

}

userService接口

package com.example.demo.remote;

import com.example.demo.HttpCall;
import com.example.demo.User;


@HttpCall
public interface UserService {

    @HttpCall(httpMethod = "GET", url = "lcoalhost:8080/users")
    User findUserById(String userId);
}

User 類:

package com.example.demo;

public class User {

   
}

Registrar類,這個類是關鍵

package com.example.demo;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.BeanClassLoaderAware;
import org.springframework.beans.factory.BeanFactory;
import org.springframework.beans.factory.BeanFactoryAware;
import org.springframework.beans.factory.annotation.AnnotatedBeanDefinition;
import org.springframework.beans.factory.config.BeanDefinition;
import org.springframework.beans.factory.support.BeanDefinitionRegistry;
import org.springframework.beans.factory.support.DefaultListableBeanFactory;
import org.springframework.context.EnvironmentAware;
import org.springframework.context.annotation.ClassPathScanningCandidateComponentProvider;
import org.springframework.context.annotation.ImportBeanDefinitionRegistrar;
import org.springframework.core.env.Environment;
import org.springframework.core.type.AnnotationMetadata;
import org.springframework.core.type.filter.AnnotationTypeFilter;
import org.springframework.util.ClassUtils;

import java.lang.annotation.Annotation;
import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.lang.reflect.Proxy;
import java.util.Map;
import java.util.Set;


public class HttpRemoteCallRegsitrar implements ImportBeanDefinitionRegistrar, EnvironmentAware,
        BeanFactoryAware, BeanClassLoaderAware {


    private static final Logger LOGGER = LoggerFactory.getLogger(HttpRemoteCallRegsitrar.class);

    private Environment environment;

    private BeanFactory beanFactory;

    private ClassLoader classLoader;

    @Override
    public void registerBeanDefinitions(AnnotationMetadata importingClassMetadata, BeanDefinitionRegistry registry) {

        Map<String, Object> map = importingClassMetadata.getAnnotationAttributes(EnableHttpRemoteCall.class.getName());
        String pacakge = (String) map.get("scanBasePackage");

        ClassPathScanningCandidateComponentProvider classScanner =
                new ClassPathScanningCandidateComponentProvider(false, this.environment) {

                    @Override
                    protected boolean isCandidateComponent(AnnotatedBeanDefinition beanDefinition) {

                        if (beanDefinition.getMetadata().isInterface()) {
                            try {
                                Class<?> target = ClassUtils.forName(
                                        beanDefinition.getMetadata().getClassName(),
                                        classLoader);
                                return !target.isAnnotation();
                            } catch (Exception ex) {
                                LOGGER.error("load class exception:", ex);
                            }
                        }
                        return false;
                    }
                };
        AnnotationTypeFilter annotationTypeFilter = new AnnotationTypeFilter(HttpCall.class);
        classScanner.addIncludeFilter(annotationTypeFilter);
        Set<BeanDefinition> beanDefinitions = classScanner.findCandidateComponents(pacakge);

        for (BeanDefinition b : beanDefinitions) {

            try {
                Class<?> target = Class.forName(b.getBeanClassName());
                Object proxy = Proxy.newProxyInstance(this.getClass().getClassLoader(), new Class[]{target},
                        new InvocationHandler() {
                            @Override
                            public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
                                Annotation annotation = method.getAnnotation(HttpCall.class);
                                String httpMethod = ((HttpCall) annotation).httpMethod();
                                String url = ((HttpCall) annotation).url();
                                LOGGER.info("Http call start: httpMethod:[{}], url:[{}]", httpMethod, url);
                                return null;
                            }
                        });
                ((DefaultListableBeanFactory) beanFactory).registerSingleton(b.getBeanClassName(), proxy);
            } catch (ClassNotFoundException e) {
                e.printStackTrace();
            }
        }
        System.out.println(map.get("scanBasePackage"));
    }

    @Override
    public void setEnvironment(Environment environment) {
        this.environment = environment;
    }

    @Override
    public void setBeanFactory(BeanFactory beanFactory) throws BeansException {
        this.beanFactory = beanFactory;
    }

    @Override
    public void setBeanClassLoader(ClassLoader classLoader) {
        this.classLoader = classLoader;
    }
}

啓動註解類

package com.example.demo;

import org.springframework.context.annotation.Import;

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

/**
 * 啓用HttpRemoteCall, 需配合configuration註解一起用
 */
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.TYPE)
@Import(HttpRemoteCallRegsitrar.class)
public @interface EnableHttpRemoteCall {
    String scanBasePackage() default "";
}

註解類:

package com.example.demo;

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

@Retention(RetentionPolicy.RUNTIME)
@Target({ElementType.METHOD,ElementType.TYPE})
public @interface HttpCall {

    String httpMethod() default "GET";

    String url() default "";

}

到此結束,其實流程可以總結爲一下幾步

  1. 創建啓動註解 在這裏可以認爲是@EnableHttpCall註解,改註解定義需要掃描的類
  2. 根據業務邏輯創建實現ImportBeanDefinitionRegistrar接口類
  3. 在啓動註解上import 2步驟中實現類
  4. 定義元數據註解,在上面的可認爲是HttpCall註解,改註解附帶了運行時需要的數據如請求方法,請求url
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章