mybatisplus自定义mapper不生效问题

系统管理员 2022-09-02 11:52 285阅读 0赞

package com.hanshow.wise.base.powerbi.config;
    
    import com.alibaba.druid.spring.boot.autoconfigure.DruidDataSourceBuilder;
    import com.baomidou.mybatisplus.annotation.DbType;
    import com.baomidou.mybatisplus.core.MybatisConfiguration;
    import com.baomidou.mybatisplus.extension.plugins.PaginationInterceptor;
    import com.baomidou.mybatisplus.extension.plugins.PerformanceInterceptor;
    import com.baomidou.mybatisplus.extension.spring.MybatisSqlSessionFactoryBean;
    import com.hanshow.wise.base.powerbi.enums.DataSourceEnum;
    import com.hanshow.wise.base.powerbi.multiple.MultipleDataSource;
    import org.apache.ibatis.plugin.Interceptor;
    import org.apache.ibatis.session.SqlSessionFactory;
    import org.apache.ibatis.type.JdbcType;
    import org.mybatis.spring.annotation.MapperScan;
    import org.springframework.beans.factory.annotation.Qualifier;
    import org.springframework.boot.context.properties.ConfigurationProperties;
    import org.springframework.context.annotation.Bean;
    import org.springframework.context.annotation.Configuration;
    import org.springframework.context.annotation.Primary;
    import org.springframework.context.annotation.Profile;
    import org.springframework.core.io.Resource;
    import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
    import org.springframework.core.io.support.ResourcePatternResolver;
    
    import javax.sql.DataSource;
    import java.io.IOException;
    import java.util.HashMap;
    import java.util.Map;
    import java.util.Optional;
    import java.util.stream.Stream;
    
    @Configuration
    @MapperScan("com.hanshow.wise.base.powerbi.mapper")
    public class MyBatiesPlusConfiguration {
        @Bean
        public PaginationInterceptor paginationInterceptor() {
            PaginationInterceptor paginationInterceptor = new PaginationInterceptor();
            paginationInterceptor.setDialectType(DbType.MYSQL.getDb());
            return paginationInterceptor;
        }
    
        /**
         * SQL执行效率插件
         */
        @Bean
        @Profile({"dev", "qa"})// 设置 dev test 环境开启
        public PerformanceInterceptor performanceInterceptor() {
            PerformanceInterceptor performanceInterceptor = new PerformanceInterceptor();
            performanceInterceptor.setMaxTime(1000);
            performanceInterceptor.setFormat(true);
            return performanceInterceptor;
        }
    
        @Bean(name = "db1")
        @ConfigurationProperties(prefix = "spring.datasource.druid.db1")
        public DataSource db1() {
            return DruidDataSourceBuilder.create().build();
        }
    
        @Bean(name = "db2")
        @ConfigurationProperties(prefix = "spring.datasource.druid.db2")
        public DataSource db2() {
            return DruidDataSourceBuilder.create().build();
        }
    
        @Bean(name = "db3")
        @ConfigurationProperties(prefix = "spring.datasource.druid.db3")
        public DataSource db3() {
            return DruidDataSourceBuilder.create().build();
        }
    
        /**
         * 动态数据源配置
         *
         * @return
         */
        @Bean
        @Primary
        public DataSource multipleDataSource(@Qualifier("db1") DataSource db1, @Qualifier("db2") DataSource db2, @Qualifier("db3") DataSource db3) {
            MultipleDataSource multipleDataSource = new MultipleDataSource();
            Map<Object, Object> targetDataSources = new HashMap<>();
            targetDataSources.put(DataSourceEnum.DB1.getValue(), db1);
            targetDataSources.put(DataSourceEnum.DB2.getValue(), db2);
            targetDataSources.put(DataSourceEnum.DB3.getValue(), db3);
            //添加数据源
            multipleDataSource.setTargetDataSources(targetDataSources);
            //设置默认数据源
            multipleDataSource.setDefaultTargetDataSource(db1);
            return multipleDataSource;
        }
        //解决mybitaspuls自定义mapper不生效问题
        @Bean("sqlSessionFactory")
        public SqlSessionFactory sqlSessionFactory() throws Exception {
            MybatisSqlSessionFactoryBean sqlSessionFactory = new MybatisSqlSessionFactoryBean();
            sqlSessionFactory.setDataSource(multipleDataSource(db1(), db2(), db3()));
            //sqlSessionFactory.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("classpath:/mapper/*/*Mapper.xml"));
    
            MybatisConfiguration configuration = new MybatisConfiguration();
            //configuration.setDefaultScriptingLanguage(MybatisXMLLanguageDriver.class);
            configuration.setJdbcTypeForNull(JdbcType.NULL);
            configuration.setMapUnderscoreToCamelCase(true);
            configuration.setCacheEnabled(false);
            configuration.addMappers("com.hanshow.wise.base.powerbi.mapper");
            sqlSessionFactory.setConfiguration(configuration);
            String[] mapperLocations = new String[]{"mapper/*.xml"};
            sqlSessionFactory.setMapperLocations(resolveMapperLocations(mapperLocations));
            sqlSessionFactory.setPlugins(new Interceptor[]{ //PerformanceInterceptor(),OptimisticLockerInterceptor()
                    paginationInterceptor() //添加分页功能
            });
            //sqlSessionFactory.setGlobalConfig(globalConfiguration());
            return sqlSessionFactory.getObject();
        }
    
        private Resource[] resolveMapperLocations(String[] mapperLocations) {
            return Stream.of(Optional.ofNullable(mapperLocations).orElse(new String[0]))
                    .flatMap(location -> Stream.of(getResources(location))).toArray(Resource[]::new);
        }
    
        ResourcePatternResolver resourceResolver = new PathMatchingResourcePatternResolver();
    
        private Resource[] getResources(String location) {
            try {
                return resourceResolver.getResources(location);
            } catch (IOException e) {
                return new Resource[0];
            }
        }
        /*@Bean
        public GlobalConfiguration globalConfiguration() {
            GlobalConfiguration conf = new GlobalConfiguration(new LogicSqlInjector());
            conf.setLogicDeleteValue("-1");
            conf.setLogicNotDeleteValue("1");
            conf.setIdType(0);
            //conf.setMetaObjectHandler(new MyMetaObjectHandler());
            conf.setDbColumnUnderline(true);
            conf.setRefresh(true);
            return conf;
        }*/
    }