SpringBoot2+Mybatis两个数据源实现 (二)

2024-04-05 07:18

本文主要是介绍SpringBoot2+Mybatis两个数据源实现 (二),希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

之前转过一篇《SpringBoot2+Mybatis多个数据源实现》,今天补充一个其他的解决方案(原理差不多),这里是使用mysql+sqlserver作为数据源,可以供大家参考一下。

一、依赖引入

首先引入必要的jar包,有省略,自行补充
pom.xml

<!--连接池 -->
<dependency><groupId>com.zaxxer</groupId><artifactId>HikariCP</artifactId><version>3.2.0</version>
</dependency>
<!--mysql -->
<dependency><groupId>mysql</groupId><artifactId>mysql-connector-java</artifactId><version>5.1.31</version>
</dependency>
<!-- sqlserver-->
<dependency><groupId>com.microsoft.sqlserver</groupId><artifactId>sqljdbc4</artifactId><version>4.0</version>
</dependency>
<!-- mybatis -->
<dependency><groupId>org.mybatis.spring.boot</groupId><artifactId>mybatis-spring-boot-starter</artifactId><version>2.0.0</version>
</dependency>
二、代码实现

先创建 DatabaseType,定义两个数据源的名称

public enum DatabaseType {mysqlDataSource,sqlServerDataSource
}

实现DatabaseContextHolder

public class DatabaseContextHolder {private static final ThreadLocal<DatabaseType> contextHolder = new ThreadLocal<>();public static void setDatabaseType(DatabaseType type) {contextHolder.set(type);}public static DatabaseType getDatabaseType() {return contextHolder.get();}
}

实现DynamicDataSource

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;public class DynamicDataSource extends AbstractRoutingDataSource {@Overrideprotected Object determineCurrentLookupKey() {return DatabaseContextHolder.getDatabaseType();}
}

实现MyBatisConfig

import org.apache.ibatis.session.SqlSessionFactory;
import org.mybatis.spring.SqlSessionFactoryBean;
import org.mybatis.spring.annotation.MapperScan;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.boot.jdbc.DataSourceBuilder;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import org.springframework.core.env.Environment;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
import org.springframework.jdbc.datasource.DataSourceTransactionManager;import javax.sql.DataSource;
import java.util.HashMap;
import java.util.Map;@Configuration // 该注解类似于spring配置文件
@MapperScan(basePackages = "com.xxx.dao")
public class MyBatisConfig {@Autowiredprivate Environment env;// 读取主数据源@Beanpublic DataSource mysqlDataSource(Environment env) {Map<String, Object> dsMap = new HashMap<>();dsMap.put("type", env.getProperty("spring.datasource.type"));dsMap.put("driver", env.getProperty("spring.datasource.driver-class-name"));dsMap.put("url", env.getProperty("spring.datasource.url"));dsMap.put("username", env.getProperty("spring.datasource.username"));dsMap.put("password", env.getProperty("spring.datasource.password"));return buildDataSource(dsMap);}// 读取从数据源@Beanpublic DataSource sqlServerDataSource(Environment env) {Map<String, Object> dsMap = new HashMap<>();dsMap.put("type", env.getProperty("slave.datasource.type"));dsMap.put("driver", env.getProperty("slave.datasource.driver-class-name"));dsMap.put("url", env.getProperty("slave.datasource.url"));dsMap.put("username", env.getProperty("slave.datasource.username"));dsMap.put("password", env.getProperty("slave.datasource.password"));return buildDataSource(dsMap);}/*** @Primary 该注解表示在同一个接口有多个实现类可以注入的时候,默认选择哪一个,而不是让@autowire注解报错* @Qualifier 根据名称进行注入,通常是在具有相同的多个类型的实例的一个注入(例如有多个DataSource类型的实例)*/@Bean@Primarypublic DynamicDataSource dataSource(@Qualifier("mysqlDataSource") DataSource mysqlDataSource,@Qualifier("sqlServerDataSource") DataSource sqlServerDataSource) {Map<Object, Object> targetDataSources = new HashMap<>();targetDataSources.put(DatabaseType.mysqlDataSource, mysqlDataSource);targetDataSources.put(DatabaseType.sqlServerDataSource, sqlServerDataSource);DynamicDataSource dataSource = new DynamicDataSource();dataSource.setTargetDataSources(targetDataSources);// 该方法是AbstractRoutingDataSource的方法dataSource.setDefaultTargetDataSource(mysqlDataSource);// 默认的datasource设return dataSource;}/*** 根据数据源创建SqlSessionFactory*/@Beanpublic SqlSessionFactory sqlSessionFactory(DynamicDataSource ds) throws Exception {SqlSessionFactoryBean fb = new SqlSessionFactoryBean();fb.setDataSource(ds);// 指定数据源(这个必须有,否则报错)// 下边两句仅仅用于*.xml文件,如果整个持久层操作不需要使用到xml文件的话(只用注解就可以搞定),则不加//fb.setTypeAliasesPackage(env.getProperty("mybatis.typeAliasesPackage"));// 指定基包fb.setConfigLocation(new PathMatchingResourcePatternResolver().getResource(env.getProperty("mybatis.config-location")));fb.setMapperLocations(new PathMatchingResourcePatternResolver().getResources(env.getProperty("mybatis.mapper-locations")));return fb.getObject();}/*** 配置事务管理器*/@Beanpublic DataSourceTransactionManager transactionManager(DynamicDataSource dataSource) throws Exception {return new DataSourceTransactionManager(dataSource);}/*** 创建DataSource* @param dataSourceMap* @return*/public DataSource buildDataSource(Map<String, Object> dataSourceMap) {try {Object type = dataSourceMap.get("type");Class<? extends DataSource> dataSourceType = (Class<? extends DataSource>) Class.forName((String) type);String driverClassName = dataSourceMap.get("driver").toString();String url = dataSourceMap.get("url").toString();String username = dataSourceMap.get("username").toString();String password = dataSourceMap.get("password").toString();// 自定义DataSource配置DataSourceBuilder factory = DataSourceBuilder.create().driverClassName(driverClassName).url(url).username(username).password(password).type(dataSourceType);return factory.build();} catch (ClassNotFoundException e) {e.printStackTrace();}return null;}
}
三、yml文件配置

