Spring加载早期获取BasePackage

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);
});

猜你喜欢

转载自www.cnblogs.com/zxporz/p/11452026.html