标记注解

Author: yungyu16

src/main/java/com/github/yungyu16/spring/stub/StubBeanDefinitionRegistrar.java

@@ -1,7 +1,9 @@
 package com.github.yungyu16.spring.stub;
 
+import com.github.yungyu16.spring.stub.annotation.ProxyStub;
 import com.github.yungyu16.spring.stub.annotation.ProxyStubScan;
 import com.github.yungyu16.spring.stub.support.ClassPathStubBeanDefinitionScanner;
+import lombok.SneakyThrows;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.support.BeanDefinitionRegistry;
 import org.springframework.context.EnvironmentAware;
@@ -10,9 +12,11 @@
 import org.springframework.core.annotation.AnnotationAttributes;
 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 org.springframework.util.StringUtils;
 
+import java.lang.annotation.Annotation;
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.Map;
@@ -26,6 +30,7 @@ public class StubBeanDefinitionRegistrar implements ImportBeanDefinitionRegistra
 
     private Environment environment;
 
+    @SneakyThrows
     @Override
     public void registerBeanDefinitions(AnnotationMetadata importingClassMetadata, BeanDefinitionRegistry registry) {
         Map<String, Object> annotationAttributes = importingClassMetadata.getAnnotationAttributes(ProxyStubScan.class.getName(), true);
@@ -46,6 +51,14 @@ public void registerBeanDefinitions(AnnotationMetadata importingClassMetadata, B
             pkgList.add(ClassUtils.getPackageName(importingClassMetadata.getClassName()));
         }
         ClassPathBeanDefinitionScanner scanner = new ClassPathStubBeanDefinitionScanner(registry, environment);
+        String markAnnotation = attrs.getString("markAnnotation");
+        if (!StringUtils.isEmpty(markAnnotation)) {
+            scanner.resetFilters(false);
+            AnnotationTypeFilter proxyStubFilter = new AnnotationTypeFilter(ProxyStub.class, true, false);
+            @SuppressWarnings("unchecked")
+            AnnotationTypeFilter markAnnotationFilter = new AnnotationTypeFilter((Class<? extends Annotation>) ClassUtils.forName(markAnnotation, null), true, false);
+            scanner.addIncludeFilter((metadataReader, metadataReaderFactory) -> proxyStubFilter.match(metadataReader, metadataReaderFactory) && markAnnotationFilter.match(metadataReader, metadataReaderFactory));
+        }
         int bdCnt = scanner.scan(StringUtils.toStringArray(pkgList));
         log.debug("本轮从{}中扫描到{}个SpringStub", pkgList, bdCnt);
     }

src/main/java/com/github/yungyu16/spring/stub/annotation/ProxyStubScan.java

@@ -14,6 +14,8 @@
 @Retention(RetentionPolicy.RUNTIME)
 @Documented
 @Import(StubBeanDefinitionRegistrar.class)
+//TODO 支持重复注解
+//@Repeatable(ProxyStubScans.class)
 public @interface ProxyStubScan {
     @AliasFor("basePackages")
     String[] value() default {};
@@ -22,4 +24,11 @@
     String[] basePackages() default {};
 
     Class<?>[] basePackageClasses() default {};
+
+    /**
+     * 标记注解
+     *
+     * @return
+     */
+    Class<? extends Annotation> markAnnotation() default Annotation.class;
 }

src/main/java/com/github/yungyu16/spring/stub/annotation/ProxyStubScans.java

@@ -0,0 +1,17 @@
+package com.github.yungyu16.spring.stub.annotation;
+
+import com.github.yungyu16.spring.stub.StubBeanDefinitionRegistrar;
+import org.springframework.context.annotation.Import;
+
+import java.lang.annotation.*;
+
+/**
+ * CreatedDate: 2020/11/24
+ * Author: songjialin
+ */
+@Target(ElementType.TYPE)
+@Retention(RetentionPolicy.RUNTIME)
+@Documented
+public @interface ProxyStubScans {
+    ProxyStubScan[] value() default {};
+}