From 6ee2a947a786edad9eb19a5401de7fcc456b90a6 Mon Sep 17 00:00:00 2001 From: cloudroam <cloudroam> Date: 星期二, 10 十二月 2024 18:23:11 +0800 Subject: [PATCH] add: 供应商子账号4 --- src/main/java/com/mzl/flower/service/supplier/SupplierSubService.java | 17 ++++++++++++++++- 1 files changed, 16 insertions(+), 1 deletions(-) diff --git a/src/main/java/com/mzl/flower/service/supplier/SupplierSubService.java b/src/main/java/com/mzl/flower/service/supplier/SupplierSubService.java index 7a61aef..aa302a3 100644 --- a/src/main/java/com/mzl/flower/service/supplier/SupplierSubService.java +++ b/src/main/java/com/mzl/flower/service/supplier/SupplierSubService.java @@ -3,6 +3,7 @@ import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.mzl.flower.base.cache.StringCacheClient; +import com.mzl.flower.config.GlobalSupplierVariables; import com.mzl.flower.config.exception.ValidationException; import com.mzl.flower.config.security.SecurityUtils; import com.mzl.flower.constant.Constants; @@ -71,6 +72,8 @@ private final PartnerService partnerService; private final StationService stationService; private final UserWechatMapper wechatMapper; + + private final GlobalSupplierVariables globalSupplierVariables; @@ -274,10 +277,22 @@ String userId = ""; if (type == 1) { userId = SecurityUtils.getUserId(); - }else { + SupplierDTO currentSupplier = supplierMapper.getCurrentSupplier(userId); + //切换主账号 + String supplier = globalSupplierVariables.getSupplier(String.valueOf(currentSupplier.getId())); + if (StringUtils.isEmpty(supplier)) { + globalSupplierVariables.removeSupplier(String.valueOf(currentSupplier.getId())); + } + } else { SupplierSub supplierSub = supplierSubMapper.selectById(id); supplierSub.setType(0); userId = supplierSub.getUserId(); + //切换子账号 + String supplier = globalSupplierVariables.getSupplier(String.valueOf(supplierSub.getSupplierId())); + if (StringUtils.isEmpty(supplier)) { + globalSupplierVariables.removeSupplier(String.valueOf(supplierSub.getSupplierId())); + } + globalSupplierVariables.setSupplier(String.valueOf(supplierSub.getSupplierId()), userId); } CurrentUserDTO result = new CurrentUserDTO(); -- Gitblit v1.9.3