SpringBoot动态从数据库中获取数据源,动态切换数据源

最新推荐文章于 2024-08-06 10:17:52 发布
最新推荐文章于 2024-08-06 10:17:52 发布
阅读量6.6k 收藏 8
点赞数
分类专栏： springboot 文章标签： spring boot 数据库 mybatis java spring
原文链接：https://blog.51cto.com/u_15069498/4547371
版权

1.简介

SpringBoot静态数据源指的是将多个数据源信息配置在配置文件中,在项目启动时加载配置文件中的多个数据源,并实例化多个数据源Bean,再通过分包/Aop达到切换数据源的目的

如果想要新增或者修改数据源,必须修改配置文件,并修改对应的代码(增加对应的DataSource Bean)重启项目,重新实例化数据源,才能使用

动态数据源指的是将数据源信息配置在关系型数据库中或者缓存数据库中,在项目启动时只初始化一个默认数据源,在项目运行过程中动态的从数据库中读取数据源信息,实例化为DataSource,并使用该数据源获取连接,执行SQL

此处研究的ORM框架为Mybatis

2.实现方法

动态数据源的实现方法有两种

2.1 重写DataSource中的getConnection方法

Mybatis中数据库的连接都是在执行sql的时候才触发并创建连接

由此可以通过重写数据源的getConnection()方法,当Mybatis需要创建对应的数据库连接时,跟据要使用的数据源,修改当前使用的数据源,达到动态数据源的目的

2.2 通过AbstractRoutingDataSource类

AbstractRoutingDataSource类是jdbc提供的轻量级的切换数据源的方案,内部维护了一个数据源的集合,提供了维护这个数据源集合的方法,这样我们在动态的创建完对应的数据源后,就可以通过这个类提供的方法,将数据源维护进这个集合,再通过重写 determineCurrentLookupKey()方法,来告诉 AbstractRoutingDataSource我们要的是哪个数据源,最终达到切换数据源的目的

3.具体代码

3.1 方案1:重写DataSource中的getConnection方法

3.1.1 主配置类 DataSourceConfig

@Configuration

@MapperScan(basePackages = "com.zhangyao.springboot.mapper",sqlSessionFactoryRef = "sqlSessionFactory")

public class DataSourceConfig {

/**

* 主数据源配置

* @return

*/

@Bean(name = "primaryDataSource")

@Primary

@ConfigurationProperties(prefix = "spring.datasource.test1")

public DataSource getDataSource1(){

HikariDataSource datasource = DataSourceBuilder.create().type(MyDynamicDataSource.class).build();

if(datasource==null){

datasource = new MyDynamicDataSource().initDataSource("default");

}

//设置默认的数据源

DataSourceCache.put("default", datasource);

ThreadLocalDataSource.setLocalSource("default");

return datasource;

}

@Bean("sqlSessionFactory")

@Primary

public SqlSessionFactory getSqlSessionFactory(@Qualifier("primaryDataSource") DataSource primaryDataSource) throws Exception {

SqlSessionFactoryBean sqlSessionFactoryBean = new SqlSessionFactoryBean();

sqlSessionFactoryBean.setDataSource(primaryDataSource);

return sqlSessionFactoryBean.getObject();

}

}

这个类里配置了默认的数据源default,这个数据源是从配置文件中读取到的,并且实例化了Mybatis的SqlSessionFactory,默认注入default数据源

3.1.2 MyDynamicDataSource 覆盖HikariDataSource的getConnection()

package com.zhangyao.springboot.config;

import com.zaxxer.hikari.HikariDataSource;

import com.zhangyao.springboot.domin.Databaseinfo;

import com.zhangyao.springboot.service.DataBaseService;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.core.io.ClassPathResource;

import org.springframework.core.io.support.EncodedResource;

import org.springframework.core.io.support.PropertiesLoaderUtils;

import java.io.IOException;

import java.sql.Connection;

import java.sql.SQLException;

import java.util.Map;

import java.util.Properties;

import java.util.concurrent.ConcurrentHashMap;

/**

* @author: zhangyao

* @create:2020-11-03 21:07

* @Description:

**/

