springboot双数据源配置注解方式+aop
我配置的示例两个都是德鲁伊的,如需更改自行更换数据源即可
配置类:
package com.xxx.xxxl.jdbc;
import com.alibaba.druid.pool.DruidDataSource;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import org.springframework.context.annotation.PropertySource;
import org.springframework.jdbc.datasource.DataSourceTransactionManager;
import org.springframework.transaction.PlatformTransactionManager;
import javax.sql.DataSource;
import java.util.HashMap;
import java.util.Map;
@Configuration
//读取yml配置文件
@PropertySource("classpath:config/application-${spring.profiles.active}.yml")
public class DataSourceConfig {
public static final String DriverClassName = "com.mysql.cj.jdbc.Driver";
public static final String ConnectionProperties = "druid.stat.mergeSql=true;druid.stat.slowSqlMillis=5000";
@Value("${db1.datasource.jdbc-url}")
String url;
@Value("${db1.datasource.username}")
String username;
@Value("${db1.datasource.password}")
String password;
@Value("${db2.datasource.jdbc-url}")
String url2;
@Value("${db2.datasource.username}")
String username2;
@Value("${db2.datasource.password}")
String password2;
@Bean(name = "db1")
public DataSource db1Source() {
DruidDataSource dataSource = new DruidDataSource();
dataSource.setDriverClassName(DriverClassName);
dataSource.setUrl(url);
dataSource.setUsername(username);
dataSource.setPassword(password);
//<!-- 配置初始化大小、最小、最大 -->
dataSource.setInitialSize(20);
dataSource.setMinIdle(20);
dataSource.setMaxActive(200);
//<!-- 配置从连接池获取连接等待超时的时间 -->
dataSource.setMaxWait(20000);
//<!-- 配置间隔多久启动一次DestroyThread,对连接池内的连接才进行一次检测,单位是毫秒。
// 检测时:1.如果连接空闲并且超过minIdle以外的连接,如果空闲时间超过minEvictableIdleTimeMillis设置的值则直接物理关闭。2.在minIdle以内的不处理。-->
dataSource.setTimeBetweenEvictionRunsMillis(60000);
//<!-- 配置一个连接在池中最大空闲时间,单位是毫秒 -->
dataSource.setMinEvictableIdleTimeMillis(30000);
//<!-- 设置从连接池获取连接时是否检查连接有效性,true时,每次都检查;false时,不检查 -->
dataSource.setTestOnBorrow(false);
//<!-- 设置往连接池归还连接时是否检查连接有效性,true时,每次都检查;false时,不检查 -->
dataSource.setTestOnReturn(false);
//<!-- 设置从连接池获取连接时是否检查连接有效性,true时,如果连接空闲时间超过minEvictableIdleTimeMillis进行检查,否则不检查;false时,不检查 -->
dataSource.setTestWhileIdle(true);
//<!-- 连接泄露检查,打开removeAbandoned功能 , 连接从连接池借出后,长时间不归还,将触发强制回连接。回收周期随timeBetweenEvictionRunsMillis进行,
//如果连接为从连接池借出状态,并且未执行任何sql,并且从借出时间起已超过removeAbandonedTimeout时间,则强制归还连接到连接池中。 -->
dataSource.setRemoveAbandoned(true);
//<!-- 超时时间,秒 -->
dataSource.setRemoveAbandonedTimeout(80);
// <!-- 打开PSCache,并且指定每个连接上PSCache的大小,Oracle等支持游标的数据库,打开此开关,会以数量级提升性能,缓存sql数 -->
dataSource.setPoolPreparedStatements(true);
dataSource.setMaxPoolPreparedStatementPerConnectionSize(100);
//<!-- 根据自身业务及事务大小来设置 -->
dataSource.setConnectionProperties(ConnectionProperties);
return dataSource;
}
@Bean(name = "db2")
public DataSource db2Source() {
DruidDataSource dataSource = new DruidDataSource();
dataSource.setDriverClassName(DriverClassName);
dataSource.setUrl(url2);
dataSource.setUsername(username2);
dataSource.setPassword(password2);
//<!-- 配置初始化大小、最小、最大 -->
dataSource.setInitialSize(20);
dataSource.setMinIdle(20);
dataSource.setMaxActive(200);
//<!-- 配置从连接池获取连接等待超时的时间 -->
dataSource.setMaxWait(20000);
//<!-- 配置间隔多久启动一次DestroyThread,对连接池内的连接才进行一次检测,单位是毫秒。
// 检测时:1.如果连接空闲并且超过minIdle以外的连接,如果空闲时间超过minEvictableIdleTimeMillis设置的值则直接物理关闭。2.在minIdle以内的不处理。-->
dataSource.setTimeBetweenEvictionRunsMillis(60000);
//<!-- 配置一个连接在池中最大空闲时间,单位是毫秒 -->
dataSource.setMinEvictableIdleTimeMillis(30000);
//<!-- 设置从连接池获取连接时是否检查连接有效性,true时,每次都检查;false时,不检查 -->
dataSource.setTestOnBorrow(false);
//<!-- 设置往连接池归还连接时是否检查连接有效性,true时,每次都检查;false时,不检查 -->
dataSource.setTestOnReturn(false);
//<!-- 设置从连接池获取连接时是否检查连接有效性,true时,如果连接空闲时间超过minEvictableIdleTimeMillis进行检查,否则不检查;false时,不检查 -->
dataSource.setTestWhileIdle(true);
//<!-- 连接泄露检查,打开removeAbandoned功能 , 连接从连接池借出后,长时间不归还,将触发强制回连接。回收周期随timeBetweenEvictionRunsMillis进行,
//如果连接为从连接池借出状态,并且未执行任何sql,并且从借出时间起已超过removeAbandonedTimeout时间,则强制归还连接到连接池中。 -->
dataSource.setRemoveAbandoned(true);
//<!-- 超时时间,秒 -->
dataSource.setRemoveAbandonedTimeout(80);
// <!-- 打开PSCache,并且指定每个连接上PSCache的大小,Oracle等支持游标的数据库,打开此开关,会以数量级提升性能,缓存sql数 -->
dataSource.setPoolPreparedStatements(true);
dataSource.setMaxPoolPreparedStatementPerConnectionSize(100);
//<!-- 根据自身业务及事务大小来设置 -->
dataSource.setConnectionProperties(ConnectionProperties);
return dataSource;
}
/**
* 动态数据库配置
*/
@Primary
@Bean(name = "dynamicDataSource")
public DataSource dynamicDataSource() {
DynamicDataSource dynamicDataSource = new DynamicDataSource();
// 默认数据源
dynamicDataSource.setDefaultTargetDataSource(db1Source());
// 配置多数据源
Map<Object, Object> dsMap = new HashMap(5);
dsMap.put("db1", db1Source());
dsMap.put("db2", db2Source());
dynamicDataSource.setTargetDataSources(dsMap);
return dynamicDataSource;
}
/**
* 配置@Transactional注解事物
*
* @return
*/
@Bean
public PlatformTransactionManager transactionManager() {
return new DataSourceTransactionManager(dynamicDataSource());
}
}
get和set以及clear方法
package com.xxx.util.jdbc;
public class DataSourceContext {
// 默认数据源
public static final String DEFAULT= "db1";
//threadlocal而是一个线程内部的存储类,可以在指定线程内存储数据,数据存储以后,只有指定线程可以得到存储数据
private static final ThreadLocal<String> threadLocal= new ThreadLocal<>();
public static void setDataSource(String dbType) {
threadLocal.set(dbType);
}
public static String getDataSource() {
return (threadLocal.get());
}
public static void clearDataSource() {
threadLocal.remove();
}
}
继承AbstractRoutingDataSource实现动态数据源切换
package com.xxxx.util.jdbc;
import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
public class DynamicDataSource extends AbstractRoutingDataSource {
@Override
protected Object determineCurrentLookupKey() {
return DataSourceContext.getDataSource();
}
}
//aop拦截切换
package com.xxx.util.jdbc;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.After;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.stereotype.Component;
import java.lang.reflect.Method;
@Aspect
@Component
public class DataBaseAspect {
//拦截注解
@Pointcut("@annotation(com.xxx.xxx.jdbc.DataBase)")
public void dbPointCut() {
}
@Before("dbPointCut()")
public void beforeSwitchDS(JoinPoint point){
//获得当前访问的class
Class<?> className = point.getTarget().getClass();
//获得访问的方法名
String methodName = point.getSignature().getName();
//得到方法的参数的类型
Class[] argClass = ((MethodSignature)point.getSignature()).getParameterTypes();
String dataSource = DataSourceContext.DEFAULT;
try {
// 得到访问的方法对象
Method method = className.getMethod(methodName, argClass);
// 判断是否存在@DateBase注解
if (method.isAnnotationPresent(DataBase.class)) {
DataBase annotation = method.getAnnotation(DataBase.class);
// 取出注解中的数据源名
dataSource = annotation.value();
}
} catch (Exception e) {
e.printStackTrace();
}
// 切换数据源
DataSourceContext.setDataSource(dataSource);
}
@After("dbPointCut()")
public void afterSwitchDS(JoinPoint point){
DataSourceContext.clearDataSource();
}
}
自定义注解
package com.xxx.util.jdbc;
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
@Retention(RetentionPolicy.RUNTIME)
@Target({ElementType.METHOD,ElementType.TYPE})
public @interface DataBase {
String value() default "db1";
}
最后去除在启动类中springboot的默认数据源
@SpringBootApplication(exclude = {DataSourceAutoConfiguration.class})
另外附上一份yml中配置示例
db1:
datasource:
jdbc-url: xxx
username: xxx
password: xxx
db2:
datasource:
jdbc-url: xxx
username: xxx
password: xxx