com.github.dreamroute.mybatis.pro.autoconfiguration.MyBatisProAutoConfiguration Maven / Gradle / Ivy

Go to download
package com.github.dreamroute.mybatis.pro.autoconfiguration;

import com.github.dreamroute.mybatis.pro.base.typehandler.EnumTypeHandler;
import com.github.dreamroute.mybatis.pro.core.consts.MyBatisProProperties;
import com.github.dreamroute.mybatis.pro.interceptor.LogicalDeleteInterceptor;
import org.apache.ibatis.mapping.DatabaseIdProvider;
import org.apache.ibatis.plugin.Interceptor;
import org.apache.ibatis.scripting.LanguageDriver;
import org.apache.ibatis.session.SqlSessionFactory;
import org.apache.ibatis.type.TypeHandler;
import org.mybatis.spring.SqlSessionFactoryBean;
import org.mybatis.spring.annotation.MapperScan;
import org.mybatis.spring.boot.autoconfigure.*;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.BeanWrapperImpl;
import org.springframework.beans.factory.ObjectProvider;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.boot.autoconfigure.AutoConfigureAfter;
import org.springframework.boot.autoconfigure.AutoConfigureBefore;
import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
import org.springframework.boot.autoconfigure.condition.ConditionalOnSingleCandidate;
import org.springframework.boot.autoconfigure.jdbc.DataSourceAutoConfiguration;
import org.springframework.boot.context.properties.EnableConfigurationProperties;
import org.springframework.context.ApplicationContext;
import org.springframework.context.annotation.Bean;
import org.springframework.core.annotation.AnnotationUtils;
import org.springframework.core.io.Resource;
import org.springframework.core.io.ResourceLoader;
import org.springframework.util.CollectionUtils;
import org.springframework.util.ObjectUtils;
import org.springframework.util.StopWatch;
import org.springframework.util.StringUtils;

import javax.sql.DataSource;
import java.beans.PropertyDescriptor;
import java.util.*;
import java.util.stream.Stream;

import static com.github.dreamroute.mybatis.pro.core.consts.ToLineThreadLocal.TO_LINE;
import static com.github.dreamroute.mybatis.pro.core.util.MyBatisProUtil.buildMyBatisPro;
import static java.util.Arrays.asList;
import static java.util.Arrays.stream;
import static java.util.Optional.ofNullable;
import static java.util.stream.Collectors.toSet;
import static org.springframework.util.ObjectUtils.isEmpty;

/**
 * mybatis启动类
 *
 * @author w.dehai
 */
@ConditionalOnSingleCandidate(DataSource.class)
@AutoConfigureBefore({MybatisAutoConfiguration.class})
@ConditionalOnClass({SqlSessionFactory.class, SqlSessionFactoryBean.class})
@EnableConfigurationProperties({MybatisProperties.class, MyBatisProProperties.class})
@AutoConfigureAfter({DataSourceAutoConfiguration.class, MybatisLanguageDriverAutoConfiguration.class})
public class MyBatisProAutoConfiguration {

    private static final Logger logger = LoggerFactory.getLogger(MyBatisProAutoConfiguration.class);
    private final MybatisProperties properties;
    private final List interceptors;
    private final TypeHandler[] typeHandlers;
    private final LanguageDriver[] languageDrivers;
    private final ResourceLoader resourceLoader;
    private final DatabaseIdProvider databaseIdProvider;
    private final List configurationCustomizers;

    @Autowired
    private ApplicationContext context;
    @Value("${mybatis.configuration.map-underscore-to-camel-case:false}")
    private boolean toLine;

    public MyBatisProAutoConfiguration(
            MybatisProperties properties,
            ResourceLoader resourceLoader,
            ObjectProvider> interceptorsProvider,
            ObjectProvider typeHandlersProvider,
            ObjectProvider languageDriversProvider,
            ObjectProvider databaseIdProvider,
            ObjectProvider> configurationCustomizersProvider) {

        this.properties = properties;
        this.resourceLoader = resourceLoader;
        this.typeHandlers = typeHandlersProvider.getIfAvailable();
        this.databaseIdProvider = databaseIdProvider.getIfAvailable();
        this.languageDrivers = languageDriversProvider.getIfAvailable();
        this.configurationCustomizers = configurationCustomizersProvider.getIfAvailable();
        this.interceptors = ofNullable(interceptorsProvider.getIfAvailable()).orElseGet(ArrayList::new);
    }