application.yml

server:tomcat:uri-encoding: UTF-8max-threads: 1000min-spare-threads: 30port: 8888spring:profiles:active: dev#mybatis
mybatis:mapper-locations: classpath:mybatis/mappers/*.xmlconfig-location: classpath:mybatis/mybatis-config.xml

定义application-dev.yml

#mysql
spring:datasource:driver-class-name: com.mysql.jdbc.Driverurl: jdbc:mysql://127.0.0.1:33306/test1?useAffectedRows=true&characterEncoding=utf8username: rootpassword: roottype: com.zaxxer.hikari.HikariDataSourcehikari:connection-timeout: 30000idle-timeout: 600000max-lifetime: 1800000maximum-pool-size: 10#sqlserver
slave:datasource:driver-class-name: com.microsoft.sqlserver.jdbc.SQLServerDriverurl: jdbc:sqlserver://127.0.0.1:1433;DatabaseName=test2;username: sapassword: 123456type: com.zaxxer.hikari.HikariDataSourcehikari:connection-timeout: 30000idle-timeout: 600000max-lifetime: 1800000maximum-pool-size: 10
四、使用方式

建议在service实现类里面使用

public List<Demo> queryList(){//切换数据源DatabaseContextHolder.setDatabaseType(DatabaseType.sqlServerDataSource);Map<String,Object> map = new HashMap<>();return demoDao.queryList(map);
}public List<Demo> selectList(){//切换数据源DatabaseContextHolder.setDatabaseType(DatabaseType.mysqlDataSource);Map<String,Object> map = new HashMap<>();return demoDao.queryList(map);
}

20190709更新
之前hikari的配置未生效

hikari:connection-timeout: 30000idle-timeout: 600000max-lifetime: 1800000maximum-pool-size: 10

今天优化了一下MyBatisConfig,增加了配置Hikari dataSource的超时设置,详细看代码

package com.xxx.config;import com.zaxxer.hikari.HikariConfig;
import com.zaxxer.hikari.HikariDataSource;
import org.apache.ibatis.session.SqlSessionFactory;
import org.mybatis.spring.SqlSessionFactoryBean;
import org.mybatis.spring.annotation.MapperScan;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import org.springframework.core.env.Environment;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
import org.springframework.jdbc.datasource.DataSourceTransactionManager;import javax.sql.DataSource;
import java.util.HashMap;
import java.util.Map;@Configuration // 该注解类似于spring配置文件
@MapperScan(basePackages = "com.xxx.dao")
public class MyBatisConfig {@Autowiredprivate Environment env;// 读取主数据源@Beanpublic DataSource mysqlDataSource(Environment env) {HikariConfig config = new HikariConfig();config.setDriverClassName(env.getProperty("spring.datasource.driver-class-name"));config.setJdbcUrl(env.getProperty("spring.datasource.url"));config.setUsername(env.getProperty("spring.datasource.username"));config.setPassword(env.getProperty("spring.datasource.password"));config.setConnectionTestQuery(env.getProperty("spring.datasource.ConnectionTestQuery"));config.setMaximumPoolSize(Integer.parseInt(env.getProperty("spring.datasource.hikari.maximum-pool-size")));config.setIdleTimeout(Integer.parseInt(env.getProperty("spring.datasource.hikari.idle-timeout")));config.setMaxLifetime(Integer.parseInt(env.getProperty("spring.datasource.hikari.max-lifetime")));config.setConnectionTimeout(Integer.parseInt(env.getProperty("spring.datasource.hikari.connection-timeout")));DataSource dataSource = new HikariDataSource(config);return dataSource;}// 读取从数据源@Beanpublic DataSource sqlServerDataSource(Environment env) {HikariConfig config = new HikariConfig();config.setDriverClassName(env.getProperty("slave.datasource.driver-class-name"));config.setJdbcUrl(env.getProperty("slave.datasource.url"));config.setUsername(env.getProperty("slave.datasource.username"));config.setPassword(env.getProperty("slave.datasource.password"));config.setConnectionTestQuery(env.getProperty("slave.datasource.ConnectionTestQuery"));config.setMaximumPoolSize(Integer.parseInt(env.getProperty("slave.datasource.hikari.maximum-pool-size")));config.setIdleTimeout(Integer.parseInt(env.getProperty("slave.datasource.hikari.idle-timeout")));config.setMaxLifetime(Integer.parseInt(env.getProperty("slave.datasource.hikari.max-lifetime")));config.setConnectionTimeout(Integer.parseInt(env.getProperty("slave.datasource.hikari.connection-timeout")));DataSource dataSource = new HikariDataSource(config);return dataSource;}/*** @Primary 该注解表示在同一个接口有多个实现类可以注入的时候,默认选择哪一个,而不是让@autowire注解报错* @Qualifier 根据名称进行注入,通常是在具有相同的多个类型的实例的一个注入(例如有多个DataSource类型的实例)*/@Bean@Primarypublic DynamicDataSource dataSource(@Qualifier("mysqlDataSource") DataSource mysqlDataSource,@Qualifier("sqlServerDataSource") DataSource sqlServerDataSource) {Map<Object, Object> targetDataSources = new HashMap<>();targetDataSources.put(DatabaseType.mysqlDataSource, mysqlDataSource);targetDataSources.put(DatabaseType.sqlServerDataSource, sqlServerDataSource);DynamicDataSource dataSource = new DynamicDataSource();dataSource.setTargetDataSources(targetDataSources);// 该方法是AbstractRoutingDataSource的方法dataSource.setDefaultTargetDataSource(mysqlDataSource);// 默认的datasource设return dataSource;}/*** 根据数据源创建SqlSessionFactory*/@Beanpublic SqlSessionFactory sqlSessionFactory(DynamicDataSource ds) throws Exception {SqlSessionFactoryBean fb = new SqlSessionFactoryBean();fb.setDataSource(ds);// 指定数据源(这个必须有,否则报错)// 下边两句仅仅用于*.xml文件,如果整个持久层操作不需要使用到xml文件的话(只用注解就可以搞定),则不加//fb.setTypeAliasesPackage(env.getProperty("mybatis.typeAliasesPackage"));// 指定基包fb.setConfigLocation(new PathMatchingResourcePatternResolver().getResource(env.getProperty("mybatis.config-location")));fb.setMapperLocations(new PathMatchingResourcePatternResolver().getResources(env.getProperty("mybatis.mapper-locations")));//return fb.getObject();}/*** 配置事务管理器*/@Beanpublic DataSourceTransactionManager transactionManager(DynamicDataSource dataSource) throws Exception {return new DataSourceTransactionManager(dataSource);}
}

