Spring4.0 + druid 配置动态配置数据源以及多数据源切换功能实现

Stella981
• 阅读 727

数据源连接池使用druid 其他的数据源基本原理相同

spring中配置默认数据源连接池如下:

<!-- 数据源配置, 使用 BoneCP 数据库连接池 -->
    <bean id="dataSourceOne" class="com.alibaba.druid.pool.DruidDataSource" init-method="init" destroy-method="close"> 
        <property name="name" value="dataSourceOne"/>
        <!-- 数据源驱动类可不写,Druid默认会自动根据URL识别DriverClass -->
        <property name="driverClassName" value="${jdbc.driver}" />
        
        <!-- 基本属性 url、user、password -->
        <property name="url" value="${jdbc.url}" />
        <property name="username" value="${jdbc.username}" />
        <property name="password" value="${jdbc.password}" />
        
        <!-- 配置初始化大小、最小、最大 -->
        <property name="initialSize" value="${jdbc.pool.init}" />
        <property name="minIdle" value="${jdbc.pool.minIdle}" /> 
        <property name="maxActive" value="${jdbc.pool.maxActive}" />
        
        <!-- 配置获取连接等待超时的时间 -->
        <property name="maxWait" value="60000" />
        
        <!-- 配置间隔多久才进行一次检测,检测需要关闭的空闲连接,单位是毫秒 -->
        <property name="timeBetweenEvictionRunsMillis" value="60000" />
        
        <!-- 配置一个连接在池中最小生存的时间,单位是毫秒 -->
        <property name="minEvictableIdleTimeMillis" value="300000" />
        
        <property name="validationQuery" value="${jdbc.testSql}" />
        <property name="testWhileIdle" value="true" />
        <property name="testOnBorrow" value="false" />
        <property name="testOnReturn" value="false" />
        
        <!-- 打开PSCache,并且指定每个连接上PSCache的大小(Oracle使用)
        <property name="poolPreparedStatements" value="true" />
        <property name="maxPoolPreparedStatementPerConnectionSize" value="20" /> -->
        
        <!-- 配置监控统计拦截的filters -->
        <property name="filters" value="stat" /> 
    </bean>

接下来配置多数据源bean

<!-- 多数据源配置 -->
     <bean id="dynamicDataSource" class="com.XXX.datasource.DynamicDataSource" >  
        <property name="targetDataSources">  
            <map>  
                <entry value-ref="dataSourceOne" key="dataSourceOne"></entry>  

                <!--此处是对数据源的引用-->
               <!--  <entry value-ref="dataSourceTow" key="dataSourceTow"></entry> -->
            </map>  
        </property>  
        <property name="defaultTargetDataSource" ref="dataSourceOne" />  
        <property name="debug"  value="true"/>
    </bean> 

这个类 com.XXX.datasource.DynamicDataSource 需要手动创建

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.util.Map;
import java.util.Properties;
import java.util.Set;

import javax.sql.DataSource;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
import org.springframework.util.StringUtils;

import com.alibaba.druid.pool.DruidConnectionHolder;
import com.alibaba.druid.pool.DruidDataSource;
import com.alibaba.druid.pool.DruidDataSourceFactory;
import com.alibaba.druid.pool.DruidPooledConnection;
import com.alibaba.druid.stat.DruidDataSourceStatManager;
import com.alibaba.druid.util.DruidDataSourceUtils;

/**
 * @author zh
 */
public class DynamicDataSource extends AbstractRoutingDataSource{

    private boolean debug = false;
    Logger log = LoggerFactory.getLogger(this.getClass());
    private Map<Object, Object> dynamicTargetDataSources;

    private Object dynamicDefaultTargetDataSource;
    /* (non-Javadoc)
@see org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource#determineCurrentLookupKey()
     */
    @Override
    protected Object determineCurrentLookupKey() {
         String datasource=DBContextHolder.getDataSource();
         if(debug)
         {
             if(StringUtils.isEmpty(datasource)){
                log.info("---当前数据源:默认数据源---"); 
             }else{
                 log.info("---当前数据源:"+datasource+"---"); 
             }
         }
        
         return datasource;
    }
    