public class MyDynamicDataSource extends HikariDataSource {

@Autowired

DataBaseService dataBaseService;

/**

* 定义缓存数据源的变量

*/

public static final Map<Object, Object> DataSourceCache = new ConcurrentHashMap<Object, Object>();

@Override

public Connection getConnection() throws SQLException {

String localSourceKey = ThreadLocalDataSource.getLocalSource();

HikariDataSource dataSource = (HikariDataSource) DataSourceCache.get(localSourceKey);

if(dataSource==null){

try {

dataSource = initDataSource(localSourceKey);

} catch (IOException e) {

e.printStackTrace();

}

}

return dataSource.getConnection();

}

/**

* 初始化DataSource

* 当缓存中没有对应的数据源时,需要去默认数据源查询数据库

*

* @param key

* @return

*/

public HikariDataSource initDataSource(String key) throws IOException {

HikariDataSource dataSource = new HikariDataSource();

if ("default".equals(key)) {

Properties properties = PropertiesLoaderUtils.loadProperties(new EncodedResource(new ClassPathResource("application.properties"), "UTF-8"));

dataSource.setJdbcUrl(properties.getProperty("spring.datasource.test1.jdbc-url"));

dataSource.setUsername(properties.getProperty("spring.datasource.test1.username"));

dataSource.setPassword(properties.getProperty("spring.datasource.test1.password"));

dataSource.setDriverClassName(properties.getProperty("spring.datasource.test1.driver-class-name"));

} else {

//查询数据库

ThreadLocalDataSource.setLocalSource("default");

Databaseinfo dataBaseInfo = dataBaseService.getDataBaseInfo(key);

dataSource.setJdbcUrl(dataBaseInfo.getUrl());

dataSource.setUsername(dataBaseInfo.getUserName());

dataSource.setPassword(dataBaseInfo.getPassword());

dataSource.setDriverClassName(dataBaseInfo.getDriverClassName());

ThreadLocalDataSource.setLocalSource(key);

}

DataSourceCache.put(key, dataSource);

return dataSource;

}

}

这个类重写了HikariDatasource类的getConnection()方法,当Mybatis使用连接时,就会调用MyDynamicDataSource的getConnection()方法,然后通过获取ThreadLoacal中存放的当前使用的数据源的key,进而从自定义的缓存变量 DataSourceCache 中获取对应的数据源,如果获取不到,就使用默认数据源查询数据库,如果再获取不到,就抛出异常,查询到数据源后,初始化完再放入到缓存中DataSourceCache

3.1.3 ThreadLocalDataSource 存放当前线程使用数据源的key
package com.zhangyao.springboot.config;

import lombok.extern.slf4j.Slf4j;

import javax.xml.crypto.Data;

/**

* ThreadLocal保存数据源的key,并切换清除

* @author: zhangyao

* @create:2020-04-07 09:24

**/

@Slf4j

public class ThreadLocalDataSource {

//使用threadLocal保证切换数据源时的线程安全 不会在多线程的情况下导致切换错数据源

private static final ThreadLocal<String> TYPE = new ThreadLocal<String>();

/**

* 修改当前线程内的数据源id

* @param key

*/

public static void setLocalSource(String key){

TYPE.set(key);

}

/**

* 获取当前线程内的数据源类型

* @return

*/

public static String getLocalSource(){

return TYPE.get();

}

/**

* 清空ThreadLocal中的TYPE

*/

public void clear(){

TYPE.remove();

}

}

提供了对ThreadLocal的set/get操作方法,ThreadLocal中存放的是数据源的key,这个key与 MyDynamicDataSource中的DataSourceCache中的key一致,ThreadLocal的作用就是保证在当前线程内可以取到唯一的数据源的key

3.1.4 DataSourceAop 切面 用于解析请求中的数据源的key


@Aspect

@Component

@Slf4j