这篇关于SpringBoot2+Mybatis两个数据源实现 (二)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/877958

相关文章

C++中unordered_set哈希集合的实现

《C++中unordered_set哈希集合的实现》std::unordered_set是C++标准库中的无序关联容器,基于哈希表实现,具有元素唯一性和无序性特点,本文就来详细的介绍一下unorder... 目录一、概述二、头文件与命名空间三、常用方法与示例1. 构造与析构2. 迭代器与遍历3. 容量相关4

C++中悬垂引用(Dangling Reference) 的实现

《C++中悬垂引用(DanglingReference)的实现》C++中的悬垂引用指引用绑定的对象被销毁后引用仍存在的情况,会导致访问无效内存,下面就来详细的介绍一下产生的原因以及如何避免,感兴趣... 目录悬垂引用的产生原因1. 引用绑定到局部变量,变量超出作用域后销毁2. 引用绑定到动态分配的对象,对象

MyBatis常用XML语法详解

《MyBatis常用XML语法详解》文章介绍了MyBatis常用XML语法,包括结果映射、查询语句、插入语句、更新语句、删除语句、动态SQL标签以及ehcache.xml文件的使用,感兴趣的朋友跟随小... 目录1、定义结果映射2、查询语句3、插入语句4、更新语句5、删除语句6、动态 SQL 标签7、ehc

