com.circustar.mybatis_accessor.relation.ScanRelationOnStartup Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of mybatis-accessor Show documentation
enhancement based on mybatis plus
The newest version!
package com.circustar.mybatis_accessor.relation;

import com.circustar.mybatis_accessor.annotation.scan.DtoEntityRelation;
import com.circustar.mybatis_accessor.annotation.scan.DtoEntityRelations;
import com.circustar.mybatis_accessor.annotation.scan.RelationScanPackages;
import com.circustar.mybatis_accessor.annotation.scan.EnableMybatisAccessor;
import org.springframework.beans.factory.config.BeanDefinition;
import org.springframework.boot.ApplicationArguments;
import org.springframework.boot.ApplicationRunner;
import org.springframework.context.ApplicationContext;
import org.springframework.context.annotation.ClassPathScanningCandidateComponentProvider;
import org.springframework.core.type.filter.AnnotationTypeFilter;

import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;

public class ScanRelationOnStartup implements ApplicationRunner {
    private final IEntityDtoServiceRelationMap relationMap;
    private final ApplicationContext context;

    public ScanRelationOnStartup(ApplicationContext context, IEntityDtoServiceRelationMap relationMap) {
        this.context = context;
        this.relationMap = relationMap;
    }

    private Class getTargetClass(Class clazz) {
        if(clazz == null) {
            return null;
        } else if(clazz.getName().contains("$$")) {
            return getTargetClass(clazz.getSuperclass());
        }
        return clazz;
    }

    @Override
    public void run(ApplicationArguments args) throws Exception {
        Map beans = context.getBeansWithAnnotation(RelationScanPackages.class);
        for(Map.Entry target : beans.entrySet()) {
            Class targetClass = getTargetClass(target.getValue().getClass());
            RelationScanPackages[] annotations = targetClass.getAnnotationsByType(RelationScanPackages.class);
            scanForRelationMap(Arrays.stream(annotations).map(x -> x.value())
                    .flatMap(Arrays::stream)
                    .collect(Collectors.toList()));
        }
        Map enableAutoController = context.getBeansWithAnnotation(EnableMybatisAccessor.class);
        for(Map.Entry classEntry : enableAutoController.entrySet()) {
            Class targetClass = getTargetClass(classEntry.getValue().getClass());
            EnableMybatisAccessor[] annotations = targetClass.getAnnotationsByType(EnableMybatisAccessor.class);
            List scannList = Arrays.stream(annotations).map(x -> x.relationScan().value())
                    .flatMap(Arrays::stream)
                    .collect(Collectors.toList());
            scanForRelationMap(scannList);
        }
    }

    protected void scanForRelationMap(List packageNameList) throws ClassNotFoundException {
        ClassPathScanningCandidateComponentProvider componentProvider = new ClassPathScanningCandidateComponentProvider(false);
        componentProvider.addIncludeFilter(new AnnotationTypeFilter(DtoEntityRelation.class));
        componentProvider.addIncludeFilter(new AnnotationTypeFilter(DtoEntityRelations.class));
        for(String packageName : packageNameList) {
            Set definitionSet = componentProvider.findCandidateComponents(packageName);
            for(BeanDefinition bd: definitionSet) {
                Class clazz = Class.forName(bd.getBeanClassName());
                DtoEntityRelation[] dtoEntityRelations = clazz.getAnnotationsByType(DtoEntityRelation.class);
                Arrays.stream(dtoEntityRelations)
                        .forEach(x -> relationMap.addRelation(new EntityDtoServiceRelation(
                                x.dtoClass() == Void.class?clazz : x.dtoClass()
                                , x.entityClass() == Void.class?clazz : x.entityClass()
                                , x.name()
                                , x.service(), x.convertDtoToEntityClazz(), x.convertEntityToDtoClazz())));

            }
        }
    }
}