我想实现类似于 Spring Data 的东西。

开发人员可以定义一些接口(interface)，向接口(interface)添加自定义注释以标记它们，(我的代码将为接口(interface)创建 Proxy 实例)并通过@Autowire 将它们用于必要的服务。

在 spring 初始化期间，我需要获取所有接口(interface)的列表(正确注释)<>

代理创建，创建的 bean 注入(inject)没问题。现在的问题:

如何找到所有接口(interface)的列表？

它们可以放在任何包中(甚至放在单独的 jar 中)并具有任何名称。扫描类路径中存在的所有类需要太多时间。

我找到了 the question但它需要基础包才能启动。

尝试了一个基于反射的解决方案，但同样它需要基础包，或者在从根目录开始的情况下需要大量时间来扫描所有可用的类。

Reflections reflections = new Reflections("...");
Set<Class<?>> annotated = reflections.getTypesAnnotatedWith(<annotation>);

所以我需要 Spring 扫描的基本包的完整列表以在包中找到我的接口(interface)(必须快得多)。

该信息肯定在 SpringContext 中可用。我尝试调试并查看 basePackages[] 是如何初始化的，但是有很多私有(private)类/方法用于初始化，我只是看不到如何从 ApplicationContext 正确访问 basePackages。

最佳答案

方案一: Spring 方式

最简单的答案是遵循 spring 子项目(boot、data...)如何实现此类要求。他们通常定义一个自定义组合注释来启用该功能并定义一组要扫描的包。

例如给出这个注解:

@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
@Import({MyInterfaceScanRegistrar.class})
public @interface MyInterfaceScan {

  String[] value() default {};
}

其中 value 定义要扫描的包，@Import 启用 MyInterfaceScan 检测。

然后创建ImportBeanDefinitionRegistrar .这个类将能够创建bean定义

Interface to be implemented by types that register additional bean definitions when processing @Configuration classes. Useful when operating at the bean definition level (as opposed to @Bean method/instance level) is desired or necessary.

public class MyInterfaceScanRegistrar implements ImportBeanDefinitionRegistrar, EnvironmentAware {
  private Environment environment;

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

  @Override
  public void registerBeanDefinitions(AnnotationMetadata metadata, BeanDefinitionRegistry registry) {
    // Get the MyInterfaceScan annotation attributes
    Map<String, Object> annotationAttributes = metadata.getAnnotationAttributes(MyInterfaceScan.class.getCanonicalName());

    if (annotationAttributes != null) {
      String[] basePackages = (String[]) annotationAttributes.get("value");

      if (basePackages.length == 0){
        // If value attribute is not set, fallback to the package of the annotated class
        basePackages = new String[]{((StandardAnnotationMetadata) metadata).getIntrospectedClass().getPackage().getName()};
      }

      // using these packages, scan for interface annotated with MyCustomBean
      ClassPathScanningCandidateComponentProvider provider = new ClassPathScanningCandidateComponentProvider(false, environment){
        // Override isCandidateComponent to only scan for interface
        @Override
        protected boolean isCandidateComponent(AnnotatedBeanDefinition beanDefinition) {
          AnnotationMetadata metadata = beanDefinition.getMetadata();
          return metadata.isIndependent() && metadata.isInterface();
        }
      };
      provider.addIncludeFilter(new AnnotationTypeFilter(MyCustomBean.class));

      // Scan all packages
      for (String basePackage : basePackages) {
        for (BeanDefinition beanDefinition : provider.findCandidateComponents(basePackage)) {
          // Do the stuff about the bean definition
          // For example, redefine it as a bean factory with custom atribute... 
          // then register it
          registry.registerBeanDefinition(generateAName() , beanDefinition);
          System.out.println(beanDefinition);
        }
      }
    }
  }
}

这是逻辑的核心。 bean 定义可以被操作并重新定义为具有属性的 bean 工厂，或者使用从接口(interface)生成的类重新定义。

MyCustomBean 是一个简单的注解:

@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
public @interface MyCustomBean {

}

它可以注释一个接口(interface):

@MyCustomBean
public interface Class1 {

}

方案二:提取组件扫描

提取@ComponentScan中定义的包的代码会更复杂。

你应该创建一个 BeanDefinitionRegistryPostProcessor并模仿 ConfigurationClassPostProcessor :

• 使用具有 ComponentScan 属性的已声明类迭代 bean 注册表以获取 bean 定义，例如(从 ConfigurationClassPostProcessor 中提取。):

  public void postProcessBeanDefinitionRegistry(BeanDefinitionRegistry registry) {
    List<BeanDefinitionHolder> configCandidates = new ArrayList<BeanDefinitionHolder>();
    String[] candidateNames = registry.getBeanDefinitionNames();
    for (String beanName : candidateNames) {
      if (ConfigurationClassUtils.checkConfigurationClassCandidate(beanDef, this.metadataReaderFactory)) {
        // Extract component scan
      }
    }
  }
  

• 像 Spring 那样提取这些属性

  Set<AnnotationAttributes> componentScans = AnnotationConfigUtils.attributesForRepeatable(
          sourceClass.getMetadata(), ComponentScans.class, ComponentScan.class);
  

• 然后像第一个解决方案一样扫描包并注册bean定义

关于java - 如何从@ComponentScan 包中获取接口(interface)列表，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/43538429/