济南网站建设要多少钱,wordpress rss,天津做网站需要多少钱,网站建设团队架构目录1 DS实现动态切换数据源原理2 不可在事务中切换数据库分析解决3 原因解析1 DS实现动态切换数据源原理
首先mybatis-plus使用com.baomidou.dynamic.datasource.AbstractRoutingDataSource继承 AbstractDataSource接管数据源#xff1b;具体实现类为com.baomidou.dynamic.d…
目录1 DS实现动态切换数据源原理2 不可在事务中切换数据库分析解决3 原因解析1 DS实现动态切换数据源原理
首先mybatis-plus使用com.baomidou.dynamic.datasource.AbstractRoutingDataSource继承 AbstractDataSource接管数据源具体实现类为com.baomidou.dynamic.datasource.DynamicRoutingDataSource。项目初始化调用public synchronized void addDataSource(String ds, DataSource dataSource)加载数据源数据源存进dataSourceMap中。
private MapString, DataSource dataSourceMap new LinkedHashMap();private MapString, DynamicGroupDataSource groupDataSources new ConcurrentHashMap();public synchronized void addDataSource(String ds, DataSource dataSource) {if (p6spy) {dataSource new P6DataSource(dataSource);}dataSourceMap.put(ds, dataSource);if (ds.contains(UNDERLINE)) {String group ds.split(UNDERLINE)[0];if (groupDataSources.containsKey(group)) {groupDataSources.get(group).addDatasource(dataSource);} else {try {DynamicGroupDataSource groupDatasource new DynamicGroupDataSource(group,strategy.newInstance());groupDatasource.addDatasource(dataSource);groupDataSources.put(group, groupDatasource);} catch (Exception e) {log.error(dynamic-datasource - add the datasource named [{}] error, ds, e);dataSourceMap.remove(ds);}}}log.info(dynamic-datasource - load a datasource named [{}] success, ds);}进行数据操作时方法会被com.baomidou.dynamic.datasource.aop.DynamicDataSourceAnnotationInterceptor拦截
public class DynamicDataSourceAnnotationInterceptor implements MethodInterceptor {/*** The identification of SPEL.*/private static final String DYNAMIC_PREFIX #;private static final DynamicDataSourceClassResolver RESOLVER new DynamicDataSourceClassResolver();Setterprivate DsProcessor dsProcessor;Overridepublic Object invoke(MethodInvocation invocation) throws Throwable {try {DynamicDataSourceContextHolder.push(determineDatasource(invocation));return invocation.proceed();} finally {DynamicDataSourceContextHolder.poll();}}private String determineDatasource(MethodInvocation invocation) throws Throwable {Method method invocation.getMethod();DS ds method.isAnnotationPresent(DS.class)? method.getAnnotation(DS.class): AnnotationUtils.findAnnotation(RESOLVER.targetClass(invocation), DS.class);String key ds.value();return (!key.isEmpty() key.startsWith(DYNAMIC_PREFIX)) ? dsProcessor.determineDatasource(invocation, key) : key;}
}拦截器首先从被拦截的方法或者类一般DS注解用于Service也可用于Mapper和Controller上寻找DS注解获取到DS注解的值后将其存入com.baomidou.dynamic.datasource.toolkit.DynamicDataSourceContextHolderDynamicDataSourceContextHolder使用ThreadLocal存储当前线程的数据源名。
public final class DynamicDataSourceContextHolder {/*** 为什么要用链表存储(准确的是栈)* 为了支持嵌套切换如ABC三个service都是不同的数据源* 其中A的某个业务要调B的方法B的方法需要调用C的方法。一级一级调用切换形成了链。* 传统的只设置当前线程的方式不能满足此业务需求必须模拟栈后进先出。*/SuppressWarnings(unchecked)private static final ThreadLocalDequeString LOOKUP_KEY_HOLDER new ThreadLocal() {Overrideprotected Object initialValue() {return new ArrayDeque();}};private DynamicDataSourceContextHolder() {}/*** 获得当前线程数据源* return 数据源名称*/public static String peek() {return LOOKUP_KEY_HOLDER.get().peek();}/*** 设置当前线程数据源* 如非必要不要手动调用调用后确保最终清除* param ds 数据源名称*/public static void push(String ds) {LOOKUP_KEY_HOLDER.get().push(StringUtils.isEmpty(ds) ? : ds);}/*** 清空当前线程数据源* 如果当前线程是连续切换数据源 只会移除掉当前线程的数据源名称*/public static void poll() {DequeString deque LOOKUP_KEY_HOLDER.get();deque.poll();if (deque.isEmpty()) {LOOKUP_KEY_HOLDER.remove();}}/*** 强制清空本地线程* 防止内存泄漏如手动调用了push可调用此方法确保清除*/public static void clear() {LOOKUP_KEY_HOLDER.remove();}
}进行数据操作时会调用org.springframework.jdbc.datasource.getConnection()方法getConnection()方法最终调用了com.baomidou.dynamic.datasource.AbstractRoutingDataSource的getConnection()方法 Overridepublic Connection getConnection() throws SQLException {return determineDataSource().getConnection();}determineDataSource()由子类com.baomidou.dynamic.datasource.DynamicRoutingDataSource实现可以看到DynamicRoutingDataSource从DynamicDataSourceContextHolder获取数据源名称这个在之前拦截器处理存进ThreadLocal中如果有数据源名称则从dataSourceMap中获取没有则获取默认的primary数据源。
public DataSource determineDataSource() {return getDataSource(DynamicDataSourceContextHolder.peek());
}public DataSource getDataSource(String ds) {if (StringUtils.isEmpty(ds)) {return determinePrimaryDataSource();} else if (!groupDataSources.isEmpty() groupDataSources.containsKey(ds)) {log.debug(dynamic-datasource switch to the datasource named [{}], ds);return groupDataSources.get(ds).determineDataSource();} else if (dataSourceMap.containsKey(ds)) {log.debug(dynamic-datasource switch to the datasource named [{}], ds);return dataSourceMap.get(ds);}if (strict) {throw new RuntimeException(dynamic-datasource could not find a datasource named ds);}return determinePrimaryDataSource();
}private DataSource determinePrimaryDataSource() {log.debug(dynamic-datasource switch to the primary datasource);return groupDataSources.containsKey(primary) ? groupDataSources.get(primary).determineDataSource() : dataSourceMap.get(primary);
}此时的数据源已经切换成了我们需要的数据源。
数据操作完成后方法返回第二步中的拦截器执行DynamicDataSourceContextHolder.poll();清除掉此次的数据源避免影响后续数据操作。
附上动态数据源相关配置
spring:application:name: datasource:dynamic:primary: dataSource1datasource:dataSource1:type: com.alibaba.druid.pool.DruidDataSourcedriverClassName: com.microsoft.sqlserver.jdbc.SQLServerDriverurl: jdbc:sqlserver://localhost:1433;databasedataSource1username: password: dataSource2:type: com.alibaba.druid.pool.DruidDataSourcedriverClassName: com.microsoft.sqlserver.jdbc.SQLServerDriverurl: jdbc:sqlserver://localhost:1433;instanceNamesqlserver2017;DatabaseNamedataSource2username: password: pom.xml
dependencygroupIdcom.baomidou/groupIdartifactIddynamic-datasource-spring-boot-starter/artifactIdversion2.5.6/version
/dependency相应类
Service
//DS(dataSource2) 放在类上就是类下所有方法都使用这个数据源。
public class XXXServiceImpl extends BaseServiceImplXXXMapper, XXXBean implements XXXService {DS(dataSource1)public void selectDataFromSource1() {// do somethinng;}DS(dataSource2)public void selectDataFromSource1() {// do somethinng;}
}**注意**不可在事务中切换数据库保证事务需要方法使用同一连接使用DS(dataSource1)方法调用DS(dataSource2)无法切换连接会导致方法报错。
2 不可在事务中切换数据库分析解决
现在我们先来看一下是怎么解决问题的。
添加DS(“quartz”)注解只用支持多数据源切换value是你配置文件里面datasource数据源的名称即可。
DS(quartz)
Transactional(propagation Propagation.REQUIRES_NEW, rollbackFor Exception.class)
public interface JobInfoService extends IServiceJobInfo {
}相关的业务代码如下简单列举 JobController 中的createJob方法这里吐槽一下代码很low当前版本仅为了快速实现功能 RestController
RequestMapping(job)
Slf4j
public class JobController {Autowiredprivate final JobService jobService;PostMapping(add)Transactional(rollbackFor Exception.class)public JsonResult createJob(RequestBody JobDto dto) {jobService...//同样的事务数据源错误}
}到这里就可以实现在保证事务的同时解决多数据源切换的问题了。 首先在JobController 中的createJob方法上加 Transactional(rollbackFor Exception.class)默认事务的传播机制是PROPAGATION_REQUIRED 可以不指定。 然后在JobInfoServic 类上添加 Transactional(propagation Propagation.REQUIRES_NEW, rollbackFor Exception.class) 指定事务的传播机制是Propagation.REQUIRES_NEW。 至此冲突解决
3 原因解析
原本的一个事务拆分成两个事务。只在JobController 中的createJob方法加事务你会发现在切面里看数据源切换了但事务内的数据源依然是旧的这样就会报出XXX表找不到的问题。 因为spring在开启事务的同时会去数据库连接池拿数据库连接。如果仅在JobController 中的createJob方法上添加Transactional那么TransactionInterceptor 会使用 Spring DataSourceTransactionManager 创建事务并将事务信息获取数据源connection连接此时获取到的数据源是默认配置的base数据源信息)连接信息通过 ThreadLocal 绑定在当前线程。 此时当前线程事务绑定的连接信息是base数据源当我们在内层JobInfoServic使用DS切换数据源并没有重新开启新事务没有改变当前线程事务的连接信息仅仅是做了一次拦截改变了DataSourceHolder的栈顶dataSource对于整个事务的连接是没有影响的所以会产生数据源没有切换的问题。 所以我这里的解决办法是将保证createJob操作数据完整性的事务拆解成两个事务在JobInfoServic 类上除了添加切换数据源的注解DS(“quartz”)再添加Transactional(propagation Propagation.REQUIRES_NEW, rollbackFor Exception.class)新建开一个事务获取新数据源connection连接。
