wsr/orange-admin
1
0
Fork 0
mirror of https://gitee.com/orangeform/orange-admin.git synced 2026-01-17 10:36:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

修复多数据源问题

Browse Source 
Signed-off-by: orange-form <3510245832@qq.com>
This commit is contained in:
orange-form
2024-08-12 04:39:39 +00:00
committed by Gitee
parent 6c1940042d
commit 361fa35e6d
1 changed files with 3 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
OrangeFormsOpen-MybatisPlus/common/common-flow/src/main/java/com/orangeforms/common/flow/config/CustomEngineConfigurator.java
View File

@@ -1,39 +1,23 @@
package com.orangeforms.common.flow.config; package com.orangeforms.common.flow.config;
import com.orangeforms.common.core.config.DynamicDataSource; import com.orangeforms.common.core.config.DataSourceContextHolder;
import com.orangeforms.common.core.constant.ApplicationConstant; import com.orangeforms.common.core.constant.ApplicationConstant;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.flowable.common.engine.impl.AbstractEngineConfiguration; import org.flowable.common.engine.impl.AbstractEngineConfiguration;
import org.flowable.common.engine.impl.EngineConfigurator; import org.flowable.common.engine.impl.EngineConfigurator;
import org.springframework.jdbc.datasource.TransactionAwareDataSourceProxy;
import javax.sql.DataSource;
import java.util.Map;
/** /**
* 服务启动过程中动态切换flowable引擎内置表所在的数据源。 * 服务启动过程中动态切换flowable引擎内置表所在的数据源。
* *
* @author Jerry * @author Jerry
* @date 2024-07-02 * @date 2024-04-15
*/ */
@Slf4j @Slf4j
public class CustomEngineConfigurator implements EngineConfigurator { public class CustomEngineConfigurator implements EngineConfigurator {
@Override @Override
public void beforeInit(AbstractEngineConfiguration engineConfiguration) { public void beforeInit(AbstractEngineConfiguration engineConfiguration) {
DataSource dataSource = engineConfiguration.getDataSource(); DataSourceContextHolder.setDataSourceType(ApplicationConstant.COMMON_FLOW_AND_ONLINE_DATASOURCE_TYPE);
if (dataSource instanceof TransactionAwareDataSourceProxy) {
TransactionAwareDataSourceProxy proxy = (TransactionAwareDataSourceProxy) dataSource;
DataSource targetDataSource = proxy.getTargetDataSource();
if (targetDataSource instanceof DynamicDataSource) {
DynamicDataSource dynamicDataSource = (DynamicDataSource) targetDataSource;
Map<Object, DataSource> dynamicDataSourceMap = dynamicDataSource.getResolvedDataSources();
DataSource flowDataSource = dynamicDataSourceMap.get(ApplicationConstant.COMMON_FLOW_AND_ONLINE_DATASOURCE_TYPE);
if (flowDataSource != null) {
engineConfiguration.setDataSource(flowDataSource);
}
}
}
} }
@Override @Override