    @Bean
    @ConditionalOnMissingBean
    public SqlSessionFactory sqlSessionFactory(DataSource dataSource, MyBatisProProperties props) throws Exception {
        SqlSessionFactoryBean factory = new SqlSessionFactoryBean();
        factory.setDataSource(dataSource);
        factory.setVfs(SpringBootVFS.class);
        if (StringUtils.hasText(this.properties.getConfigLocation())) {
            factory.setConfigLocation(this.resourceLoader.getResource(this.properties.getConfigLocation()));
        }
        applyConfiguration(factory);
        if (this.properties.getConfigurationProperties() != null) {
            factory.setConfigurationProperties(this.properties.getConfigurationProperties());
        }

        // 如果枚举处理器开启，那么加入到configuratin中
        if (props.isEnableEnumTypeHandler()) {
            properties.getConfiguration().getTypeHandlerRegistry().register(new EnumTypeHandler<>());
        }

        // 如果逻辑删除开启，这里将逻辑删除插件加入到插件列表
        if (props.isEnableLogicalDelete()) {
            interceptors.add(new LogicalDeleteInterceptor(props));
        }

        if (!ObjectUtils.isEmpty(this.interceptors)) {
            factory.setPlugins(this.interceptors.toArray(new Interceptor[0]));
        }

        if (this.databaseIdProvider != null) {
            factory.setDatabaseIdProvider(this.databaseIdProvider);
        }
        if (StringUtils.hasLength(this.properties.getTypeAliasesPackage())) {
            factory.setTypeAliasesPackage(this.properties.getTypeAliasesPackage());
        }
        if (this.properties.getTypeAliasesSuperType() != null) {
            factory.setTypeAliasesSuperType(this.properties.getTypeAliasesSuperType());
        }
        if (StringUtils.hasLength(this.properties.getTypeHandlersPackage())) {
            factory.setTypeHandlersPackage(this.properties.getTypeHandlersPackage());
        }
        if (!ObjectUtils.isEmpty(this.typeHandlers)) {
            factory.setTypeHandlers(this.typeHandlers);
        }

        Resource[] resources = this.properties.resolveMapperLocations();

        // -- mybatis-pro begin.
        logger.info("织入mybatis-pro开始 ......");
        StopWatch watch = new StopWatch();
        watch.start();

        TO_LINE.set(toLine);

        Set mapperPackages = getMapperPackages();
        if (!(isEmpty(resources) && isEmpty(mapperPackages))) {
            Resource[] rs = buildMyBatisPro(resources, mapperPackages);
            factory.setMapperLocations(rs);
        }

        TO_LINE.remove();

        logger.info("织入mybatis-pro结束 ......");
        watch.stop();
        logger.info("织入mybatis-pro耗时: {}", watch.getTotalTimeSeconds());
        // -- mybatis-pro end.

        Set factoryPropertyNames = Stream
                .of(new BeanWrapperImpl(SqlSessionFactoryBean.class).getPropertyDescriptors()).map(PropertyDescriptor::getName)
                .collect(toSet());
        Class defaultLanguageDriver = this.properties.getDefaultScriptingLanguageDriver();
        if (factoryPropertyNames.contains("scriptingLanguageDrivers") && !ObjectUtils.isEmpty(this.languageDrivers)) {
            // Need to mybatis-spring 2.0.2+
            factory.setScriptingLanguageDrivers(this.languageDrivers);
            if (defaultLanguageDriver == null && this.languageDrivers.length == 1) {
                defaultLanguageDriver = this.languageDrivers[0].getClass();
            }
        }
        if (factoryPropertyNames.contains("defaultScriptingLanguageDriver")) {
            // Need to mybatis-spring 2.0.2+
            factory.setDefaultScriptingLanguageDriver(defaultLanguageDriver);
        }

        return factory.getObject();
    }

    private void applyConfiguration(SqlSessionFactoryBean factory) {
        // 这里特殊说明一下：configuration是properties的一个嵌套属性，如果application.properties里面配置了configuration的属性，那么就会被初始化，否则这里的configuration就是null
        org.apache.ibatis.session.Configuration configuration = this.properties.getConfiguration();
        if (configuration == null && !StringUtils.hasText(this.properties.getConfigLocation())) {
            configuration = new org.apache.ibatis.session.Configuration();
        }
        if (configuration != null && !CollectionUtils.isEmpty(this.configurationCustomizers)) {
            for (ConfigurationCustomizer customizer : this.configurationCustomizers) {
                customizer.customize(configuration);
            }
        }
        factory.setConfiguration(configuration);
    }

    /**
     * 获取mapper接口的包路径集合
     */
    private Set getMapperPackages() {
        Set mapperPackages = new HashSet<>();
        // mapperScanMap如果为空会返回一个size = 0的Map
        Map mapperScanMap = context.getBeansWithAnnotation(MapperScan.class);
        mapperScanMap.values().forEach(scan -> {
            Class scanCls = scan.getClass();
            MapperScan ms = AnnotationUtils.findAnnotation(scanCls, MapperScan.class);
            if (ms != null) {
                String[] value = ms.value() != null ? ms.value() : new String[0];
                String[] basePackages = ms.basePackages() != null ? ms.basePackages() : new String[0];
                Class[] basePackageClasses = ms.basePackageClasses() != null ? ms.basePackageClasses() : new Class[0];

                mapperPackages.addAll(asList(value));
                mapperPackages.addAll(asList(basePackages));
                mapperPackages.addAll(stream(basePackageClasses).map(cls -> cls.getPackage().getName()).collect(toSet()));
            }
        });
        logger.info("MyBatis-Pro检测出Mapper路径包括: {}", mapperPackages);
        return mapperPackages;
    }

}