Mybatis plus 配置多數據源的實現示例

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

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

一. mybatis plus

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

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

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

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

二.多數據源配置開始

思路：

1、yml中配置多個數據源信息2、通過AOP切換不同數據源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、啟動多個數據源

@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(); } /** * 動態數據源配置 * * @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); // 程序默認數據源，這個要根據程序調用數據源頻次，經常把常調用的數據源作為默認 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、動態數據源決策

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('使用數據源 {}', datasource); return datasource;}}

5、設置、獲取數據源

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

6、AOP實現的數據源切換

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

/** * aop的實現的數據源切換<br> * aop切點，實現mapper類找尋，找到所屬大本營以后，如db1Aspect(),則會調用<br> * db1()前面之前的操作，進行數據源的切換。 */@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 數據源...'); DataSourceContextHolder.setDbType(DBTypeEnum.db1); } @Before('db2Aspect()') public void db2() { log.info('切換到db2 數據源...'); 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、測試結果

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

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