public class DataSourceAop {

/**

* 定义切入点

* 切入点为有该注解的方法

* 此注解用于数据源TEST1

*/

@Pointcut("@annotation(com.zhangyao.springboot.annotation.DataSourceServiceAop)")

public void serviceTest1DatasourceAspect(){};

/**

* 在切入service方法之前执行

* 设置数据源

*/

@Before("serviceTest1DatasourceAspect()")

public void beforeAspect(){

log.info("切入方法,开始设置数据源");

ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();

String database_key = attributes.getRequest().getHeader("database_key");

ThreadLocalDataSource.setLocalSource(database_key);

}

/**

* 在切入service方法之后执行

* 设置回默认数据源

*/

@After("serviceTest1DatasourceAspect()")

public void afterAspect(){

log.info("切入方法后,开始切换默认数据源");

ThreadLocalDataSource.setLocalSource("default");

}

}

需要自定义一个注解@DataSourceServiceAop,标识在使用动态数据源的方法上

这里是跟据前台传输的数据源的key来设置ThreadLocal中的key,如果前台传输的数据源的key不在header中,再跟据实际情况调整

切完方法之后,切换回default数据源

3.2 方案2: 通过AbstractRoutingDataSource类
3.2.1 主配置类 DataSourceConfig
package com.zhangyao.springboot.config;

import com.alibaba.druid.pool.DruidDataSource;

import com.zaxxer.hikari.HikariDataSource;

import org.apache.ibatis.session.SqlSessionFactory;

import org.mybatis.spring.SqlSessionFactoryBean;

import org.mybatis.spring.SqlSessionTemplate;

import org.springframework.beans.factory.annotation.Qualifier;

import org.springframework.boot.context.properties.ConfigurationProperties;

import org.springframework.boot.jdbc.DataSourceBuilder;

import org.springframework.context.annotation.Bean;

import org.springframework.context.annotation.Configuration;

import org.springframework.context.annotation.Primary;

import tk.mybatis.spring.annotation.MapperScan;

import javax.sql.DataSource;

import java.io.IOException;

import java.util.HashMap;

import java.util.Map;

import static com.zhangyao.springboot.config.MyDynamicDataSource.DataSourceCache;

/**

*

* aop多数据源动态切换配置

* @author: zhangyao

* @create:2020-04-06 22:17

**/

@Configuration

@MapperScan(basePackages = "com.zhangyao.springboot.mapper",sqlSessionFactoryRef = "sqlSessionFactory")

public class DataSourceConfig {

/**

* 主数据源配置

* @return

*/

@Bean(name = "primaryDataSource")

@Primary

@ConfigurationProperties(prefix = "spring.datasource.test1")

public DataSource getDataSource1(){

HikariDataSource datasource = DataSourceBuilder.create().type(HikariDataSource.class).build();

//设置默认的数据源

DataSourceCache.put("default", datasource);

ThreadLocalDataSource.setLocalSource("default");

return datasource;

}

/**

* 动态装配所有的数据源

* @param primaryDataSource

* @return

*/

@Bean("dynamicDataSource")

public DynamicChangeDataSourceConfig setDynamicDataSource(@Qualifier("primaryDataSource") DataSource primaryDataSource){

//定义所有的数据源

Map<Object,Object> allDataSource = new HashMap<Object, Object>();

//把配置的多数据源放入map

allDataSource.put("default", primaryDataSource);

//定义实现了AbstractDataSource的自定义aop切换类

DynamicChangeDataSourceConfig dynamicChangeDataSourceConfig = new DynamicChangeDataSourceConfig();

//把上面的所有的数据源的map放进去

dynamicChangeDataSourceConfig.setTargetDataSources(allDataSource);

//设置默认的数据源

dynamicChangeDataSourceConfig.setDefaultTargetDataSource(primaryDataSource);

return dynamicChangeDataSourceConfig;

}

@Bean("sqlSessionFactory")

@Primary

public SqlSessionFactory getSqlSessionFactory(@Qualifier("dynamicDataSource") DataSource dynamicDataSource) throws Exception {

SqlSessionFactoryBean sqlSessionFactoryBean = new SqlSessionFactoryBean();

sqlSessionFactoryBean.setDataSource(dynamicDataSource);

return sqlSessionFactoryBean.getObject();

}

}

