關於MybatisPlus配置雙數據庫驅動連接數據庫問題
最近項目中需要用到2種數據庫驅動連接數據庫,下面我們基於MybatisPlus實現一下
具體實現
1、在pom.xml中添加如下依賴:
<properties> <java.version>1.8</java.version> <lombok.version>1.18.2</lombok.version> <mybatis-plus.version>3.2.0</mybatis-plus.version> <druid.version>1.1.9</druid.version> <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding> <project.reporting.outputEncoding>UTF-8</project.reporting.outputEncoding> </properties> <dependencies> <dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-web</artifactId> </dependency> <!-- mysql--> <dependency> <groupId>mysql</groupId> <artifactId>mysql-connector-java</artifactId> <scope>runtime</scope> </dependency> <!-- postgrepsql--> <dependency> <groupId>org.postgresql</groupId> <artifactId>postgresql</artifactId> <scope>runtime</scope> </dependency> <dependency> <groupId>com.baomidou</groupId> <artifactId>mybatis-plus-boot-starter</artifactId> <version>${mybatis-plus.version}</version> </dependency> <dependency> <groupId>com.baomidou</groupId> <artifactId>mybatis-plus</artifactId> <version>${mybatis-plus.version}</version> </dependency> <dependency> <groupId>org.projectlombok</groupId> <artifactId>lombok</artifactId> <version>${lombok.version}</version> </dependency> <dependency> <groupId>com.alibaba</groupId> <artifactId>druid-spring-boot-starter</artifactId> <version>${druid.version}</version> </dependency> <dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-aop</artifactId> </dependency> </dependencies>
2、在yml配置文件中添加如下配置:
server: port: 8080 spring: application: name: xxxx datasource: druid: # mysql數據源配置 db1: driver-class-name: com.mysql.jdbc.Driver url: jdbc:mysql://127.0.0.1:3306/db1?useUnicode=true&characterEncoding=UTF-8&useSSL=false username: ${username} password: ${password} initial-size: 5 min-idle: 5 max-active: 50 # postgresql 數據源配置 db2: driver-class-name: org.postgresql.Driver url: jdbc:postgresql://127.0.0.1:5432/db2?useUnicode=true&characterEncoding=utf-8 username: ${username} password: ${password} initial-size: 5 min-idle: 5 max-active: 50 # mybatis-plus配置 mybatis-plus: type-aliases-package: com.dms.gateway.api.entity mapper-locations: classpath:/mapper/*Mapper.xml global-config: db-config: id-type: auto field-strategy: not_empty logic-delete-value: 1 logic-not-delete-value: 0 configuration: map-underscore-to-camel-case: true cache-enabled: false call-setters-on-nulls: true log-impl: org.apache.ibatis.logging.stdout.StdOutImpl
3、新建DataSourceEnum枚舉類,如下:
public enum DataSourceEnum { DB1("db1"), DB2("db2"); private String value; DataSourceEnum(String value){this.value=value;} public String getValue() { return value; } }
4、新建DataSourceContextHolder類,如下:
public class DataSourceContextHolder { // 默認數據源 public static final String DEFAULT_DS = DataSourceEnum.DB1.getValue(); private static final ThreadLocal<String> contextHolder = new InheritableThreadLocal<>(); /** * 設置數據源 * @param db */ public static void setDataSource(String db){ contextHolder.set(db); } /** * 取得當前數據源 * @return */ public static String getDataSource(){ return contextHolder.get(); } /** * 清除上下文數據 */ public static void clear(){ contextHolder.remove(); } }
5、新建MultipleDataSource類,如下:
public class MultipleDataSource extends AbstractRoutingDataSource { @Override protected Object determineCurrentLookupKey() { return DataSourceContextHolder.getDataSource(); } }
6、新建DataSource註解,如下:
@Target({ElementType.METHOD,ElementType.TYPE}) @Retention(RetentionPolicy.RUNTIME) @Documented public @interface DataSource { DataSourceEnum value() default DataSourceEnum.DB1; }
7、新建面向類和方法級別的切面,如下:
@Component @Slf4j @Aspect @Order(-6) public class DataSourceClassAspect { @Before("@within(dataSource)") public void doBefore(JoinPoint point, DataSource dataSource){ log.info("切換到數據源[{}]", dataSource.value().getValue()); DataSourceContextHolder.setDataSource(dataSource.value().getValue()); } @After("@within(dataSource)") public void doAfter(JoinPoint point, DataSource dataSource){ log.info("回收數據源[{}]", dataSource.value().getValue()); DataSourceContextHolder.clear(); } }
@Component @Slf4j @Aspect @Order(-5) public class DataSourceMethodAspect { @Before("@annotation(dataSource)") public void doBefore(JoinPoint point, DataSource dataSource){ log.info("切換到數據源[{}]", dataSource.value().getValue()); DataSourceContextHolder.setDataSource(dataSource.value().getValue()); } @After("@annotation(dataSource)") public void doAfter(JoinPoint point, DataSource dataSource){ log.info("回收數據源[{}]", dataSource.value().getValue()); DataSourceContextHolder.clear(); } }
8、新建多數據源配置類,如下:
@Configuration @MapperScan("com.dms.gateway.api.mapper") public class MybatisPlusConfig { /* * 分頁插件,自動識別數據庫類型 * 多租戶,請參考官網【插件擴展】 */ @Bean public PaginationInterceptor paginationInterceptor() { PaginationInterceptor paginationInterceptor = new PaginationInterceptor(); 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) { MultipleDataSource multipleDataSource = new MultipleDataSource(); Map< Object, Object > targetDataSources = new HashMap<>(); targetDataSources.put(DataSourceEnum.DB1.getValue(), db1); targetDataSources.put(DataSourceEnum.DB2.getValue(), db2); //添加數據源 multipleDataSource.setTargetDataSources(targetDataSources); //設置默認數據源 multipleDataSource.setDefaultTargetDataSource(db1); return multipleDataSource; } @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); //添加分頁功能 sqlSessionFactory.setPlugins(paginationInterceptor()); return sqlSessionFactory.getObject(); } }
接下來需要具體的業務邏輯,在service層的類或者方法上面添加@DataSource
註解來指定該業務需要用到的數據源,如下:
@Service @DataSource(DataSourceEnum.DB2) public class GatewayLogService { @Autowired private GatewayLogMapper mapper; @Override public Result<IPage<GatewayLog>> pageList(GatewayLogDTO gatewayLogDTO) { LambdaQueryWrapper<GatewayLog> wrapper = Wrappers.lambdaQuery(); wrapper.like(StringUtils.isNotEmpty(gatewayLogDTO.getPath()), GatewayLog::getPath, gatewayLogDTO.getPath()); wrapper.like(StringUtils.isNotEmpty(gatewayLogDTO.getSourceServer()), GatewayLog::getSourceServer, gatewayLogDTO.getSourceServer()); wrapper.like(StringUtils.isNotEmpty(gatewayLogDTO.getTargetServer()), GatewayLog::getTargetServer, gatewayLogDTO.getTargetServer()); wrapper.eq(StringUtils.isNotEmpty(gatewayLogDTO.getMethod()), GatewayLog::getMethod, gatewayLogDTO.getMethod()); wrapper.like(StringUtils.isNotEmpty(gatewayLogDTO.getRequestBody()), GatewayLog::getRequestBody, gatewayLogDTO.getRequestBody()); wrapper.like(StringUtils.isNotEmpty(gatewayLogDTO.getResponse()), GatewayLog::getResponse, gatewayLogDTO.getResponse()); wrapper.orderByDesc(GatewayLog::getId); IPage<GatewayLog> page = new Page<>(gatewayLogDTO.getPageNo(), gatewayLogDTO.getPageSize()); return Result.success(mapper.selectPage(page, wrapper)); } }
啟動服務,調用相關的接口,我們會在控制臺看到如下信息:
說明數據源切換成功!!!
到此這篇關於關於MybatisPlus配置雙數據庫驅動連接數據庫問題的文章就介紹到這瞭,更多相關MybatisPlus配置雙數據庫內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet!
推薦閱讀:
- 解決SpringBoot整合MybatisPlus分模塊管理遇到的bug
- Springboot動態切換數據源的具體實現與原理分析
- Spring整合Mybatis思路梳理總結
- springboot整合多數據源配置方式
- springboot 整合fluent mybatis的過程,看這篇夠瞭