From 6bc568611d0430c94f30e41736d7809e32a64273 Mon Sep 17 00:00:00 2001
From: cloudroam <cloudroam>
Date: 星期三, 11 九月 2024 15:18:20 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/master-v2' into master-v2

---
 src/main/java/com/mzl/flower/dto/response/current/CurrentUserDTO.java |    3 +++
 src/main/java/com/mzl/flower/service/supplier/StationService.java     |    4 ++++
 src/main/java/com/mzl/flower/service/system/UserService.java          |    5 +++++
 3 files changed, 12 insertions(+), 0 deletions(-)

diff --git a/src/main/java/com/mzl/flower/dto/response/current/CurrentUserDTO.java b/src/main/java/com/mzl/flower/dto/response/current/CurrentUserDTO.java
index 7d12c3b..c3cdf51 100644
--- a/src/main/java/com/mzl/flower/dto/response/current/CurrentUserDTO.java
+++ b/src/main/java/com/mzl/flower/dto/response/current/CurrentUserDTO.java
@@ -53,4 +53,7 @@
 
     @ApiModelProperty(value = "当前积分")
     private Integer currentPoint;
+
+    @ApiModelProperty(value = "是否总仓权限")
+    private Boolean mainWarehouse;
 }
diff --git a/src/main/java/com/mzl/flower/service/supplier/StationService.java b/src/main/java/com/mzl/flower/service/supplier/StationService.java
index 198eb34..cfc730d 100644
--- a/src/main/java/com/mzl/flower/service/supplier/StationService.java
+++ b/src/main/java/com/mzl/flower/service/supplier/StationService.java
@@ -96,4 +96,8 @@
         List<StationDTO> list = stationMapper.queryList( name);
         return list;
     }
+
+    public Boolean getMainWarehouse(String userId) {
+        return stationMapper.selectCount(new LambdaQueryWrapper<Station>().like(Station::getUserIds, userId))>0;
+    }
 }
diff --git a/src/main/java/com/mzl/flower/service/system/UserService.java b/src/main/java/com/mzl/flower/service/system/UserService.java
index 3ddf79c..e5286b5 100644
--- a/src/main/java/com/mzl/flower/service/system/UserService.java
+++ b/src/main/java/com/mzl/flower/service/system/UserService.java
@@ -21,6 +21,7 @@
 import com.mzl.flower.service.customer.CustomerService;
 import com.mzl.flower.service.partner.PartnerService;
 import com.mzl.flower.service.point.CustomerPointService;
+import com.mzl.flower.service.supplier.StationService;
 import com.mzl.flower.service.supplier.SupplierService;
 import com.mzl.flower.utils.TreeBuilderUtil;
 import com.mzl.flower.utils.UUIDGenerator;
@@ -79,6 +80,8 @@
 
     @Autowired
     private CustomerPointMapper customerPointMapper;
+    @Autowired
+    private StationService stationService;
 
 
     public User findUserByOpenId(String openId, String sessionKey, String unionId
@@ -208,6 +211,8 @@
             result.setSupplierDTO(supplierService.getCurrentSupplier());
         }else if(Constants.USER_TYPE.partner.name().equals(user.getType())){
             result.setPartnerDTO(partnerService.getCurrentPartner());
+        }else if (Constants.USER_TYPE.admin.name().equals(user.getType())){
+            result.setMainWarehouse(stationService.getMainWarehouse(user.getId()));
         }
         result.setBindWechat(wechatMapper.selectCount(new LambdaQueryWrapper<UserWechat>()
                 .eq(UserWechat::getUserId, userId)) > 0);

--
Gitblit v1.9.3