与方案一一样需要先实例化默认数据源default,但是MyBatis的SqlSessionFactory中注入的就不是默认数据源了,而是AbstractRoutingDataSource的实现类,并将默认数据源放入AbstractRoutingDataSource的targetDataSources中

3.2.2 DynamicChangeDataSourceConfig AbstractRoutingDataSource的子类
package com.zhangyao.springboot.config;

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;

/**

* 继承AbStractRoutingDataSource

* 动态切换数据源

* @author: zhangyao

* @create:2020-04-07 09:23

**/

public class DynamicChangeDataSourceConfig extends AbstractRoutingDataSource {

@Override

protected Object determineCurrentLookupKey() {

return ThreadLocalDataSource.getLocalSource();

}

}

重写了AbstractRoutingDataSource的determineCurrentLookupKey方法,改为返回ThreadLocal中的数据源的key

3.2.3 ThreadLocalDataSource 存放当前线程使用数据源的key
 

与方案1一摸一样

3.2.3 DataSourceAop 切面 用于解析请求中的数据源的key
package com.zhangyao.springboot.config;

import com.zaxxer.hikari.HikariDataSource;

import com.zhangyao.springboot.domin.Databaseinfo;

import com.zhangyao.springboot.service.DataBaseService;

import lombok.extern.slf4j.Slf4j;

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.beans.factory.annotation.Autowired;

import org.springframework.stereotype.Component;

import org.springframework.web.context.request.RequestAttributes;

import org.springframework.web.context.request.RequestContextHolder;

import org.springframework.web.context.request.ServletRequestAttributes;

import javax.annotation.Resource;

import static com.zhangyao.springboot.config.MyDynamicDataSource.DataSourceCache;

/**

* @author: zhangyao

* @create:2020-04-07 11:20

**/

@Aspect

@Component

@Slf4j

public class DataSourceAop {

@Autowired

DataBaseService dataBaseService;

@Resource(name = "dynamicDataSource")

DynamicChangeDataSourceConfig dynamicChangeDataSourceConfig;

/**

* 定义切入点

* 切入点为有该注解的方法

* 此注解用于数据源TEST1

*/

@Pointcut("@annotation(com.zhangyao.springboot.annotation.DataSourceServiceAop)")

public void serviceTest1DatasourceAspect(){};

/**

* 在切入service方法之前执行

* 设置数据源

*/

@Before("serviceTest1DatasourceAspect()")

public void beforeAspect(){

log.info("切入方法,开始设置数据源");

ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();

String database_key = attributes.getRequest().getHeader("database_key");

initDataSource(database_key);

ThreadLocalDataSource.setLocalSource(database_key);

}

/**

* 在切入service方法之后执行

* 设置回默认数据源

*/

@After("serviceTest1DatasourceAspect()")

public void afterAspect(){

log.info("切入方法后,开始切换默认数据源");

ThreadLocalDataSource.setLocalSource("default");

}

public HikariDataSource initDataSource(String key) {

HikariDataSource dataSource = new HikariDataSource();

if ("default".equals(key)) {

Properties properties = PropertiesLoaderUtils.loadProperties(new EncodedResource(new ClassPathResource("application.properties"), "UTF-8"));

dataSource.setJdbcUrl(properties.getProperty("spring.datasource.test1.jdbc-url"));

dataSource.setUsername(properties.getProperty("spring.datasource.test1.username"));

dataSource.setPassword(properties.getProperty("spring.datasource.test1.password"));

dataSource.setDriverClassName(properties.getProperty("spring.datasource.test1.driver-class-name"));

} else {

//查询数据库

ThreadLocalDataSource.setLocalSource("default");

Databaseinfo dataBaseInfo = dataBaseService.getDataBaseInfo(key);

dataSource.setJdbcUrl(dataBaseInfo.getUrl());

dataSource.setUsername(dataBaseInfo.getUserName());

dataSource.setPassword(dataBaseInfo.getPassword());

dataSource.setDriverClassName(dataBaseInfo.getDriverClassName());

DataSourceCache.put(key, dataSource);

dynamicChangeDataSourceConfig.setTargetDataSources(DataSourceCache);

dynamicChangeDataSourceConfig.afterPropertiesSet();

ThreadLocalDataSource.setLocalSource(key);

}

return dataSource;

}

}

