德鲁伊druid动态切换多数据源配置 mybatis+boot

女爷i 2022-12-26 04:56 111阅读 0赞

参照：[https://www.cnblogs.com/hsbt2333/p/9347249.html][https_www.cnblogs.com_hsbt2333_p_9347249.html]

1.pom文件，必备jar

<dependency>
                <groupId>mysql</groupId>
                <artifactId>mysql-connector-java</artifactId>
                <version>8.0.13</version>
            </dependency>
    
            <dependency>
                <groupId>com.baomidou</groupId>
                <artifactId>mybatis-plus-boot-starter</artifactId>
            </dependency>
    
            <dependency>
                <groupId>com.alibaba</groupId>
                <artifactId>druid-spring-boot-starter</artifactId>
                <version>${druid-version}</version>
            </dependency>
            <dependency>
                <groupId>org.projectlombok</groupId>
                <artifactId>lombok</artifactId>
                <version>1.18.12</version>
            </dependency>

1.1 yml文件配置数据源的实例

spring:
      datasource:
        druid:
          complex:
            platform: mysql
            url: jdbc:mysql://localhost:3306/complex_bus?useUnicode=true&characterEncoding=utf8&useSSL=false
            username: roo4
            password: 961151
            driver-class-name: com.mysql.cj.jdbc.Driver
            type: com.alibaba.druid.pool.DruidDataSource
            initialSize: 5
            minIdle: 5
            maxActive: 20
            maxWait: 60000
            timeBetweenEvictionRunsMillis: 60000
            minEvictableIdleTimeMillis: 300000
            validationQuery: SELECT 1 FROM DUAL
            testWhileIdle: true
            testOnBorrow: false
            testOnReturn: false
          complex-bus:
            platform: mysql
            url: jdbc:mysql://localhost:3306/complex_bus?useUnicode=true&characterEncoding=utf8&useSSL=false
            username: root
            password: 1561658
            driver-class-name: com.mysql.cj.jdbc.Driver
            type: com.alibaba.druid.pool.DruidDataSource
            initialSize: 5
            minIdle: 5
            maxActive: 20
            maxWait: 60000
            timeBetweenEvictionRunsMillis: 60000
            minEvictableIdleTimeMillis: 300000
            validationQuery: SELECT 1 FROM DUAL
            testWhileIdle: true
            testOnBorrow: false
            testOnReturn: false

2. 配置声明动态数据源

package com.linkcheers.mark.config.datasource;
    
    import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
    
    /**
     * 声明动态数据源
     * fl
     */
    public class MultipleDataSource extends AbstractRoutingDataSource {
    
        @Override
        protected Object determineCurrentLookupKey() {
            return DataSourceContextHolder.getDataSource();
        }
    }

3.设置线程池本地变量操作数据源

public class DataSourceContextHolder {
    
        private static final ThreadLocal<String> contextHolder = new InheritableThreadLocal<>();
    
        /**
         *  设置数据源
         * @param db
         */
        public static void setDataSource(String db){
            contextHolder.set(db);
        }
    
        /**
         * 取得当前数据源
         * @return
         */
        public static String getDataSource(){
            return contextHolder.get();
        }
    
        /**
         * 清除上下文数据
         */
        public static void clear(){
            contextHolder.remove();
        }
    
    }

4.德鲁伊数据库sql监控配置类

package com.linkcheers.mark.config.datasource;
    
    import com.alibaba.druid.support.http.StatViewServlet;
    import com.alibaba.druid.support.http.WebStatFilter;
    import org.springframework.boot.web.servlet.FilterRegistrationBean;
    import org.springframework.boot.web.servlet.ServletRegistrationBean;
    import org.springframework.context.annotation.Bean;
    import org.springframework.context.annotation.Configuration;
    
    import java.util.Arrays;
    import java.util.HashMap;
    import java.util.Map;
    
    @Configuration
    public class DruidSourceConfig {
    
        /**
         * 配置 Druid 监控 之  管理后台的 Servlet
         */
        @Bean
        public ServletRegistrationBean statViewServlet() {
            ServletRegistrationBean servletRegistrationBean = new ServletRegistrationBean(new StatViewServlet(), "/druid/*");
    
            servletRegistrationBean.addInitParameter("resetEnable", "false");
    
            /*loginUsername：Druid 后台管理界面的登录账号
            loginPassword：Druid 后台管理界面的登录密码
            allow：Druid 后台允许谁可以访问
            initParams.put("allow", "localhost")：表示只有本机可以访问
            initParams.put("allow", "")：为空或者为null时，表示允许所有访问
            deny：Druid 后台拒绝谁访问
            initParams.put("deny", "192.168.1.20");表示禁止此ip访问*/
    
            //设置初始化参数
            Map<String, String> initParams = new HashMap<>();
            initParams.put("loginUsername", "admin");
            initParams.put("loginPassword", "admin");
            //是否能够重置数据 禁用HTML页面上的“Reset All”功能
            servletRegistrationBean.addInitParameter("resetEnable", "false");
    
            servletRegistrationBean.setInitParameters(initParams);
            return servletRegistrationBean;
        }
    
        /**
         * 配置 Druid 监控 之  web 监控的 filter
         * WebStatFilter：用于配置Web和Druid数据源之间的管理关联监控统计
         */
        @Bean
        public FilterRegistrationBean webStatFilter() {
            FilterRegistrationBean bean = new FilterRegistrationBean();
            bean.setFilter(new WebStatFilter());
    
            //exclusions：设置哪些请求进行过滤排除掉，从而不进行统计
             /*Map<String, String> initParams = new HashMap<>();
             initParams.put("exclusions", "*.js,*.css,/druid/*");
             bean.setInitParameters(initParams);*/
    
            //表示过滤所有请求
            bean.setUrlPatterns(Arrays.asList("/*"));
            return bean;
        }
    
    }

5.数据库定义为枚举值

package com.linkcheers.mark.config.datasource;
    
    /**
     * 枚举类定义数据库名称
     */
    public enum DataSourceEnum {
    
        Complex("complex"),
        ComplexBus("complex_bus");
    
        private String value;
    
        DataSourceEnum(String value){this.value=value;}
    
        public String getValue() {
            return value;
        }
    }

6.定义注解植入在serviceImpl中的方法上

package com.linkcheers.mark.config.datasource;
    
    import java.lang.annotation.*;
    
    @Target({ElementType.METHOD,ElementType.TYPE})
    @Retention(RetentionPolicy.RUNTIME)
    @Documented
    public @interface DataSource {
    
        DataSourceEnum value() default DataSourceEnum.ComplexBus;
    }

7.定义切面类，注解生效

package com.linkcheers.mark.config.datasource;
    
    import lombok.extern.slf4j.Slf4j;
    import org.apache.logging.log4j.core.config.Order;
    import org.aspectj.lang.annotation.After;
    import org.aspectj.lang.annotation.Aspect;
    import org.aspectj.lang.annotation.Before;
    import org.aspectj.lang.annotation.Pointcut;
    import org.springframework.stereotype.Component;
    
    @Component
    @Slf4j
    @Aspect
    @Order(-1)
    public class DataSourceAspect {
    
        @Pointcut("@within(com.linkcheers.mark.config.datasource.DataSource) || @annotation(com.linkcheers.mark.config.datasource.DataSource)")
        public void pointCut(){
    
        }
    
        @Before("pointCut() && @annotation(dataSource)")
        public void doBefore(DataSource dataSource){
            log.info("选择数据源---"+dataSource.value().getValue());
            DataSourceContextHolder.setDataSource(dataSource.value().getValue());
        }
    
        @After("pointCut()")
        public void doAfter(){
            DataSourceContextHolder.clear();
        }
    }

8.多数据配置

package com.linkcheers.mark.config;
    
    import com.alibaba.druid.spring.boot.autoconfigure.DruidDataSourceBuilder;
    import com.baomidou.mybatisplus.core.MybatisConfiguration;
    import com.baomidou.mybatisplus.extension.plugins.PaginationInterceptor;
    import com.baomidou.mybatisplus.extension.spring.MybatisSqlSessionFactoryBean;
    import com.linkcheers.mark.config.datasource.DataSourceEnum;
    import com.linkcheers.mark.config.datasource.MultipleDataSource;
    import org.apache.ibatis.plugin.Interceptor;
    import org.apache.ibatis.plugin.Invocation;
    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.jdbc.datasource.DataSourceTransactionManager;
    
    import java.util.HashMap;
    import javax.sql.DataSource;
    import java.util.Map;
    
    @Configuration
    @MapperScan(basePackages = "com.linkcheers.mark.dao")
    public class mybatisConfig {
    
        @Bean
        public PaginationInterceptor paginationInnerInterceptor(){
            return new PaginationInterceptor();
        }
    
        @Bean(name = "complex")
        @ConfigurationProperties(prefix = "spring.datasource.druid.complex" )
        public DataSource complex() {
            return DruidDataSourceBuilder.create().build();
        }
    
        @Bean(name = "complexBus")
        @ConfigurationProperties(prefix = "spring.datasource.druid.complex-bus" )
        public DataSource complexBus() {
            return DruidDataSourceBuilder.create().build();
        }
    
        /**
         * 数据源添加
         * @return
         */
        @Bean
        @Primary
        public DataSource multipleDataSource(@Qualifier("complex") DataSource complex, @Qualifier("complexBus") DataSource complexBus) {
            MultipleDataSource multipleDataSource = new MultipleDataSource();
            Map< Object, Object > targetDataSources = new HashMap<>();
            targetDataSources.put(DataSourceEnum.Complex.getValue(), complex);
            targetDataSources.put(DataSourceEnum.ComplexBus.getValue(), complexBus);
            //添加数据源
            multipleDataSource.setTargetDataSources(targetDataSources);
            //设置默认数据源
            multipleDataSource.setDefaultTargetDataSource(complexBus);
            return multipleDataSource;
        }
    
    
        @Bean("sqlSessionFactory")
        public SqlSessionFactory sqlSessionFactory() throws Exception {
            MybatisSqlSessionFactoryBean sqlSessionFactory = new MybatisSqlSessionFactoryBean();
            sqlSessionFactory.setDataSource(multipleDataSource(complex(),complexBus()));
            //这里注意 如果你用了mybatis的xml的文件查询，路径要改
            sqlSessionFactory.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("classpath:/mapper/*/*Mapper.xml"));
    
            MybatisConfiguration configuration = new MybatisConfiguration();
            configuration.setJdbcTypeForNull(JdbcType.NULL);
            configuration.setMapUnderscoreToCamelCase(true);
            configuration.setCacheEnabled(false);
            sqlSessionFactory.setConfiguration(configuration);
            sqlSessionFactory.setPlugins(new Interceptor() {
                @Override
                public Object intercept(Invocation invocation){
                    return paginationInnerInterceptor();
                }
            });
            return sqlSessionFactory.getObject();
        }
    
        // 配置事物管理器
        @Bean(name="transactionManager")
        public DataSourceTransactionManager transactionManager(){
            DataSourceTransactionManager transactionManager = new DataSourceTransactionManager();
            transactionManager.setDataSource(complex());
            transactionManager.setDataSource(complexBus());
            return transactionManager;
        }
    
        /**
         * SQL执行效率插件
         */
        /*@Bean
        @Profile({"dev","test"})// 设置 dev test 环境开启
        public PerformanceInterce performanceInterceptor() {
            PerformanceInterceptor performanceInterceptor = new PerformanceInterceptor();
            performanceInterceptor.setMaxTime(1000);
            performanceInterceptor.setFormat(true);
            return performanceInterceptor;
        }*/
    }

