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