MyCat实现读写分离与动态数据源切换

上一篇 <<<MySQL自带主从复制原理
下一篇 >>>分表分库与分区的区别及拆分策略

1.MyCAT读写分离的原理

2.SpringBoot项目整合动态数据源(读写分离)原理

3.SpringBoot项目整合动态数据源(读写分离)核心代码

• 引入多数据源

spring:
  datasource:
    ###可读数据源
    select:
      jdbc-url: jdbc:mysql://10.211.55.16:8066/mycat_testdb
      driver-class-name: com.mysql.jdbc.Driver
      username: user
      password: user
    ####可写数据源
    update:
      jdbc-url: jdbc:mysql://10.211.55.16:8066/mycat_testdb
      driver-class-name: com.mysql.jdbc.Driver
      username: root
      password: 123456
    type: com.alibaba.druid.pool.DruidDataSource
/**
 * 多数据源配置
 */
@Configuration
public class DataSourceConfig {

   // 创建可读数据源
   @Bean(name = "selectDataSource")
   @ConfigurationProperties(prefix = "spring.datasource.select") // application.properteis中对应属性的前缀
   public DataSource dataSource1() {
      return DataSourceBuilder.create().build();
   }

   // 创建可写数据源
   @Bean(name = "updateDataSource")
   @ConfigurationProperties(prefix = "spring.datasource.update") // application.properteis中对应属性的前缀
   public DataSource dataSource2() {
      return DataSourceBuilder.create().build();
   }

}

• 引入AOP判断，通过DataSourceContextHolder传输

/**
 * 1、lazy(false)和order(0)--指定启动的时候就加载，且AOP的执行顺序在数据库事务之上
 * 2、采用AOP技术进行拦截业务逻辑层方法，判断方法的前缀是否需要写或者读的操作设置路由key
 *
 */
@Aspect
@Component
@Lazy(false)
@Order(0) // Order设定AOP执行顺序 使之在数据库事务上先执行
public class SwitchDataSourceAOP {

   @Before("execution(* com.jgspx.service.*.*(..))")
   public void process(JoinPoint joinPoint) {
      String methodName = joinPoint.getSignature().getName();
      if (methodName.startsWith("get") || methodName.startsWith("count") || methodName.startsWith("find")
            || methodName.startsWith("list") || methodName.startsWith("select") || methodName.startsWith("check")) {
         DataSourceContextHolder.setDbType("selectDataSource");
      } else {
         // 切换dataSource
         DataSourceContextHolder.setDbType("updateDataSource");
      }
   }
}
/**
 * 线程传递
 */
@Component
@Lazy(false)
public class DataSourceContextHolder {
   // 采用ThreadLocal 保存本地多数据源
   private static final ThreadLocal<String> contextHolder = new ThreadLocal<>();

   // 设置数据源类型
   public static void setDbType(String dbType) {
      contextHolder.set(dbType);
   }

   public static String getDbType() {
      return contextHolder.get();
   }

   public static void clearDbType() {
      contextHolder.remove();
   }

}

• 引入AbstractRoutingDataSource路由中介

/**
 * AbstractRoutingDataSource路由中介，能在运行时根据某种key值来动态切换到真正的DataSource上
 */
@Component
@Primary
public class DynamicDataSource extends AbstractRoutingDataSource {
   @Autowired
   @Qualifier("selectDataSource")
   private DataSource selectDataSource;

   @Autowired
   @Qualifier("updateDataSource")
   private DataSource updateDataSource;

   /**
    *返回生效的数据源名称
    */
   @Override
   protected Object determineCurrentLookupKey() {
      System.out.println("DataSourceContextHolder：：：" + DataSourceContextHolder.getDbType());
      return DataSourceContextHolder.getDbType();
   }

   /**
    * 将所有的数据源放到map中
    */
   @Override
   public void afterPropertiesSet() {
      Map<Object, Object> map = new HashMap<>();
      map.put("selectDataSource", selectDataSource);
      map.put("updateDataSource", updateDataSource);
      setTargetDataSources(map);
      setDefaultTargetDataSource(updateDataSource);
      super.afterPropertiesSet();
   }
}

推荐阅读：
<<<MySQL自带主从复制原理
<<<分表分库与分区的区别及拆分策略
<<<MyCat的分片查询原理
<<<Sharding-Jdbc实现读写分离
<<<Sharding-Jdbc的分片算法及分表分库
<<<Sharding-Jdbc与MyCat区别