9.多数据的使用

package com.linkcheers.mark.service.impl;
    
    import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
    import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
    import com.linkcheers.mark.config.datasource.DataSource;
    import com.linkcheers.mark.config.datasource.DataSourceEnum;
    import com.linkcheers.mark.dao.TbPoliceDao;
    import com.linkcheers.mark.entity.TbPolice;
    import com.linkcheers.mark.service.TbPoliceService;
    import com.linkcheers.mark.vo.TbPoliceVo;
    import org.apache.commons.lang3.StringUtils;
    import org.springframework.stereotype.Service;
    
    
    /**
     * (TbPolice)表服务实现类
     *
     * @author fl
     * @since 2020-12-03 10:48:28
     */
    @Service("tbPoliceService")
    public class TbPoliceServiceImpl extends ServiceImpl<TbPoliceDao, TbPolice> implements TbPoliceService {
    
        @DataSource(value = DataSourceEnum.ComplexBus)
        @Override
        public TbPoliceVo oneByPoliceNo(String policeNo) {
            QueryWrapper<TbPolice> wrapper = new QueryWrapper();
            wrapper.lambda().eq(TbPolice::getJycode,policeNo);
            return this.baseMapper.oneByPoliceNo(wrapper);
        }
    
    }

完结.

[https_www.cnblogs.com_hsbt2333_p_9347249.html]: https://www.cnblogs.com/hsbt2333/p/9347249.html