当进入切点方法后,获取到前台传输的数据源key,去缓存中取,如果取不到,就查询数据库,并实例化放置到缓存中,并设置ThreadLocal的key为前台传输的key

4.总结
其实两种方案本质上是一种方法,第一种方法相当于自己把AbstractRoutingDataSource这个类的功能再手动的实现一遍,好处是更加灵活,可以针对自己的业务做定制

Zark渊
关注
专栏目录
SpringBoot多数据源动态切换源码
07-07
SpringBoot的jpa多数据源的动态切换项目源代码，适用于不同的用户操作不同的数据库，但是其对应的repository是一样的，可以进行http请求的动态数据源切换
SpringBoot框架Datasource注入
04-20
该项目采用标签形式对Datasource进行注入将Datasource组件交给容器进行统一管理
DynamicDatasource动态数据源实现及使用
最新发布
努力提高技术水平
08-06 1441
baomidou开源的多数据源扩展插件DynamicDatasource使用及具体实现，附完整源码。
spring boot获取应用配置的数据源
夜半离人
04-15 4224
注入应用上下文 @Autowired protected ApplicationContext applicationContext; 通过应用上下文获取数据源配置 // 获取配置的数据源 DataSource dataSource = applicationContext.getBean(DataSource.class); Connection connection = null; try { connection = dataSource.getConne...
Spring Boot配置多个DataSource
jaesonchen
04-29 557
使用Spring Boot时，默认情况下，配置DataSource非常容易。Spring Boot会自动为我们配置好一个DataSource。 如果在application.yml中指定了spring.datasource的相关配置，Spring Boot就会使用该配置创建一个DataSource。如果在application.yml中没有指定任何spring.datasource的相关配置，...
Spring中获得dataSurce对象的方法
漂泊一剑客的专栏
03-20 1025
转载地址：http://www.oschina.net/code/snippet_117958_4699 在Spring框架中有如下3种获得DataSource对象的方法:  1.从JNDI获得DataSource.  2.从第三方的连接池获得DataSource.  3.使用DriverManagerDataSource获得DataSource. 代码片段(3)
Spring Boot数据访问—Springboot 多数据源—官方原版
深圳市多克创新科技有限公司
02-28 1565
Spring Boot数据访问—（springboot 多数据源）—官方原版
SpringBoot整合Oracle数据库使用双数据源
05-22
在实际应用中，你可能还需要考虑数据源切换策略、读写分离、分布式事务等问题，以满足更复杂的需求。提供的源码包（如`synchro`）可能包含了这些高级功能的实现，建议根据实际需求进行研究和调整。 总的来说，...
SpringBoot整合mybatis-plus实现多数据源的动态切换且支持分页查询.pdf
10-07
在实际的业务代码中，可以通过`DynamicDataSourceContextHolder`来切换数据源，如在Service层根据业务逻辑选择合适的数据源，而在Mapper层则可以直接使用Mybatis-Plus的分页查询功能，如`BaseMapper.selectPage()`...
SpringBoot实现动态切换数据源（含源码）
01-25
在Spring Boot应用中，动态切换数据源是一项关键功能，尤其在多数据库环境或者读写分离的架构中。这里我们将深入探讨如何使用`ThreadLocal`和`AbstractRoutingDataSource`来实现这一目标。 `ThreadLocal`是Java提供...
基于 SpringBoot 多数据源 动态数据源 主从分离 快速启动器 支持分布式事务
01-05
基于 SpringBoot 多数据源 动态数据源 主从分离 快速启动器 支持分布式事务。一个基于springboot的快速集成多数据源的启动器。支持 数据源分组 ，适用于多种场景 纯粹多库 读写分离 一主多从 混合模式。支持数据库...
Spring获取数据源代码
03-07
Spring获取数据源代码，简单使用Spring数据源。Spring获取数据源代码，简单使用Spring数据源。
SpringBoot配置多数据源实现动态切换数据源
06-29
接下来是动态切换数据源的关键，这通常通过AOP（面向切面编程）和ThreadLocal来实现。创建一个自定义的`Aspect`，并在方法执行前根据业务逻辑选择合适的数据源： ```java @Aspect @Component public class ...
Spring Boot
LMY0210的博客
11-23 1365
Spring Boot是Spring项目中的一个子工程，与我们所熟知的Spring-framework 同属于spring的产品:其最主要作用就是帮助开发人员快速的构建庞大的spring项目，并且尽可能的减少一切xml配置，做到开箱即用，迅速上手，让开发人员关注业务而非配置。主要特点:设计目的： 用来简化 Spring 应用的初始搭建以及开发过程。从最根本上来讲，Spring Boot 就是一些库的集合，它能够被任意项目所使用。它使用 “习惯优于配置”的理念让你的项目快速运行起来。
springboot获取当前数据库连接
qq_34068440的博客
06-01 1398
通过上述方法，可以在 Spring 应用中获取当前的 DataSource URL，具体选用哪种方法取决于你项目的具体配置和使用的连接池类型。
springboot 链接多数据库，动态获取数据源信息
热门推荐
z357904947的博客
04-09 1万+
springboot 链接多数据库，动态获取数据源信息 前言 最近公司有个需求需要后端访问多个数据库，在网上查了半天资料，结果发现大部分都是配置的主从库这种数据源固定的情况，不符合我的需求，假设我们有这样一种需求，后端需要从一个数据库的配置表里动态的读取其它mysql数据库的链接配置信息，并根据链接信息动态创建数据库链接，发起请求，而且还要能使用现在的一些链接池。 最后找到了这篇博客 https:...
Spring boot 基于注解方式配置datasource
m0_54845475的博客
08-13 1457
我们先来回顾下，使用xml配置数据源。
SpringBoot整合Druid数据源
m0_47475986的博客
12-19 340
maven仓库：https://mvnrepository.com/artifact/com.alibaba/druid 1.导入Druid依赖 <dependency> <groupId>com.alibaba</groupId> <artifactId>druid</artifactId> <version>1.2.5</version> </dependency> 2.配置appl
配置Spring数据源
weixin_33674437的博客
07-31 596
配置一个数据源Spring在第三方依赖包中包含了两个数据源的实现类包，其一是Apache的DBCP，其二是C3P0。可以在Spring配置文件中利用这两者中任何一个配置数据源。 DBCP数据源DBCP类包位于/lib/jakarta-commons/commons-dbcp.jar，DBCP是一个依赖Jakartacommons-pool对象池机...
@Configuration获取datasource
08-04
在使用@Configuration注解的类中获取DataSource的方式有多种。一种常见的方式是在@Configuration类中定义一个方法，并使用@Bean注解将其声明为一个Spring bean。在这个方法中，你可以实例化、配置和返回一个DataSource对象。例如，你可以这样定义一个EmbeddedDatabaseConfig类和一个ProductionDatabaseConfig类，分别用于开发环境和生产环境的DataSource配置。在这些类中，你可以使用@Bean注解的方法来创建和配置DataSource对象。[1] 另一种方式是使用@Import注解将其他@Configuration类导入到当前的@Configuration类中。这样，被导入的@Configuration类中的@Bean方法也会被注入到当前的@Configuration类中。例如，你可以创建一个DatabaseConfig类来配置DataSource，并在AppConfig类中使用@Import注解将DatabaseConfig类导入。然后，在AppConfig类中，你可以通过构造函数注入DatabaseConfig对象，并在@Bean方法中引用DatabaseConfig类中的dataSource()方法来获取DataSource对象。[2] 总之，通过在@Configuration类中定义@Bean方法或使用@Import注解导入其他@Configuration类，你可以获取到DataSource对象。这样，你就可以在Spring应用程序中使用这个DataSource对象了。如果你想指定一个主要的DataSource实现类，你可以使用@Primary注解来标记这个实现类。[3]
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值