SpringBoot配置多數(shù)據(jù)源的四種方式分享
更新時間：2023年07月03日 11:50:54 作者：中國胖子風清揚
在日常開發(fā)中我們都是以單個數(shù)據(jù)庫進行開發(fā)，在小型項目中是完全能夠滿足需求的，但是，當我們牽扯到大型項目的時候，單個數(shù)據(jù)庫就難以承受用戶的CRUD操作，那么此時，我們就需要使用多個數(shù)據(jù)源進行讀寫分離的操作，本文就給大家介紹SpringBoot配置多數(shù)據(jù)源的方式
1、所需的資源
- Spring boot
- Mybatis-plus
- Alibab Druid數(shù)據(jù)庫連接池
- MySql 數(shù)據(jù)庫
2、Spring Boot配置多數(shù)據(jù)源
數(shù)據(jù)庫
在YAML文件中定義數(shù)據(jù)源所需的數(shù)據(jù)
spring:
datasource:
type: com.alibaba.druid.pool.DruidDataSource ## 聲明數(shù)據(jù)源的類型
mysql-datasource1: ## 聲明第一個數(shù)據(jù)源所需的數(shù)據(jù)
url: jdbc:mysql://localhost:3306/mybatis?useSSL=true&serverTimezone=Asia/Shanghai
username: root
password: 123456
driver-class-name: com.mysql.cj.jdbc.Driver
mysql-datasource2: ## 聲明第二個數(shù)據(jù)源所需的數(shù)據(jù)
url: jdbc:mysql://localhost:3306/bookstore?useSSL=true&serverTimezone=Asia/Shanghai
username: root
password: 123456
driver-class-name: com.mysql.cj.jdbc.Driver
druid: ## druid數(shù)據(jù)庫連接池的基本初始化屬性
initial-size: 5 ## 連接池初始化的大小
min-idle: 1 ## 最小空閑的線程數(shù)
max-active: 20 ## 最大活動的線程數(shù)
mybatis-plus:
mapper-locations: classpath:/mapper/*.xml ## 配置MyBatis-Plus掃描Mapper文件的位置
type-aliases-package: com.example.sqlite.entity ## 創(chuàng)建別名的類所在的包mysql-datasource1、mysql-datasource2是自定義的數(shù)據(jù)。
定義多個數(shù)據(jù)源
@Configuration
public class DataSourceConfig {
@Bean(name = "mysqlDataSource1")
@ConfigurationProperties(prefix = "spring.datasource.mysql-datasource1")
public DataSource dataSource1(){
DruidDataSource build = DruidDataSourceBuilder.create().build();
return build;
}
@Bean(name = "mysqlDataSource2")
@ConfigurationProperties(prefix = "spring.datasource.mysql-datasource2")
public DataSource dataSource2(){
DruidDataSource build = DruidDataSourceBuilder.create().build();
return build;
}
}@ConfigurationProperties注解用于將YAML中指定的數(shù)據(jù)創(chuàng)建成指定的對象，但是，YAML中的數(shù)據(jù)必須要與對象對象中的屬性同名，不然無法由Spring Boot完成賦值。
由于我們要定義多個數(shù)據(jù)源，所以在Spring Boot數(shù)據(jù)源自動配置類中就無法確定導入哪個數(shù)據(jù)源來完成初始化，所以我們就需要禁用掉Spring Boot的數(shù)據(jù)源自動配置類，然后使用我們自定義的數(shù)據(jù)源配置類來完成數(shù)據(jù)源的初始化與管理。
@SpringBootApplication(exclude = {DataSourceAutoConfiguration.class})
public class DatasourceDomeApplication {
public static void main(String[] args) {
SpringApplication.run(DatasourceDomeApplication.class, args);
}
}在啟動類上聲明需要禁用的自動配置類：exclude = {DataSourceAutoConfiguration.class}
2.1、實現(xiàn)DataSource接口
缺點：產(chǎn)生大量的代碼冗余，在代碼中存在硬編碼。
2.1.1、代碼
@Component
@Primary
public class DynamicDataSource implements DataSource {
//使用ThreadLocal而不是String，可以在多線程的時候保證數(shù)據(jù)的可靠性
public static ThreadLocal<String> flag = new ThreadLocal<>();
@Resource
private DataSource mysqlDataSource1; // 注入第一個數(shù)據(jù)源
@Resource
private DataSource mysqlDataSource2; // 注入第二個數(shù)據(jù)源
public DynamicDataSource(){ // 使用構造方法初始化ThreadLocal的值
flag.set("r");
}
@Override
public Connection getConnection() throws SQLException {
// 通過修改ThreadLocal來修改數(shù)據(jù)源，
// 為什么通過修改狀態(tài)就能改變已經(jīng)注入的數(shù)據(jù)源？ 這就得看源碼了。
if(flag.get().equals("r")){
return mysqlDataSource1.getConnection();
}
return mysqlDataSource2.getConnection();
}
@Override
public Connection getConnection(String username, String password) throws SQLException {
return null;
}
@Override
public PrintWriter getLogWriter() throws SQLException {
return null;
}
@Override
public void setLogWriter(PrintWriter out) throws SQLException {
}
@Override
public void setLoginTimeout(int seconds) throws SQLException {
}
@Override
public int getLoginTimeout() throws SQLException {
return 0;
}
@Override
public Logger getParentLogger() throws SQLFeatureNotSupportedException {
return null;
}
@Override
public <T> T unwrap(Class<T> iface) throws SQLException {
return null;
}
@Override
public boolean isWrapperFor(Class<?> iface) throws SQLException {
return false;
}
}實現(xiàn)DataSource接口我們本質上只使用了一個方法，就是getConnection()這個無參的方法，但是DataSource接口中所有的方法我們也都需要實現(xiàn)，只是不用寫方法體而已，也就是存在了很多的 “廢方法” 。
@Primary注解 == @Order(1)，用于設置此類的注入順序。
2.1.2、使用
// 訪問第一個數(shù)據(jù)庫的t_user表
@RestController
public class UserController {
@Resource
private UserService userService;
@GetMapping(value = "/user_list")
public List<User> showUserList(){
DynamicDataSource.flag.set("read"); // 修改數(shù)據(jù)源的狀態(tài)
List<User> list = userService.list();
return list;
}
}// 訪問第二個數(shù)據(jù)庫的Book表
@RestController
public class BookController {
@Resource
private BookService BookService;
@GetMapping(value = "/Book_list")
public List<Book> getBookList(){
DynamicDataSource.flag.set("write"); // 修改數(shù)據(jù)源的狀態(tài)
List<Book> list = BookService.list();
return list;
}
}2.2、繼承AbstrictRoutingDataSource類
減少了代碼的冗余，但是還是會存在硬編碼。
2.2.1、代碼
@Primary
@Component
public class DynamicDataSource extends AbstractRoutingDataSource {
public static ThreadLocal<String> flag = new ThreadLocal<>();
@Resource
private DataSource mysqlDataSource1;
@Resource
private DataSource mysqlDataSource2;
public DynamicDataSource(){
flag.set("read");
}
@Override
protected Object determineCurrentLookupKey() { // 通過Key來得到數(shù)據(jù)源
return flag.get();
}
@Override
public void afterPropertiesSet() {
Map<Object,Object> targetDataSource = new ConcurrentHashMap<>();
targetDataSource.put("read",mysqlDataSource1);
// 將第一個數(shù)據(jù)源設置為默認的數(shù)據(jù)源。
super.setDefaultTargetDataSource(mysqlDataSource1);
targetDataSource.put("write",mysqlDataSource2);
// 將Map對象賦值給AbstrictRoutingDataSource內(nèi)部的Map對象中。
super.setTargetDataSources(targetDataSource);
super.afterPropertiesSet();
}
}AbstrictRoutingDataSource的本質就是利用一個Map將數(shù)據(jù)源存儲起來，然后通過Key來得到Value來修改數(shù)據(jù)源。
2.2.2、使用
// 訪問第一個數(shù)據(jù)庫的t_user表
@RestController
public class UserController {
@Resource
private UserService userService;
@GetMapping(value = "/user_list")
public List<User> showUserList(){
DynamicDataSource.flag.set("read"); // 修改數(shù)據(jù)源的狀態(tài)
List<User> list = userService.list();
return list;
}
}// 訪問第二個數(shù)據(jù)庫的Book表
@RestController
public class BookController {
@Resource
private BookService BookService;
@GetMapping(value = "/Book_list")
public List<Book> getBookList(){
DynamicDataSource.flag.set("write"); // 修改數(shù)據(jù)源的狀態(tài)
List<Book> list = BookService.list();
return list;
}
}2.3、使用Spring AOP + 自定義注解的形式
Spring AOP + 自定義注解的形式是一種推薦的寫法，減少代碼的冗余且不存在硬編碼。
此方法適合對指定功能操作指定數(shù)據(jù)庫的模式。
2.3.1、導入依賴
<dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-aop</artifactId> </dependency>
2.3.2、開啟AOP支持
@SpringBootApplication(exclude = {DataSourceAutoConfiguration.class})
@EnableAspectJAutoProxy //開啟Spring Boot對AOP的支持
public class AopDatasourceApplication {
public static void main(String[] args) {
SpringApplication.run(AopDatasourceApplication.class, args);
}
}2.3.3、定義枚舉來表示數(shù)據(jù)源的標識
public enum DataSourceType {
MYSQL_DATASOURCE1,
MYSQL_DATASOURCE2,
}2.3.4、繼承AbstractRoutingDataSource類
@Primary
@Component
public class DataSourceManagement extends AbstractRoutingDataSource {
public static ThreadLocal<String> flag = new ThreadLocal<>();
@Resource
private DataSource mysqlDataSource1;
@Resource
private DataSource mysqlDataSource2;
public DataSourceManagement(){
flag.set(DataSourceType.MYSQL_DATASOURCE1.name());
}
@Override
protected Object determineCurrentLookupKey() {
return flag.get();
}
@Override
public void afterPropertiesSet() {
Map<Object,Object> targetDataSource = new ConcurrentHashMap<>();
targetDataSource.put(DataSourceType.MYSQL_DATASOURCE1.name(),mysqlDataSource1);
targetDataSource.put(DataSourceType.MYSQL_DATASOURCE2.name(),mysqlDataSource2);
super.setTargetDataSources(targetDataSource);
super.setDefaultTargetDataSource(mysqlDataSource1);
super.afterPropertiesSet();
}
}2.3.5、自定義注解
@Target({ElementType.TYPE,ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface TargetDataSource {
DataSourceType value() default DataSourceType.MYSQL_DATASOURCE1;
}2.3.6、定義注解的實現(xiàn)類
@Component
@Aspect
@Slf4j
public class TargetDataSourceAspect {
@Before("@within(TargetDataSource) || @annotation(TargetDataSource)")
public void beforeNoticeUpdateDataSource(JoinPoint joinPoint){
TargetDataSource annotation = null;
Class<? extends Object> target = joinPoint.getTarget().getClass();
if(target.isAnnotationPresent(TargetDataSource.class)){
// 判斷類上是否標注著注解
annotation = target.getAnnotation(TargetDataSource.class);
log.info("類上標注了注解");
}else{
Method method = ((MethodSignature) joinPoint.getSignature()).getMethod();
if(method.isAnnotationPresent(TargetDataSource.class)){
// 判斷方法上是否標注著注解，如果類和方法上都沒有標注，則報錯
annotation = method.getAnnotation(TargetDataSource.class);
log.info("方法上標注了注解");
}else{
throw new RuntimeException("@TargetDataSource注解只能用于類或者方法上, 錯誤出現(xiàn)在:[" +
target.toString() +" " + method.toString() + "];");
}
}
// 切換數(shù)據(jù)源
DataSourceManagement.flag.set(annotation.value().name());
}
}在有的博客中也會使用@Around環(huán)繞通知的方式，但是環(huán)繞通知需要執(zhí)行joinPoint.process()方法來調用目標對象的方法，最后返回執(zhí)行的值，不然得不到所需要的數(shù)據(jù)。
我這里使用了@Before前置通知，效果是一樣的，因為@Around就會包含@Before。
@Around("@within(TargetDataSource) || @annotation(TargetDataSource)")
public Object beforeNoticeUpdateDataSource(ProceedingJoinPoint joinPoint){
// 省略邏輯代碼
Object result = null;
try {
result = joinPoint.proceed();
} catch (Throwable e) {
e.printStackTrace();
}
return result;
}ProceedingJoinPoint 對象只能在@Around環(huán)繞通知中使用，在其他通知中使用就會報錯。
2.3.7、使用
// 訪問第一個數(shù)據(jù)源。
@RestController
// 將注解標注在類上，表示本類中所有的方法都是使用數(shù)據(jù)源1
@TargetDataSource(value = DataSourceType.MYSQL_DATASOURCE1)
public class UserController {
@Resource
private UserService userService;
@GetMapping(value = "/user_list")
public List<User> showUserList(){
System.out.println(DataSourceType.MYSQL_DATASOURCE1.name());
List<User> list = userService.list();
return list;
}
}// 訪問第二個數(shù)據(jù)源
@RestController
public class BookController {
@Resource
private BookService BookService;
@GetMapping(value = "/Book_list")
// 將注解標注在方法上，表示此方法使用數(shù)據(jù)源2
@TargetDataSource(value = DataSourceType.MYSQL_DATASOURCE2)
public List<Book> getBookList(){
List<Book> list = BookService.list();
return list;
}
}2.4、通過SqlSessionFactory指定的數(shù)據(jù)源來操作指定目錄的XML文件
使用此方法則不會與上面所述的類有任何關系，本方法會重新定義類。本方法也是一種推薦的方法，適用于對指定數(shù)據(jù)庫的操作，也就是適合讀寫分離。不會存在代碼冗余和存在硬編碼。
2.4.1、項目的目錄結構
對所需要操作的數(shù)據(jù)庫的Mapper層和dao層分別建立一個文件夾。
2.4.2、配置YAML文件
spring:
datasource:
type: com.alibaba.druid.pool.DruidDataSource
mysql-datasource:
jdbc-url: jdbc:mysql://localhost:3306/mybatis?useSSL=true&serverTimezone=Asia/Shanghai
username: root
password: 123456
driver-class-name: com.mysql.cj.jdbc.Driver
sqlite-datasource:
jdbc-url: jdbc:mysql://localhost:3306/bookstore?useSSL=true&serverTimezone=Asia/Shanghai
username: root
password: 123456
driver-class-name: com.mysql.cj.jdbc.Driver
druid:
initial-size: 5
min-idle: 1
max-active: 20
mybatis-plus:
mapper-locations: classpath:/mapper/*.xml
type-aliases-package: com.example.sqlite.entity2.4.3、針對Mapper層通過SqlSessionFactory指定數(shù)據(jù)源來操作
2.4.3.1、創(chuàng)建MySql數(shù)據(jù)源
@Configuration
@MapperScan(basePackages = "com.example.sqlite.dao.mysql", sqlSessionFactoryRef = "MySQLSqlSessionFactory")
public class MySQLDataSourceConfig {
@Bean(name = "MySQLDataSource")
@Primary
@ConfigurationProperties(prefix = "spring.datasource.mysql-datasource")
public DataSource getDateSource1() {
return DataSourceBuilder.create().build();
}
@Bean(name = "MySQLSqlSessionFactory")
@Primary
public SqlSessionFactory test1SqlSessionFactory(
@Qualifier("MySQLDataSource") DataSource datasource) throws Exception {
MybatisSqlSessionFactoryBean bean = new MybatisSqlSessionFactoryBean ();
bean.setDataSource(datasource);
bean.setMapperLocations(// 設置mybatis的xml所在位置
new PathMatchingResourcePatternResolver().getResources("classpath*:mapper/mysql/*.xml"));
return bean.getObject();
}
@Bean("MySQLSqlSessionTemplate")
@Primary
public SqlSessionTemplate test1SqlSessionTemplate(
@Qualifier("MySQLSqlSessionFactory") SqlSessionFactory sessionFactory) {
return new SqlSessionTemplate(sessionFactory);
}
@Bean
public PlatformTransactionManager transactionManager(@Qualifier("MySQLDataSource")DataSource dataSource) {
return new DataSourceTransactionManager(dataSource);
}
}2.4.3.2、創(chuàng)建Sqlite數(shù)據(jù)源
@Configuration
@MapperScan(basePackages = "com.example.sqlite.dao.sqlite", sqlSessionFactoryRef = "SqliteSqlSessionFactory")
public class SqliteDataSourceConfig {
@Bean(name = "SqliteDateSource")
@ConfigurationProperties(prefix = "spring.datasource.sqlite-datasource")
public DataSource getDateSource1() {
return DataSourceBuilder.create().build();
}
@Bean(name = "SqliteSqlSessionFactory")
public SqlSessionFactory test1SqlSessionFactory(
@Qualifier("SqliteDateSource") DataSource datasource) throws Exception {
MybatisSqlSessionFactoryBean bean = new MybatisSqlSessionFactoryBean();
bean.setDataSource(datasource);
bean.setMapperLocations(
new PathMatchingResourcePatternResolver().getResources("classpath*:mapper/sqlite/*.xml"));
return bean.getObject();
}
@Bean("SqliteSqlSessionTemplate")
public SqlSessionTemplate test1SqlSessionTemplate(
@Qualifier("SqliteSqlSessionFactory") SqlSessionFactory sessionFactory) {
return new SqlSessionTemplate(sessionFactory);
}
@Bean
public PlatformTransactionManager transactionManager(@Qualifier("SqliteDateSource")DataSource dataSource) {
return new DataSourceTransactionManager(dataSource);
}
}- @MapperScan注解中的basePackages指向的是指定的Dao層。
- @MapperScan注解中sqlSessionFactoryRef 用來指定使用某個SqlSessionFactory來操作數(shù)據(jù)源。
- bean.setMapperLocations(
new PathMatchingResourcePatternResolver()
.getResources(“classpath*:mapper/sqlite/*.xml”)); 指向的是操作執(zhí)行數(shù)據(jù)庫的Mapper層。
如果使用SQLite數(shù)據(jù)庫，那么就必須在項目中內(nèi)嵌SQLite數(shù)據(jù)庫，這個一個輕量級的數(shù)據(jù)庫，不同于Mysql，SQLite不需要服務器，SQLite適合使用于移動APP開發(fā)。
像微信，用戶的聊天記錄就是使用這個數(shù)據(jù)庫進行存儲。SQLite也可以使用在Web端，只是不太方便。
2.4.4、使用
// 訪問第一個數(shù)據(jù)庫
@RestController
public class UserController {
@Resource
private UserService userService;
@GetMapping(value = "/user_list")
public List<User> showUserList(){
List<User> list = userService.list();
return list;
}
}// 訪問第二個數(shù)據(jù)庫
@RestController
public class AddressController {
@Resource
private AddressService addressService;
@GetMapping(value = "/address_list")
public List<Address> getAddressList(){
List<Address> list = addressService.list();
return list;
}
}使用此種方法不會存在任何代碼的冗余以及硬編碼的存在，但是需要分層明確。
唯一的不足就是添加一個數(shù)據(jù)源就需要重新寫一個類，而這個類中的代碼大部分又是相同的。
3、總結
- 實現(xiàn)DataSource接口這種寫法是不推薦的。
- 推薦使用Spring Boot + 自定義注解的方式與SqlSessionFactory方式。
另外，Spring AOP中各種通知的執(zhí)行順序如下圖所示：
以上就是SpringBoot配置多數(shù)據(jù)源的四種方式分享的詳細內(nèi)容，更多關于SpringBoot配置多數(shù)據(jù)源的資料請關注腳本之家其它相關文章！
相關文章
大家好，本篇文章主要講的是Java字符串拼接詳解，感興趣的同學趕快來看一看吧，對你有幫助的話記得收藏一下2022-02-02
springboot與vue詳解實現(xiàn)短信發(fā)送流程
隨著人工智能的不斷發(fā)展，機器學習這門技術也越來越重要，很多人都開啟了學習機器學習，本文就介紹了機器學習的基礎內(nèi)容2022-06-06
java并發(fā)編程之深入理解Synchronized的使用
文詳細講述了線程、進程的關系及在操作系統(tǒng)中的表現(xiàn)，這是多線程學習必須了解的基礎。本文將接著講一下Java線程同步中的一個重要的概念synchronized，希望能夠給你有所幫助2021-06-06
使用cmd根據(jù)WSDL網(wǎng)址生成java客戶端代碼的實現(xiàn)
這篇文章主要介紹了使用cmd根據(jù)WSDL網(wǎng)址生成java客戶端代碼的實現(xiàn)方式，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2021-03-03
這篇文章主要介紹了Java如何把int類型轉換成byte,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2020-02-02
在Web應用開發(fā)過程中,接口重復提交問題一直是一個需要重點關注和解決的難題,本文將從SpringBoot應用的角度出發(fā),探討在單機環(huán)境和分布式環(huán)境下如何有效防止接口重復提交,希望通過本文的介紹,讀者能夠掌握在SpringBoot應用中防止接口重復提交的有效方法2024-05-05
這篇文章主要給大家介紹了關于JAVA多線程的使用場景與注意事項的相關資料，文中通過示例代碼介紹的非常詳細，對大家的學習或者使用java具有一定的參考學習價值，需要的朋友們下面來一起學習學習吧2019-03-03
這篇文章介紹了Java下載文件的四種方式，文中通過示例代碼介紹的非常詳細。對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下2021-12-12
這篇文章主要介紹了講解了在面試中jvm如何實現(xiàn)多態(tài)，怎樣回答才能得到高分的問題分析，有需要的朋友可以借鑒參考下，祝大家早日升職加薪多多進步2022-01-01
Mybatis是一個優(yōu)秀的持久層框架，它對JDBC操作數(shù)據(jù)庫的過程進行封裝，使開發(fā)者只需要關注sql本身。我們原來使用JDBC操作數(shù)據(jù)庫，需要手動的寫代碼去注冊驅動、獲取connection、獲取statement等等，現(xiàn)在Mybaits幫助我們把這些事情做了，我們只需要關注我們的業(yè)務sql即可2021-06-06