SpringBoot基于注解实现数据库字段回填的完整方案

《SpringBoot基于注解实现数据库字段回填的完整方案》这篇文章主要为大家详细介绍了SpringBoot如何基于注解实现数据库字段回填的相关方法,文中的示例代码讲解详细,感兴趣的小伙伴可以了解... 目录数据库表pom.XMLRelationFieldRelationFieldMapping基础的一些代

Java HashMap的底层实现原理深度解析

《JavaHashMap的底层实现原理深度解析》HashMap基于数组+链表+红黑树结构,通过哈希算法和扩容机制优化性能,负载因子与树化阈值平衡效率,是Java开发必备的高效数据结构,本文给大家介绍... 目录一、概述:HashMap的宏观结构二、核心数据结构解析1. 数组(桶数组)2. 链表节点(Node

Java AOP面向切面编程的概念和实现方式

《JavaAOP面向切面编程的概念和实现方式》AOP是面向切面编程,通过动态代理将横切关注点(如日志、事务)与核心业务逻辑分离,提升代码复用性和可维护性,本文给大家介绍JavaAOP面向切面编程的概... 目录一、AOP 是什么?二、AOP 的核心概念与实现方式核心概念实现方式三、Spring AOP 的关

详解SpringBoot+Ehcache使用示例

《详解SpringBoot+Ehcache使用示例》本文介绍了SpringBoot中配置Ehcache、自定义get/set方式,并实际使用缓存的过程,文中通过示例代码介绍的非常详细,对大家的学习或者... 目录摘要概念内存与磁盘持久化存储:配置灵活性:编码示例引入依赖:配置ehcache.XML文件:配置

Python实现字典转字符串的五种方法

《Python实现字典转字符串的五种方法》本文介绍了在Python中如何将字典数据结构转换为字符串格式的多种方法,首先可以通过内置的str()函数进行简单转换;其次利用ison.dumps()函数能够... 目录1、使用json模块的dumps方法:2、使用str方法:3、使用循环和字符串拼接:4、使用字符

MyBatis延迟加载与多级缓存全解析

《MyBatis延迟加载与多级缓存全解析》文章介绍MyBatis的延迟加载与多级缓存机制,延迟加载按需加载关联数据提升性能,一级缓存会话级默认开启,二级缓存工厂级支持跨会话共享,增删改操作会清空对应缓... 目录MyBATis延迟加载策略一对多示例一对多示例MyBatis框架的缓存一级缓存二级缓存MyBat

Linux下利用select实现串口数据读取过程

《Linux下利用select实现串口数据读取过程》文章介绍Linux中使用select、poll或epoll实现串口数据读取,通过I/O多路复用机制在数据到达时触发读取,避免持续轮询,示例代码展示设... 目录示例代码(使用select实现)代码解释总结在 linux 系统里,我们可以借助 select、