SpringBoot实现多数据源的切换实践
前言
我们在进行软件开发的过程中,刚开始的时候因为无法估量系统后期的访问量和并发量,所以一开始会采用单体架构,后期如果网站流量变大, 并发量变大,那么就可能会将架构扩展为微服务架构,各个微服务对应一个数据库,不过这样的成本就有点大了,可能只是有些模块用的人比较多, 有些模块没什么人用,如果都进行服务拆分,其实也没那个必要,如果有些模块用的人比较多,那么我们可以采用读写分离来减轻压力,这样的话, 可以在一定程度上提升系统的用户体验,不过这只是在数据库的I/O上面做方案,如果系统的压力很大,那么肯定要做负载均衡,我们今天就先说 实现数据库的读写分离。我们要在代码层面实现数据库的读写分离,那么核心就是数据源的切换,本文基于AOP来实现数据源的切换。
工程结构
└─com └─steak └─transaction │ TransactionDemoApplication.java │ ├─datasource │ │ DatasourceChooser.java │ │ DatasourceConfiguration.java │ │ DatasourceContext.java │ │ DatasourceScope.java │ │ DynamicDatasourceAspect.java │ │ │ └─properties │ DynamicDatasourceProperties.java │ MainDatasourceProperties.java │ ├─execute │ PlaceOrderExecute.java │ ├─rest │ PlaceOrderApi.java │ ├─result │ R.java │ └─service IntegralService.java OrderService.java StockService.java
在下面的实现中,我们一个有三个数据源,其中有一个是默认的,如果不指定具体的数据源,那么就使用默认的,我们是基于申明式的方式来切换 数据源,只需要在具体的接口上加上注解便能实现数据源的切换。
编码实现
yml文件
主数据源直接使用spring的配置,其他数据源采用自定义的方式,这里采用一个map结构来定义,方便进行解析,可以在yml文件里进行添加多个,在代码 逻辑层面不用改动。
spring: datasource: type: com.alibaba.druid.pool.DruidDataSource druid: username: root password: 123456 url: jdbc:mysql://127.0.0.1:3306/db?useUnicode=true&characterEncoding=UTF-8&useSSL=false&serverTimezone=UTC driver-class-name: com.mysql.cj.jdbc.Driver dynamic: datasource: { slave1: { username: 'root', password: '123456', url: ' url: jdbc:mysql://127.0.0.1:3306/db?useUnicode=true&characterEncoding=UTF-8&useSSL=false&serverTimezone=UTC', driver-class-name: 'com.mysql.cj.jdbc.Driver' }, slave2: { username: 'root', password: '123456', url: ' url: jdbc:mysql://127.0.0.1:3306/db?useUnicode=true&characterEncoding=UTF-8&useSSL=false&serverTimezone=UTC', driver-class-name: 'com.mysql.cj.jdbc.Driver' } }
主数据源MainDatasourceProperties
对于主数据源,我们单独拿出来放在一个类里面,不过其实也可以放到dynamic里面,只是需要做一定的处理,我们就简单的放几个连接属性。
/** * @author 刘牌 * @date 2022/3/220:14 */ @Component @ConfigurationProperties(prefix = "spring.datasource.druid") @Data public class MainDatasourceProperties { private String username; private String password; private String url; private String driverClassName; }
其他数据源DynamicDatasourceProperties
其他数据源使用一个Map来接受yml文件中的数据源配置
/** * @author 刘牌 * @date 2022/3/213:47 */ @Component @ConfigurationProperties(prefix = "dynamic") @Data public class DynamicDatasourceProperties { private Map<String,Map<String,String>> datasource; }
数据源配置类DatasourceConfiguration
配置类中主要对DataSource进行配置,主数据源我们按照正常的bean来定义连接属性,而其他数据数据源则使用反射的方式来进行连接属性的 配置,因为主数据源一般是不会变动的,但是其他数据源可能会发生变动,可能会添加,这时候如果通过硬编码取配置,那么每增加一个数据源,就需要 增加一个配置,显然不太行,所以就使用反射来进行赋值。
/** * @author 刘牌 * @date 2022/3/111:34 */ @Configuration @AllArgsConstructor public class DatasourceConfiguration { final MainDatasourceProperties mainDatasourceProperties; final DynamicDatasourceProperties dynamicDatasourceProperties; @Bean public DataSource datasource(){ Map<Object,Object> datasourceMap = new HashMap<>(); DatasourceChooser datasourceChooser = new DatasourceChooser(); /** * main database */ DruidDataSource mainDataSource = new DruidDataSource(); mainDataSource.setUsername(mainDatasourceProperties.getUsername()); mainDataSource.setPassword(mainDatasourceProperties.getPassword()); mainDataSource.setUrl(mainDatasourceProperties.getUrl()); mainDataSource.setDriverClassName(mainDatasourceProperties.getDriverClassName()); datasourceMap.put("main",mainDataSource); /** * other database */ Map<String, Map<String, String>> sourceMap = dynamicDatasourceProperties.getDatasource(); sourceMap.forEach((datasourceName,datasourceMaps) -> { DruidDataSource dataSource = new DruidDataSource(); datasourceMaps.forEach((K,V) -> { String setField = "set" + K.substring(0, 1).toUpperCase() + K.substring(1); //转换yml文件中带有-符号的属性 String[] strings = setField.split(""); StringBuilder newStr = new StringBuilder(); for (int i = 0; i < strings.length; i++) { if (strings[i].equals("-")) strings[i + 1] = strings[i + 1].toUpperCase(); if (!strings[i].equals("-")) newStr.append(strings[i]); } try { DruidDataSource.class.getMethod(newStr.toString(),String.class).invoke(dataSource,V); } catch (Exception e) { e.printStackTrace(); } }); datasourceMap.put(datasourceName,dataSource); }); //设置目标数据源 datasourceChooser.setTargetDataSources(datasourceMap); //设置默认数据源 datasourceChooser.setDefaultTargetDataSource(mainDataSource); return datasourceChooser; } }
上面使用数据源配置类中使用反射对其他数据源进行连接属性的设置,然后设置目标数据源和默认数据源,里面有一个DatasourceChooser
。
DatasourceChooser
DatasourceChooser
继承自AbstractRoutingDataSource
,AbstractRoutingDataSource
可以实现数据源的切换,它里面的 determineCurrentLookupKey()
方法需要我们返回一个数据源的名称,它会自动给我们匹配上数据源。
/** * @author 刘牌 * @date 2022/3/112:21 */ public class DatasourceChooser extends AbstractRoutingDataSource { @Override protected Object determineCurrentLookupKey() { return DatasourceContext.getDatasource(); } }
如下是AbstractRoutingDataSource
的部分源码,我们可以看出数据源是一个Map结构,可以通过数据源名称查找到对应的数据源。
package org.springframework.jdbc.datasource.lookup; public abstract class AbstractRoutingDataSource extends AbstractDataSource implements InitializingBean { @Nullable private Map<Object, DataSource> resolvedDataSources; public AbstractRoutingDataSource() { } protected DataSource determineTargetDataSource() { Object lookupKey = this.determineCurrentLookupKey(); DataSource dataSource = (DataSource)this.resolvedDataSources.get(lookupKey); } @Nullable protected abstract Object determineCurrentLookupKey(); }
DatasourceContext
DatasourceContext
内部是一个ThreadLocal
,主要是用来存储每一个线程的数据源名称和获取数据源名称,而数据源的名称我们用过AOP切面 来获取。
/** * @author 刘牌 * @date 2022/3/112:22 */ public class DatasourceContext { private static final ThreadLocal<String> threadLocal = new ThreadLocal<>(); public static void setDatasource(String key){ threadLocal.set(key); } public static String getDatasource(){ return threadLocal.get(); } }
数据源注解DatasourceScope
DatasourceScope
标准在方法上面,通过scope
来指定数据源,不指定默认为主数据源main
。
/** * @author 刘牌 * @date 2022/3/111:22 */ @Target({ElementType.METHOD}) @Retention(RetentionPolicy.RUNTIME) @Documented public @interface DatasourceScope { String scope() default "main"; }
数据源切面DynamicDatasourceAspect
我们在访问每一个带有DatasourceScope
注解的方法时,都会经过数据源切面DynamicDatasourceAspect
,获取到注解上面的 scope
的值后,通过DatasourceContext
设置数据源名称,便可实现对数据源的切换。
/** * @author 刘牌 * @date 2022/3/111:34 */ @Aspect @Component public class DynamicDatasourceAspect { @Pointcut("@annotation(dataSourceScope)") public void dynamicPointcut(DatasourceScope dataSourceScope){} @Around(value = "dynamicPointcut(dataSourceScope)", argNames = "joinPoint,dataSourceScope") public Object dynamicAround(ProceedingJoinPoint joinPoint , DatasourceScope dataSourceScope) throws Throwable { String scope = dataSourceScope.scope(); DatasourceContext.setDatasource(scope); return joinPoint.proceed(); } }
使用
只需要在具体的方法上面标注数据源注解@DatasourceScope
,并指定scope
的值,便可实现切换,如果不指定,那么就使用主数据源。
/** * @author 刘牌 * @date 2022/3/19:49 */ @Service @AllArgsConstructor public class OrderService { private JdbcTemplate jdbcTemplate; @DatasourceScope(scope = "slave1") public R saveOrder(Integer userId , Integer commodityId){ String sql = "INSERT INTO `order`(user_id,commodity_id) VALUES("+userId+","+commodityId+")"; jdbcTemplate.execute(sql); return R.builder().code(200).msg("save order success").build(); } }
到此这篇关于SpringBoot实现多数据源的切换实践的文章就介绍到这了,更多相关SpringBoot多数据源切换内容请搜索猪先飞以前的文章或继续浏览下面的相关文章希望大家以后多多支持猪先飞!
原文出处:https://juejin.cn/post/7070701713145102343
相关文章
解决springboot使用logback日志出现LOG_PATH_IS_UNDEFINED文件夹的问题
这篇文章主要介绍了解决springboot使用logback日志出现LOG_PATH_IS_UNDEFINED文件夹的问题,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2021-04-28- 下面小编就为大家带来一篇利用JS实现点击按钮后图片自动切换的简单方法。小编觉得挺不错的,现在就分享给大家,也给大家做个参考。一起跟随小编过来看看吧...2016-10-25
- 这篇文章主要为大家详细介绍了SpringBoot实现excel文件生成和下载,文中示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下...2021-02-09
- 这篇文章主要介绍了SpringBoot集成Redis实现消息队列的方法,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2021-02-10
- 这篇文章主要介绍了详解springBoot启动时找不到或无法加载主类解决办法,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2020-09-16
- 这篇文章主要介绍了js组件SlotMachine实现图片切换效果制作抽奖系统的相关资料,需要的朋友可以参考下...2016-04-19
- 这篇文章主要介绍了解决Springboot get请求是参数过长的情况,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2020-09-17
- 这篇文章主要介绍了Springboot+TCP监听服务器搭建过程,本文通过图文并茂的形式给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下...2020-10-28
Spring Boot项目@RestController使用重定向redirect方式
这篇文章主要介绍了Spring Boot项目@RestController使用重定向redirect方式,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教...2021-09-02- 这篇文章主要介绍了springBoot 项目排除数据库启动方式,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教...2021-09-10
- 最近项目不多忙,于是抽点时间巩固下切换窗口问题,感兴趣的朋友跟着小编一起学习吧...2020-06-25
springboot中使用@Transactional注解事物不生效的坑
这篇文章主要介绍了springboot中使用@Transactional注解事物不生效的原因,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2021-01-26详解SpringBoot之访问静态资源(webapp...)
这篇文章主要介绍了详解SpringBoot之访问静态资源(webapp...),文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2020-09-14- 这篇文章主要介绍了SpringBoot接口接收json参数解析,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教...2021-10-19
Springboot mybatis plus druid多数据源解决方案 dynamic-datasource的使用详解
这篇文章主要介绍了Springboot mybatis plus druid多数据源解决方案 dynamic-datasource的使用,本文给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下...2020-11-18- 下面小编就为大家带来一篇javascript的列表切换【实现代码】。小编觉得挺不错的,现在分享给大家,也给大家做个参考。...2016-05-05
ShardingSphere jdbc集成多数据源的实现步骤
本文主要介绍了ShardingSphere jdbc集成多数据源的实现步骤,文中通过示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下...2021-10-21- 复制代码 代码如下: <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd"> <html xmlns="http://www.w3.org/1999/xhtml"> <head> <title...2013-10-13
- 这篇文章主要介绍了springboot多模块包扫描问题的解决方法,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2020-09-16
Springboot+MDC+traceId日志中打印唯一traceId
本文主要介绍了Springboot+MDC+traceId日志中打印唯一traceId,文中通过示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下...2021-10-17