記得面試時候，有面試官會問道，你們多數(shù)據(jù)源是怎么實現(xiàn)的呀。.......，一陣蒙蔽中，然后說道我們之前項目中，沒有用到多數(shù)據(jù)源。

所幸，目前做得項目中有一個業(yè)務邏輯中，用到多個數(shù)據(jù)庫數(shù)據(jù)情況，多數(shù)據(jù)源華麗上線。

一. mybatis plus

因為我們項目是springboot+mybatis plus，有些人一看，mybatis還知道對吧，mybatis plus是什么鬼，其實字面意思可以理解，就是對mybatis進行一些功能改造，一些封裝升級，然后用起來特別方便。

核心功能的升級主要是以下三點：

支持通用的 CRUD、代碼生成器與條件構造器。

通用 CRUD：定義好 Mapper 接口后，只需要繼承 BaseMapper<T> 接口即可獲得通用的增刪改查功能，無需編寫任何接口方法與配置文件條件構造器：通過 EntityWrapper<T> （實體包裝類），可以用于拼接 SQL 語句，并且支持排序、分組查詢等復雜的 SQL代碼生成器：支持一系列的策略配置與全局配置，比 MyBatis 的代碼生成更好用

二.多數(shù)據(jù)源配置開始

思路：

1、yml中配置多個數(shù)據(jù)源信息2、通過AOP切換不同數(shù)據(jù)源3、配合mybatis plus使用

1、yml配置

spring: aop: proxy-target-class: true auto: true datasource: druid: db1: url: jdbc:mysql://localhost:3306/eboot username: root password: root driver-class-name: com.mysql.jdbc.Driver initialSize: 5 minIdle: 5 maxActive: 20 db2: url: jdbc:oracle:thin:@192.168.136.222:ORCL username: sa password: sa123456 driver-class-name: oracle.jdbc.OracleDriver initialSize: 5 minIdle: 5 maxActive: 20

2、啟動多個數(shù)據(jù)源

@EnableTransactionManagement //開啟事務@Configuration //spring中常用到注解，與xml配置相對立。是兩種加載bean方式@MapperScan('com.df.openapi.**.mapper.db*') // 掃描mapperdao的地址public class MybatisPlusConfig { @Bean public PaginationInterceptor paginationInterceptor() { PaginationInterceptor paginationInterceptor = new PaginationInterceptor();// paginationInterceptor.setLocalPage(true); // 由于版本問題，有些類可能招不到這個方法，需要升級jar包 return paginationInterceptor; } @Bean(name = 'db1') @ConfigurationProperties(prefix = 'spring.datasource.druid.db1') public DataSource db1() { return DruidDataSourceBuilder.create().build(); } @Bean(name = 'db2') @ConfigurationProperties(prefix = 'spring.datasource.druid.db2') public DataSource db2() { return DruidDataSourceBuilder.create().build(); } /** * 動態(tài)數(shù)據(jù)源配置 * * @return */ @Bean @Primary public DataSource multipleDataSource(@Qualifier('db1') DataSource db1, @Qualifier('db2') DataSource db2) { DynamicDataSource dynamicDataSource = new DynamicDataSource(); Map<Object, Object> targetDataSources = new HashMap<>(); targetDataSources.put(DBTypeEnum.db1.getValue(), db1); targetDataSources.put(DBTypeEnum.db2.getValue(), db2); dynamicDataSource.setTargetDataSources(targetDataSources); dynamicDataSource.setDefaultTargetDataSource(db2); // 程序默認數(shù)據(jù)源，這個要根據(jù)程序調用數(shù)據(jù)源頻次，經常把常調用的數(shù)據(jù)源作為默認 return dynamicDataSource; } @Bean('sqlSessionFactory') public SqlSessionFactory sqlSessionFactory() throws Exception { MybatisSqlSessionFactoryBean sqlSessionFactory = new MybatisSqlSessionFactoryBean(); sqlSessionFactory.setDataSource(multipleDataSource(db1(), db2())); MybatisConfiguration configuration = new MybatisConfiguration(); configuration.setJdbcTypeForNull(JdbcType.NULL); configuration.setMapUnderscoreToCamelCase(true); configuration.setCacheEnabled(false); sqlSessionFactory.setConfiguration(configuration); //PerformanceInterceptor(),OptimisticLockerInterceptor() //添加分頁功能 sqlSessionFactory.setPlugins(new Interceptor[]{paginationInterceptor() });// sqlSessionFactory.setGlobalConfig(globalConfiguration()); //注釋掉全局配置，因為在xml中讀取就是全局配置 return sqlSessionFactory.getObject(); } /* @Bean public GlobalConfiguration globalConfiguration() { GlobalConfiguration conf = new GlobalConfiguration(new LogicSqlInjector()); conf.setLogicDeleteValue('-1'); conf.setLogicNotDeleteValue('1'); conf.setIdType(0); conf.setMetaObjectHandler(new MyMetaObjectHandler()); conf.setDbColumnUnderline(true); conf.setRefresh(true); return conf; }*/}

