将数据库相关依赖引入
<dependency><groupId>com.mysql</groupId><artifactId>mysql-connector-j</artifactId></dependency><dependency><groupId>com.baomidou</groupId><artifactId>mybatis-plus-boot-starter</artifactId><version>3.4.2</version></dependency><dependency><groupId>org.projectlombok</groupId><artifactId>lombok</artifactId></dependency><dependency><groupId>com.alibaba</groupId><artifactId>druid-spring-boot-starter</artifactId><version>1.2.16</version></dependency><dependency><groupId>com.baomidou</groupId><artifactId>dynamic-datasource-spring-boot-starter</artifactId><version>3.4.1</version></dependency>
application.yml配置文件(!!!不要配错datasource)
spring:datasource:dynamic:enabled: true #启用动态数据源,默认trueprimary: db1 #设置默认的数据源或者数据源组,默认值即为masterstrict: false #严格匹配数据源,默认false. true未匹配到指定数据源时抛异常,false使用默认数据源grace-destroy: false #是否优雅关闭数据源,默认为false,设置为true时,关闭数据源时如果数据源中还存在活跃连接,至多等待10s后强制关闭datasource: #这个还有一个datasource属性,不要丢了db1:driver-class-name: com.mysql.cj.jdbc.Driverurl: jdbc:mysql://localhost:3306/db1?serverTimezone=Asia/Shanghai&useUnicode=true&characterEncoding=utf-8&zeroDateTimeBehavior=convertToNull&useSSL=false&allowPublicKeyRetrieval=trueusername: rootpassword: rootdb2:driver-class-name: com.mysql.cj.jdbc.Driverurl: jdbc:mysql://localhost:3306/db2?serverTimezone=Asia/Shanghai&useUnicode=true&characterEncoding=utf-8&zeroDateTimeBehavior=convertToNull&useSSL=false&allowPublicKeyRetrieval=trueusername: rootpassword: root
动态切换数据源
有两种情况:
- 1、不同访问的用户需要连接不同的数据源,但一个用户只使用到一个数据源
- 2、一个用户需要同时使用不同的数据源,不同的业务逻辑操作不同的数据源
第一种情况,假设每个用户有唯一的省份,每个省份又有对应的数据库,用户在发送请求时携带省份唯一标识,根据省份标识来连接对应的数据源。
- 将省份标识作为数据源名称,如同上面配置文件的db1,db2
- 在用户登录时获取到其携带的省份标识,并将其保存到ThreadLocal中,登录过期/注销时清除
DynamicDataSourceContextHolder.push(获取到的数据源标识);//保存到ThreadLocal
DynamicDataSourceContextHolder.peek(db);//从ThreadLocal获取不删除
DynamicDataSourceContextHolder.poll(db);//从ThreadLocal获取数据,并删除该数据
DynamicDataSourceContextHolder.peek(db);//从ThreadLocal中删除
DynamicDataSourceContextHolder是Dynamic组件提供的工具,底层是ThreadLocal里维护了一个Deque双向队列,然后数据存放在队列中。自己直接用Thread提供的set,get,remove方法是一样的效果。
- dynamic会自动获取ThreadLocal中的省份标识,并自动查找连接对应的数据源
第二种情况,一个用户需要处理多个数据源的数据
Dynamic提供了 @DS("数据源名称")
注解,该注解可以作用在方法上,类上,接口上
对于某个接口、类、方法操作的是哪个数据源,直接使用@DS注解声明对应的数据源就行了
示例:在调用getName方法时,就会查询db1
数据源
@DS("db1")public String getName(String id) {return dynamicMapper.queryName(id);}
注:使用ThreadLocal的方式更灵活,适用于复杂的情况,但使用后一定要清除ThreadLocal中的数据,否则会出现线程安全问题和内存泄漏问题