enable作為模塊驅(qū)動在Spring Farmework、Spring Boot、Spring Cloud使用，都是通過注解的形式以@enable作為前綴,一些常用注解如

如何自定義enable開發(fā)？

基于ImportSelector實現(xiàn)注解驅(qū)動

自定義接入類型

Access為接入類型的接口，下文的RPC接入和REST接入基于這個實現(xiàn)，定義兩個接口，一個為啟動，一個停止，內(nèi)部嵌套一個枚舉用于標識是哪一種接入

public interface Access {
    /**
     * 初始化配置
     */
    void start();

    /**
     * 銷毀配置
     */
    void stop();

    enum Type{
        REST,
        RPC
    }
}

定義RPC和REST的實現(xiàn)

REST實現(xiàn)，只是簡單的打印方法

public class RestAccess implements Access{
    @Override
    public void start() {
        System.out.println("rest接入配置");
    }

    @Override
    public void stop() {
        System.out.println("rest接入銷毀配置");
    }
}

RPC實現(xiàn)

public class RpcAccess implements Access{
    @Override
    public void start() {
        System.out.println("rpc接入配置");
    }

    @Override
    public void stop() {
        System.out.println("rpc接入銷毀配置");
    }
}

自定義注解EnableAccess

接入類型為RPC或者REST，AccessImportSelector在下一步驟實現(xiàn)

@Target({ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Inherited
@Import(AccessImportSelector.class)
public @interface EnableAccess {
    /**
     * 接入類型
     * @return
     */
    Access.Type type();
}

實現(xiàn)ImportSelector

定義AccessImportSelector實現(xiàn)ImportSelector，分別獲取注解信息，根據(jù)注解獲取接入類型，根據(jù)接入類型選擇不同的接入類型

public class AccessImportSelector implements ImportSelector{
    @Override
    public String[] selectImports(AnnotationMetadata annotationMetadata) {
        //讀取EnableAccess中所有的屬性方法
        Map<String, Object> annotationAttributes =  annotationMetadata.getAnnotationAttributes(EnableAccess.class.getName());
        //獲取屬性為type的屬性方法
        Access.Type type = (Access.Type )annotationAttributes.get("type");
        //導入的類名稱數(shù)組
        String [] importClassName = new String[0];
        switch (type){
            case RPC:
                //設置為RPC，返回RpcAccess組件
                importClassName = new String[]{RpcAccess.class.getName()};
                break;
            case REST:
                //設置為REST，返回RestAccess組件
                importClassName = new String[]{RestAccess.class.getName()};
        }
        return importClassName;
    }
}

使用

在primarySource也就是這里的DemoApplication上使用注解EnableAccess，選擇接入方式，就會初始化不通的接入組件

@SpringBootApplication
@EnableAccess(type=Access.Type.REST)
public class DemoApplication {

    public static void main(String[] args) {
        ConfigurableApplicationContext context = SpringApplication.run(DemoApplication.class, args);
        Access access = context.getBean(Access.class);
        access.start();
        access.stop();
    }

}

基于ImportBeanDefinitionRegistrar實現(xiàn)注解驅(qū)動

這里其它步驟一樣，主要區(qū)別是注解里面Import的類變了，這里是基于基于ImportBeanDefinitionRegistrar實現(xiàn)注解驅(qū)動實現(xiàn)

自定義ImportBeanDefinitionRegistrar

public class AccessImportBeanDefinitionRegistrar implements ImportBeanDefinitionRegistrar {
    @Override
    public void registerBeanDefinitions(AnnotationMetadata annotationMetadata, BeanDefinitionRegistry beanDefinitionRegistry) {
        ImportSelector importSelector = new AccessImportSelector();
        //篩選class名稱集合
        String[] selectedClassNames = importSelector.selectImports(annotationMetadata);
        Stream.of(selectedClassNames)
                .map(BeanDefinitionBuilder::genericBeanDefinition)
                .map(BeanDefinitionBuilder::getBeanDefinition)
                .forEach(beanDefinition ->{
                    //注冊beanDefinition到beanDefinitionRegistry
                    BeanDefinitionReaderUtils.registerWithGeneratedName(beanDefinition,beanDefinitionRegistry);
                });
    }
}

EnableAccess注解變更

這里import導入了AccessImportBeanDefinitionRegistrar

@Target({ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Inherited
@Import(AccessImportBeanDefinitionRegistrar.class)
public @interface EnableAccess {
    /**
     * 接入類型
     * @return
     */
    Access.Type type();
}

實現(xiàn)

RPC接入

type=Access.Type.RPC

@SpringBootApplication
@EnableAccess(type=Access.Type.RPC)
public class DemoApplication {

    public static void main(String[] args) {
        ConfigurableApplicationContext context = SpringApplication.run(DemoApplication.class, args);
        Access access = context.getBean(Access.class);
        access.start();
        access.stop();
    }

}

REST接入

type=Access.Type.REST

@SpringBootApplication
@EnableAccess(type=Access.Type.REST)
public class DemoApplication {

    public static void main(String[] args) {
        ConfigurableApplicationContext context = SpringApplication.run(DemoApplication.class, args);
        Access access = context.getBean(Access.class);
        access.start();
        access.stop();
    }

}

本文由博客一文多發(fā)平臺 OpenWrite 發(fā)布！