Spring 多数据源方法级别注解实现过程
更新时间：2023年07月26日 14:16:12 作者：东皋长歌
多数据源管理是Spring框架中非常重要的一部分，它可以提高应用程序的灵活性和可靠性，从而更好地满足业务需求，这篇文章主要介绍了Spring 多数据源方法级别注解实现,需要的朋友可以参考下
Spring框架提供了多种数据源管理方式，其中多数据源管理是其中之一。多数据源管理允许应用程序使用多个数据源，而不是只使用一个数据源，从而提高了应用程序的灵活性和可靠性。
多数据源管理的主要目的是让应用程序能够在不同的数据库之间切换，以满足不同的业务需求。例如，如果应用程序需要访问一个高可用性的数据库，但是该数据库当前不可用，那么应用程序可以自动切换到备用数据库，以保证业务的连续性。
在多数据源管理中，Spring框架通过使用JNDI（Java命名和目录接口）来管理数据源。JNDI允许应用程序在运行时动态查找数据源，并且可以使用不同的数据源来访问不同的数据库。
在多数据源管理中，还需要考虑数据源的配置和管理。这包括创建数据源、配置数据源、管理连接池和处理连接泄漏等问题。Spring框架提供了多种数据源配置和管理的工具和技术，例如Spring JDBC Template、Spring Data JPA和Spring Boot等。
总之，多数据源管理是Spring框架中非常重要的一部分，它可以提高应用程序的灵活性和可靠性，从而更好地满足业务需求。
废话少说，直接开干！
下面详细地介绍一下多数据源的实现过程，同时实现一个方法级别的注解，便于灵活使用。
1， Pom配置
引入mysql，spring-boot-starter-jdbc
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-jdbc</artifactId>
<version>2.4.2</version>
</dependency>
<dependency>
<groupId>mysql</groupId>
<artifactId>mysql-connector-java</artifactId>
<version>${mysql-connector.version}</version>
</dependency>
<dependency>
<groupId>com.baomidou</groupId>
<artifactId>mybatis-plus-boot-starter</artifactId>
<version>${mybatis-plus.version}</version>
</dependency>2，配置数据源
可以在项目里，或者在apollo,nacos等配置中心上
#数据源master spring.dynamic.datasource.master.url = jdbc:mysql://127.0.0.1:3306/aaa?characterEncoding=utf-8&rewriteBatchedStatements=true&tinyInt1isBit=false&allowMultiQueries=true spring.dynamic.datasource.master.username = root spring.dynamic.datasource.master.password = xxxxxxxx spring.dynamic.datasource.master.driver-class-name = com.mysql.cj.jdbc.Driver spring.dynamic.datasource.master.type = com.zaxxer.hikari.HikariDataSource #数据源slave spring.dynamic.datasource.slave.url = jdbc:mysql://127.0.0.1:3306/bbb?characterEncoding=utf-8&rewriteBatchedStatements=true&tinyInt1isBit=false&allowMultiQueries=true spring.dynamic.datasource.slave.username = root spring.dynamic.datasource.slave.password = yyyyyyyy spring.dynamic.datasource.slave.driver-class-name = com.mysql.cj.jdbc.Driver spring.dynamic.datasource.slave.type = com.zaxxer.hikari.HikariDataSource
3，读取数据源配置
3.1 先看整个项目结构
3.2 新增读取数据源配置类
import lombok.Data;
import org.springframework.boot.autoconfigure.jdbc.DataSourceProperties;
import org.springframework.boot.context.properties.ConfigurationProperties;
import java.util.Map;
@Data
@ConfigurationProperties(prefix = "spring.dynamic")
public class DSProperties {
private Map<String, DataSourceProperties> datasource;
}3.3 新增线程和数据源mapping的类
import com.xxx.gateway.common.Constant;
import lombok.extern.slf4j.Slf4j;
@Slf4j
public class DynamicDataSourceHolder {
/**
* 保存动态数据源名称
*/
private static final ThreadLocal<String> DYNAMIC_DATASOURCE_KEY = new ThreadLocal<>();
/**
* 设置/切换数据源，决定当前线程使用哪个数据源
*/
public static void setDynamicDataSourceKey(String key){
log.info("数据源切换为：{}",key);
DYNAMIC_DATASOURCE_KEY.set(key);
}
/**
* 获取动态数据源名称，默认使用mater数据源
*/
public static String getDynamicDataSourceKey(){
String key = DYNAMIC_DATASOURCE_KEY.get();
return key == null ? Constant.MASTER_DATASOURCE : key;
}
/**
* 移除当前数据源
*/
public static void removeDynamicDataSourceKey(){
log.info("移除数据源：{}",DYNAMIC_DATASOURCE_KEY.get());
DYNAMIC_DATASOURCE_KEY.remove();
}
}3.4 继承路由多数据源的抽象类，并实现determineCurrentLookupKey方法
import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
public class DynamicDataSource extends AbstractRoutingDataSource {
@Override
protected Object determineCurrentLookupKey() {
String dataBaseType = DynamicDataSourceHolder.getDynamicDataSourceKey();
return dataBaseType;
}
}4，根据配置重写数据源和事务管理器
注意： 下面这个注解里的basePackages是你项目的路径
@MapperScan(basePackages = {"com.xxx.gateway.mapper"},
sqlSessionFactoryRef = "SqlSessionFactory")
import org.apache.ibatis.session.SqlSessionFactory;
import org.mybatis.spring.SqlSessionFactoryBean;
import org.mybatis.spring.annotation.MapperScan;
import org.springframework.boot.context.properties.EnableConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
import org.springframework.jdbc.datasource.DataSourceTransactionManager;
import org.springframework.transaction.PlatformTransactionManager;
import java.util.HashMap;
import java.util.Map;
@Configuration
@EnableConfigurationProperties(DSProperties.class)
@MapperScan(basePackages = {"com.xxx.gateway.mapper"},
sqlSessionFactoryRef = "SqlSessionFactory")
public class DynamicDataSourceConfig {
@Bean(name = "dynamicDataSource")
public DynamicDataSource DataSource(DSProperties dsProperties) {
Map targetDataSource = new HashMap<>(8);
dsProperties.getDatasource().forEach((k, v) -> {
targetDataSource.put(k, v.initializeDataSourceBuilder().build());
});
DynamicDataSource dataSource = new DynamicDataSource();
dataSource.setTargetDataSources(targetDataSource);
return dataSource;
}
/**
* 创建动态数据源的SqlSessionFactory，传入的是动态数据源
* @Primary这个注解很重要，如果项目中存在多个SqlSessionFactory，这个注解一定要加上
*/
@Primary
@Bean("SqlSessionFactory")
public SqlSessionFactory sqlSessionFactoryBean(DynamicDataSource dynamicDataSource) throws Exception {
SqlSessionFactoryBean sqlSessionFactoryBean = new SqlSessionFactoryBean();
sqlSessionFactoryBean.setDataSource(dynamicDataSource);
org.apache.ibatis.session.Configuration configuration = new org.apache.ibatis.session.Configuration();
configuration.setMapUnderscoreToCamelCase(true);
configuration.setDefaultFetchSize(100);
configuration.setDefaultStatementTimeout(30);
sqlSessionFactoryBean.setMapperLocations(
new PathMatchingResourcePatternResolver().getResources("classpath*:mapper/*.xml"));
sqlSessionFactoryBean.setConfiguration(configuration);
return sqlSessionFactoryBean.getObject();
}
/**
* 重写事务管理器，管理动态数据源
*/
@Primary
@Bean(value = "transactionManager")
public PlatformTransactionManager annotationDrivenTransactionManager(DynamicDataSource dataSource) {
return new DataSourceTransactionManager(dataSource);
}
}5，注解实现
5.1 新增注解定义
import java.lang.annotation.*;
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface DS {
String value() default Constant.MASTER_DATASOURCE;
}
5.2 新增注解实现
import com.xxx.gateway.config.DynamicDataSourceHolder;
import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.*;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.core.annotation.Order;
import org.springframework.stereotype.Component;
import java.util.Objects;
@Aspect
@Order(1)
@Component
@Slf4j
public class DsAspect {
@Pointcut("@annotation(com.xgd.gateway.aspect.DS)")
public void dsPointCut() {
}
//环绕通知
@Around("dsPointCut()")
public Object around(ProceedingJoinPoint joinPoint) throws Throwable{
String key = getDefineAnnotation(joinPoint).value();
DynamicDataSourceHolder.setDynamicDataSourceKey(key);
try {
return joinPoint.proceed();
} finally {
DynamicDataSourceHolder.removeDynamicDataSourceKey();
}
}
/**
* 先判断方法的注解，后判断类的注解，以方法的注解为准
* @param joinPoint
* @return
*/
private DS getDefineAnnotation(ProceedingJoinPoint joinPoint){
MethodSignature methodSignature = (MethodSignature) joinPoint.getSignature();
DS dataSourceAnnotation = methodSignature.getMethod().getAnnotation(DS.class);
if (Objects.nonNull(methodSignature)) {
return dataSourceAnnotation;
} else {
Class<?> dsClass = joinPoint.getTarget().getClass();
return dsClass.getAnnotation(DS.class);
}
}
}6, 通过注解使用多数据源
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.xxx.gateway.aspect.DS;
import com.xxx.gateway.domain.User;
import org.apache.ibatis.annotations.Param;
import java.util.List;
public interface UserMapper extends BaseMapper<User> {
public User queryByCode(@Param("code") String code);
@DS(value="master")
public User queryByCodeNoPass(@Param("code") String code);
@DS(value="slave1")
public List<String> queryBanks();
}注意：
1，@DS注解里的value值是可以随意的，只要和配置中心里spring.dynamic.datasource.后面的一致就可以。
2，如果方法上没有配置数据源，那么就是走默认的master数据源，所以项目里必须至少配置了master数据源。
3，各个数据源可以重复，因为不会校验是否重复。
到此这篇关于Spring 多数据源方法级别注解实现的文章就介绍到这了,更多相关Spring 多数据源内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
在SpringCloud项目中集成Sentinel,可以实现流量控制、熔断降级等功能,提升系统稳定性和可用性,集成步骤包括添加Sentinel依赖、配置控制台地址、启动控制台、配置限流熔断规则、使用注解和集成SpringCloudGateway,这有助于处理高并发场景,保护服务稳定运行2024-10-10
这篇文章主要介绍了JAVA Stream流的相关资料，文中讲解非常细致，代码帮助大家更好的理解和学习，感兴趣的朋友可以了解下2020-07-07
这篇文章主要介绍了Gradle构建多模块项目的方法步骤,小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2018-05-05
这篇文章主要给大家介绍了关于Java Calendar类使用之日期和时间处理指南的相关资料,Calendar类是Java中用于处理日期和时间的抽象类,它提供了一种独立于特定日历系统的方式来处理日期和时间,需要的朋友可以参考下2023-12-12
线程要经历开始（等待）、运行、挂起和停止四种不同的状态。这四种状态都可以通过Thread类中的方法进行控制。下面给出了Thread类中和这四种状态相关的方法2014-01-01
Spring security 如何开放 Swagger 访问权限
这篇文章主要介绍了Spring security 如何开放 Swagger 访问权限操作，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2021-09-09
这篇文章主要介绍了springboot跨域CORS处理代码解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-12-12
由于网络或服务器问题,Redis连接可能会断开,导致应用程序无法继续正常工作,本文主要介绍了springboot自动重连Redis的实现方法,具有一定的参考价值,感兴趣的可以了解一下2024-02-02
本文主要介绍了Mybatis中自定义TypeHandler处理枚举的相关知识。具有很好的参考价值，下面跟着小编一起来看下吧2017-02-02
Java的ThreadPoolExecutor业务线程池详细解析
这篇文章主要介绍了Java线程池ThreadPoolExecutor详细解析,任务刚开始进来的时候就创建核心线程,核心线程满了会把任务放到阻塞队列,阻塞队列满了之后才会创建空闲线程,达到最大线程数之后,再有任务进来,就只能执行拒绝策略了,需要的朋友可以参考下2024-01-01
最新评论