3、DBType枚舉類

package com.df.openapi.config.db; public enum DBTypeEnum { db1('db1'), db2('db2'); private String value; DBTypeEnum(String value) { this.value = value; } public String getValue() { return value; }}

4、動態(tài)數(shù)據(jù)源決策

package com.df.openapi.config.db; import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource; public class DynamicDataSource extends AbstractRoutingDataSource {private static final Logger LOGGER = LoggerFactory.getLogger(DynamicDataSource.class);@Overrideprotected Object determineCurrentLookupKey() { String datasource = DataSourceContextHolder.getDbType(); LOGGER.debug('使用數(shù)據(jù)源 {}', datasource); return datasource;}}

5、設置、獲取數(shù)據(jù)源

public class DataSourceContextHolder { private static final Logger LOGGER = LoggerFactory.getLogger(DataSourceContextHolder.class); private static final ThreadLocal contextHolder = new ThreadLocal<>(); //實際上就是開啟多個線程，每個線程進行初始化一個數(shù)據(jù)源 /** * 設置數(shù)據(jù)源 * @param dbTypeEnum */ public static void setDbType(DBTypeEnum dbTypeEnum) { contextHolder.set(dbTypeEnum.getValue()); } /** * 取得當前數(shù)據(jù)源 * @return */ public static String getDbType() { return (String) contextHolder.get(); } /** * 清除上下文數(shù)據(jù) */ public static void clearDbType() { contextHolder.remove(); }}

6、AOP實現(xiàn)的數(shù)據(jù)源切換

@Order設置的足夠小是為了讓他先執(zhí)行

/** * aop的實現(xiàn)的數(shù)據(jù)源切換<br> * aop切點，實現(xiàn)mapper類找尋，找到所屬大本營以后，如db1Aspect(),則會調用<br> * db1()前面之前的操作，進行數(shù)據(jù)源的切換。 */@Component@Order(value = -100)@Slf4j@Aspectpublic class DataSourceAspect { @Pointcut('execution(* com.zwyl.bazhong.dao.mapper.db1..*.*(..))') private void db1Aspect() { } @Pointcut('execution(* com.zwyl.bazhong.dao.mapper.db2..*.*(..))') private void db2Aspect() { } @Before('db1Aspect()') public void db1() { log.info('切換到db1 數(shù)據(jù)源...'); DataSourceContextHolder.setDbType(DBTypeEnum.db1); } @Before('db2Aspect()') public void db2() { log.info('切換到db2 數(shù)據(jù)源...'); DataSourceContextHolder.setDbType(DBTypeEnum.db2); }}

7、mapper層結構

8、寫一個service測試一下

@Servicepublic class DictServiceImpl implements IDictService { @Resource private PtDictMapper ptDictMapper; //來自db1 @Resource private SysDictMapper sysDictMapper; // 來自db2 @Override public void getById(String id) { PtDict dict = ptDictMapper.selectById('2bf6257fc8fe483c84c1ad7e89d632f6'); SysDict sysDict = sysDictMapper.getById('49'); System.out.println('123'); }}

9、測試結果

總結： 其實整個過程可以理解成，配置多數(shù)據(jù)源 xml中 -------> 然后通過加載多數(shù)源到spring工廠中-------->然后創(chuàng)建多線程，每個數(shù)據(jù)源對應一個數(shù)據(jù)源--------->然后實際調用時候，會先通過aop匹配到某一具體數(shù)據(jù)源------------->然后實例化當前數(shù)據(jù)源

到此這篇關于Mybatis plus 配置多數(shù)據(jù)源的實現(xiàn)示例的文章就介紹到這了,更多相關Mybatis plus 多數(shù)據(jù)源內容請搜索好吧啦網以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持好吧啦網！