Ver Fonte

查询用户切换数据源

24282 há 2 anos atrás
pai
commit
5585900f02

+ 5 - 1
hx-common/src/main/java/com/fjhx/common/service/coding/impl/CodingRuleServiceImpl.java

@@ -4,10 +4,12 @@ import cn.hutool.core.convert.Convert;
 import cn.hutool.core.date.DateUtil;
 import cn.hutool.core.util.StrUtil;
 import com.alibaba.fastjson2.JSON;
+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.area.entity.po.CustomizeArea;
 import com.fjhx.area.service.CustomizeAreaService;
+import com.fjhx.common.constant.SourceConstant;
 import com.fjhx.common.entity.coding.bo.CustomerCodeAndCountryId;
 import com.fjhx.common.entity.coding.dto.CodingRuleDto;
 import com.fjhx.common.entity.coding.dto.CodingRuleSelectDto;
@@ -255,10 +257,12 @@ public class CodingRuleServiceImpl extends ServiceImpl<CodingRuleMapper, CodingR
      * 获取业务员编码
      */
     private String getUserCode() {
+        DynamicDataSourceContextHolder.push(SourceConstant.BASE);
         SysUser sysUser = sysUserService.getById(SecurityUtils.getUserId());
+        DynamicDataSourceContextHolder.poll();
         String userCode = sysUser.getUserCode();
         if (StrUtil.isBlank(userCode)) {
-            throw new ServiceException("业务代码未配置,生成编码失败");
+            throw new ServiceException("业务代码未配置,生成编码失败");
         }
         return userCode;
     }