public class GetBasePackage { private Class<? extends Annotation> annotation; public GetBasePackage(Class<? extends Annotation> annotation) { this.annotation = annotation; } public Stream getBasePackage(AnnotationMetadata annotationMetadata){ Map<String, Object> annotationAttributes = annotationMetadata.getAnnotationAttributes(annotation.getName()); AnnotationAttributes attributes = new AnnotationAttributes(annotationAttributes); String[] value = attributes.getStringArray("value");//annotationg中的注解 String[] basePackages = attributes.getStringArray("basePackages");//annotationg中的注解 String[] entityPaths = attributes.getStringArray("entityPath");//annotationg中的注解 //没配注解 if (value.length == 0 && basePackages.length == 0 && entityPaths.length == 0) { String className = annotationMetadata.getClassName(); return Stream.of(ClassUtils.getPackageName(className)); } //配了注解 Set<String> packages = new HashSet<>(); packages.addAll(Arrays.asList(value)); packages.addAll(Arrays.asList(basePackages)); packages.addAll(Arrays.asList(entityPaths)); return Stream.of(packages); } }
GetBasePackage getBasePackage = new GetBasePackage(EnableESTools.class); getBasePackage.getBasePackage(annotationMetadata).forEach((s) -> { System.out.println(s); });