    @Override
    public void setTargetDataSources(Map<Object, Object> targetDataSources) {
        super.setTargetDataSources(targetDataSources);
        this.dynamicTargetDataSources = targetDataSources;
    }
    //创建数据源
    public boolean createDataSource(String key,String driveClass,String url,String username,String password){
        try {
            try {    //排除连接不上的错误
                Class.forName(driveClass);  
                DriverManager.getConnection(url, username, password);
            } catch (Exception e) {
                return false;
            }
            @SuppressWarnings("resource")
            DruidDataSource druidDataSource = new DruidDataSource();
            druidDataSource.setName(key);
            druidDataSource.setDriverClassName(driveClass);
            druidDataSource.setUrl(url);
            druidDataSource.setUsername(username);
            druidDataSource.setPassword(password);
            druidDataSource.setMaxWait(60000);
            druidDataSource.setFilters("stat");
            DataSource createDataSource = (DataSource)druidDataSource;
            druidDataSource.init();
            Map<Object, Object> dynamicTargetDataSources2 =  this.dynamicTargetDataSources;
            dynamicTargetDataSources2.put(key, createDataSource);//加入map
            setTargetDataSources(dynamicTargetDataSources2);//将map赋值给父类的TargetDataSources
            super.afterPropertiesSet();//将TargetDataSources中的连接信息放入resolvedDataSources管理
            return true;
        } catch (Exception e) {
            log.error(e+"");
            return false;
        }
    }
    //删除数据源
    public boolean delDatasources(String datasourceid){
        Map<Object, Object> dynamicTargetDataSources2 =  this.dynamicTargetDataSources;
        if(dynamicTargetDataSources2.containsKey(datasourceid)){
            Set<DruidDataSource> druidDataSourceInstances = DruidDataSourceStatManager.getDruidDataSourceInstances();
            for(DruidDataSource l:druidDataSourceInstances){
                if(datasourceid.equals(l.getName())){
                    System.out.println(l);
                    dynamicTargetDataSources2.remove(datasourceid);
                    DruidDataSourceStatManager.removeDataSource(l);
                    setTargetDataSources(dynamicTargetDataSources2);//将map赋值给父类的TargetDataSources
                    super.afterPropertiesSet();//将TargetDataSources中的连接信息放入resolvedDataSources管理
                    return true;
                }
            }
            return false;
        }else{
            return false;
        }
    }
    
    //测试数据源连接是否有效
    public boolean testDatasource(String key,String driveClass,String url,String username,String password){
        try {
            Class.forName(driveClass);  
            DriverManager.getConnection(url, username, password);  
            return true;
        } catch (Exception e) {
            return false;
        }
    }
    /**
     * Specify the default target DataSource, if any.
     * <p>The mapped value can either be a corresponding {@link javax.sql.DataSource}
     * instance or a data source name String (to be resolved via a
     * {@link #setDataSourceLookup DataSourceLookup}).
     * <p>This DataSource will be used as target if none of the keyed
     * {@link #setTargetDataSources targetDataSources} match the
     * {@link #determineCurrentLookupKey()} current lookup key.
     */
    public void setDefaultTargetDataSource(Object defaultTargetDataSource) {
        super.setDefaultTargetDataSource(defaultTargetDataSource);
        this.dynamicDefaultTargetDataSource = defaultTargetDataSource;
    }
    /**
     * @param debug the debug to set
     */
    public void setDebug(boolean debug) {
        this.debug = debug;
    }

    /**
     * @return the debug
     */
    public boolean isDebug() {
        return debug;
    }

    /**
     * @return the dynamicTargetDataSources
     */
    public Map<Object, Object> getDynamicTargetDataSources() {
        return dynamicTargetDataSources;
    }

    /**
     * @param dynamicTargetDataSources the dynamicTargetDataSources to set
     */
    public void setDynamicTargetDataSources(
            Map<Object, Object> dynamicTargetDataSources) {
        this.dynamicTargetDataSources = dynamicTargetDataSources;
    }

    /**
     * @return the dynamicDefaultTargetDataSource
     */
    public Object getDynamicDefaultTargetDataSource() {
        return dynamicDefaultTargetDataSource;
    }
    
    /**
     * @param dynamicDefaultTargetDataSource the dynamicDefaultTargetDataSource to set
     */
    public void setDynamicDefaultTargetDataSource(
            Object dynamicDefaultTargetDataSource) {
        this.dynamicDefaultTargetDataSource = dynamicDefaultTargetDataSource;
    }
    
}

其中该类继承了spring的AbstractRoutingDataSource 查看其源码,发现所有的数据源都是通过

afterPropertiesSet() 将存放在targetDataSources 这个Map中的数据源赋值给resolvedDataSources

对象的,spring是从resolvedDataSources对象中获取数据源对象的,我们能需要把自己的数据源放入

resolvedDataSources这个Map中就ok了。

接下来创建数据源切换工具类

/**
 * 数据源切换
 * @author zh
 *
 */
public class DBContextHolder {
     private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();  
     
         //调用此方法,切换数据源
        public static void setDataSource(String dataSource) {  
            contextHolder.set(dataSource);  
        }  
          
        public static String getDataSource() {  
            return contextHolder.get();  
        }  
          
