springboot项目自定义注解实现的多数据源切换 springboot项目自定义注解实现的

一、主要依赖

org.springframework.bootspring-boot-starter-parent2.1.4.RELEASEorg.springframework.bootspring-boot-starter-aop2.1.4.RELEASEdependency>
com.alibabadruid-spring-boot-starter1.1.10

二、yml

# 数据源配置
spring:
datasource:
type: com.alibaba.druid.pool.DruidDataSource
driverClassName: com.mysql.jdbc.Driver
druid:
# 主库数据源
master:
url: jdbc:mysql://127.0.0.1:3306/master?characterEncoding=UTF-8username: root
password: root
#树熊数据源
slave:
enabled : trueurl: jdbc:mysql:////127.0.0.1:3306/slave?characterEncoding=UTF-8username: root
password: root

# 初始连接数
initial-size: 10# 最大连接池数量
max-active: 100# 最小连接池数量
min-idle: 10# 配置获取连接等待超时的时间
max-wait: 60000# 打开PSCache，并且指定每个连接上PSCache的大小
pool-prepared-statements:truemax-pool-prepared-statement-per-connection-size: 20# 配置间隔多久才进行一次检测，检测需要关闭的空闲连接，单位是毫秒
timeBetweenEvictionRunsMillis: 60000# 配置一个连接在池中最小生存的时间，单位是毫秒
min-evictable-idle-time-millis: 300000validation-query: SELECT 1 FROM DUAL
test-while-idle:truetest-on-borrow:falsetest-on-return:falsestat-view-servlet:
enabled: trueurl-pattern: /druid/*filter:
stat:
log-slow-sql: true
slow-sql-millis: 1000
merge-sql: false
wall:
config:
multi-statement-allow: true

三、实现
3.1、@DataSource和DataSourceType

/** * 数据源
* @author DUCHONG
*/publicenum DataSourceType
{
/*** 主库
*/MASTER,
/*** 从库
*/SLAVE
}

import java.lang.annotation.ElementType; import java.lang.annotation.Retention; import java.lang.annotation.RetentionPolicy; import java.lang.annotation.Target; /** * 自定义多数据源切换注解
*
* @author DUCHONG
*/@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)public@interface DataSource
{
/*** 切换数据源名称
*/publicDataSourceType value()default DataSourceType.MASTER;
}

3.2、DynamicDataSourceContextHolder

import org.slf4j.Logger; import org.slf4j.LoggerFactory; /** * 数据源切换处理
*
* @author DUCHONG
*/publicclass DynamicDataSourceContextHolder
{
publicstaticfinalLogger log = LoggerFactory.getLogger(DynamicDataSourceContextHolder.class);
/*** 使用ThreadLocal维护变量，ThreadLocal为每个使用该变量的线程提供独立的变量副本，
*所以每一个线程都可以独立地改变自己的副本，而不会影响其它线程所对应的副本。
*/privatestaticfinalThreadLocal CONTEXT_HOLDER =newThreadLocal<>();
/*** 设置数据源的变量
*/publicstaticvoid setDateSourceType(String dsType)
{
log.info("切换到{}数据源", dsType);
CONTEXT_HOLDER.set(dsType);
}
/*** 获得数据源的变量
*/publicstatic String getDateSourceType()
{
return CONTEXT_HOLDER.get();
}
/*** 清空数据源变量
*/publicstaticvoid clearDateSourceType()
{
CONTEXT_HOLDER.remove();
}
}

3.3、继承AbstractRoutingDataSource

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource; import javax.sql.DataSource; import java.util.Map; /** * 动态数据源
*
* @author DUCHONG
*/publicclassDynamicDataSourceextends AbstractRoutingDataSource
{
publicDynamicDataSource(DataSource defaultTargetDataSource, Map targetDataSources)
{
【springboot项目自定义注解实现的多数据源切换】super.setDefaultTargetDataSource(defaultTargetDataSource);
super.setTargetDataSources(targetDataSources);
super.afterPropertiesSet();
}
@Override
protected Object determineCurrentLookupKey()
{
return DynamicDataSourceContextHolder.getDateSourceType();
}
}

3.4、定义切面

