如何在Springboot中使用enable模塊驅(qū)動(dòng)

本篇文章給大家分享的是有關(guān)如何在Springboot中使用enable模塊驅(qū)動(dòng)，小編覺得挺實(shí)用的，因此分享給大家學(xué)習(xí)，希望大家閱讀完這篇文章后可以有所收獲，話不多說，跟著小編一起來看看吧。

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

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

基于ImportSelector實(shí)現(xiàn)注解驅(qū)動(dòng)

自定義接入類型

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

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

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

  enum Type{
    REST,
    RPC
  }
}

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

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

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

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

RPC實(shí)現(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在下一步驟實(shí)現(xiàn)

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

實(shí)現(xiàn)ImportSelector

定義AccessImportSelector實(shí)現(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");
    //導(dǎo)入的類名稱數(shù)組
    String [] importClassName = new String[0];
    switch (type){
      case RPC:
        //設(shè)置為RPC，返回RpcAccess組件
        importClassName = new String[]{RpcAccess.class.getName()};
        break;
      case REST:
        //設(shè)置為REST，返回RestAccess組件
        importClassName = new String[]{RestAccess.class.getName()};
    }
    return importClassName;
  }
}

使用

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

@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實(shí)現(xiàn)注解驅(qū)動(dòng)

這里其它步驟一樣，主要區(qū)別是注解里面Import的類變了，這里是基于基于ImportBeanDefinitionRegistrar實(shí)現(xiàn)注解驅(qū)動(dòng)實(shí)現(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 ->{
          //注冊(cè)beanDefinition到beanDefinitionRegistry
          BeanDefinitionReaderUtils.registerWithGeneratedName(beanDefinition,beanDefinitionRegistry);
        });
  }
}

EnableAccess注解變更

這里import導(dǎo)入了AccessImportBeanDefinitionRegistrar

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

實(shí)現(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();
  }

}

以上就是如何在Springboot中使用enable模塊驅(qū)動(dòng)，小編相信有部分知識(shí)點(diǎn)可能是我們?nèi)粘９ぷ鲿?huì)見到或用到的。希望你能通過這篇文章學(xué)到更多知識(shí)。更多詳情敬請(qǐng)關(guān)注億速云行業(yè)資訊頻道。