        public static void clearDataSource() {  
            contextHolder.remove();  
        }  
}

具体实动态新增数据源,需要创建数据库用以存储 数据库连接信息,以及数据源key信息。

初始话数据库连接数据源,可以使用spring监听 实现ApplicationListener即可,如下

import java.util.List;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.ApplicationEvent;
import org.springframework.context.ApplicationListener;

import com.casking.cdds.modules.datasource.entity.CNDatasources;

public class InitDatasourcesLS implements ApplicationListener<ApplicationEvent>{

    @Autowired
    private CNDatasourcesService service;
    
    @Override
    public void onApplicationEvent(ApplicationEvent event) {
        List<CNDatasources> list = service.findList(new CNDatasources());
        for(CNDatasources li:list){

            //这里调用创建数据源的方法即可
            service.addDataSourceDynamic(li.getDatasource(),li);
        }
    }

}
点赞
收藏
评论区
推荐文章
blmius blmius
2年前
MySQL:[Err] 1292 - Incorrect datetime value: ‘0000-00-00 00:00:00‘ for column ‘CREATE_TIME‘ at row 1
文章目录问题用navicat导入数据时,报错:原因这是因为当前的MySQL不支持datetime为0的情况。解决修改sql\mode:sql\mode:SQLMode定义了MySQL应支持的SQL语法、数据校验等,这样可以更容易地在不同的环境中使用MySQL。全局s
Jacquelyn38 Jacquelyn38
2年前
2020年前端实用代码段,为你的工作保驾护航
有空的时候,自己总结了几个代码段,在开发中也经常使用,谢谢。1、使用解构获取json数据let jsonData  id: 1,status: "OK",data: 'a', 'b';let  id, status, data: number   jsonData;console.log(id, status, number )
Wesley13 Wesley13
2年前
SSH实现动态数据源切换,事务场景下使用AOP
上周写代码遇到了切换数据源的问题,在同一个方法中向两个不同数据源做一些操作,但是这个方法使用了事务,所以网上一般动态切换数据源的方法就失效了。框架是spirngmvchibernate,数据库是oracle,连接池druid。一般情况下,操作数据都是在DAO层进行处理。一种办法是使用多个DataSource然后创建多个SessionFa
Stella981 Stella981
2年前
Spring Boot 集成 Mybatis 实现双数据源
这里用到了SpringBootMybatisDynamicDataSource配置动态双数据源,可以动态切换数据源实现数据库的读写分离。添加依赖加入Mybatis启动器,这里添加了Druid连接池、Oracle数据库驱动为例。<dependency<groupIdorg.mybatis.spring
Stella981 Stella981
2年前
BeetlSQL 3.0.10 发布,多数据源分布式sega事务支持
本次发布主要增加了分布式Sega事务支持,适合多数据源按照社区建议,修改了了springboot的yml配置方式修改了@Jackson和@UpdateTime,本来是用来作为例子,但社区开发者提供了较好的完整实现增加Sega支持<dependency<groupIdcom.ibeetl</gr
Stella981 Stella981
2年前
KVM调整cpu和内存
一.修改kvm虚拟机的配置1、virsheditcentos7找到“memory”和“vcpu”标签,将<namecentos7</name<uuid2220a6d1a36a4fbb8523e078b3dfe795</uuid
可莉 可莉
2年前
0018SpringBoot连接docker中的mysql并使用druid数据源
由于druid数据源自带监控功能,所以引用druid数据源1、centos7中安装并启动docker2、docker安装并启动mysql3、pom.xml中引入druid依赖4、application.yml中配置数据库连接及druid数据源信息5、编写DruidConfig配置文件,绑定4中所配置的数据源信息6、编写HelloCon
Stella981 Stella981
2年前
Django中Admin中的一些参数配置
设置在列表中显示的字段,id为django模型默认的主键list_display('id','name','sex','profession','email','qq','phone','status','create_time')设置在列表可编辑字段list_editable
Wesley13 Wesley13
2年前
MySQL部分从库上面因为大量的临时表tmp_table造成慢查询
背景描述Time:20190124T00:08:14.70572408:00User@Host:@Id:Schema:sentrymetaLast_errno:0Killed:0Query_time:0.315758Lock_
Python进阶者 Python进阶者
4个月前
Excel中这日期老是出来00:00:00,怎么用Pandas把这个去除
大家好,我是皮皮。一、前言前几天在Python白银交流群【上海新年人】问了一个Pandas数据筛选的问题。问题如下:这日期老是出来00:00:00,怎么把这个去除。二、实现过程后来【论草莓如何成为冻干莓】给了一个思路和代码如下:pd.toexcel之前把这