在前面两节我们已经完成springboot操作mysql数据库，但是在实际业务场景中，数据量迅速增长，一个库一个表已经满足不了我们的需求的时候，我们就会考虑分库分表的操作，那么接下来我们就去学习一下，在springboot中如何实现多数据源，动态数据源切换，读写分离等操作。
实现1、建库建表首先，我们在本地新建三个数据库名分别为master,slave1,slave2，我们的目前就是写入操作都是在master，查询是 slave1,slave2
因此我们在上一篇也就是【SpringBoot2.0系列05】SpringBoot之整合Mybatis基础上进行改动，
我们在master slave1 slave2中都创建user表 其中初始化salve1库的user表数据为
image.png
初始化
slave2库的user表
image.png
具体的数据库脚本如下
create table master.user ( id bigint auto_increment comment '主键' primary key, age int null comment '年龄', password varchar(32) null comment '密码', sex int null comment '性别', username varchar(32) null comment '用户名')engine=MyISAM collate=utf8mb4_bin ;create table slave1.user ( id bigint auto_increment comment '主键' primary key, age int null comment '年龄', password varchar(32) null comment '密码', sex int null comment '性别', username varchar(32) null comment '用户名')engine=MyISAM collate=utf8mb4_bin ;INSERT INTO slave1.user (id, age, password, sex, username) VALUES (2, 22, 'admin', 1, 'admin');create table slave2.user ( id bigint auto_increment comment '主键' primary key, age int null comment '年龄', password varchar(32) null comment '密码', sex int null comment '性别', username varchar(32) null comment '用户名')engine=MyISAM collate=utf8mb4_bin ;INSERT INTO slave2.user (id, age, password, sex, username) VALUES (3, 19, 'uuu', 2, 'user');INSERT INTO slave2.user (id, age, password, sex, username) VALUES (4, 18, 'bbbb', 1, 'zzzz');2、配置多数据源
经过上面初始化 我们的master.user是一张空表，我们等下的插入与更新操作就在这上面，那么我们的查询操作就是在slave1.user跟slave2.user上面了。
上面我们的数据库初始化工作完成了，接下来就是实现动态数据源的过程
首先我们需要在我们的application.yml配置我们的三个数据源
server:
port: 8989spring:
datasource:
master:
password: root url: jdbc:mysql://127.0.0.1:3306/master?useUnicode=true&characterEncoding=UTF-8
driver-class-name: com.mysql.jdbc.Driver
username: root type: com.zaxxer.hikari.HikariDataSource cluster:
- key: slave1 password: root url: jdbc:mysql://127.0.0.1:3306/slave1?useUnicode=true&characterEncoding=UTF-8
idle-timeout: 20000
driver-class-name: com.mysql.jdbc.Driver
username: root type: com.zaxxer.hikari.HikariDataSource
- key: slave2 password: root url: jdbc:mysql://127.0.0.1:3306/slave2?useUnicode=true&characterEncoding=UTF-8
driver-class-name: com.mysql.jdbc.Driver
username: rootmybatis:
mapper-locations: classpath:/mybatis/mapperpublic class DynamicDataSourceRegister implements importBeanDefinitionRegistrar, EnvironmentAware { private static final Logger logger = LoggerFactory.getLogger(DynamicDataSourceRegister.class);
private Environment evn;
private final static ConfigurationPropertyNameAliases aliases = new ConfigurationPropertyNameAliases();
static {
aliases.addAliases("url", new String[]{"jdbc-url"});
aliases.addAliases("username", new String[]{"user"});
}
private Map customDataSources = new HashMap();
private Binder binder;
@Override
public void registerBeanDefinitions(Annotationmetadata annotationmetadata, BeanDefinitionRegistry beanDefinitionRegistry) { // 获取所有数据源配置
Map config, defauleDataSourceProperties;
defauleDataSourceProperties = binder.bind("spring.datasource.master", Map.class).get(); // 获取数据源类型
String typeStr = evn.getProperty("spring.datasource.master.type"); // 获取数据源类型
Class clazz = getDataSourceType(typeStr); // 绑定默认数据源参数 也就是主数据源
DataSource consumerDatasource, defaultDatasource = bind(clazz, defauleDataSourceProperties);
DynamicDataSourceContextHolder.dataSourceIds.add("master");
logger.info("注册默认数据源成功"); // 获取其他数据源配置
List 上面代码需要注意的是在springboot2.x系列中用于绑定的工具类如RelaxedPropertyResolver已经无法现在使用Binder代替。上面代码主要是读取application中数据源的配置，先读取spring.datasource.master构建默认数据源,然后在构建cluster中的数据源。
在这里注册完数据源之后，我们需要通过@import注解把我们的数据源注册器导入到spring中 在启动类Chapter5Application.java加上如下注解@import(DynamicDataSourceRegister.class)。
其中我们用到了一个DynamicDataSourceContextHolder 中的静态变量来保存我们已经注册成功的数据源的key,至此我们的数据源注册就已经完成了。
我们需要新建一个数据源上下文，用户记录当前线程使用的数据源的key是什么，以及记录所有注册成功的数据源的key的集合。对于线程级别的私有变量，我们首先ThreadLocal来实现。
com.yukong.chapter5.config.DynamicDataSourceContextHolder代码取下
public class DynamicDataSourceContextHolder { private static Logger logger = LoggerFactory.getLogger(DynamicDataSourceContextHolder.class);
public static List dataSourceIds = new ArrayList<>();
private static final ThreadLocal HOLDER = new ThreadLocal<>(); public static String getDataSourceRouterKey () { return HOLDER.get();
} public static void setDataSourceRouterKey (String dataSourceRouterKey) {
logger.info("切换至{}数据源", dataSourceRouterKey);
HOLDER.set(dataSourceRouterKey);
}
public static void removeDataSourceRouterKey () {
HOLDER.remove();
}
public static boolean containsDataSource(String dataSourceId){ return dataSourceIds.contains(dataSourceId);
}
} 5、动态数据源路由前面我们以及新建了数据源上下文，用于存储我们当前线程的数据源key那么怎么通知spring用key当前的数据源呢，查阅资料可知，spring提供一个接口，名为AbstractRoutingDataSource的抽象类，我们只需要重写determineCurrentLookupKey方法就可以，这个方法看名字就知道，就是返回当前线程的数据源的key，那我们只需要从我们刚刚的数据源上下文中取出我们的key即可，那么具体代码取下。
com.yukong.chapter5.config.DynamicRoutingDataSource
public class DynamicRoutingDataSource extends AbstractRoutingDataSource { private static Logger logger = LoggerFactory.getLogger(DynamicRoutingDataSource.class); @Override
protected Object determineCurrentLookupKey() {
String dataSourceName = DynamicDataSourceContextHolder.getDataSourceRouterKey();
logger.info("当前数据源是：{}", dataSourceName); return DynamicDataSourceContextHolder.getDataSourceRouterKey();
}
}6、通过aop+注解实现动态数据源的切换现在spring也已经知道通过key来取对应的数据源，我们现在只需要实现给对应的类或者方法设置他们的数据源的key，并且保存在数据源上下文中即可。这里我们采用注解来设置数据源，通过aop拦截并且保存到数据源上下中。
我们新建一个标识数据源的注解@DataSource具体代码取下
com.yukong.chapter5.annotation.DataSource
@Target({ElementType.METHOD, ElementType.TYPE, ElementType.PARAMETER})@Retention(RetentionPolicy.RUNTIME)@documentedpublic @interface DataSource { String value() default "master"; //该值即key值}其中他的默认值是master,因为我们默认数据源的key也是master。也就是说如果你直接用注解，而不指定value的话，那么默认就使用master默认数据源。
然后我们新建一个aop类来拦截。代码如下
com.yukong.chapter5.aop
package com.yukong.chapter5.aop;import com.yukong.chapter5.annotation.DataSource;import com.yukong.chapter5.config.DynamicDataSourceContextHolder;import org.aspectj.lang.JoinPoint;import org.aspectj.lang.annotation.After;import org.aspectj.lang.annotation.Aspect;import org.aspectj.lang.annotation.Before;import org.slf4j.Logger;import org.slf4j.LoggerFactory;import org.springframework.stereotype.Component;import org.springframework.util.StringUtils;@Aspect@Componentpublic class DynamicDataSourceAspect { private static final Logger logger = LoggerFactory.getLogger(DynamicDataSourceAspect.class); @Before("@annotation(ds)") public void changeDataSource(JoinPoint point, DataSource ds) throws Throwable {
String dsId = ds.value(); if (DynamicDataSourceContextHolder.dataSourceIds.contains(dsId)) {
logger.debug("Use DataSource :{} >", dsId, point.getSignature());
} else {
logger.info("数据源[{}]不存在，使用默认数据源 >{}", dsId, point.getSignature());
DynamicDataSourceContextHolder.setDataSourceRouterKey(dsId);
}
} @After("@annotation(ds)") public void restoreDataSource(JoinPoint point, DataSource ds) {
logger.debug("Revert DataSource : " + ds.value() + " > " + point.getSignature());
DynamicDataSourceContextHolder.removeDataSourceRouterKey();
}
}通过aop拦截，获取注解上面的value的值key，然后取判断我们注册的keys集合中是否有这个key,如果没有，则使用默认数据源，如果有，则设置上下文中当前数据源的key为注解的value。
7、测试
最后我们在对应的方法上面加上注解来测试一下即可
我们在UserMapper.java上面加上注解，并且进行测试。
public interface UserMapper {
@DataSource //默认数据源
int save(User user);
@DataSource //默认数据源
int update(User user);
@DataSource //默认数据源
int deleteById(Long id);
@DataSource("slave1") //slave1
User selectById(Long id);
@DataSource("slave2") //slave2
List selectAll();
} 上面代码可以知道，我们的新增，修改，删除方法都是在默认数据master上，我们的id查询是在slave1，我们的查询所有在slave2，我们编写测试类来测试把。
@SpringBootTest@RunWith(SpringJUnit4ClassRunner.class)public class UserMapperTest { @Autowired
private UserMapper userMapper; @Test
public void save() {
User user = new User();
user.setUsername("master");
user.setPassword("master");
user.setSex(1);
user.setAge(18);
Assert.assertEquals(1,userMapper.save(user));
} @Test
public void update() {
User user = new User();
user.setId(8L);
user.setPassword("newpassword"); // 返回插入的记录数 ，期望是1条 如果实际不是一条则抛出异常
Assert.assertEquals(1,userMapper.update(user));
} @Test
public void selectById() {
User user = userMapper.selectById(2L);
System.out.println("id:" + user.getId());
System.out.println("name:" + user.getUsername());
System.out.println("password:" + user.getPassword());
} @Test
public void deleteById() {
Assert.assertEquals(1,userMapper.deleteById(1L));
} @Test
public void selectAll() {
List users= userMapper.selectAll();
users.forEach(user -> {
System.out.println("id:" + user.getId());
System.out.println("name:" + user.getUsername());
System.out.println("password:" + user.getPassword());
});
}
} 首先测试save方法，它将会把数据存到master库的user表，
现在user表是空的，如图
image.png
运行save方法。
image.png
绿色，测试通过，并且日志提示数据源注册成功，一共三个。并且当前使用的master数据源，我们再去master数据库看看有没有数据。
image.png
如上图，插入成功。
新增方法测试完成了。我们在测试一下修改与删除。
image.png
修改方法也测试通过，查看数据库。
image.png
修改成功，删除方法我就不测试， 我们在测试测试，slave1跟slave2数据源的方法，
首先测试slave1的主键查询方法，先看数据库 slave1有哪些数据。
image.png
slave1.user就一条id为2 的数据并且id为2 的数据就slave1才有，我们测试一下能不能查到。
image.png
运行通过，数据源为slave1并且数据也正确显示。
最后我们来测试一下slave2的selectAll方法把，同样先看看slave2.user中有什么数据。
image.png
从图中，得知slave2.user中有两条数据，id分别为3，4。接下来运行测试方法。
结果如图。
image.png
日志提示数据源切换值slave2，并且id为3，4的数据也成功打印。
那么至此我们的多数据源动态数据源就完成了。
主要的思路就是
配置文件中配置多个数据源
启动类注册动态数据源
在需要的方法上使用注解指定数据源
作者：余空啊
链接：https://www.jianshu.com/p/cac4759b2684
