赋值流程id切库

hx-flow/src/main/java/com/fjhx/flow/service/flow/impl/FlowExampleServiceImpl.java

@@ -1,5 +1,6 @@
 package com.fjhx.flow.service.flow.impl;
 
+import com.baomidou.dynamic.datasource.toolkit.DynamicDataSourceContextHolder;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.fjhx.flow.entity.flow.dto.FlowExampleSelectDto;
@@ -13,6 +14,7 @@ import com.fjhx.flow.enums.HandleTypeEnum;
 import com.fjhx.flow.enums.NodeTypeEnum;
 import com.fjhx.flow.mapper.flow.FlowExampleMapper;
 import com.fjhx.flow.service.flow.*;
+import com.ruoyi.common.constant.BaseSourceConstant;
 import com.ruoyi.common.constant.StatusConstant;
 import com.ruoyi.common.core.domain.BaseIdPo;
 import com.ruoyi.common.core.domain.BasePo;
@@ -217,10 +219,13 @@ public class FlowExampleServiceImpl extends ServiceImpl<FlowExampleMapper, FlowE
             return;
         }
 
+        DynamicDataSourceContextHolder.push(BaseSourceConstant.BASE);
+
         List<Long> idList = list.stream().map(BaseIdPo::getId).distinct().collect(Collectors.toList());
         List<FlowExample> flowExampleList = list(q -> q.in(FlowExample::getBusinessId, idList));
 
         if (flowExampleList.size() == 0) {
+            DynamicDataSourceContextHolder.poll();
             return;
         }
 
@@ -229,6 +234,7 @@ public class FlowExampleServiceImpl extends ServiceImpl<FlowExampleMapper, FlowE
 
         list.forEach(item -> fun.accept(item, map.get(item.getId())));
 
+        DynamicDataSourceContextHolder.poll();
     }
 
     private List<ApprovalRecordVo.ButtonInfo> getButtonInfoList(FlowExample flowExample,