SpringBoot实现多数据源的切换实践

更新时间：2022年03月03日 16:39:15 作者：渣男小四

这篇主要介绍了SpringBoot实现多数据源的切换，本文基于AOP来实现数据源的切换，文中通过示例代码介绍的非常详细，感兴趣的小伙伴们可以参考一下

前言

我们在进行软件开发的过程中，刚开始的时候因为无法估量系统后期的访问量和并发量，所以一开始会采用单体架构，后期如果网站流量变大，并发量变大，那么就可能会将架构扩展为微服务架构，各个微服务对应一个数据库，不过这样的成本就有点大了，可能只是有些模块用的人比较多，有些模块没什么人用，如果都进行服务拆分，其实也没那个必要，如果有些模块用的人比较多，那么我们可以采用读写分离来减轻压力，这样的话，可以在一定程度上提升系统的用户体验，不过这只是在数据库的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多数据源切换内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:

mybatis 多表关联mapper文件写法操作
这篇文章主要介绍了mybatis 多表关联mapper文件写法操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-12-12
ssm项目实现用户登陆持久化(token)
这篇文章主要介绍了ssm项目实现用户登陆持久化(token)，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2021-04-04
Java网络编程TCP实现文件上传功能
这篇文章主要为大家详细介绍了Java网络编程TCP实现文件上传功能，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2021-07-07
彻底搞懂Java多线程(二)
这篇文章主要给大家介绍了关于Java面试题之多线程和高并发的相关资料，文中通过示例代码介绍的非常详细，对大家学习或者使用java具有一定的参考学习价值，需要的朋友们下面来一起学习学习吧
2021-07-07
JVM中的GC初识
GC（Garbage Collection）称之为垃圾回收，是对内存中的垃圾对象,采用一定的算法进行内存回收的一个动作，这篇文章主要介绍了JVM中的GC初识,需要的朋友可以参考下
2022-05-05
Java虚拟机内存分配与回收策略问题精细解读
Java技术体系中所提倡的自动内存管理最终可以归结为自动化地解决了两个问题：给对象分配内存以及回收分配给对象的内存，本文让我们来详细了解
2021-11-11
SpringBoot之自定义Banner详解
这篇文章主要介绍了SpringBoot之自定义Banner详解,本篇文章通过简要的案例,讲解了该项技术的了解与使用,以下就是详细内容,需要的朋友可以参考下
2021-09-09
JAVA Stack详细介绍和示例学习
JAVA Stack是栈。它的特性是：先进后出(FILO, First In Last Out)。
2013-11-11
IDEA配置Maven教程的超详细讲解版
IntelliJ IDEA是当前最流行的Java IDE(集成开发环境)之一,也是业界公认最好用的Java开发工具之一,这篇文章主要给大家介绍了关于IDEA配置Maven教程的超详细讲解版,需要的朋友可以参考下
2023-11-11
Java中工具Jstack的使用实例
jstack用于生成java虚拟机当前时刻的线程快照,下面这篇文章主要给大家介绍了关于Java中工具Jstack使用的相关资料,文中通过实例代码介绍的非常详细,需要的朋友可以参考下
2022-04-04