import com.starfast.admin.common.annotation.DataSource; import com.starfast.admin.datasource.DynamicDataSourceContextHolder; import org.aspectj.lang.ProceedingJoinPoint; import org.aspectj.lang.annotation.Around; import org.aspectj.lang.annotation.Aspect; import org.aspectj.lang.annotation.Pointcut; import org.aspectj.lang.reflect.MethodSignature; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.core.annotation.Order; import org.springframework.stereotype.Component; import java.lang.reflect.Method; /** * 多数据源处理
* @author DUCHONG
*/@Aspect
@Order(1)
@Componentpublicclass DataSourceAspect
{
protectedLogger logger = LoggerFactory.getLogger(getClass());
@Pointcut("@annotation(com.duchong.common.annotation.DataSource)")
publicvoid dsPointCut()
{
}
@Around("dsPointCut()")
publicObject around(ProceedingJoinPoint point)throws Throwable
{
MethodSignature signature = (MethodSignature) point.getSignature();
Method method = signature.getMethod();
DataSource dataSource = method.getAnnotation(DataSource.class);
if(null!=dataSource)
{
DynamicDataSourceContextHolder.setDateSourceType(dataSource.value().name());
}
try{
return point.proceed();
}
finally{
// 销毁数据源在执行方法之后DynamicDataSourceContextHolder.clearDateSourceType();
}
}
}

3.5、@Configuration

import com.alibaba.druid.spring.boot.autoconfigure.DruidDataSourceBuilder; import com.starfast.admin.common.enums.DataSourceType; import com.starfast.admin.datasource.DynamicDataSource; import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty; import org.springframework.boot.context.properties.ConfigurationProperties; import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Configuration; import org.springframework.context.annotation.Primary; import javax.sql.DataSource; import java.util.HashMap; import java.util.Map; /** * druid 配置多数据源
*
* @author DUCHONG
*/@Configurationpublicclass DruidConfig
{
@Bean
@ConfigurationProperties("spring.datasource.druid.master")
public DataSource masterDataSource()
{
return DruidDataSourceBuilder.create().build();
}
@Bean
@ConfigurationProperties("spring.datasource.druid.slave")
@ConditionalOnProperty(prefix = "spring.datasource.druid.slave", name = "enabled", havingValue = "https://www.it610.com/article/true")
public DataSource slaveDataSource()
{
return DruidDataSourceBuilder.create().build();
}
@Bean(name = "dynamicDataSource")
@Primary
public DynamicDataSource dataSource()
{
Map targetDataSources =newHashMap<>();
targetDataSources.put(DataSourceType.MASTER.name(), masterDataSource());
targetDataSources.put(DataSourceType.SLAVE.name(), slaveDataSource());
returnnew DynamicDataSource(masterDataSource(), targetDataSources);
}
}

3.6、使用
需要切换数据源的方法上加
@DataSource(value = https://www.it610.com/article/DataSourceType.SLAVE)

springboot项目自定义注解实现的多数据源切换

推荐阅读

xp3，什么是XP3

吃了紧急避孕药怀孕了孩子能要吗

福州长乐区5月20日婚姻预约登记公告福州长乐区5月20日婚姻预约登记公告

怎么舒缓痛经如何快速缓解痛经

石榴吃了有什么效果

天猫精灵APP如何查找手机具体操作步骤

佳能维修模式清零方法-佳能ip1188如何进维修模式，打印机墨盒清零

微信欢乐斗地主残局闯关专家第40关怎么过 40关通关破解技巧

网商银行账户怎么开通开通网商银行账户流程

mysql存储过程cursor mysql有存储过程没

烤箱烤栗子怎么做好吃

川车牌号是哪个省

新买的水族箱怎么养水新鱼缸第一次怎样养水

豆瓣电视剧怎么打开看豆瓣的书怎么打开

图像处理|【OpenCv】图像分割——聚类算法

江南意表达了什么情感

有没有带锁的冰箱？如何选择适合自己的带锁冰箱

赛欧amt变速箱赛欧自动挡

安卓计步传感器不灵敏,手机放置位置不同会影响传感器检测步骤

0xc000012f解决方法